@langgraph-js/sdk 1.11.0 → 1.12.0

package/dist/LangGraphClient.d.ts

@@ -35,6 +35,7 @@ export type RenderMessage = Message & {
             };
         }[];
     };
+    sub_agent_messages?: RenderMessage[];
     usage_metadata?: {
         total_tokens: number;
         input_tokens: number;
@@ -91,6 +92,8 @@ export declare class LangGraphClient extends Client {
     private streamingCallbacks;
     tools: ToolManager;
     stopController: AbortController | null;
+    /** 用于存储 subAgent 状态数据的键 */
+    subAgentsKey: string;
     constructor(config: LangGraphClientConfig);
     availableAssistants: Assistant[];
     private listAssistants;
@@ -130,6 +133,8 @@ export declare class LangGraphClient extends Client {
      * @en Messages used for streaming rendering in the UI.
      */
     get renderMessage(): RenderMessage[];
+    /** 转换 subAgent 消息为工具的子消息 */
+    private convertSubAgentMessages;
     /**
      * @zh 为消息附加额外的信息，如耗时、唯一 ID 等。
      * @en Attaches additional information to messages, such as spend time, unique ID, etc.

package/dist/LangGraphClient.js

@@ -32,6 +32,8 @@ export class LangGraphClient extends Client {
         this.streamingCallbacks = new Set();
         this.tools = new ToolManager();
         this.stopController = null;
+        /** 用于存储 subAgent 状态数据的键 */
+        this.subAgentsKey = "task_store";
         this.availableAssistants = [];
         this.streamingMessage = [];
         /** 图发过来的更新信息 */
@@ -225,7 +227,56 @@ export class LangGraphClient extends Client {
                 result.unshift(message);
             }
         }
-        return this.attachInfoForMessage(this.composeToolMessages(result));
+        return this.convertSubAgentMessages(this.attachInfoForMessage(this.composeToolMessages(result)));
+    }
+    /** 转换 subAgent 消息为工具的子消息 */
+    convertSubAgentMessages(messages) {
+        const origin_task_store = this.graphState[this.subAgentsKey];
+        if (!origin_task_store)
+            return messages;
+        const task_store = JSON.parse(JSON.stringify(origin_task_store));
+        console.log(messages);
+        /** 获取 subAgent 消息的 id，用于流式过程中对数据进行标记 */
+        messages
+            .filter((i) => {
+            var _a;
+            return (_a = i.node_name) === null || _a === void 0 ? void 0 : _a.startsWith("subagent_");
+        })
+            .forEach((i) => {
+            const tool_call_id = i.node_name.replace("subagent_", "");
+            const store = task_store[tool_call_id];
+            if (store) {
+                // 根据 id 进行去重
+                const exists = store.messages.some((msg) => msg.id === i.id);
+                if (!exists) {
+                    store.messages.push(i);
+                }
+            }
+            else {
+                task_store[tool_call_id] = {
+                    messages: [i],
+                };
+            }
+        });
+        const ignoreIds = new Set();
+        Object.values(task_store).forEach((task) => {
+            task.messages.forEach((message) => {
+                ignoreIds.add(message.id);
+            });
+        });
+        const result = [];
+        for (const message of messages) {
+            if (message.type === "tool" && message.tool_call_id) {
+                const task = task_store[message.tool_call_id];
+                if (task) {
+                    message.sub_agent_messages = this.attachInfoForMessage(this.composeToolMessages(task.messages));
+                }
+            }
+            if (message.id && ignoreIds.has(message.id))
+                continue;
+            result.push(message);
+        }
+        return result;
     }
     /**
      * @zh 为消息附加额外的信息，如耗时、唯一 ID 等。

package/dist/TestKit.d.ts

@@ -158,6 +158,7 @@ export declare class TestLangGraphChat {
                 };
             }[];
         };
+        sub_agent_messages?: RenderMessage[];
         usage_metadata?: {
             total_tokens: number;
             input_tokens: number;
@@ -182,6 +183,7 @@ export declare class TestLangGraphChat {
                 };
             }[];
         };
+        sub_agent_messages?: RenderMessage[];
         usage_metadata?: {
             total_tokens: number;
             input_tokens: number;
@@ -206,6 +208,7 @@ export declare class TestLangGraphChat {
                 };
             }[];
         };
+        sub_agent_messages?: RenderMessage[];
         usage_metadata?: {
             total_tokens: number;
             input_tokens: number;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "1.11.0",
+  "version": "1.12.0",
   "description": "The UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces",
   "main": "dist/index.js",
   "type": "module",

package/src/LangGraphClient.ts

@@ -35,6 +35,7 @@ export type RenderMessage = Message & {
             };
         }[];
     };
+    sub_agent_messages?: RenderMessage[];
     usage_metadata?: {
         total_tokens: number;
         input_tokens: number;
@@ -102,6 +103,8 @@ export class LangGraphClient extends Client {
     private streamingCallbacks: Set<StreamingUpdateCallback> = new Set();
     tools: ToolManager = new ToolManager();
     stopController: AbortController | null = null;
+    /** 用于存储 subAgent 状态数据的键 */
+    subAgentsKey = "task_store";
 
     constructor(config: LangGraphClientConfig) {
         super(config);
@@ -296,7 +299,54 @@ export class LangGraphClient extends Client {
             }
         }
 
-        return this.attachInfoForMessage(this.composeToolMessages(result as RenderMessage[]));
+        return this.convertSubAgentMessages(this.attachInfoForMessage(this.composeToolMessages(result as RenderMessage[])));
+    }
+    /** 转换 subAgent 消息为工具的子消息 */
+    private convertSubAgentMessages(messages: RenderMessage[]) {
+        const origin_task_store = this.graphState[this.subAgentsKey];
+        if (!origin_task_store) return messages;
+
+        const task_store = JSON.parse(JSON.stringify(origin_task_store));
+        console.log(messages);
+        /** 获取 subAgent 消息的 id，用于流式过程中对数据进行标记 */
+        messages
+            .filter((i) => {
+                return i.node_name?.startsWith("subagent_");
+            })
+            .forEach((i) => {
+                const tool_call_id = i.node_name!.replace("subagent_", "");
+                const store = task_store[tool_call_id];
+                if (store) {
+                    // 根据 id 进行去重
+                    const exists = (store.messages as RenderMessage[]).some((msg) => msg.id === i.id);
+                    if (!exists) {
+                        (store.messages as RenderMessage[]).push(i);
+                    }
+                } else {
+                    task_store[tool_call_id] = {
+                        messages: [i],
+                    };
+                }
+            });
+
+        const ignoreIds = new Set<string>();
+        Object.values(task_store).forEach((task: any) => {
+            task.messages.forEach((message: RenderMessage) => {
+                ignoreIds.add(message.id!);
+            });
+        });
+        const result: RenderMessage[] = [];
+        for (const message of messages) {
+            if (message.type === "tool" && message.tool_call_id) {
+                const task = task_store[message.tool_call_id];
+                if (task) {
+                    message.sub_agent_messages = this.attachInfoForMessage(this.composeToolMessages(task.messages));
+                }
+            }
+            if (message.id && ignoreIds.has(message.id)) continue;
+            result.push(message);
+        }
+        return result;
     }
     /**
      * @zh 为消息附加额外的信息，如耗时、唯一 ID 等。