@axiom-lattice/pg-stores 3.1.12 → 3.1.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "3.1.12",
+  "version": "3.1.14",
   "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.12",
-    "@axiom-lattice/protocols": "4.2.3"
+    "@axiom-lattice/core": "4.2.14",
+    "@axiom-lattice/protocols": "4.3.0"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/__tests__/PostgreSQLAgentWebAppStore.test.ts

@@ -288,4 +288,22 @@ describe("PostgreSQLAgentWebAppStore", () => {
       "Invalid agent web app row",
     );
   });
+
+  it("round-trips appearance quickPrompts and rejects malformed entries", async () => {
+    const appearance = {
+      quickPrompts: [
+        { label: "Pricing", text: "How much does it cost?" },
+        { label: "Support", text: "How do I contact support?" },
+      ],
+    };
+    const store = new PostgreSQLAgentWebAppStore({ pool });
+
+    mockQuery.mockResolvedValueOnce({ rows: [{ ...row, appearance }] });
+    await expect(store.getById("tenant-1", "webapp_123")).resolves.toMatchObject({ appearance });
+
+    mockQuery.mockResolvedValueOnce({ rows: [{ ...row, appearance: { quickPrompts: [{ label: "missing text" }] } }] });
+    await expect(store.getById("tenant-1", "webapp_123")).rejects.toThrow(
+      "Invalid agent web app row",
+    );
+  });
 });

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/PostgreSQLAgentWebAppStore.ts

@@ -3,6 +3,7 @@ import type {
   AgentWebApp,
   AgentWebAppAppearance,
   AgentWebAppFeatures,
+  AgentWebAppQuickPrompt,
   AgentWebAppScope,
   AgentWebAppStatus,
   AgentWebAppStore,
@@ -70,11 +71,19 @@ function isFeatures(value: unknown): value is AgentWebAppFeatures {
     && (value.files === undefined || typeof value.files === "boolean");
 }
 
+function isQuickPrompt(value: unknown): value is AgentWebAppQuickPrompt {
+  return isRecord(value)
+    && typeof value.label === "string"
+    && typeof value.text === "string";
+}
+
 function isAppearance(value: unknown): value is AgentWebAppAppearance {
   return isRecord(value)
     && (value.title === undefined || typeof value.title === "string")
     && (value.welcomeMessage === undefined || typeof value.welcomeMessage === "string")
-    && (value.primaryColor === undefined || typeof value.primaryColor === "string");
+    && (value.primaryColor === undefined || typeof value.primaryColor === "string")
+    && (value.quickPrompts === undefined
+      || (Array.isArray(value.quickPrompts) && value.quickPrompts.every(isQuickPrompt)));
 }
 
 function isIdentity(value: unknown): boolean {

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];