@mingxy/cerebro 1.5.4 → 1.5.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mingxy/cerebro",
-  "version": "1.5.4",
+  "version": "1.5.6",
   "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

@@ -353,6 +353,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 +364,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 () => {
@@ -375,6 +381,7 @@ export function sessionIdleHook(
 
         const messages = response.data;
         const conversationMessages: Array<{ role: string; content: string }> = [];
+        const newMessageIds: string[] = [];
         let hasNewMessages = false;
 
         for (const msg of messages) {
@@ -391,7 +398,7 @@ export function sessionIdleHook(
           if (!text) continue;
 
           hasNewMessages = true;
-          processedMessageIds.add(msgId);
+          newMessageIds.push(msgId);
           conversationMessages.push({ role, content: text });
         }
 
@@ -399,8 +406,13 @@ 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) {

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;