@axiom-lattice/pg-stores 1.0.33 → 1.0.35

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/pg-stores@1.0.33 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     130.75 KB
-CJS dist/index.js.map 243.67 KB
-CJS ⚡️ Build success in 327ms
-ESM dist/index.mjs     127.43 KB
-ESM dist/index.mjs.map 243.62 KB
-ESM ⚡️ Build success in 329ms
+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 11993ms
+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,21 @@
 # @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
+
+- e8fc7ec: fix subagent issue
+- Updated dependencies [e8fc7ec]
+  - @axiom-lattice/core@2.1.44
+
 ## 1.0.33
 
 ### 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: any, type?: "human" | "system"): 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: any, type?: "human" | "system"): Promise<PendingMessage>;
+    addMessageAtHead(params: AddMessageParams): Promise<PendingMessage>;
     /**
      * Get pending messages for thread
      */

package/dist/index.js

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -398,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(
       `
@@ -409,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
@@ -4067,6 +4085,7 @@ var createThreadMessageQueueTable = {
 };
 
 // src/stores/ThreadMessageQueueStore.ts
+var import_crypto = __toESM(require("crypto"));
 var import_core4 = require("@axiom-lattice/core");
 
 // src/migrations/add_priority_command_columns.ts
@@ -4094,6 +4113,48 @@ 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,
+  name: "alter_message_queue_id_column",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_thread_message_queue 
+      ALTER COLUMN id DROP DEFAULT
+    `);
+    await client.query(`
+      COMMENT ON COLUMN lattice_thread_message_queue.id IS 
+      'Message ID - must be provided explicitly (UUID format). Used for consistent ID across client and server.'
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_thread_message_queue 
+      ALTER COLUMN id SET DEFAULT gen_random_uuid()
+    `);
+    await client.query(`
+      COMMENT ON COLUMN lattice_thread_message_queue.id IS NULL
+    `);
+  }
+};
+
 // src/stores/ThreadMessageQueueStore.ts
 var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   constructor() {
@@ -4117,6 +4178,8 @@ 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;
     }
@@ -4132,7 +4195,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
    */
   async addMessage(params) {
     const pool = this.getPool();
-    const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command } = 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 
@@ -4142,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 
-       (thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $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, $8, $9, $10)
        RETURNING *`,
-      [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]);
   }
@@ -4153,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") {
+  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 
@@ -4177,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 
-       (thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority)
-       VALUES ($1, $2, $3, $4, $5, $6, 100)
+       (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 *`,
-      [threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq]
+      [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]);
   }
@@ -4314,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
     };
   }
 };