@axiom-lattice/pg-stores 1.0.34 → 1.0.35

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/pg-stores@1.0.34 build /home/runner/work/agentic/agentic/packages/pg-stores
+> @axiom-lattice/pg-stores@1.0.35 build /home/runner/work/agentic/agentic/packages/pg-stores
 > tsup src/index.ts --format cjs,esm --dts --sourcemap
 
 CLI Building entry: src/index.ts
@@ -8,13 +8,13 @@
 CLI Target: es2020
 CJS Build start
 ESM Build start
-CJS dist/index.js     132.35 KB
-CJS dist/index.js.map 245.86 KB
-CJS ⚡️ Build success in 373ms
-ESM dist/index.mjs     128.40 KB
-ESM dist/index.mjs.map 245.79 KB
-ESM ⚡️ Build success in 377ms
+ESM dist/index.mjs     129.31 KB
+ESM dist/index.mjs.map 247.58 KB
+ESM ⚡️ Build success in 343ms
+CJS dist/index.js     133.26 KB
+CJS dist/index.js.map 247.65 KB
+CJS ⚡️ Build success in 357ms
 DTS Build start
-DTS ⚡️ Build success in 11712ms
-DTS dist/index.d.ts  34.60 KB
-DTS dist/index.d.mts 34.60 KB
+DTS ⚡️ Build success in 11746ms
+DTS dist/index.d.ts  34.53 KB
+DTS dist/index.d.mts 34.53 KB

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @axiom-lattice/pg-stores
 
+## 1.0.35
+
+### Patch Changes
+
+- 4c4ca60: fix subagent state issue
+- Updated dependencies [4c4ca60]
+  - @axiom-lattice/core@2.1.45
+
 ## 1.0.34
 
 ### Patch Changes

package/dist/index.d.mts

@@ -50,7 +50,7 @@ declare class PostgreSQLThreadStore implements ThreadStore {
     /**
      * Get all threads for a specific tenant and assistant
      */
-    getThreadsByAssistantId(tenantId: string, assistantId: string): Promise<Thread[]>;
+    getThreadsByAssistantId(tenantId: string, assistantId: string, metadataFilter?: Record<string, string>): Promise<Thread[]>;
     /**
      * Get a thread by ID for a specific tenant
      */
@@ -1113,7 +1113,7 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      * Add message at head of queue (high priority, e.g., STEER/Command messages)
      * Uses priority=100 to ensure message is processed first
      */
-    addMessageAtHead(threadId: string, content: PendingMessage["content"], type?: "human" | "system", id?: string, command?: PendingMessage["command"]): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     /**
      * Get pending messages for thread
      */

package/dist/index.d.ts

@@ -50,7 +50,7 @@ declare class PostgreSQLThreadStore implements ThreadStore {
     /**
      * Get all threads for a specific tenant and assistant
      */
-    getThreadsByAssistantId(tenantId: string, assistantId: string): Promise<Thread[]>;
+    getThreadsByAssistantId(tenantId: string, assistantId: string, metadataFilter?: Record<string, string>): Promise<Thread[]>;
     /**
      * Get a thread by ID for a specific tenant
      */
@@ -1113,7 +1113,7 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      * Add message at head of queue (high priority, e.g., STEER/Command messages)
      * Uses priority=100 to ensure message is processed first
      */
-    addMessageAtHead(threadId: string, content: PendingMessage["content"], type?: "human" | "system", id?: string, command?: PendingMessage["command"]): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     /**
      * Get pending messages for thread
      */

package/dist/index.js

@@ -408,7 +408,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(
       `
@@ -419,7 +419,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
@@ -4105,6 +4113,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,
@@ -4153,6 +4178,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;
@@ -4169,7 +4195,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 
@@ -4179,10 +4205,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 || import_crypto.default.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, priority, command ? JSON.stringify(command) : null]
+      [id || import_crypto.default.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]);
   }
@@ -4190,21 +4216,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 
@@ -4214,10 +4230,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 || import_crypto.default.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, command ? JSON.stringify(command) : null]
+      [id || import_crypto.default.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]);
   }
@@ -4351,7 +4367,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
     };
   }
 };