fix: close memory index and refresh protocol outputs

src/memory/index.test.ts

@@ -4,7 +4,7 @@ import path from "node:path";
 
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
-import { getMemorySearchManager } from "./index.js";
+import { getMemorySearchManager, type MemoryIndexManager } from "./index.js";
 
 vi.mock("./embeddings.js", () => {
   const embedText = (text: string) => {
@@ -29,6 +29,7 @@ vi.mock("./embeddings.js", () => {
 describe("memory index", () => {
   let workspaceDir: string;
   let indexPath: string;
+  let manager: MemoryIndexManager | null = null;
 
   beforeEach(async () => {
     workspaceDir = await fs.mkdtemp(path.join(os.tmpdir(), "clawdbot-mem-"));
@@ -45,6 +46,10 @@ describe("memory index", () => {
   });
 
   afterEach(async () => {
+    if (manager) {
+      await manager.close();
+      manager = null;
+    }
     await fs.rm(workspaceDir, { recursive: true, force: true });
   });
 
@@ -67,6 +72,7 @@ describe("memory index", () => {
     const result = await getMemorySearchManager({ cfg, agentId: "main" });
     expect(result.manager).not.toBeNull();
     if (!result.manager) throw new Error("manager missing");
+    manager = result.manager;
     await result.manager.sync({ force: true });
     const results = await result.manager.search("alpha");
     expect(results.length).toBeGreaterThan(0);
@@ -91,6 +97,7 @@ describe("memory index", () => {
     const result = await getMemorySearchManager({ cfg, agentId: "main" });
     expect(result.manager).not.toBeNull();
     if (!result.manager) throw new Error("manager missing");
+    manager = result.manager;
     await expect(
       result.manager.readFile({ relPath: "NOTES.md" }),
     ).rejects.toThrow("path required");

src/memory/index.ts

@@ -56,6 +56,7 @@ const SNIPPET_MAX_CHARS = 700;
 const INDEX_CACHE = new Map<string, MemoryIndexManager>();
 
 export class MemoryIndexManager {
+  private readonly cacheKey: string;
   private readonly cfg: ClawdbotConfig;
   private readonly agentId: string;
   private readonly workspaceDir: string;
@@ -67,6 +68,7 @@ export class MemoryIndexManager {
   private watcher: FSWatcher | null = null;
   private watchTimer: NodeJS.Timeout | null = null;
   private intervalTimer: NodeJS.Timeout | null = null;
+  private closed = false;
   private dirty = false;
   private sessionWarm = new Set<string>();
   private syncing: Promise<void> | null = null;
@@ -91,6 +93,7 @@ export class MemoryIndexManager {
       local: settings.local,
     });
     const manager = new MemoryIndexManager({
+      cacheKey: key,
       cfg,
       agentId,
       workspaceDir,
@@ -102,12 +105,14 @@ export class MemoryIndexManager {
   }
 
   private constructor(params: {
+    cacheKey: string;
     cfg: ClawdbotConfig;
     agentId: string;
     workspaceDir: string;
     settings: ResolvedMemorySearchConfig;
     providerResult: EmbeddingProviderResult;
   }) {
+    this.cacheKey = params.cacheKey;
     this.cfg = params.cfg;
     this.agentId = params.agentId;
     this.workspaceDir = params.workspaceDir;
@@ -234,6 +239,25 @@ export class MemoryIndexManager {
     };
   }
 
+  async close(): Promise<void> {
+    if (this.closed) return;
+    this.closed = true;
+    if (this.watchTimer) {
+      clearTimeout(this.watchTimer);
+      this.watchTimer = null;
+    }
+    if (this.intervalTimer) {
+      clearInterval(this.intervalTimer);
+      this.intervalTimer = null;
+    }
+    if (this.watcher) {
+      await this.watcher.close();
+      this.watcher = null;
+    }
+    this.db.close();
+    INDEX_CACHE.delete(this.cacheKey);
+  }
+
   private openDatabase(): DatabaseSync {
     const dbPath = resolveUserPath(this.settings.store.path);
     const dir = path.dirname(dbPath);