agent-inbox 0.2.2 → 0.2.4

package/test/sqlite-storage.test.ts

@@ -240,4 +240,110 @@ describe("SqliteStorage", () => {
       expect(threads[0].root_turn_id).toBe("turn-1");
     });
   });
+
+  describe("setMessageConversationId / touchConversation / addParticipant", () => {
+    it("should set conversation_id without rewriting recipients", () => {
+      storage.putMessage(
+        makeMessage({
+          id: "m1",
+          recipients: [
+            { agent_id: "bob", kind: "to" },
+            { agent_id: "carol", kind: "cc" },
+          ],
+        })
+      );
+
+      storage.setMessageConversationId("m1", "conv-99");
+      const m = storage.getMessage("m1")!;
+      expect(m.conversation_id).toBe("conv-99");
+      expect(m.recipients).toHaveLength(2);
+    });
+
+    it("should touchConversation without disturbing participants", () => {
+      storage.putConversation({
+        id: "conv-1", scope: "default", status: "active",
+        participants: [
+          { agent_id: "alice", joined_at: "2025-01-01T00:00:00Z" },
+        ],
+        metadata: {}, created_at: "2025-01-01T00:00:00Z",
+        updated_at: "2025-01-01T00:00:00Z",
+      });
+
+      storage.touchConversation("conv-1", "2025-02-01T00:00:00Z");
+      const c = storage.getConversation("conv-1")!;
+      expect(c.updated_at).toBe("2025-02-01T00:00:00Z");
+      expect(c.participants).toHaveLength(1);
+    });
+
+    it("should addParticipant idempotently", () => {
+      storage.putConversation({
+        id: "conv-1", scope: "default", status: "active",
+        participants: [], metadata: {},
+        created_at: "2025-01-01T00:00:00Z",
+        updated_at: "2025-01-01T00:00:00Z",
+      });
+
+      storage.addParticipant("conv-1", {
+        agent_id: "alice",
+        joined_at: "2025-01-01T00:00:00Z",
+      });
+      storage.addParticipant("conv-1", {
+        agent_id: "alice",
+        joined_at: "2025-01-02T00:00:00Z",
+      });
+
+      const c = storage.getConversation("conv-1")!;
+      expect(c.participants).toHaveLength(1);
+    });
+  });
+
+  describe("pruneMessagesOlderThan", () => {
+    it("should remove old messages, recipients, turns and orphan threads", () => {
+      storage.putConversation({
+        id: "c1", scope: "default", status: "active", participants: [],
+        metadata: {}, created_at: "2024-01-01T00:00:00Z",
+        updated_at: "2026-01-01T00:00:00Z",
+      });
+      storage.putMessage(
+        makeMessage({ id: "old", created_at: "2024-01-01T00:00:00Z" })
+      );
+      storage.putMessage(
+        makeMessage({ id: "new", created_at: "2026-01-01T00:00:00Z" })
+      );
+
+      storage.addTurn({
+        id: "turn-old", conversation_id: "c1", participant_id: "alice",
+        source_message_id: "old", content_type: "text",
+        content: { type: "text", text: "x" },
+        created_at: "2024-01-01T00:00:00Z",
+      });
+      storage.addTurn({
+        id: "turn-new", conversation_id: "c1", participant_id: "alice",
+        source_message_id: "new", content_type: "text",
+        content: { type: "text", text: "y" },
+        created_at: "2026-01-01T00:00:00Z",
+      });
+      storage.putThread({
+        id: "thread-old", conversation_id: "c1",
+        root_turn_id: "turn-old",
+        created_at: "2024-01-01T00:00:00Z",
+      });
+      storage.putThread({
+        id: "thread-new", conversation_id: "c1",
+        root_turn_id: "turn-new",
+        created_at: "2026-01-01T00:00:00Z",
+      });
+
+      const removed = storage.pruneMessagesOlderThan("2025-01-01T00:00:00Z");
+      expect(removed).toBe(1);
+      expect(storage.getMessage("old")).toBeUndefined();
+      expect(storage.getMessage("new")).toBeDefined();
+
+      const turns = storage.getTurns("c1").map((t) => t.id);
+      expect(turns).toEqual(["turn-new"]);
+
+      const threads = storage.getThreadsByConversation("c1").map((t) => t.id);
+      expect(threads).toEqual(["thread-new"]);
+    });
+  });
 });

package/test/storage.test.ts

@@ -193,4 +193,96 @@ describe("InMemoryStorage", () => {
       expect(storage.getTurns("conv-2")).toHaveLength(0);
     });
   });
+
+  describe("setMessageConversationId", () => {
+    it("should set conversation_id on an existing message", () => {
+      storage.putMessage(makeMessage({ id: "m1" }));
+      storage.setMessageConversationId("m1", "conv-99");
+      expect(storage.getMessage("m1")?.conversation_id).toBe("conv-99");
+    });
+
+    it("should be a no-op for unknown message", () => {
+      expect(() =>
+        storage.setMessageConversationId("missing", "conv-99")
+      ).not.toThrow();
+    });
+  });
+
+  describe("touchConversation", () => {
+    it("should bump updated_at without disturbing participants", () => {
+      const conv: Conversation = {
+        id: "conv-1",
+        scope: "default",
+        status: "active",
+        participants: [
+          { agent_id: "alice", joined_at: "2025-01-01T00:00:00Z" },
+        ],
+        metadata: {},
+        created_at: "2025-01-01T00:00:00Z",
+        updated_at: "2025-01-01T00:00:00Z",
+      };
+      storage.putConversation(conv);
+
+      storage.touchConversation("conv-1", "2025-02-01T00:00:00Z");
+      const updated = storage.getConversation("conv-1");
+      expect(updated?.updated_at).toBe("2025-02-01T00:00:00Z");
+      expect(updated?.participants).toHaveLength(1);
+    });
+  });
+
+  describe("addParticipant", () => {
+    it("should add a participant once and ignore duplicates", () => {
+      storage.putConversation({
+        id: "conv-1", scope: "default", status: "active",
+        participants: [], metadata: {},
+        created_at: "2025-01-01T00:00:00Z",
+        updated_at: "2025-01-01T00:00:00Z",
+      });
+
+      storage.addParticipant("conv-1", {
+        agent_id: "alice",
+        joined_at: "2025-01-01T00:00:00Z",
+      });
+      storage.addParticipant("conv-1", {
+        agent_id: "alice",
+        joined_at: "2025-01-02T00:00:00Z",
+      });
+
+      const conv = storage.getConversation("conv-1");
+      expect(conv?.participants).toHaveLength(1);
+      expect(conv?.participants[0].agent_id).toBe("alice");
+    });
+  });
+
+  describe("pruneMessagesOlderThan", () => {
+    it("should remove messages and their turns older than cutoff", () => {
+      storage.putMessage(makeMessage({ id: "old", created_at: "2024-01-01T00:00:00Z" }));
+      storage.putMessage(makeMessage({ id: "new", created_at: "2026-01-01T00:00:00Z" }));
+      storage.addTurn({
+        id: "turn-old",
+        conversation_id: "c1",
+        participant_id: "alice",
+        source_message_id: "old",
+        content_type: "text",
+        content: { type: "text", text: "x" },
+        created_at: "2024-01-01T00:00:00Z",
+      });
+      storage.addTurn({
+        id: "turn-new",
+        conversation_id: "c1",
+        participant_id: "alice",
+        source_message_id: "new",
+        content_type: "text",
+        content: { type: "text", text: "y" },
+        created_at: "2026-01-01T00:00:00Z",
+      });
+
+      const removed = storage.pruneMessagesOlderThan("2025-01-01T00:00:00Z");
+      expect(removed).toBe(1);
+      expect(storage.getMessage("old")).toBeUndefined();
+      expect(storage.getMessage("new")).toBeDefined();
+      const turns = storage.getTurns("c1");
+      expect(turns.map((t) => t.id)).toEqual(["turn-new"]);
+    });
+  });
 });

package/vitest.bench.config.ts

@@ -0,0 +1,8 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  test: {
+    include: ["bench/**/*.bench.ts"],
+    testTimeout: 120_000,
+  },
+});