fix(memory-core): force exit after memory search --json output

bladin · bladin · commit f546d05d11a4 · 2026-06-10T14:22:14.000+08:00
Fixes #91821
diff --git a/extensions/memory-core/src/cli.runtime.ts b/extensions/memory-core/src/cli.runtime.ts
@@ -1271,6 +1271,23 @@ export async function runMemorySearch(
         typeof (manager as { status?: () => { workspaceDir?: string } }).status === "function"
           ? manager.status().workspaceDir
           : undefined;
+      if (opts.json) {
+        defaultRuntime.writeJson({ results });
+        if (dreamingEnabled && workspaceDir) {
+          void recordShortTermRecalls({
+            workspaceDir,
+            query,
+            results,
+            timezone: dreaming.timezone,
+          }).catch(() => {
+            // Recall tracking is best-effort and must not block normal search results.
+          });
+        }
+        if (!process.env.VITEST) {
+          setImmediate(() => process.exit(process.exitCode ?? 0));
+        }
+        return;
+      }
       if (dreamingEnabled) {
         void recordShortTermRecalls({
           workspaceDir,
@@ -1281,10 +1298,6 @@ export async function runMemorySearch(
           // Recall tracking is best-effort and must not block normal search results.
         });
       }
-      if (opts.json) {
-        defaultRuntime.writeJson({ results });
-        return;
-      }
       const identityWarning =
         typeof manager.status === "function"
           ? formatMemoryIndexIdentityWarning(manager.status(), agentId)
