@mingxy/cerebro 1.5.5 → 1.5.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mingxy/cerebro",
-  "version": "1.5.5",
+  "version": "1.5.7",
   "description": "Cerebro persistent memory plugin for OpenCode — auto-recall, auto-capture, 9 memory tools with clustering",
   "type": "module",
   "main": "src/index.ts",

package/src/hooks.ts

@@ -18,25 +18,6 @@ const firstMessages = new Map<string, string>();
 const sessionMessages = new Map<string, Array<{ role: string; content: string }>>();
 const profileInjectedSessions = new Set<string>();
 
-function extractMemoryIds(result: unknown): string[] {
-  if (!result) return [];
-  if (Array.isArray(result)) {
-    return (result as Array<{ id?: string }>).map((m) => m.id).filter(Boolean) as string[];
-  }
-  if (typeof result === "object" && result !== null) {
-    const r = result as Record<string, unknown>;
-    if (Array.isArray(r.memories)) {
-      return (r.memories as Array<{ id?: string }>).map((m) => m.id).filter(Boolean) as string[];
-    }
-    if (Array.isArray(r.results)) {
-      return (r.results as Array<{ id?: string; memory?: { id?: string } }>)
-        .map((m) => m.id ?? m.memory?.id)
-        .filter(Boolean) as string[];
-    }
-  }
-  return [];
-}
-
 function formatRelativeAge(isoDate: string): string {
   const diffMs = Date.now() - new Date(isoDate).getTime();
   const minutes = Math.floor(diffMs / 60_000);
@@ -152,8 +133,6 @@ export function autoRecallHook(client: OmemClient, containerTags: string[], tui:
       const query_text = userMessages[userMessages.length - 1]?.content || firstMessages.get(input.sessionID) || "";
       const last_query_text = userMessages.length >= 2 ? userMessages[userMessages.length - 2].content : undefined;
 
-      if (!query_text.trim()) return;
-
       const projectTags = containerTags.filter(t => t.startsWith("omem_project_"));
       const shouldRecallRes = await client.shouldRecall(query_text, last_query_text, input.sessionID, similarityThreshold, maxRecallResults, projectTags.length > 0 ? projectTags : undefined);
 
@@ -280,7 +259,7 @@ export function autoRecallHook(client: OmemClient, containerTags: string[], tui:
   };
 }
 
-export function keywordDetectionHook(client: OmemClient, containerTags: string[], threshold: number, tui: any, ingestMode: "smart" | "raw" = "smart") {
+export function keywordDetectionHook(_client: OmemClient, _containerTags: string[], threshold: number, _tui: any, _ingestMode: "smart" | "raw" = "smart") {
   return async (
     input: { sessionID: string; messageID?: string },
     output: { message: UserMessage; parts: Part[] },
@@ -309,38 +288,47 @@ export function keywordDetectionHook(client: OmemClient, containerTags: string[]
     });
 
     const messages = sessionMessages.get(input.sessionID)!;
+    // Ingest is now handled by sessionIdleHook (session.idle → sessionIngest API).
+    // This hook only collects messages and detects keywords for recall.
     if (messages.length >= threshold) {
-      try {
-        const result = await client.ingestMessages(messages, {
-          mode: ingestMode,
-          tags: [...containerTags, "auto-capture"],
-          sessionId: input.sessionID,
-        });
-        if (result === null) {
-          showToast(tui, "🔴 Capture Failed", `Memory capture blocked · check API Key and spiritual connection`, "error");
-        } else {
-          showToast(tui, "🧠 Memory Sealed", `${messages.length} dialogues captured · entrusted to the heavens for refinement`, "success");
-          const memoryIds = extractMemoryIds(result);
-          if (memoryIds.length > 0) {
-            const recordResult = await client.recordSessionRecall(
-              input.sessionID,
-              memoryIds,
-              "auto",
-              firstMessages.get(input.sessionID) || "",
-              0,
-              0,
-            );
-            if (recordResult) {
-              showToast(tui, "📦 Capture Recorded", `${memoryIds.length} memory(s) saved to session history`, "success");
-            } else {
-              showToast(tui, "🔴 Capture Record Failed", `Failed to save capture record · check API connection`, "error");
-            }
+      // Threshold reached — messages will be processed on next session.idle
+    }
+  };
+}
+
+export function compactingHook(client: OmemClient, containerTags: string[], tui: any, ingestMode: "smart" | "raw" = "smart") {
+  return async (
+    input: { sessionID?: string },
+    output: { context: string[]; prompt?: string },
+  ) => {
+    if (input.sessionID && sessionMessages.has(input.sessionID)) {
+      const messages = sessionMessages.get(input.sessionID)!;
+      if (messages.length > 0) {
+        try {
+          const result = await client.ingestMessages(messages, {
+            mode: ingestMode,
+            tags: [...containerTags, "auto-capture"],
+            sessionId: input.sessionID,
+          });
+          if (result === null) {
+            showToast(tui, "🔴 Archive Failed", "Session archive blocked · check spiritual realm status", "error");
+          } else {
+            showToast(tui, "📦 Session Archived", `${messages.length} residual dialogues archived · merged into the realm`, "success");
           }
-          sessionMessages.delete(input.sessionID);
+        } catch {
+          showToast(tui, "🔴 Archive Failed", "Session archive blocked · spiritual pulse anomaly", "error");
         }
-      } catch {
-        showToast(tui, "🔴 Capture Failed", "Memory capture blocked · spiritual pulse anomaly", "error");
+        sessionMessages.delete(input.sessionID);
+      }
+    }
+
+    try {
+      const results = await client.searchMemories("*", 20, undefined, containerTags);
+      const block = buildContextBlock(results);
+      if (block) {
+        output.context.push(block);
       }
+    } catch {
     }
   };
 }
@@ -353,6 +341,7 @@ export function sessionIdleHook(
   tui: any,
   sdkClient: any,
   _ingestMode: "smart" | "raw" = "smart",
+  getMainSessionId?: () => string | undefined,
 ) {
   let idleTimeout: ReturnType<typeof setTimeout> | null = null;
   let isCapturing = false;
@@ -363,6 +352,11 @@ export function sessionIdleHook(
     const sessionID = input.event.properties?.sessionID;
     if (!sessionID) return;
 
+    if (getMainSessionId) {
+      const mainId = getMainSessionId();
+      if (mainId && sessionID !== mainId) return;
+    }
+
     if (idleTimeout) clearTimeout(idleTimeout);
 
     idleTimeout = setTimeout(async () => {
@@ -400,13 +394,11 @@ export function sessionIdleHook(
 
         try {
           await omemClient.sessionIngest(conversationMessages, sessionID);
-          // Only mark as processed after successful ingest to prevent message loss on server errors
           for (const id of newMessageIds) {
             processedMessageIds.add(id);
           }
           showToast(tui, "🧠 Memory Sealed", `${conversationMessages.length} dialogues captured · entrusted to the heavens for refinement`, "success");
         } catch (err) {
-          // Do NOT mark as processed — messages will be retried on next idle event
           showToast(tui, "🔴 Session Capture Failed", String(err).substring(0, 100), "error");
         }
       } catch (err) {
@@ -419,40 +411,3 @@ export function sessionIdleHook(
     }, 10000);
   };
 }
-
-export function compactingHook(client: OmemClient, containerTags: string[], tui: any, ingestMode: "smart" | "raw" = "smart") {
-  return async (
-    input: { sessionID?: string },
-    output: { context: string[]; prompt?: string },
-  ) => {
-    if (input.sessionID && sessionMessages.has(input.sessionID)) {
-      const messages = sessionMessages.get(input.sessionID)!;
-      if (messages.length > 0) {
-        try {
-          const result = await client.ingestMessages(messages, {
-            mode: ingestMode,
-            tags: [...containerTags, "auto-capture"],
-            sessionId: input.sessionID,
-          });
-          if (result === null) {
-            showToast(tui, "🔴 Archive Failed", "Session archive blocked · check spiritual realm status", "error");
-          } else {
-            showToast(tui, "📦 Session Archived", `${messages.length} residual dialogues archived · merged into the realm`, "success");
-          }
-        } catch {
-          showToast(tui, "🔴 Archive Failed", "Session archive blocked · spiritual pulse anomaly", "error");
-        }
-        sessionMessages.delete(input.sessionID);
-      }
-    }
-
-    try {
-      const results = await client.searchMemories("*", 20, undefined, containerTags);
-      const block = buildContextBlock(results);
-      if (block) {
-        output.context.push(block);
-      }
-    } catch {
-    }
-  };
-}

package/src/index.ts

@@ -96,7 +96,7 @@ const OmemPlugin: Plugin = async (input) => {
     "chat.message": keywordDetectionHook(omemClient, containerTags, config.autoCaptureThreshold, tui, config.ingestMode),
     "experimental.session.compacting": compactingHook(omemClient, containerTags, tui, config.ingestMode),
     tool: buildTools(omemClient, containerTags, { agentId, getSessionId: () => currentSessionId }),
-    event: sessionIdleHook(omemClient, containerTags, tui, client, config.ingestMode),
+    event: sessionIdleHook(omemClient, containerTags, tui, client, config.ingestMode, () => currentSessionId),
     "shell.env": async (_input: any, output: any) => {
       if (directory) {
         output.env.OMEM_PROJECT_DIR = directory;