@axiom-lattice/pg-stores 1.0.34 → 1.0.36

package/dist/index.mjs

@@ -336,7 +336,7 @@ var PostgreSQLThreadStore = class {
   /**
    * Get all threads for a specific tenant and assistant
    */
-  async getThreadsByAssistantId(tenantId, assistantId) {
+  async getThreadsByAssistantId(tenantId, assistantId, metadataFilter) {
     await this.ensureInitialized();
     const result = await this.pool.query(
       `
@@ -347,7 +347,15 @@ var PostgreSQLThreadStore = class {
     `,
       [tenantId, assistantId]
     );
-    return result.rows.map(this.mapRowToThread);
+    let threads = result.rows.map(this.mapRowToThread);
+    if (metadataFilter && Object.keys(metadataFilter).length > 0) {
+      threads = threads.filter(
+        (thread) => Object.entries(metadataFilter).every(
+          ([key, value]) => thread.metadata?.[key] === value
+        )
+      );
+    }
+    return threads;
   }
   /**
    * Get a thread by ID for a specific tenant
@@ -4039,6 +4047,23 @@ var addPriorityAndCommandColumns = {
   }
 };
 
+// src/migrations/add_custom_run_config_column.ts
+var addCustomRunConfigColumn = {
+  version: 102,
+  name: "add_custom_run_config_column",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_thread_message_queue
+      ADD COLUMN IF NOT EXISTS custom_run_config JSONB
+    `);
+  },
+  down: async (client) => {
+    await client.query(
+      "ALTER TABLE lattice_thread_message_queue DROP COLUMN IF EXISTS custom_run_config"
+    );
+  }
+};
+
 // src/migrations/alter_message_queue_id_column.ts
 var alterMessageQueueIdColumn = {
   version: 102,
@@ -4087,6 +4112,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
       const migrationManager = new MigrationManager(pool);
       migrationManager.register(createThreadMessageQueueTable);
       migrationManager.register(addPriorityAndCommandColumns);
+      migrationManager.register(addCustomRunConfigColumn);
       migrationManager.register(alterMessageQueueIdColumn);
       await migrationManager.migrate();
       this.initialized = true;
@@ -4103,7 +4129,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
    */
   async addMessage(params) {
     const pool = this.getPool();
-    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command, id } = params;
+    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command, custom_run_config, id } = params;
     const seqResult = await pool.query(
       `SELECT COALESCE(MAX(sequence_order), 0) + 1 as next_seq
        FROM lattice_thread_message_queue 
@@ -4113,10 +4139,10 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
     const nextSeq = seqResult.rows[0].next_seq;
     const result = await pool.query(
       `INSERT INTO lattice_thread_message_queue 
-       (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
+       (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command, custom_run_config)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
        RETURNING *`,
-      [id || crypto.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, priority, command ? JSON.stringify(command) : null]
+      [id || crypto.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, priority, command ? JSON.stringify(command) : null, custom_run_config ? JSON.stringify(custom_run_config) : null]
     );
     return this.rowToMessage(result.rows[0]);
   }
@@ -4124,21 +4150,11 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
    * Add message at head of queue (high priority, e.g., STEER/Command messages)
    * Uses priority=100 to ensure message is processed first
    */
-  async addMessageAtHead(threadId, content, type = "system", id, command) {
+  async addMessageAtHead(params) {
     const pool = this.getPool();
-    const threadResult = await pool.query(
-      `SELECT tenant_id, assistant_id 
-       FROM lattice_thread_message_queue 
-       WHERE thread_id = $1 
-       LIMIT 1`,
-      [threadId]
-    );
-    let tenantId = "default";
-    let assistantId = "";
-    if (threadResult.rows.length > 0) {
-      tenantId = threadResult.rows[0].tenant_id;
-      assistantId = threadResult.rows[0].assistant_id;
-    }
+    const { threadId, tenantId, assistantId, content, type = "human", command, custom_run_config, id } = params;
+    const resolvedTenantId = tenantId;
+    const resolvedAssistantId = assistantId;
     const seqResult = await pool.query(
       `SELECT COALESCE(MAX(sequence_order), 0) + 1 as next_seq
        FROM lattice_thread_message_queue 
@@ -4148,10 +4164,10 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
     const nextSeq = seqResult.rows[0].next_seq;
     const result = await pool.query(
       `INSERT INTO lattice_thread_message_queue 
-       (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, 100, $8)
+       (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command, custom_run_config)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, 100, $8, $9)
        RETURNING *`,
-      [id || crypto.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, command ? JSON.stringify(command) : null]
+      [id || crypto.randomUUID(), threadId, resolvedTenantId, resolvedAssistantId, JSON.stringify(content), type, nextSeq, command ? JSON.stringify(command) : null, custom_run_config ? JSON.stringify(custom_run_config) : null]
     );
     return this.rowToMessage(result.rows[0]);
   }
@@ -4285,7 +4301,8 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
       sequence: row.sequence_order,
       createdAt: new Date(row.created_at),
       priority: row.priority || 0,
-      command: row.command ? typeof row.command === "string" ? JSON.parse(row.command) : row.command : void 0
+      command: row.command ? typeof row.command === "string" ? JSON.parse(row.command) : row.command : void 0,
+      custom_run_config: row.custom_run_config ? typeof row.custom_run_config === "string" ? JSON.parse(row.custom_run_config) : row.custom_run_config : void 0
     };
   }
 };