@axiom-lattice/gateway 2.1.48 → 2.1.49

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.48",
+  "version": "2.1.49",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -38,8 +38,8 @@
     "pg": "^8.11.0",
     "redis": "^5.0.1",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.42",
-    "@axiom-lattice/pg-stores": "1.0.32",
+    "@axiom-lattice/core": "2.1.43",
+    "@axiom-lattice/pg-stores": "1.0.33",
     "@axiom-lattice/protocols": "2.1.23",
     "@axiom-lattice/queue-redis": "1.0.22"
   },

package/src/controllers/run.ts

@@ -169,7 +169,6 @@ export const resumeStream = async (
 
     try {
       // Create Agent instance for accessing chunk buffer
-      // Note: assistant_id, tenant_id are not required for chunkStream
       // Get or create Agent instance (ensures single instance per thread)
       const agent = agentInstanceManager.getAgent({
         assistant_id,
@@ -180,7 +179,6 @@ export const resumeStream = async (
       });
 
       // Get the stream from agent's chunk buffer
-      console.log("[UI]message_id", message_id)
       const stream = agent.chunkStream(message_id, [MessageChunkTypes.THREAD_IDLE]);
 
       // Stream the chunks to the client

package/src/index.ts

@@ -19,6 +19,7 @@ import {
   sandboxLatticeManager,
   sqlDatabaseManager,
   getStoreLattice,
+  agentInstanceManager,
 } from "@axiom-lattice/core";
 import type { DatabaseConfigStore } from "@axiom-lattice/protocols";
 import {
@@ -260,6 +261,16 @@ const start = async (config?: LatticeGatewayConfig) => {
         agentTaskConsumer.startPollingQueue();
       }
     }
+
+    // Restore agent instances with pending messages
+    try {
+      logger.info("Starting agent instance recovery...");
+      const restoreStats = await agentInstanceManager.restore();
+      logger.info(`Agent recovery complete: ${restoreStats.restored} threads restored, ${restoreStats.errors} errors`);
+    } catch (error) {
+      logger.error("Agent recovery failed", { error });
+      // Don't exit - server can still function even if recovery fails
+    }
   } catch (err) {
     logger.error("Server start failed", { error: err });
     process.exit(1);

package/src/services/agent_task_consumer.ts

@@ -38,9 +38,9 @@ const handleAgentTask = async (
     );
 
     // 检查API服务器是否可用
-    const apiUrl = AgentTaskConsumer.agent_run_endpoint;
+    //    const apiUrl = AgentTaskConsumer.agent_run_endpoint;
 
-    console.log(`apiUrl: ${apiUrl}`);
+    //   console.log(`apiUrl: ${apiUrl}`);
     const agent = agentInstanceManager.getAgent({ assistant_id, thread_id, tenant_id, workspace_id: runConfig?.workspaceId, project_id: runConfig?.projectId, custom_run_config: runConfig })
     await agent.addMessage({ input, command }, QueueMode.STEER);
     if (callback_event) {
@@ -51,6 +51,13 @@ const handleAgentTask = async (
           config: { assistant_id, thread_id, tenant_id },
         });
       })
+      agent.subscribeOnce("message:interrupted", (evt) => {
+        eventBus.publish(callback_event, {
+          success: true,
+          state: evt.state,
+          config: { assistant_id, thread_id, tenant_id },
+        });
+      })
     }
 
     return true;