@axiom-lattice/pg-stores 3.1.12 → 3.1.14

package/dist/index.mjs

@@ -6605,8 +6605,11 @@ function isScope(value) {
 function isFeatures(value) {
   return isRecord(value) && typeof value.projectSelector === "boolean" && typeof value.modelSelector === "boolean" && typeof value.threadManagement === "boolean" && typeof value.attachments === "boolean" && typeof value.hitl === "boolean" && typeof value.genUI === "boolean" && (value.files === void 0 || typeof value.files === "boolean");
 }
+function isQuickPrompt(value) {
+  return isRecord(value) && typeof value.label === "string" && typeof value.text === "string";
+}
 function isAppearance(value) {
-  return isRecord(value) && (value.title === void 0 || typeof value.title === "string") && (value.welcomeMessage === void 0 || typeof value.welcomeMessage === "string") && (value.primaryColor === void 0 || typeof value.primaryColor === "string");
+  return isRecord(value) && (value.title === void 0 || typeof value.title === "string") && (value.welcomeMessage === void 0 || typeof value.welcomeMessage === "string") && (value.primaryColor === void 0 || typeof value.primaryColor === "string") && (value.quickPrompts === void 0 || Array.isArray(value.quickPrompts) && value.quickPrompts.every(isQuickPrompt));
 }
 function isIdentity(value) {
   if (value === null || value === void 0) return true;
@@ -9884,6 +9887,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 +10747,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 +10948,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) => {