@langgraph-js/sdk 1.10.4 → 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;
@@ -57,6 +58,7 @@ export type SendMessageOptions = {
         streamResponse?: any;
     };
     command?: Command;
+    joinRunId?: string;
 };
 export interface LangGraphClientConfig {
     apiUrl?: string;
@@ -90,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;
@@ -115,7 +119,8 @@ export declare class LangGraphClient extends Client {
      * @zh 从历史中恢复 Thread 数据。
      * @en Resets the Thread data from history.
      */
-    resetThread(agent: string, threadId: string): Promise<void>;
+    resetThread(agent: string, threadId: string): Promise<Thread<import("@langchain/langgraph-sdk").DefaultValues>>;
+    resetStream(): Promise<void>;
     streamingMessage: RenderMessage[];
     /** 图发过来的更新信息 */
     graphMessages: RenderMessage[];
@@ -128,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.
@@ -168,7 +175,7 @@ export declare class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    sendMessage(input: string | Message[], { extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
+    sendMessage(input: string | Message[], { joinRunId, extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
     /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
     private graphPosition;
     getGraphPosition(): {

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 = [];
         /** 图发过来的更新信息 */
@@ -112,10 +114,11 @@ export class LangGraphClient extends Client {
      * @en Resets the Thread data from history.
      */
     async resetThread(agent, threadId) {
+        var _a;
         await this.initAssistant(agent);
         this.currentThread = await this.threads.get(threadId);
         this.graphState = this.currentThread.values;
-        this.graphMessages = this.graphState.messages;
+        this.graphMessages = ((_a = this.graphState) === null || _a === void 0 ? void 0 : _a.messages) || [];
         this.emitStreamingUpdate({
             type: "value",
             data: {
@@ -125,6 +128,15 @@ export class LangGraphClient extends Client {
                 },
             },
         });
+        return this.currentThread;
+    }
+    // 从历史中恢复时，应该恢复流式状态
+    async resetStream() {
+        const runs = await this.runs.list(this.currentThread.thread_id);
+        const runningRun = runs === null || runs === void 0 ? void 0 : runs.find((run) => run.status === "running" || run.status === "pending");
+        if (runningRun) {
+            await this.sendMessage([], { joinRunId: runningRun.run_id });
+        }
     }
     cloneMessage(message) {
         return JSON.parse(JSON.stringify(message));
@@ -215,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 等。
@@ -352,7 +413,7 @@ export class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    async sendMessage(input, { extraParams, _debug, command } = {}) {
+    async sendMessage(input, { joinRunId, extraParams, _debug, command } = {}) {
         var _a;
         if (!this.currentAssistant) {
             throw new Error("Thread or Assistant not initialized");
@@ -377,8 +438,14 @@ export class LangGraphClient extends Client {
                     content: input,
                 },
             ];
-        const streamResponse = (_debug === null || _debug === void 0 ? void 0 : _debug.streamResponse) ||
-            this.runs.stream(this.currentThread.thread_id, this.currentAssistant.assistant_id, {
+        const createStreamResponse = async () => {
+            if (_debug === null || _debug === void 0 ? void 0 : _debug.streamResponse) {
+                return _debug.streamResponse;
+            }
+            if (joinRunId) {
+                return this.runs.joinStream(this.currentThread.thread_id, joinRunId);
+            }
+            return this.runs.stream(this.currentThread.thread_id, this.currentAssistant.assistant_id, {
                 input: {
                     ...this.graphState,
                     ...this.extraParams,
@@ -390,6 +457,8 @@ export class LangGraphClient extends Client {
                 streamSubgraphs: true,
                 command,
             });
+        };
+        const streamResponse = await createStreamResponse();
         const streamRecord = [];
         this.emitStreamingUpdate({
             type: "start",

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/dist/ui-store/createChatStore.d.ts

@@ -84,7 +84,7 @@ export declare const createChatStore: (initClientName: string, config: LangGraph
          */
         toHistoryChat(thread: Thread<{
             messages: Message[];
-        }>): void;
+        }>): Promise<Thread<import("@langchain/langgraph-sdk").DefaultValues> | undefined>;
         /**
          * @zh 删除指定的历史聊天会话。
          * @en Deletes the specified historical chat session.

package/dist/ui-store/createChatStore.js

@@ -283,11 +283,15 @@ export const createChatStore = (initClientName, config, context = {}) => {
              * @zh 切换到指定的历史聊天会话。
              * @en Switches to the specified historical chat session.
              */
-            toHistoryChat(thread) {
-                var _a, _b;
+            async toHistoryChat(thread) {
+                var _a, _b, _c;
                 inChatError.set(null);
                 loading.set(false);
-                (_a = client.get()) === null || _a === void 0 ? void 0 : _a.resetThread((_b = thread.metadata) === null || _b === void 0 ? void 0 : _b.graph_id, thread.thread_id);
+                const nowThread = await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.resetThread((_b = thread.metadata) === null || _b === void 0 ? void 0 : _b.graph_id, thread.thread_id));
+                if (nowThread) {
+                    (_c = client.get()) === null || _c === void 0 ? void 0 : _c.resetStream();
+                }
+                return nowThread;
             },
             /**
              * @zh 删除指定的历史聊天会话。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "1.10.4",
+  "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;
@@ -55,6 +56,7 @@ export type SendMessageOptions = {
     extraParams?: Record<string, any>;
     _debug?: { streamResponse?: any };
     command?: Command;
+    joinRunId?: string;
 };
 export interface LangGraphClientConfig {
     apiUrl?: string;
@@ -101,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);
@@ -182,7 +186,7 @@ export class LangGraphClient extends Client {
         await this.initAssistant(agent);
         this.currentThread = await this.threads.get(threadId);
         this.graphState = this.currentThread.values;
-        this.graphMessages = this.graphState.messages;
+        this.graphMessages = this.graphState?.messages || [];
         this.emitStreamingUpdate({
             type: "value",
             data: {
@@ -192,6 +196,15 @@ export class LangGraphClient extends Client {
                 },
             },
         });
+        return this.currentThread;
+    }
+    // 从历史中恢复时，应该恢复流式状态
+    async resetStream() {
+        const runs = await this.runs.list(this.currentThread!.thread_id);
+        const runningRun = runs?.find((run) => run.status === "running" || run.status === "pending");
+        if (runningRun) {
+            await this.sendMessage([], { joinRunId: runningRun.run_id });
+        }
     }
 
     streamingMessage: RenderMessage[] = [];
@@ -286,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 等。
@@ -432,7 +492,7 @@ export class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    async sendMessage(input: string | Message[], { extraParams, _debug, command }: SendMessageOptions = {}) {
+    async sendMessage(input: string | Message[], { joinRunId, extraParams, _debug, command }: SendMessageOptions = {}) {
         if (!this.currentAssistant) {
             throw new Error("Thread or Assistant not initialized");
         }
@@ -457,20 +517,29 @@ export class LangGraphClient extends Client {
                       content: input,
                   } as HumanMessage,
               ];
-        const streamResponse =
-            _debug?.streamResponse ||
-            this.runs.stream(this.currentThread!.thread_id, this.currentAssistant.assistant_id, {
+        const createStreamResponse = async () => {
+            if (_debug?.streamResponse) {
+                return _debug.streamResponse;
+            }
+            if (joinRunId) {
+                return this.runs.joinStream(this.currentThread!.thread_id, joinRunId);
+            }
+
+            return this.runs.stream(this.currentThread!.thread_id, this.currentAssistant!.assistant_id, {
                 input: {
                     ...this.graphState,
                     ...this.extraParams,
                     ...(extraParams || {}),
                     messages: messagesToSend,
-                    fe_tools: await this.tools.toJSON(this.currentAssistant.graph_id),
+                    fe_tools: await this.tools.toJSON(this.currentAssistant!.graph_id),
                 },
                 streamMode: ["messages", "values"],
                 streamSubgraphs: true,
                 command,
             });
+        };
+        const streamResponse = await createStreamResponse();
+
         const streamRecord: any[] = [];
         this.emitStreamingUpdate({
             type: "start",

package/src/ui-store/createChatStore.ts

@@ -289,14 +289,18 @@ export const createChatStore = (
              * @zh 切换到指定的历史聊天会话。
              * @en Switches to the specified historical chat session.
              */
-            toHistoryChat(
+            async toHistoryChat(
                 thread: Thread<{
                     messages: Message[];
                 }>
             ) {
                 inChatError.set(null);
                 loading.set(false);
-                client.get()?.resetThread(thread.metadata?.graph_id as string, thread.thread_id);
+                const nowThread = await client.get()?.resetThread(thread.metadata?.graph_id as string, thread.thread_id);
+                if (nowThread) {
+                    client.get()?.resetStream();
+                }
+                return nowThread;
             },
             /**
              * @zh 删除指定的历史聊天会话。