@axiom-lattice/gateway 2.1.46 → 2.1.48

package/dist/index.mjs

@@ -200,9 +200,9 @@ var getAgentGraph = async (request, reply) => {
 // src/controllers/run.ts
 import { v4 } from "uuid";
 import {
-  Agent,
   agentInstanceManager as agentInstanceManager2
 } from "@axiom-lattice/core";
+import { MessageChunkTypes } from "@axiom-lattice/protocols";
 var createRun = async (request, reply) => {
   try {
     const {
@@ -247,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)}
 
 `);
@@ -292,11 +291,14 @@ var createRun = async (request, reply) => {
 };
 var resumeStream = async (request, reply) => {
   try {
-    const { thread_id, message_id, known_content, poll_interval } = request.body;
-    if (!thread_id || !message_id || known_content === void 0) {
+    const { thread_id, message_id, assistant_id, known_content, poll_interval } = request.body;
+    const tenant_id = request.headers["x-tenant-id"];
+    const workspace_id = request.headers["x-workspace-id"];
+    const project_id = request.headers["x-project-id"];
+    if (!thread_id || !message_id || !assistant_id || known_content === void 0) {
       reply.status(400).send({
         success: false,
-        error: "thread_id, message_id, and known_content are required"
+        error: "thread_id, message_id, assistant_id, and known_content are required"
       });
       return;
     }
@@ -308,12 +310,15 @@ var resumeStream = async (request, reply) => {
       "Access-Control-Allow-Origin": "*"
     });
     try {
-      const agent = new Agent({
-        assistant_id: "",
+      const agent = agentInstanceManager2.getAgent({
+        assistant_id,
         thread_id,
-        tenant_id: ""
+        tenant_id,
+        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)}
 
@@ -446,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) {