@axiom-lattice/gateway 2.1.47 → 2.1.48

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,8 @@ var resumeStream = async (request, reply) => {
         workspace_id,
         project_id
       });
-      const stream = agent.chunkStream(message_id, known_content);
+      console.log("[UI]message_id", message_id);
+      const stream = agent.chunkStream(message_id, [MessageChunkTypes.THREAD_IDLE]);
       for await (const chunk of stream) {
         reply.raw.write(`data: ${JSON.stringify(chunk)}
 
@@ -450,9 +451,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) {