@axiom-lattice/gateway 2.1.47 → 2.1.49

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.47",
+  "version": "2.1.49",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -38,10 +38,10 @@
     "pg": "^8.11.0",
     "redis": "^5.0.1",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.41",
-    "@axiom-lattice/pg-stores": "1.0.31",
-    "@axiom-lattice/protocols": "2.1.22",
-    "@axiom-lattice/queue-redis": "1.0.21"
+    "@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"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/controllers/memory.ts

@@ -167,9 +167,17 @@ export const getAgentState = async (
     }
 
     // 3. 合并返回
+    // Normalize pendingMessages content to string for consistent frontend format
+    const normalizedPendingMessages = pendingMessages.map(msg => ({
+      ...msg,
+      content: typeof msg.content === 'object' && msg.content !== null
+        ? (msg.content.message || JSON.stringify(msg.content))
+        : String(msg.content || '')
+    }));
+
     const mergedResult = {
       ...result,
-      pendingMessages
+      pendingMessages: normalizedPendingMessages
     };
 
     reply.send(mergedResult);

package/src/controllers/run.ts

@@ -7,6 +7,7 @@ import {
   agentLatticeManager,
   agentInstanceManager,
 } from "@axiom-lattice/core";
+import { MessageChunkTypes } from "@axiom-lattice/protocols";
 
 interface ResumeStreamRequest {
   thread_id: string;
@@ -87,11 +88,10 @@ export const createRun = async (
         //   return
         // }
 
-        const stream = agent.chunkStream((result).messageId)
+        const stream = agent.chunkStream((result).messageId, [MessageChunkTypes.MESSAGE_COMPLETED])
 
         // Forward all chunks to SSE
         for await (const chunk of stream) {
-          console.log(input.message, chunk.data.content);
           const success = reply.raw.write(`data: ${JSON.stringify(chunk)}\n\n`);
           if (!success) {
             await new Promise(resolve => reply.raw.once('drain', resolve));
@@ -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,10 +179,11 @@ export const resumeStream = async (
       });
 
       // Get the stream from agent's chunk buffer
-      const stream = agent.chunkStream(message_id, known_content);
+      const stream = agent.chunkStream(message_id, [MessageChunkTypes.THREAD_IDLE]);
 
       // Stream the chunks to the client
       for await (const chunk of stream) {
+        // console.log(chunk);
         reply.raw.write(`data: ${JSON.stringify(chunk)}\n\n`);
       }
     } catch (error: any) {

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;