@axiom-lattice/gateway 2.1.46 → 2.1.48

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.46",
+  "version": "2.1.48",
   "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.40",
-    "@axiom-lattice/pg-stores": "1.0.30",
-    "@axiom-lattice/protocols": "2.1.22",
-    "@axiom-lattice/queue-redis": "1.0.21"
+    "@axiom-lattice/core": "2.1.42",
+    "@axiom-lattice/pg-stores": "1.0.32",
+    "@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,9 +7,11 @@ import {
   agentLatticeManager,
   agentInstanceManager,
 } from "@axiom-lattice/core";
+import { MessageChunkTypes } from "@axiom-lattice/protocols";
 
 interface ResumeStreamRequest {
   thread_id: string;
+  assistant_id: string
   message_id: string;
   known_content: string;
   poll_interval?: number;
@@ -86,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));
@@ -136,14 +137,21 @@ export const resumeStream = async (
   reply: FastifyReply
 ): Promise<void> => {
   try {
-    const { thread_id, message_id, known_content, poll_interval } =
+
+
+
+
+    const { thread_id, message_id, assistant_id, known_content, poll_interval } =
       request.body as ResumeStreamRequest;
+    const tenant_id = request.headers["x-tenant-id"] as string;
+    const workspace_id = request.headers["x-workspace-id"] as string;
+    const project_id = request.headers["x-project-id"] as string;
 
     // Validate request data
-    if (!thread_id || !message_id || known_content === undefined) {
+    if (!thread_id || !message_id || !assistant_id || known_content === undefined) {
       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;
     }
@@ -162,17 +170,22 @@ export const resumeStream = async (
     try {
       // Create Agent instance for accessing chunk buffer
       // Note: assistant_id, tenant_id are not required for chunkStream
-      const agent = new Agent({
-        assistant_id: '',
+      // Get or create Agent instance (ensures single instance per thread)
+      const agent = agentInstanceManager.getAgent({
+        assistant_id,
         thread_id,
-        tenant_id: '',
+        tenant_id,
+        workspace_id,
+        project_id,
       });
 
       // Get the stream from agent's chunk buffer
-      const stream = agent.chunkStream(message_id, known_content);
+      console.log("[UI]message_id", message_id)
+      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) {