refactor: keep runtime context in memory

.opencode/tools/memory_manager.ts

@@ -1,13 +1,12 @@
 import { tool } from "@opencode-ai/plugin";
 import { MemoryStore } from "../../src/memory/store.js";
-import { SessionRuntimeContextStore } from "../../src/sessions/runtimeContextStore.js";
+import {
+  getRuntimeSessionContext,
+  setRuntimeSessionContext,
+} from "../../src/runtime/sessionContext.js";
 
 const memoryStore = new MemoryStore();
-const toolContextStore = new SessionRuntimeContextStore();
-const initializePromise = Promise.all([
-  memoryStore.initialize(),
-  toolContextStore.initialize(),
-]);
+const initializePromise = memoryStore.initialize();
 
 export default tool({
   description:
@@ -37,7 +36,7 @@ export default tool({
   },
   async execute(args, context) {
     await initializePromise;
-    const sessionContext = await toolContextStore.read(context.sessionID);
+    const sessionContext = getRuntimeSessionContext(context.sessionID);
     if (!sessionContext) {
       throw new Error(`session context not found for ${context.sessionID}`);
     }
@@ -57,10 +56,10 @@ export default tool({
     }
     if (sessionContext.allowLearningWrite === false && args.action !== "list") {
       return JSON.stringify({
-      ok: true,
-      kind: "memory",
-      decision: "rejected",
-      detail: "memory writes are disabled for this session",
+        ok: true,
+        kind: "memory",
+        decision: "rejected",
+        detail: "memory writes are disabled for this session",
       });
     }
 
@@ -71,17 +70,17 @@ export default tool({
         ...(sessionContext.memoryListReadScopes ?? {}),
         [scope]: true,
       };
-      await toolContextStore.write({
+      setRuntimeSessionContext({
         ...sessionContext,
         memoryListReadScopes: readScopes,
       });
       return JSON.stringify({
-      ok: true,
-      kind: "memory",
-      decision: "accepted",
-      detail: "memory listed",
-      items: await memoryStore.list(scope, scopeKey),
-      target: scope,
+        ok: true,
+        kind: "memory",
+        decision: "accepted",
+        detail: "memory listed",
+        items: await memoryStore.list(scope, scopeKey),
+        target: scope,
       });
     }
 
@@ -96,46 +95,55 @@ export default tool({
         });
       }
       const result = await memoryStore.upsert(scope, scopeKey, {
-      content: args.content ?? "",
-      sessionId: sessionContext.clientSessionId,
-      source: "tool",
-      traceId: sessionContext.traceId,
+        content: args.content ?? "",
+        sessionId: sessionContext.clientSessionId,
+        source: "tool",
+        traceId: sessionContext.traceId,
       });
       if (!result.entry) {
+        return JSON.stringify({
+          ok: true,
+          kind: "memory",
+          decision: "rejected",
+          detail: "content rejected by persistence policy",
+        });
+      }
       return JSON.stringify({
         ok: true,
         kind: "memory",
-        decision: "rejected",
-        detail: "content rejected by persistence policy",
-      });
-      }
-      return JSON.stringify({
-      ok: true,
-      kind: "memory",
-      decision: result.changed ? "accepted" : "deduped",
-      detail: result.detail,
-      entry: result.entry,
-      target: scope,
+        decision: result.changed ? "accepted" : "deduped",
+        detail: result.detail,
+        entry: result.entry,
+        target: scope,
       });
     }
 
     if (args.action === "replace") {
-      const result = await memoryStore.replace(scope, scopeKey, args.target_id ?? "", {
-      content: args.content ?? "",
-      sessionId: sessionContext.clientSessionId,
-      source: "tool",
-      traceId: sessionContext.traceId,
-      });
+      const result = await memoryStore.replace(
+        scope,
+        scopeKey,
+        args.target_id ?? "",
+        {
+          content: args.content ?? "",
+          sessionId: sessionContext.clientSessionId,
+          source: "tool",
+          traceId: sessionContext.traceId,
+        },
+      );
       return JSON.stringify({
-      ok: true,
-      kind: "memory",
-      decision: result.changed ? "accepted" : "rejected",
-      detail: result.detail,
-      target: scope,
+        ok: true,
+        kind: "memory",
+        decision: result.changed ? "accepted" : "rejected",
+        detail: result.detail,
+        target: scope,
       });
     }
 
-    const result = await memoryStore.remove(scope, scopeKey, args.target_id ?? "");
+    const result = await memoryStore.remove(
+      scope,
+      scopeKey,
+      args.target_id ?? "",
+    );
     return JSON.stringify({
       ok: true,
       kind: "memory",