opencode-codex-memory 0.1.7 → 0.1.8

package/dist/opencode.json

@@ -5,6 +5,7 @@
       "mode": "subagent",
       "prompt": "You are a memory consolidation agent. Read the workspace diff file and update MEMORY.md, memory_summary.md, and skills/ to reflect the latest memories. Keep memory_summary.md under 10000 chars (2500 tokens). Prune stale entries. Do not access the network.",
       "permission": {
+        "*": "deny",
         "bash": "deny",
         "webfetch": "deny",
         "websearch": "deny",
@@ -21,6 +22,7 @@
       "mode": "subagent",
       "prompt": "You are a memory extraction agent. Read the session transcript and extract raw_memory, rollout_summary, and rollout_slug as JSON. Exclude AGENTS.md/instruction content. Redact secrets.",
       "permission": {
+        "*": "deny",
         "bash": "deny",
         "webfetch": "deny",
         "websearch": "deny",

package/dist/src/index.d.ts

@@ -1,5 +1,7 @@
+import { MemoryStore } from "./store.js";
 import type { PluginInput, PluginOptions } from "@opencode-ai/plugin";
 export declare function takeNewCitations(partKey: string, ids: string[]): string[];
+export declare function handleSessionDeleted(sessionId: string, store?: Pick<MemoryStore, "deleteSessionMemory">, schedulePhase2?: () => void): void;
 declare const _default: {
     id: string;
     server(input: PluginInput, opts?: PluginOptions): Promise<{

package/dist/src/index.js

@@ -52,6 +52,10 @@ export function takeNewCitations(partKey, ids) {
         seen.add(id);
     return fresh;
 }
+export function handleSessionDeleted(sessionId, store = getStore(), schedulePhase2 = () => { void triggerPhase2(); }) {
+    if (store.deleteSessionMemory(sessionId))
+        schedulePhase2();
+}
 export default {
     id: "opencode-codex-memory",
     async server(input, opts) {
@@ -267,13 +271,13 @@ function buildHooks() {
                 }
                 if (ev.type === "session.deleted") {
                     // Mirrors codex delete_thread_memory: drop the extracted memory and
-                    // its job when the session is deleted; the file disappears at the
-                    // next phase-2 rebuild and the diff drives forgetting.
+                    // its job when the session is deleted. If phase 2 had consumed it,
+                    // enqueue and attempt consolidation so the diff drives forgetting.
                     const props = ev.properties;
                     const sid = props?.info?.id;
                     if (sid) {
                         try {
-                            getStore().deleteSessionMemory(sid);
+                            handleSessionDeleted(sid);
                         }
                         catch (e) {
                             console.error("[opencode-codex-memory] deleteSessionMemory failed:", e);

package/dist/src/store.d.ts

@@ -56,6 +56,11 @@ export declare class MemoryStore {
     /** Extraction succeeded but produced nothing worth keeping: finish the job and drop any stale output. */
     markStage1SucceededNoOutput(sessionId: string, ownershipToken: string, sourceUpdatedAt: number): void;
     markStage1Failed(sessionId: string, ownershipToken: string, error: string): void;
+    /**
+     * Enqueues global consolidation after stage-1 state changes. If phase 2 is
+     * already running, preserve its lease and advance only the input watermark.
+     */
+    private enqueueGlobalConsolidation;
     claimGlobalPhase2Job(): Phase2ClaimResult;
     heartbeatPhase2Job(ownershipToken: string): boolean;
     /**
@@ -74,8 +79,11 @@ export declare class MemoryStore {
      * - ranked by usage, then recency
      */
     getPhase2InputSelection(maxRaw: number, maxUnusedDays: number): Stage1Output[];
-    /** Mirrors codex delete_thread_memory: remove a deleted session's output + job. */
-    deleteSessionMemory(sessionId: string): void;
+    /**
+     * Mirrors codex delete_thread_memory: remove a deleted session's output and
+     * job, then enqueue forgetting if phase 2 had consumed that output.
+     */
+    deleteSessionMemory(sessionId: string): boolean;
     /**
      * codex clear_memory_data deletes extracted memories and jobs but explicitly
      * preserves per-session memory modes: a reset must not re-enable sessions

package/dist/src/store.js

@@ -146,8 +146,10 @@ export class MemoryStore {
                 .run(nowSec(), out.source_updated_at, sessionId, ownershipToken);
             // Ownership lost (lease expired, job re-claimed): do not clobber the new
             // owner's output. Mirrors codex mark_stage1_job_succeeded.
-            if (res.changes > 0)
+            if (res.changes > 0) {
                 this.upsertStage1Output(out);
+                this.enqueueGlobalConsolidation(out.source_updated_at);
+            }
         }).immediate();
     }
     /** Extraction succeeded but produced nothing worth keeping: finish the job and drop any stale output. */
@@ -158,8 +160,11 @@ export class MemoryStore {
             last_success_watermark=?, retry_at=NULL
            WHERE kind='memory_stage1' AND job_key=? AND status='running' AND ownership_token=?`)
                 .run(nowSec(), sourceUpdatedAt, sessionId, ownershipToken);
-            if (res.changes > 0)
-                this.db.prepare("DELETE FROM memory_stage1_outputs WHERE session_id = ?").run(sessionId);
+            if (res.changes === 0)
+                return;
+            const deleted = this.db.prepare("DELETE FROM memory_stage1_outputs WHERE session_id = ?").run(sessionId);
+            if (deleted.changes > 0)
+                this.enqueueGlobalConsolidation(sourceUpdatedAt);
         }).immediate();
     }
     markStage1Failed(sessionId, ownershipToken, error) {
@@ -174,6 +179,32 @@ export class MemoryStore {
          WHERE kind='memory_stage1' AND job_key=? AND status='running' AND ownership_token=?`)
             .run(error.slice(0, 4000), nowSec() + STAGE1_RETRY_DELAY_SECONDS, nowSec(), sessionId, ownershipToken);
     }
+    /**
+     * Enqueues global consolidation after stage-1 state changes. If phase 2 is
+     * already running, preserve its lease and advance only the input watermark.
+     */
+    enqueueGlobalConsolidation(inputWatermark) {
+        this.db
+            .prepare(`INSERT INTO memory_jobs
+          (kind, job_key, status, retry_remaining, input_watermark, last_success_watermark)
+         VALUES ('memory_consolidate_global', 'global', 'pending', ?, ?, 0)
+         ON CONFLICT(kind, job_key) DO UPDATE SET
+           status = CASE
+             WHEN memory_jobs.status = 'running' THEN 'running'
+             ELSE 'pending'
+           END,
+           retry_at = CASE
+             WHEN memory_jobs.status = 'running' THEN memory_jobs.retry_at
+             ELSE NULL
+           END,
+           retry_remaining = MAX(memory_jobs.retry_remaining, excluded.retry_remaining),
+           input_watermark = CASE
+             WHEN excluded.input_watermark > COALESCE(memory_jobs.input_watermark, 0)
+               THEN excluded.input_watermark
+             ELSE COALESCE(memory_jobs.input_watermark, 0) + 1
+           END`)
+            .run(DEFAULT_RETRY_REMAINING, inputWatermark);
+    }
     claimGlobalPhase2Job() {
         const workerId = newId();
         const ownershipToken = newId();
@@ -287,11 +318,21 @@ export class MemoryStore {
          LIMIT ?`)
             .all(cutoff, cutoff, maxRaw);
     }
-    /** Mirrors codex delete_thread_memory: remove a deleted session's output + job. */
+    /**
+     * Mirrors codex delete_thread_memory: remove a deleted session's output and
+     * job, then enqueue forgetting if phase 2 had consumed that output.
+     */
     deleteSessionMemory(sessionId) {
-        this.db.transaction(() => {
-            this.db.prepare("DELETE FROM memory_stage1_outputs WHERE session_id = ?").run(sessionId);
+        return this.db.transaction(() => {
+            const existing = this.db
+                .prepare("SELECT selected_for_phase2 FROM memory_stage1_outputs WHERE session_id = ?")
+                .get(sessionId);
+            const deleted = this.db.prepare("DELETE FROM memory_stage1_outputs WHERE session_id = ?").run(sessionId);
             this.db.prepare("DELETE FROM memory_jobs WHERE kind='memory_stage1' AND job_key = ?").run(sessionId);
+            const shouldConsolidate = deleted.changes > 0 && existing !== null && existing.selected_for_phase2 !== 0;
+            if (shouldConsolidate)
+                this.enqueueGlobalConsolidation(now());
+            return shouldConsolidate;
         }).immediate();
     }
     /**

package/opencode.json

@@ -5,6 +5,7 @@
       "mode": "subagent",
       "prompt": "You are a memory consolidation agent. Read the workspace diff file and update MEMORY.md, memory_summary.md, and skills/ to reflect the latest memories. Keep memory_summary.md under 10000 chars (2500 tokens). Prune stale entries. Do not access the network.",
       "permission": {
+        "*": "deny",
         "bash": "deny",
         "webfetch": "deny",
         "websearch": "deny",
@@ -21,6 +22,7 @@
       "mode": "subagent",
       "prompt": "You are a memory extraction agent. Read the session transcript and extract raw_memory, rollout_summary, and rollout_slug as JSON. Exclude AGENTS.md/instruction content. Redact secrets.",
       "permission": {
+        "*": "deny",
         "bash": "deny",
         "webfetch": "deny",
         "websearch": "deny",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-codex-memory",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Persistent memory plugin for opencode — ports codex's two-phase memory system (extraction → consolidation → injection → citation feedback)",
   "type": "module",
   "main": "./dist/src/index.js",