@axiom-lattice/pg-stores 1.0.32 → 1.0.34

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/pg-stores@1.0.32 build /home/runner/work/agentic/agentic/packages/pg-stores
+> @axiom-lattice/pg-stores@1.0.34 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
-ESM dist/index.mjs     126.66 KB
-ESM dist/index.mjs.map 238.53 KB
-ESM ⚡️ Build success in 369ms
-CJS dist/index.js     129.98 KB
-CJS dist/index.js.map 242.69 KB
-CJS ⚡️ Build success in 385ms
+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
 DTS Build start
-DTS ⚡️ Build success in 14418ms
-DTS dist/index.d.ts  34.21 KB
-DTS dist/index.d.mts 34.21 KB
+DTS ⚡️ Build success in 11712ms
+DTS dist/index.d.ts  34.60 KB
+DTS dist/index.d.mts 34.60 KB

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @axiom-lattice/pg-stores
 
+## 1.0.34
+
+### Patch Changes
+
+- e8fc7ec: fix subagent issue
+- Updated dependencies [e8fc7ec]
+  - @axiom-lattice/core@2.1.44
+
+## 1.0.33
+
+### Patch Changes
+
+- 21052a9: update issues
+- Updated dependencies [21052a9]
+  - @axiom-lattice/core@2.1.43
+
 ## 1.0.32
 
 ### 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,9 +1110,10 @@ 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>;
+    addMessageAtHead(threadId: string, content: PendingMessage["content"], type?: "human" | "system", id?: string, command?: PendingMessage["command"]): Promise<PendingMessage>;
     /**
      * Get pending messages for thread
      */
@@ -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,9 +1110,10 @@ 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>;
+    addMessageAtHead(threadId: string, content: PendingMessage["content"], type?: "human" | "system", id?: string, command?: PendingMessage["command"]): Promise<PendingMessage>;
     /**
      * Get pending messages for thread
      */
@@ -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

@@ -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
@@ -23,6 +33,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 +71,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 +4035,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,32 +4051,33 @@ 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");
   }
 };
 
 // src/stores/ThreadMessageQueueStore.ts
+var import_crypto = __toESM(require("crypto"));
 var import_core4 = require("@axiom-lattice/core");
 
 // src/migrations/add_priority_command_columns.ts
@@ -4072,22 +4086,47 @@ 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");
+  }
+};
+
+// 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
+    `);
   }
 };
 
@@ -4114,6 +4153,7 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
       const migrationManager = new MigrationManager(pool);
       migrationManager.register(createThreadMessageQueueTable);
       migrationManager.register(addPriorityAndCommandColumns);
+      migrationManager.register(alterMessageQueueIdColumn);
       await migrationManager.migrate();
       this.initialized = true;
     }
@@ -4129,65 +4169,57 @@ 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, id } = 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 
-       (thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
+      `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)
        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]
     );
     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") {
+  async addMessageAtHead(threadId, content, type = "system", id, command) {
     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 
+       (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)
+       RETURNING *`,
+      [id || import_crypto.default.randomUUID(), threadId, tenantId, assistantId, JSON.stringify(content), type, nextSeq, command ? JSON.stringify(command) : null]
+    );
+    return this.rowToMessage(result.rows[0]);
   }
   /**
    * Get pending messages for thread
@@ -4195,9 +4227,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 +4240,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 +4253,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 +4282,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 +4293,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 +4303,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 +4313,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 +4323,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 +4361,8 @@ var getThreadMessageQueueStore = () => ThreadMessageQueueStore.getInstance();
   AddMessageParams,
   MigrationManager,
   PendingMessage,
+  Pool,
+  PoolConfig,
   PostgreSQLAssistantStore,
   PostgreSQLDatabaseConfigStore,
   PostgreSQLMcpServerConfigStore,