@axiom-lattice/pg-stores 3.1.12 → 3.1.13

package/dist/index.mjs

@@ -9884,6 +9884,22 @@ var createProjectRoomReadStates = {
   }
 };
 
+// src/migrations/project_room_message_retention_migration.ts
+var addProjectRoomMessageRetentionIndex = {
+  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");
+  }
+};
+
 // src/migrations/user_push_subscription_migration.ts
 var createUserPushSubscriptions = {
   version: 178,
@@ -10728,6 +10744,30 @@ var PostgreSQLProjectRoomMessageStore = class {
     );
     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) {
+    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(1e3, 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;
+  }
   parameters(input) {
     return [
       input.id,
@@ -10905,6 +10945,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addAgentWebAppsIdentity);
   mm.register(createProjectRoomReadStates);
   mm.register(createUserPushSubscriptions);
+  mm.register(addProjectRoomMessageRetentionIndex);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {