@axiom-lattice/pg-stores 3.1.11 → 3.1.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "3.1.11",
+  "version": "3.1.13",
   "description": "PG stores implementation for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -25,8 +25,8 @@
     "@langchain/core": "1.1.30",
     "pg": "^8.16.3",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "4.2.11",
-    "@axiom-lattice/protocols": "4.2.2"
+    "@axiom-lattice/core": "4.2.13",
+    "@axiom-lattice/protocols": "4.2.3"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/__tests__/PostgreSQLProjectRoomMessageStore.test.ts

@@ -114,4 +114,29 @@ describe("PostgreSQLProjectRoomMessageStore", () => {
     );
     warn.mockRestore();
   });
+
+  it("deletes one bounded batch of old chat messages and returns the row count", async () => {
+    query.mockResolvedValueOnce({ rows: [], rowCount: 3 });
+    const store = new PostgreSQLProjectRoomMessageStore({ pool });
+
+    await expect(store.deleteChatOlderThan({ before: now, limit: 500 })).resolves.toBe(3);
+
+    const [sql, params] = query.mock.calls[0] as [string, unknown[]];
+    expect(sql).toContain("DELETE FROM lattice_project_room_messages");
+    expect(sql).toContain("ctid IN");
+    expect(sql).toContain("source IN ('user','agent')");
+    expect(sql).toContain("ORDER BY created_at ASC");
+    expect(params).toEqual([now, 500]);
+  });
+
+  it("rejects an invalid retention horizon and clamps the delete batch", async () => {
+    const store = new PostgreSQLProjectRoomMessageStore({ pool });
+
+    await expect(store.deleteChatOlderThan({ before: new Date("invalid"), limit: 10 }))
+      .rejects.toBeInstanceOf(RangeError);
+
+    query.mockResolvedValueOnce({ rows: [], rowCount: 0 });
+    await store.deleteChatOlderThan({ before: now, limit: 99_999 });
+    expect(query.mock.calls[0][1]).toEqual([now, 1_000]);
+  });
 });

package/src/createPgStoreConfig.ts

@@ -87,6 +87,7 @@ import { createAgentWebAppsTable } from "./migrations/agent_web_apps_migration";
 import { addAgentWebAppsIdentity } from "./migrations/add_agent_web_apps_identity_migration";
 import { createCapabilityBundlesTable } from "./migrations/capability_bundle_migration";
 import { createProjectRoomReadStates, createProjectRoomTables } from "./migrations/project_room_migration";
+import { addProjectRoomMessageRetentionIndex } from "./migrations/project_room_message_retention_migration";
 import { createUserPushSubscriptions } from "./migrations/user_push_subscription_migration";
 import { PostgreSQLCapabilityBundleStore } from "./stores/PostgreSQLCapabilityBundleStore";
 import { PostgreSQLProjectRoomStore } from "./stores/PostgreSQLProjectRoomStore";
@@ -167,6 +168,7 @@ export async function createPgStoreConfig(connectionString: string) {
   mm.register(addAgentWebAppsIdentity);         // v176
   mm.register(createProjectRoomReadStates);     // v177
   mm.register(createUserPushSubscriptions);     // v178
+  mm.register(addProjectRoomMessageRetentionIndex); // v179
 
   await mm.migrate();
 

package/src/migrations/project_room_message_retention_migration.ts

@@ -0,0 +1,23 @@
+import type { Migration } from "./migration";
+
+/**
+ * Adds a partial `created_at` index for the room chat-message retention sweep.
+ *
+ * The maintenance job deletes `source IN ('user','agent')` rows older than a horizon;
+ * the partial index keeps that global (cross-room) scan bounded without affecting
+ * task/system event rows.
+ */
+export const addProjectRoomMessageRetentionIndex: Migration = {
+  version: 179,
+  name: "add_project_room_message_retention_index",
+  up: async (client) => {
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_project_room_messages_chat_created
+      ON lattice_project_room_messages (created_at)
+      WHERE source IN ('user', 'agent')
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_lattice_project_room_messages_chat_created");
+  },
+};

package/src/stores/PostgreSQLProjectRoomMessageStore.ts

@@ -205,6 +205,31 @@ export class PostgreSQLProjectRoomMessageStore implements ProjectRoomMessageStor
     return result.rows[0]?.count ?? 0;
   }
 
+  /**
+   * Deletes one bounded batch of chat messages older than a horizon.
+   *
+   * @param input Exclusive upper time bound and requested batch size (clamped to 1..1000).
+   * @returns The number of deleted rows.
+   * @remarks Only `source IN ('user','agent')` rows are eligible; task/system events are retained.
+   */
+  async deleteChatOlderThan(input: { before: Date; limit: number }): Promise<number> {
+    if (!(input.before instanceof Date) || Number.isNaN(input.before.getTime())) {
+      throw new RangeError("Project room message retention horizon date is invalid");
+    }
+    const limit = Math.min(1_000, Math.max(1, Math.trunc(Number.isFinite(input.limit) ? input.limit : 1)));
+    const result = await this.pool.query(
+      `DELETE FROM lattice_project_room_messages
+       WHERE ctid IN (
+         SELECT ctid FROM lattice_project_room_messages
+         WHERE created_at < $1 AND source IN ('user','agent')
+         ORDER BY created_at ASC
+         LIMIT $2
+       )`,
+      [input.before, limit],
+    );
+    return result.rowCount ?? 0;
+  }
+
   private parameters(input: Omit<ProjectRoomMessage, "createdAt">): unknown[] {
     return [input.id, input.tenantId, input.workspaceId, input.projectId, input.roomId, JSON.stringify(input.author), JSON.stringify(input.content),
       JSON.stringify(input.mentions), input.replyToMessageId ?? null, input.source, input.sourceId ?? null, input.idempotencyKey ?? null];