@axiom-lattice/pg-stores 1.0.31 → 1.0.33

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/pg-stores@1.0.31 build /home/runner/work/agentic/agentic/packages/pg-stores
+> @axiom-lattice/pg-stores@1.0.33 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     129.98 KB
-CJS dist/index.js.map 242.69 KB
-CJS ⚡️ Build success in 259ms
-ESM dist/index.mjs     126.66 KB
-ESM dist/index.mjs.map 238.53 KB
-ESM ⚡️ Build success in 259ms
+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
 DTS Build start
-DTS ⚡️ Build success in 11629ms
-DTS dist/index.d.ts  34.21 KB
-DTS dist/index.d.mts 34.21 KB
+DTS ⚡️ Build success in 11993ms
+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.33
+
+### Patch Changes
+
+- 21052a9: update issues
+- Updated dependencies [21052a9]
+  - @axiom-lattice/core@2.1.43
+
+## 1.0.32
+
+### Patch Changes
+
+- Updated dependencies [9a4f338]
+  - @axiom-lattice/protocols@2.1.23
+  - @axiom-lattice/core@2.1.42
+
 ## 1.0.31
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,4 +1,5 @@
 import { PoolConfig, PoolClient, Pool } from 'pg';
+export { Pool, PoolConfig } from 'pg';
 import { ThreadStore, Thread, CreateThreadRequest, AssistantStore, Assistant, CreateAssistantRequest, ScheduleStorage, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, SkillStore, Skill, CreateSkillRequest, DatabaseConfigStore, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, MetricsServerConfigStore, MetricsServerConfigEntry, CreateMetricsServerConfigRequest, UpdateMetricsServerConfigRequest, McpServerConfigStore, McpServerConfigEntry, CreateMcpServerConfigRequest, UpdateMcpServerConfigRequest, WorkspaceStore, Workspace, CreateWorkspaceRequest, UpdateWorkspaceRequest, ProjectStore, Project, CreateProjectRequest, UpdateProjectRequest, UserStore, User, CreateUserRequest, UpdateUserRequest, TenantStore, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLinkStore, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest } from '@axiom-lattice/protocols';
 export { Assistant, AssistantStore, CreateAssistantRequest, CreateDatabaseConfigRequest, CreateMcpServerConfigRequest, CreateMetricsServerConfigRequest, CreateProjectRequest, CreateSkillRequest, CreateTenantRequest, CreateThreadRequest, CreateUserRequest, CreateUserTenantLinkRequest, CreateWorkspaceRequest, DatabaseConfig, DatabaseConfigEntry, DatabaseConfigStore, DatabaseType, McpServerConfigEntry, McpServerConfigStore, MetricsServerConfig, MetricsServerConfigEntry, MetricsServerConfigStore, Project, ProjectStore, ScheduleExecutionType, ScheduleStorage, ScheduledTaskDefinition, ScheduledTaskStatus, Skill, SkillStore, StorageType, Tenant, TenantStatus, TenantStore, Thread, ThreadStore, UpdateDatabaseConfigRequest, UpdateMcpServerConfigRequest, UpdateMetricsServerConfigRequest, UpdateProjectRequest, UpdateTenantRequest, UpdateUserRequest, UpdateUserTenantLinkRequest, UpdateWorkspaceRequest, User, UserStore, UserTenantLink, UserTenantLinkStore, UserTenantRole, Workspace, WorkspaceStore } from '@axiom-lattice/protocols';
 import { IMessageQueueStore, AddMessageParams, PendingMessage, ThreadInfo } from '@axiom-lattice/core';
@@ -1109,7 +1110,8 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      */
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
     /**
-     * Add message at head of queue (sequence = 0, shift others)
+     * 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>;
     /**
@@ -1125,7 +1127,7 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      */
     getQueueSize(threadId: string): Promise<number>;
     /**
-     * Get all threads with pending messages
+     * Get all threads with pending or processing messages
      */
     getThreadsWithPendingMessages(): Promise<ThreadInfo[]>;
     /**
@@ -1148,6 +1150,11 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      * Clear completed messages for thread
      */
     clearCompletedMessages(threadId: string): Promise<void>;
+    /**
+     * Reset all processing messages to pending state for a thread
+     * Returns the number of messages reset
+     */
+    resetProcessingToPending(threadId: string): Promise<number>;
     private rowToMessage;
 }
 declare const getThreadMessageQueueStore: () => ThreadMessageQueueStore;

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 import { PoolConfig, PoolClient, Pool } from 'pg';
+export { Pool, PoolConfig } from 'pg';
 import { ThreadStore, Thread, CreateThreadRequest, AssistantStore, Assistant, CreateAssistantRequest, ScheduleStorage, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, SkillStore, Skill, CreateSkillRequest, DatabaseConfigStore, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, MetricsServerConfigStore, MetricsServerConfigEntry, CreateMetricsServerConfigRequest, UpdateMetricsServerConfigRequest, McpServerConfigStore, McpServerConfigEntry, CreateMcpServerConfigRequest, UpdateMcpServerConfigRequest, WorkspaceStore, Workspace, CreateWorkspaceRequest, UpdateWorkspaceRequest, ProjectStore, Project, CreateProjectRequest, UpdateProjectRequest, UserStore, User, CreateUserRequest, UpdateUserRequest, TenantStore, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLinkStore, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest } from '@axiom-lattice/protocols';
 export { Assistant, AssistantStore, CreateAssistantRequest, CreateDatabaseConfigRequest, CreateMcpServerConfigRequest, CreateMetricsServerConfigRequest, CreateProjectRequest, CreateSkillRequest, CreateTenantRequest, CreateThreadRequest, CreateUserRequest, CreateUserTenantLinkRequest, CreateWorkspaceRequest, DatabaseConfig, DatabaseConfigEntry, DatabaseConfigStore, DatabaseType, McpServerConfigEntry, McpServerConfigStore, MetricsServerConfig, MetricsServerConfigEntry, MetricsServerConfigStore, Project, ProjectStore, ScheduleExecutionType, ScheduleStorage, ScheduledTaskDefinition, ScheduledTaskStatus, Skill, SkillStore, StorageType, Tenant, TenantStatus, TenantStore, Thread, ThreadStore, UpdateDatabaseConfigRequest, UpdateMcpServerConfigRequest, UpdateMetricsServerConfigRequest, UpdateProjectRequest, UpdateTenantRequest, UpdateUserRequest, UpdateUserTenantLinkRequest, UpdateWorkspaceRequest, User, UserStore, UserTenantLink, UserTenantLinkStore, UserTenantRole, Workspace, WorkspaceStore } from '@axiom-lattice/protocols';
 import { IMessageQueueStore, AddMessageParams, PendingMessage, ThreadInfo } from '@axiom-lattice/core';
@@ -1109,7 +1110,8 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      */
     addMessage(params: AddMessageParams): Promise<PendingMessage>;
     /**
-     * Add message at head of queue (sequence = 0, shift others)
+     * 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>;
     /**
@@ -1125,7 +1127,7 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      */
     getQueueSize(threadId: string): Promise<number>;
     /**
-     * Get all threads with pending messages
+     * Get all threads with pending or processing messages
      */
     getThreadsWithPendingMessages(): Promise<ThreadInfo[]>;
     /**
@@ -1148,6 +1150,11 @@ declare class ThreadMessageQueueStore implements IMessageQueueStore {
      * Clear completed messages for thread
      */
     clearCompletedMessages(threadId: string): Promise<void>;
+    /**
+     * Reset all processing messages to pending state for a thread
+     * Returns the number of messages reset
+     */
+    resetProcessingToPending(threadId: string): Promise<number>;
     private rowToMessage;
 }
 declare const getThreadMessageQueueStore: () => ThreadMessageQueueStore;

package/dist/index.js

@@ -23,6 +23,8 @@ __export(index_exports, {
   AddMessageParams: () => import_core4.AddMessageParams,
   MigrationManager: () => MigrationManager,
   PendingMessage: () => import_core4.PendingMessage,
+  Pool: () => import_pg13.Pool,
+  PoolConfig: () => import_pg13.PoolConfig,
   PostgreSQLAssistantStore: () => PostgreSQLAssistantStore,
   PostgreSQLDatabaseConfigStore: () => PostgreSQLDatabaseConfigStore,
   PostgreSQLMcpServerConfigStore: () => PostgreSQLMcpServerConfigStore,
@@ -59,6 +61,7 @@ __export(index_exports, {
   getThreadMessageQueueStore: () => getThreadMessageQueueStore
 });
 module.exports = __toCommonJS(index_exports);
+var import_pg13 = require("pg");
 
 // src/stores/PostgreSQLThreadStore.ts
 var import_pg = require("pg");
@@ -4022,10 +4025,10 @@ var PostgreSQLUserTenantLinkStore = class {
 // src/migrations/thread_message_queue_migrations.ts
 var createThreadMessageQueueTable = {
   version: 100,
-  name: "create_thread_message_queue_table",
+  name: "create_lattice_thread_message_queue_table",
   up: async (client) => {
     await client.query(`
-      CREATE TABLE IF NOT EXISTS thread_message_queue (
+      CREATE TABLE IF NOT EXISTS lattice_thread_message_queue (
         id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
         thread_id VARCHAR(255) NOT NULL,
         tenant_id VARCHAR(255) NOT NULL,
@@ -4038,28 +4041,28 @@ var createThreadMessageQueueTable = {
       )
     `);
     await client.query(`
-      CREATE INDEX IF NOT EXISTS idx_thread_message_queue_thread 
-      ON thread_message_queue(thread_id, status)
+      CREATE INDEX IF NOT EXISTS idx_lattice_thread_message_queue_thread 
+      ON lattice_thread_message_queue(thread_id, status)
     `);
     await client.query(`
-      CREATE INDEX IF NOT EXISTS idx_thread_message_queue_tenant 
-      ON thread_message_queue(tenant_id, thread_id, status)
+      CREATE INDEX IF NOT EXISTS idx_lattice_thread_message_queue_tenant 
+      ON lattice_thread_message_queue(tenant_id, thread_id, status)
     `);
     await client.query(`
-      CREATE INDEX IF NOT EXISTS idx_thread_message_queue_status 
-      ON thread_message_queue(status, created_at)
+      CREATE INDEX IF NOT EXISTS idx_lattice_thread_message_queue_status 
+      ON lattice_thread_message_queue(status, created_at)
     `);
     await client.query(`
-      CREATE UNIQUE INDEX IF NOT EXISTS idx_thread_message_queue_sequence 
-      ON thread_message_queue(thread_id, sequence_order)
+      CREATE INDEX IF NOT EXISTS idx_lattice_thread_message_queue_sequence 
+      ON lattice_thread_message_queue(thread_id, sequence_order)
     `);
   },
   down: async (client) => {
-    await client.query("DROP INDEX IF EXISTS idx_thread_message_queue_sequence");
-    await client.query("DROP INDEX IF EXISTS idx_thread_message_queue_status");
-    await client.query("DROP INDEX IF EXISTS idx_thread_message_queue_tenant");
-    await client.query("DROP INDEX IF EXISTS idx_thread_message_queue_thread");
-    await client.query("DROP TABLE IF EXISTS thread_message_queue");
+    await client.query("DROP INDEX IF EXISTS idx_lattice_thread_message_queue_sequence");
+    await client.query("DROP INDEX IF EXISTS idx_lattice_thread_message_queue_status");
+    await client.query("DROP INDEX IF EXISTS idx_lattice_thread_message_queue_tenant");
+    await client.query("DROP INDEX IF EXISTS idx_lattice_thread_message_queue_thread");
+    await client.query("DROP TABLE IF EXISTS lattice_thread_message_queue");
   }
 };
 
@@ -4072,22 +4075,22 @@ var addPriorityAndCommandColumns = {
   name: "add_priority_and_command_columns",
   up: async (client) => {
     await client.query(`
-      ALTER TABLE thread_message_queue 
+      ALTER TABLE lattice_thread_message_queue 
       ADD COLUMN IF NOT EXISTS priority INTEGER DEFAULT 0
     `);
     await client.query(`
-      ALTER TABLE thread_message_queue 
+      ALTER TABLE lattice_thread_message_queue 
       ADD COLUMN IF NOT EXISTS command JSONB
     `);
     await client.query(`
-      CREATE INDEX IF NOT EXISTS idx_thread_message_queue_priority 
-      ON thread_message_queue(thread_id, priority DESC, sequence_order ASC)
+      CREATE INDEX IF NOT EXISTS idx_lattice_thread_message_queue_priority 
+      ON lattice_thread_message_queue(thread_id, priority DESC, sequence_order ASC)
     `);
   },
   down: async (client) => {
-    await client.query("DROP INDEX IF EXISTS idx_thread_message_queue_priority");
-    await client.query("ALTER TABLE thread_message_queue DROP COLUMN IF EXISTS command");
-    await client.query("ALTER TABLE thread_message_queue DROP COLUMN IF EXISTS priority");
+    await client.query("DROP INDEX IF EXISTS idx_lattice_thread_message_queue_priority");
+    await client.query("ALTER TABLE lattice_thread_message_queue DROP COLUMN IF EXISTS command");
+    await client.query("ALTER TABLE lattice_thread_message_queue DROP COLUMN IF EXISTS priority");
   }
 };
 
@@ -4132,13 +4135,13 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
     const { threadId, tenantId, assistantId, content, type = "human", priority = 0, command } = params;
     const seqResult = await pool.query(
       `SELECT COALESCE(MAX(sequence_order), 0) + 1 as next_seq
-       FROM thread_message_queue 
+       FROM lattice_thread_message_queue 
        WHERE thread_id = $1`,
       [threadId]
     );
     const nextSeq = seqResult.rows[0].next_seq;
     const result = await pool.query(
-      `INSERT INTO thread_message_queue 
+      `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)
        RETURNING *`,
@@ -4147,47 +4150,39 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
     return this.rowToMessage(result.rows[0]);
   }
   /**
-   * Add message at head of queue (sequence = 0, shift others)
+   * 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") {
     const pool = this.getPool();
-    const client = await pool.connect();
-    try {
-      await client.query("BEGIN");
-      await client.query(
-        `UPDATE thread_message_queue 
-         SET sequence_order = sequence_order + 1
-         WHERE thread_id = $1`,
-        [threadId]
-      );
-      const threadResult = await client.query(
-        `SELECT tenant_id, assistant_id 
-         FROM 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 result = await client.query(
-        `INSERT INTO thread_message_queue 
-         (thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order)
-         VALUES ($1, $2, $3, $4, $5, 0)
-         RETURNING *`,
-        [threadId, tenantId, assistantId, JSON.stringify(content), type]
-      );
-      await client.query("COMMIT");
-      return this.rowToMessage(result.rows[0]);
-    } catch (error) {
-      await client.query("ROLLBACK");
-      throw error;
-    } finally {
-      client.release();
+    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 seqResult = await pool.query(
+      `SELECT COALESCE(MAX(sequence_order), 0) + 1 as next_seq
+       FROM lattice_thread_message_queue 
+       WHERE thread_id = $1`,
+      [threadId]
+    );
+    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)
+       RETURNING *`,
+      [threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq]
+    );
+    return this.rowToMessage(result.rows[0]);
   }
   /**
    * Get pending messages for thread
@@ -4195,9 +4190,9 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async getPendingMessages(threadId) {
     const pool = this.getPool();
     const result = await pool.query(
-      `SELECT * FROM thread_message_queue 
+      `SELECT * FROM lattice_thread_message_queue 
        WHERE thread_id = $1 AND status = 'pending'
-       ORDER BY sequence_order ASC`,
+       ORDER BY priority DESC, sequence_order ASC`,
       [threadId]
     );
     return result.rows.map((row) => this.rowToMessage(row));
@@ -4208,9 +4203,9 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async getProcessingMessages(threadId) {
     const pool = this.getPool();
     const result = await pool.query(
-      `SELECT * FROM thread_message_queue 
+      `SELECT * FROM lattice_thread_message_queue 
        WHERE thread_id = $1 AND status = 'processing'
-       ORDER BY sequence_order ASC`,
+       ORDER BY priority DESC, sequence_order ASC`,
       [threadId]
     );
     return result.rows.map((row) => this.rowToMessage(row));
@@ -4221,21 +4216,21 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async getQueueSize(threadId) {
     const pool = this.getPool();
     const result = await pool.query(
-      `SELECT COUNT(*) as count FROM thread_message_queue 
+      `SELECT COUNT(*) as count FROM lattice_thread_message_queue 
        WHERE thread_id = $1 AND status = 'pending'`,
       [threadId]
     );
     return parseInt(result.rows[0].count, 10);
   }
   /**
-   * Get all threads with pending messages
+   * Get all threads with pending or processing messages
    */
   async getThreadsWithPendingMessages() {
     const pool = this.getPool();
     const result = await pool.query(
       `SELECT DISTINCT tenant_id, assistant_id, thread_id 
-       FROM thread_message_queue 
-       WHERE status = 'pending'
+       FROM lattice_thread_message_queue 
+       WHERE status IN ('pending', 'processing')
        ORDER BY thread_id`
     );
     return result.rows.map((row) => ({
@@ -4250,7 +4245,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async removeMessage(messageId) {
     const pool = this.getPool();
     const result = await pool.query(
-      `DELETE FROM thread_message_queue WHERE id = $1 RETURNING id`,
+      `DELETE FROM lattice_thread_message_queue WHERE id = $1 RETURNING id`,
       [messageId]
     );
     return (result.rowCount ?? 0) > 0;
@@ -4261,7 +4256,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async clearMessages(threadId) {
     const pool = this.getPool();
     await pool.query(
-      `DELETE FROM thread_message_queue WHERE thread_id = $1`,
+      `DELETE FROM lattice_thread_message_queue WHERE thread_id = $1`,
       [threadId]
     );
   }
@@ -4271,7 +4266,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async markProcessing(messageId) {
     const pool = this.getPool();
     await pool.query(
-      `UPDATE thread_message_queue SET status = 'processing' WHERE id = $1`,
+      `UPDATE lattice_thread_message_queue SET status = 'processing' WHERE id = $1`,
       [messageId]
     );
   }
@@ -4281,7 +4276,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async markCompleted(messageId) {
     const pool = this.getPool();
     await pool.query(
-      `UPDATE thread_message_queue SET status = 'completed' WHERE id = $1`,
+      `UPDATE lattice_thread_message_queue SET status = 'completed' WHERE id = $1`,
       [messageId]
     );
   }
@@ -4291,11 +4286,26 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
   async clearCompletedMessages(threadId) {
     const pool = this.getPool();
     await pool.query(
-      `DELETE FROM thread_message_queue 
+      `DELETE FROM lattice_thread_message_queue 
        WHERE thread_id = $1 AND status = 'completed'`,
       [threadId]
     );
   }
+  /**
+   * Reset all processing messages to pending state for a thread
+   * Returns the number of messages reset
+   */
+  async resetProcessingToPending(threadId) {
+    const pool = this.getPool();
+    const result = await pool.query(
+      `UPDATE lattice_thread_message_queue 
+       SET status = 'pending'
+       WHERE thread_id = $1 AND status = 'processing'
+       RETURNING id`,
+      [threadId]
+    );
+    return result.rowCount ?? 0;
+  }
   rowToMessage(row) {
     return {
       id: row.id,
@@ -4314,6 +4324,8 @@ var getThreadMessageQueueStore = () => ThreadMessageQueueStore.getInstance();
   AddMessageParams,
   MigrationManager,
   PendingMessage,
+  Pool,
+  PoolConfig,
   PostgreSQLAssistantStore,
   PostgreSQLDatabaseConfigStore,
   PostgreSQLMcpServerConfigStore,