@axiom-lattice/gateway 2.1.47 → 2.1.49

package/dist/index.mjs

@@ -202,6 +202,7 @@ import { v4 } from "uuid";
 import {
   agentInstanceManager as agentInstanceManager2
 } from "@axiom-lattice/core";
+import { MessageChunkTypes } from "@axiom-lattice/protocols";
 var createRun = async (request, reply) => {
   try {
     const {
@@ -246,9 +247,8 @@ var createRun = async (request, reply) => {
           command,
           custom_run_config
         });
-        const stream = agent.chunkStream(result.messageId);
+        const stream = agent.chunkStream(result.messageId, [MessageChunkTypes.MESSAGE_COMPLETED]);
         for await (const chunk of stream) {
-          console.log(input.message, chunk.data.content);
           const success = reply.raw.write(`data: ${JSON.stringify(chunk)}
 
 `);
@@ -317,7 +317,7 @@ var resumeStream = async (request, reply) => {
         workspace_id,
         project_id
       });
-      const stream = agent.chunkStream(message_id, known_content);
+      const stream = agent.chunkStream(message_id, [MessageChunkTypes.THREAD_IDLE]);
       for await (const chunk of stream) {
         reply.raw.write(`data: ${JSON.stringify(chunk)}
 
@@ -450,9 +450,13 @@ var getAgentState = async (request, reply) => {
       reply.status(500).send(result);
       return;
     }
+    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);
   } catch (error) {
@@ -4932,8 +4936,6 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
     console.log(
       `\u5F00\u59CB\u5904\u7406\u4EFB\u52A1 [assistant_id: ${assistant_id}, thread_id: ${thread_id}]`
     );
-    const apiUrl = AgentTaskConsumer.agent_run_endpoint;
-    console.log(`apiUrl: ${apiUrl}`);
     const agent = agentInstanceManager4.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) {
@@ -4944,6 +4946,13 @@ var handleAgentTask = async (taskRequest, retryCount = 0) => {
           config: { assistant_id, thread_id, tenant_id }
         });
       });
+      agent.subscribeOnce("message:interrupted", (evt) => {
+        eventBus2.publish(callback_event, {
+          success: true,
+          state: evt.state,
+          config: { assistant_id, thread_id, tenant_id }
+        });
+      });
     }
     return true;
   } catch (error) {
@@ -5145,7 +5154,8 @@ import {
   loggerLatticeManager,
   sandboxLatticeManager as sandboxLatticeManager2,
   sqlDatabaseManager as sqlDatabaseManager2,
-  getStoreLattice as getStoreLattice12
+  getStoreLattice as getStoreLattice12,
+  agentInstanceManager as agentInstanceManager5
 } from "@axiom-lattice/core";
 import {
   LoggerType
@@ -5297,6 +5307,13 @@ var start = async (config) => {
         agentTaskConsumer.startPollingQueue();
       }
     }
+    try {
+      logger.info("Starting agent instance recovery...");
+      const restoreStats = await agentInstanceManager5.restore();
+      logger.info(`Agent recovery complete: ${restoreStats.restored} threads restored, ${restoreStats.errors} errors`);
+    } catch (error) {
+      logger.error("Agent recovery failed", { error });
+    }
   } catch (err) {
     logger.error("Server start failed", { error: err });
     process.exit(1);