memories-lite 0.9.4 → 0.10.1

package/tests/memory.todo.test.ts

@@ -0,0 +1,126 @@
+/// <reference types="jest" />
+import { MemoriesLite } from "../src";
+import { SearchResult } from "../src/types";
+import dotenv from "dotenv";
+import { createTestMemory } from "./init.mem";
+
+dotenv.config();
+
+jest.setTimeout(30000); // Increase timeout to 30 seconds
+
+describe("Memory Class TODO regression tests", () => {
+  let memory: MemoriesLite;
+  let userId: string;
+
+  beforeEach(async () => {
+    // Initialize memory via helper
+    ({ memory, userId } = createTestMemory({ customPrompt: "L'utilisateur travail pour une régie immobilière!" }));
+    // Reset all memories before each test
+    await memory.reset(userId);
+  });
+
+  afterEach(async () => {
+    // Clean up after each test
+    await memory.reset(userId);
+  });
+
+  describe("Manage TODOs (add/remove/update)", () => {
+    it("should add a TODO when explicitly asked with 'ajoute un todo'", async () => {
+      const result = (await memory.capture([
+        { role: "user", content: "Ajoute un todo: appeler la banque demain à 09:00" },
+        { role: "assistant", content: "Compris, j'ajoute un todo." }
+      ], userId, {})) as SearchResult;
+
+      expect(result).toBeDefined();
+      expect(Array.isArray(result.results)).toBe(true);
+      // At least one TODO should be captured when explicitly asked
+      expect(result.results.some(r => r.type === "todo")).toBe(true);
+    });
+
+    it("should not create a TODO when task is implicit only", async () => {
+      const result = (await memory.capture([
+        { role: "user", content: "Je dois appeler la banque demain à 09:00" },
+        { role: "assistant", content: "Très bien." }
+      ], userId, {})) as SearchResult;
+
+      expect(result).toBeDefined();
+      expect(Array.isArray(result.results)).toBe(true);
+      // Must not infer TODO without explicit user request
+      expect(result.results.every(r => r.type !== "todo")).toBe(true);
+    });
+
+    it("should not create memory for how-to business requests (procedural removed)", async () => {
+      const result = (await memory.capture([
+        { role: "user", content: "Je veux résilier un bail, comment faire ?" },
+        { role: "assistant", content: "Voici la procédure..." }
+      ], userId, {})) as SearchResult;
+
+      expect(result).toBeDefined();
+      expect(Array.isArray(result.results)).toBe(true);
+      // Business procedures should not create memories (procedural type removed)
+      expect(result.results.length).toBe(0);
+    });
+
+    it("should add then update a TODO via new capture", async () => {
+      const add = (await memory.capture([
+        { role: "user", content: "Ajoute un todo: appeler la banque demain à 09:00" },
+        { role: "assistant", content: "Compris, j'ajoute un todo." }
+      ], userId, {})) as SearchResult;
+
+      expect(add).toBeDefined();
+      const todo = add.results.find(r => r.type === "todo");
+      expect(todo?.id).toBeDefined();
+
+      const newText = "Appeler la banque lundi à 10:00";
+      const updateRes = (await memory.capture([
+        { role: "user", content: `Mets à jour ce todo: ${newText}` },
+        { role: "assistant", content: "OK, je mets à jour le todo." }
+      ], userId, {})) as SearchResult;
+
+      expect(updateRes).toBeDefined();
+      // Expect an UPDATE event captured (merger decides UPDATE vs ADD)
+      const updatedItem = updateRes.results.find(r => (r as any).metadata?.event === 'UPDATE' && r.type === 'todo');
+      expect(updatedItem).toBeDefined();
+      // Allow optional suffix from LLM like annotations removal; assert containment
+      expect(updatedItem?.memory?.toLowerCase().includes("appeler la banque lundi à 10:00".toLowerCase())).toBe(true);
+
+      const persisted = await memory.get(updatedItem!.id, userId);
+      expect(persisted).not.toBeNull();
+      expect(persisted?.memory?.toLowerCase().includes("appeler la banque lundi à 10:00".toLowerCase())).toBe(true);
+      expect(persisted?.type).toBe("todo");
+    });
+
+    it("should add then delete a TODO via new capture", async () => {
+      const add = (await memory.capture([
+        { role: "user", content: "Ajoute un todo: appeler la banque demain à 09:00" },
+        { role: "assistant", content: "Compris, j'ajoute un todo." }
+      ], userId, {})) as SearchResult;
+
+      const todo = add.results.find(r => r.type === "todo");
+      expect(todo?.id).toBeDefined();
+
+      const delRes = (await memory.capture([
+        { role: "user", content: `Supprime ce todo: ${todo!.memory}` },
+        { role: "assistant", content: "Je supprime le todo." }
+      ], userId, {})) as SearchResult;
+
+      expect(delRes).toBeDefined();
+      const deletedItem = delRes.results.find(r => (r as any).metadata?.event === 'DELETE' && r.type === 'todo');
+      expect(deletedItem).toBeDefined();
+
+      const persisted = await memory.get(deletedItem!.id, userId);
+      expect(persisted).toBeNull();
+    });
+
+    it("should add a TODO using 'tâche' wording (synonym)", async () => {
+      const result = (await memory.capture([
+        { role: "user", content: "Ajoute une tâche: appeler la banque demain à 09:00" },
+        { role: "assistant", content: "C'est noté, j'ajoute la tâche." }
+      ], userId, {})) as SearchResult;
+
+      expect(result).toBeDefined();
+      expect(Array.isArray(result.results)).toBe(true);
+      expect(result.results.some(r => r.type === "todo")).toBe(true);
+    });
+  });
+});