@langgraph-js/sdk 3.8.3 → 4.0.0

package/dist/LangGraphClient.d.ts

@@ -2,6 +2,7 @@ import type { Thread, Message, Assistant, Command } from "@langchain/langgraph-s
 import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
+import { HumanInTheLoopDecision, HumanInTheLoopState } from "./humanInTheLoop.js";
 import { type ILangGraphClient } from "@langgraph-js/pure-graph/dist/types.js";
 import { RevertChatToOptions } from "./time-travel/index.js";
 export type RenderMessage = Message & {
@@ -44,20 +45,6 @@ export type SendMessageOptions = {
     command?: Command;
     joinRunId?: string;
 };
-export type InterruptData = {
-    id: string;
-    value: {
-        actionRequests: {
-            name: string;
-            description: string;
-            args: any;
-        }[];
-        reviewConfigs: {
-            actionName: string;
-            allowedDecisions: ("approve" | "edit" | "reject" | "respond")[];
-        }[];
-    };
-}[];
 export interface LangGraphClientConfig {
     apiUrl?: string;
     apiKey?: string;
@@ -244,7 +231,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
         messages: Message[];
     }>;
     messagesMetadata: {};
-    humanInTheLoop: InterruptData | null;
+    humanInTheLoop: HumanInTheLoopState | null;
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -267,6 +254,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
      */
     private processStreamChunk;
     private runFETool;
+    private responseHumanInTheLoop;
     private callFETool;
     extraParams: Record<string, any>;
     /**
@@ -277,8 +265,14 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
     /**
      * @zh 标记前端工具等待已完成。
      * @en Marks the frontend tool waiting as completed.
+     * @deprecated 请使用 doneHumanInTheLoopWaiting 和规范的 humanInTheLoop 协议定义状态
      */
     doneFEToolWaiting(id: string, result: CallToolResult): void;
+    /**
+     * @zh 标记人机交互等待已完成。
+     * @en Marks the human in the loop waiting as completed.
+     */
+    doneHumanInTheLoopWaiting(tool_id: string, action_request_id: string, result: HumanInTheLoopDecision): void;
     /**
      * @zh 获取当前的 Thread。
      * @en Gets the current Thread.

package/dist/LangGraphClient.js

@@ -1,5 +1,6 @@
 import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
+import { createActionRequestID } from "./humanInTheLoop.js";
 import { MessageProcessor } from "./MessageProcessor.js";
 import { revertChatTo } from "./time-travel/index.js";
 import camelcaseKeys from "camelcase-keys";
@@ -281,9 +282,9 @@ export class LangGraphClient extends EventEmitter {
             }
         }
         const data = await this.runFETool();
+        await this.responseHumanInTheLoop();
         if (data)
             streamRecord.push(...data);
-        this.humanInTheLoop = null;
         this._status = "idle";
         this.emit("done", {
             event: "done",
@@ -332,13 +333,34 @@ export class LangGraphClient extends EventEmitter {
         }
         else if (chunk.event === "values") {
             const data = chunk.data;
-            if (data.__interrupt__) {
+            if (data?.__interrupt__) {
                 this._status = "interrupted";
-                this.humanInTheLoop = camelcaseKeys(data.__interrupt__, {
-                    deep: true,
+                const humanInTheLoop = data?.__interrupt__.map((i) => {
+                    // 修复 python 版本是以下划线命名的，而 js 版本是以驼峰命名的
+                    if (i && "review_configs" in i.value) {
+                        return {
+                            id: i.id,
+                            value: {
+                                /** @ts-ignore */
+                                actionRequests: i.value.action_requests,
+                                reviewConfigs: camelcaseKeys(i.value.review_configs, { deep: true }),
+                            },
+                        };
+                    }
+                    return i;
+                });
+                humanInTheLoop.forEach((i) => {
+                    i.value.actionRequests.forEach((j) => {
+                        j.id = j.id || createActionRequestID(j);
+                    });
+                    return i;
                 });
+                this.humanInTheLoop = {
+                    interruptData: humanInTheLoop,
+                    result: {},
+                };
             }
-            else if (data.messages) {
+            else if (data?.messages) {
                 const isResume = !!command?.resume;
                 const isLongerThanLocal = data.messages.length >= this.messageProcessor.getGraphMessages().length;
                 // resume 情况下，长度低于前端 message 的统统不接受
@@ -375,16 +397,27 @@ export class LangGraphClient extends EventEmitter {
             });
             this._status = "interrupted";
             this.currentThread.status = "interrupted"; // 修复某些机制下，状态不为 interrupted 与后端有差异
+            console.log("batch call tools", result.length);
             return Promise.all(result);
         }
     }
+    async responseHumanInTheLoop() {
+        if (this.humanInTheLoop) {
+            const humanInTheLoop = this.humanInTheLoop;
+            this.humanInTheLoop = null;
+            return this.resume({
+                decisions: humanInTheLoop.interruptData[0].value.actionRequests.map((i) => {
+                    return humanInTheLoop.result[i.id];
+                }),
+            });
+        }
+    }
     async callFETool(message, args) {
         const that = this; // 防止 this 被错误解析
         const result = await this.tools.callTool(message.name, args, { client: that, message });
         if (!result) {
             return;
         }
-        return this.resume(result);
     }
     extraParams = {};
     /**
@@ -401,6 +434,7 @@ export class LangGraphClient extends EventEmitter {
     /**
      * @zh 标记前端工具等待已完成。
      * @en Marks the frontend tool waiting as completed.
+     * @deprecated 请使用 doneHumanInTheLoopWaiting 和规范的 humanInTheLoop 协议定义状态
      */
     doneFEToolWaiting(id, result) {
         const done = this.tools.doneWaiting(id, result);
@@ -408,6 +442,17 @@ export class LangGraphClient extends EventEmitter {
             this.resume(result);
         }
     }
+    /**
+     * @zh 标记人机交互等待已完成。
+     * @en Marks the human in the loop waiting as completed.
+     */
+    doneHumanInTheLoopWaiting(tool_id, action_request_id, result) {
+        // 移除等待状态
+        this.tools.doneWaiting(tool_id, result);
+        if (this.humanInTheLoop) {
+            this.humanInTheLoop.result[action_request_id] = result;
+        }
+    }
     /**
      * @zh 获取当前的 Thread。
      * @en Gets the current Thread.

package/dist/ToolManager.d.ts

@@ -62,11 +62,12 @@ export declare class ToolManager {
      * @en Marks the tool waiting with the specified ID as completed and passes the result.
      */
     doneWaiting(id: string, value: CallToolResult): boolean;
+    isAllToolCompleted(): number;
     /**
      * @zh 等待指定 ID 的工具完成。
      * @en Waits for the tool with the specified ID to complete.
      */
-    waitForDone(id: string): Promise<unknown> | ((value: CallToolResult) => void) | undefined;
+    waitForDone(id: string): Promise<unknown>;
     /**
      * @zh 一个静态方法，用于在前端等待用户界面操作完成。
      * @en A static method used in the frontend to wait for user interface operations to complete.
@@ -74,5 +75,5 @@ export declare class ToolManager {
     static waitForUIDone<T>(_: T, context: {
         client: LangGraphClient;
         message: ToolMessage;
-    }): Promise<unknown> | ((value: CallToolResult) => void) | undefined;
+    }): Promise<unknown>;
 }

package/dist/ToolManager.js

@@ -92,14 +92,17 @@ export class ToolManager {
             return false;
         }
     }
+    isAllToolCompleted() {
+        return this.waitingMap.size;
+    }
     /**
      * @zh 等待指定 ID 的工具完成。
      * @en Waits for the tool with the specified ID to complete.
      */
     waitForDone(id) {
-        if (this.waitingMap.has(id)) {
-            return this.waitingMap.get(id);
-        }
+        // if (this.waitingMap.has(id)) {
+        //     return this.waitingMap.get(id);
+        // }
         const promise = new Promise((resolve, reject) => {
             this.waitingMap.set(id, resolve);
         });

package/dist/humanInTheLoop.d.ts

@@ -0,0 +1,38 @@
+export type HumanInTheLoopDecision = {
+    type: "approve" | "edit" | "reject" | "respond";
+    edited_action?: {
+        name: string;
+        args: Record<string, any>;
+    };
+    message?: string;
+};
+/**
+ * HumanInTheLoop 的标准回复格式
+ */
+export type InterruptResponse = {
+    decisions: HumanInTheLoopDecision[];
+};
+/** 由于 langchain human in the loop 没有设计调用 id，所以我们需要给一个 id */
+export declare const createActionRequestID: (j: {
+    name: string;
+    args: any;
+}) => string;
+export type HumanInTheLoopState = {
+    interruptData: InterruptData;
+    result: Record<string, HumanInTheLoopDecision>;
+};
+export type InterruptData = {
+    id: string;
+    value: {
+        actionRequests: {
+            id?: string;
+            name: string;
+            description: string;
+            args: any;
+        }[];
+        reviewConfigs: {
+            actionName: string;
+            allowedDecisions: ("approve" | "edit" | "reject" | "respond")[];
+        }[];
+    };
+}[];

package/dist/humanInTheLoop.js

@@ -0,0 +1,4 @@
+/** 由于 langchain human in the loop 没有设计调用 id，所以我们需要给一个 id */
+export const createActionRequestID = (j) => {
+    return j.name + JSON.stringify(j.args);
+};

package/dist/tool/ToolUI.d.ts

@@ -1,6 +1,6 @@
 import { RenderMessage } from "../LangGraphClient.js";
 import { LangGraphClient } from "../LangGraphClient.js";
-import { InterruptResponse } from "./createTool.js";
+import { HumanInTheLoopDecision } from "../humanInTheLoop.js";
 export type DeepPartial<T> = {
     [P in keyof T]?: T[P] extends object ? DeepPartial<T[P]> : T[P];
 };
@@ -8,16 +8,19 @@ export declare class ToolRenderData<I, D> {
     message: RenderMessage;
     client: LangGraphClient;
     constructor(message: RenderMessage, client: LangGraphClient);
+    private getToolActionRequestID;
     /**
      * 获取人机交互数据
      * 直接使用 reviewConfig 获取可以显示的按钮
      * actionRequest 获取当前工具的入参
      */
     getHumanInTheLoopData(): {
+        actionRequestIndex: number;
         config: {
             id: string;
             value: {
                 actionRequests: {
+                    id?: string;
                     name: string;
                     description: string;
                     args: any;
@@ -33,14 +36,16 @@ export declare class ToolRenderData<I, D> {
             allowedDecisions: ("approve" | "edit" | "reject" | "respond")[];
         };
         actionRequest: {
+            id?: string;
             name: string;
             description: string;
             args: any;
         };
+        result: HumanInTheLoopDecision;
     } | null;
     /** 发送恢复状态的数据 */
-    sendResumeData(response: InterruptResponse["decisions"][number]): void;
-    get state(): "idle" | "interrupted" | "done" | "loading";
+    sendResumeData(response: HumanInTheLoopDecision): void;
+    get state(): "idle" | "done" | "loading";
     get input(): I | null;
     get output(): string;
     getJSONOutput(): D;

package/dist/tool/ToolUI.js

@@ -1,5 +1,6 @@
 import { getMessageContent } from "../ui-store/createChatStore.js";
 import { jsonrepair } from "jsonrepair";
+import { createActionRequestID } from "../humanInTheLoop.js";
 export class ToolRenderData {
     message;
     client;
@@ -7,19 +8,33 @@ export class ToolRenderData {
         this.message = message;
         this.client = client;
     }
+    getToolActionRequestID() {
+        return createActionRequestID({
+            name: this.message.name,
+            args: this.getInputRepaired(),
+        });
+    }
     /**
      * 获取人机交互数据
      * 直接使用 reviewConfig 获取可以显示的按钮
      * actionRequest 获取当前工具的入参
      */
     getHumanInTheLoopData() {
-        const configOfHumanInTheLoop = this.client.humanInTheLoop?.find((i) => i.value.reviewConfigs?.some((j) => j.actionName === this.message.name));
+        const toolActionRequestID = this.getToolActionRequestID();
+        if (!this.client.humanInTheLoop)
+            return null;
+        const configOfHumanInTheLoop = this.client.humanInTheLoop.interruptData.find((i) => i.value.actionRequests.some((j) => j.id === toolActionRequestID));
         if (!configOfHumanInTheLoop)
             return null;
+        const actionRequestIndex = configOfHumanInTheLoop.value.actionRequests.findIndex((j) => j.id === toolActionRequestID);
+        if (actionRequestIndex === -1)
+            return null;
         return {
+            actionRequestIndex: actionRequestIndex,
             config: configOfHumanInTheLoop,
-            reviewConfig: configOfHumanInTheLoop.value.reviewConfigs.find((j) => j.actionName === this.message.name),
-            actionRequest: configOfHumanInTheLoop.value.actionRequests.find((j) => j.name === this.message.name),
+            reviewConfig: configOfHumanInTheLoop.value.reviewConfigs.find((j) => j.actionName === configOfHumanInTheLoop.value.actionRequests[actionRequestIndex].name),
+            actionRequest: configOfHumanInTheLoop.value.actionRequests[actionRequestIndex],
+            result: this.client.humanInTheLoop.result[toolActionRequestID],
         };
     }
     /** 发送恢复状态的数据 */
@@ -28,14 +43,18 @@ export class ToolRenderData {
             /**@ts-ignore 修复 sb 的 langchain 官方的命名不统一，我们一致采用下划线版本，而非驼峰版本 */
             response.editedAction = response.edited_action;
         }
-        return this.client.doneFEToolWaiting(this.message.id, { decisions: [response] });
+        return this.client.doneHumanInTheLoopWaiting(this.message.id, this.getToolActionRequestID(), response);
     }
     get state() {
         if (this.message.type === "tool" && this.message?.additional_kwargs?.done) {
             return "done";
         }
-        if (this.client.status === "interrupted" && this.client.humanInTheLoop?.some((i) => i.value.reviewConfigs?.some((j) => j.actionName === this.message.name))) {
-            return "interrupted";
+        const humanInTheLoopData = this.getHumanInTheLoopData();
+        if (humanInTheLoopData?.result) {
+            return "done";
+        }
+        if (this.client.status === "interrupted" && humanInTheLoopData?.actionRequest) {
+            return "done";
         }
         if (this.message.tool_input) {
             return "loading";

package/dist/tool/createTool.d.ts

@@ -2,6 +2,7 @@ import { z, ZodRawShape } from "zod";
 import { Action, Parameter } from "./copilotkit-actions.js";
 import { Message } from "@langchain/langgraph-sdk";
 import { ToolRenderData } from "./ToolUI.js";
+import { HumanInTheLoopDecision, InterruptResponse as HumanInTheLoopResponse } from "../humanInTheLoop.js";
 export interface UnionTool<Args extends ZodRawShape, Child extends Object = Object, ResponseType = any> {
     name: string;
     description: string;
@@ -22,27 +23,10 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     isPureParams?: boolean;
 }
 export type ToolCallback<Args extends ZodRawShape> = (args: z.infer<z.ZodObject<Args>>, context?: any) => CallToolResult | Promise<CallToolResult>;
-/**
- * HumanInTheLoop 的标准回复格式
- */
-export type InterruptResponse = {
-    decisions: ({
-        type: "approve";
-    } | {
-        type: "edit";
-        edited_action: {
-            name: string;
-            args: Record<string, any>;
-        };
-    } | {
-        type: "reject";
-        message?: string;
-    })[];
-};
 export type CallToolResult = string | {
     type: "text";
     text: string;
-}[] | InterruptResponse;
+}[] | HumanInTheLoopDecision | HumanInTheLoopResponse;
 /** 用于格式校验 */
 export declare const createTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => UnionTool<Args, Object, any>;
 /**
@@ -173,7 +157,7 @@ export declare const createMCPTool: <Args extends ZodRawShape>(tool: UnionTool<A
     }[];
     isError?: undefined;
 } | {
-    content: InterruptResponse | {
+    content: HumanInTheLoopDecision | HumanInTheLoopResponse | {
         type: "text";
         text: string;
     }[] | undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "3.8.3",
+  "version": "4.0.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

@@ -2,6 +2,7 @@ import type { Thread, Message, Assistant, HumanMessage, AIMessage, ToolMessage,
 import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
+import { createActionRequestID, HumanInTheLoopDecision, HumanInTheLoopState, InterruptData } from "./humanInTheLoop.js";
 import { type ILangGraphClient } from "@langgraph-js/pure-graph/dist/types.js";
 import { MessageProcessor } from "./MessageProcessor.js";
 import { revertChatTo, RevertChatToOptions } from "./time-travel/index.js";
@@ -45,20 +46,7 @@ export type SendMessageOptions = {
     command?: Command;
     joinRunId?: string;
 };
-export type InterruptData = {
-    id: string;
-    value: {
-        actionRequests: {
-            name: string;
-            description: string;
-            args: any;
-        }[];
-        reviewConfigs: {
-            actionName: string;
-            allowedDecisions: ("approve" | "edit" | "reject" | "respond")[];
-        }[];
-    };
-}[];
+
 export interface LangGraphClientConfig {
     apiUrl?: string;
     apiKey?: string;
@@ -337,7 +325,7 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
         return state;
     }
     public messagesMetadata = {};
-    public humanInTheLoop: InterruptData | null = null;
+    public humanInTheLoop: HumanInTheLoopState | null = null;
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -411,8 +399,8 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
             }
         }
         const data = await this.runFETool();
+        await this.responseHumanInTheLoop();
         if (data) streamRecord.push(...data);
-        this.humanInTheLoop = null;
         this._status = "idle";
         this.emit("done", {
             event: "done",
@@ -457,17 +445,40 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
             this.emit("message", chunk);
             return true;
         } else if (chunk.event === "values") {
-            const data = chunk.data as {
-                __interrupt__?: InterruptData;
-                messages: Message[];
-            };
+            const data = chunk.data as
+                | {
+                      __interrupt__?: InterruptData;
+                      messages: Message[];
+                  }
+                | undefined;
 
-            if (data.__interrupt__) {
+            if (data?.__interrupt__) {
                 this._status = "interrupted";
-                this.humanInTheLoop = camelcaseKeys(data.__interrupt__, {
-                    deep: true,
+                const humanInTheLoop = data?.__interrupt__.map((i) => {
+                    // 修复 python 版本是以下划线命名的，而 js 版本是以驼峰命名的
+                    if (i && "review_configs" in i.value) {
+                        return {
+                            id: i.id,
+                            value: {
+                                /** @ts-ignore */
+                                actionRequests: i.value.action_requests,
+                                reviewConfigs: camelcaseKeys(i.value.review_configs as any[], { deep: true }),
+                            },
+                        } as InterruptData[number];
+                    }
+                    return i;
                 });
-            } else if (data.messages) {
+                humanInTheLoop.forEach((i) => {
+                    i.value.actionRequests.forEach((j) => {
+                        j.id = j.id || createActionRequestID(j);
+                    });
+                    return i;
+                });
+                this.humanInTheLoop = {
+                    interruptData: humanInTheLoop,
+                    result: {},
+                };
+            } else if (data?.messages) {
                 const isResume = !!command?.resume;
                 const isLongerThanLocal = data.messages.length >= this.messageProcessor.getGraphMessages().length;
                 // resume 情况下，长度低于前端 message 的统统不接受
@@ -503,16 +514,27 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
             });
             this._status = "interrupted";
             this.currentThread!.status = "interrupted"; // 修复某些机制下，状态不为 interrupted 与后端有差异
+            console.log("batch call tools", result.length);
             return Promise.all(result);
         }
     }
+    private async responseHumanInTheLoop() {
+        if (this.humanInTheLoop) {
+            const humanInTheLoop = this.humanInTheLoop;
+            this.humanInTheLoop = null;
+            return this.resume({
+                decisions: humanInTheLoop.interruptData[0].value.actionRequests.map((i) => {
+                    return humanInTheLoop.result[i.id!] as HumanInTheLoopDecision;
+                }),
+            });
+        }
+    }
     private async callFETool(message: ToolMessage, args: any) {
         const that = this; // 防止 this 被错误解析
         const result = await this.tools.callTool(message.name!, args, { client: that, message });
         if (!result) {
             return;
         }
-        return this.resume(result);
     }
     extraParams: Record<string, any> = {};
 
@@ -530,6 +552,7 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
     /**
      * @zh 标记前端工具等待已完成。
      * @en Marks the frontend tool waiting as completed.
+     * @deprecated 请使用 doneHumanInTheLoopWaiting 和规范的 humanInTheLoop 协议定义状态
      */
     doneFEToolWaiting(id: string, result: CallToolResult) {
         const done = this.tools.doneWaiting(id, result);
@@ -538,6 +561,18 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
         }
     }
 
+    /**
+     * @zh 标记人机交互等待已完成。
+     * @en Marks the human in the loop waiting as completed.
+     */
+    doneHumanInTheLoopWaiting(tool_id: string, action_request_id: string, result: HumanInTheLoopDecision) {
+        // 移除等待状态
+        this.tools.doneWaiting(tool_id, result);
+        if (this.humanInTheLoop) {
+            this.humanInTheLoop.result[action_request_id] = result;
+        }
+    }
+
     /**
      * @zh 获取当前的 Thread。
      * @en Gets the current Thread.

package/src/ToolManager.ts

@@ -102,15 +102,18 @@ export class ToolManager {
             return false;
         }
     }
+    isAllToolCompleted() {
+        return this.waitingMap.size;
+    }
 
     /**
      * @zh 等待指定 ID 的工具完成。
      * @en Waits for the tool with the specified ID to complete.
      */
     waitForDone(id: string) {
-        if (this.waitingMap.has(id)) {
-            return this.waitingMap.get(id);
-        }
+        // if (this.waitingMap.has(id)) {
+        //     return this.waitingMap.get(id);
+        // }
         const promise = new Promise((resolve, reject) => {
             this.waitingMap.set(id, resolve);
         });

package/src/humanInTheLoop.ts

@@ -0,0 +1,40 @@
+export type HumanInTheLoopDecision = {
+    type: "approve" | "edit" | "reject" | "respond";
+    edited_action?: {
+        name: string;
+        args: Record<string, any>;
+    };
+    message?: string;
+};
+
+/**
+ * HumanInTheLoop 的标准回复格式
+ */
+export type InterruptResponse = {
+    decisions: HumanInTheLoopDecision[];
+};
+/** 由于 langchain human in the loop 没有设计调用 id，所以我们需要给一个 id */
+export const createActionRequestID = (j: { name: string; args: any }) => {
+    return j.name + JSON.stringify(j.args);
+};
+
+export type HumanInTheLoopState = {
+    interruptData: InterruptData;
+    result: Record<string, HumanInTheLoopDecision>;
+};
+
+export type InterruptData = {
+    id: string;
+    value: {
+        actionRequests: {
+            id?: string;
+            name: string;
+            description: string;
+            args: any;
+        }[];
+        reviewConfigs: {
+            actionName: string;
+            allowedDecisions: ("approve" | "edit" | "reject" | "respond")[];
+        }[];
+    };
+}[];

package/src/tool/ToolUI.ts

@@ -3,7 +3,7 @@ import { RenderMessage } from "../LangGraphClient.js";
 import { LangGraphClient } from "../LangGraphClient.js";
 import { getMessageContent } from "../ui-store/createChatStore.js";
 import { jsonrepair } from "jsonrepair";
-import { InterruptResponse } from "./createTool.js";
+import { createActionRequestID, HumanInTheLoopDecision, InterruptResponse } from "../humanInTheLoop.js";
 
 export type DeepPartial<T> = {
     [P in keyof T]?: T[P] extends object ? DeepPartial<T[P]> : T[P];
@@ -13,34 +13,53 @@ export class ToolRenderData<I, D> {
         public message: RenderMessage,
         public client: LangGraphClient
     ) {}
+    private getToolActionRequestID() {
+        return createActionRequestID({
+            name: this.message.name!,
+            args: this.getInputRepaired(),
+        });
+    }
     /**
      * 获取人机交互数据
      * 直接使用 reviewConfig 获取可以显示的按钮
      * actionRequest 获取当前工具的入参
      */
     getHumanInTheLoopData() {
-        const configOfHumanInTheLoop = this.client.humanInTheLoop?.find((i) => i.value.reviewConfigs?.some((j) => j.actionName === this.message.name));
+        const toolActionRequestID = this.getToolActionRequestID();
+        if (!this.client.humanInTheLoop) return null;
+        const configOfHumanInTheLoop = this.client.humanInTheLoop.interruptData.find((i) => i.value.actionRequests.some((j) => j.id === toolActionRequestID));
         if (!configOfHumanInTheLoop) return null;
+
+        const actionRequestIndex = configOfHumanInTheLoop.value.actionRequests.findIndex((j) => j.id === toolActionRequestID);
+        if (actionRequestIndex === -1) return null;
+
         return {
+            actionRequestIndex: actionRequestIndex,
             config: configOfHumanInTheLoop,
-            reviewConfig: configOfHumanInTheLoop.value.reviewConfigs.find((j) => j.actionName === this.message.name)!,
-            actionRequest: configOfHumanInTheLoop.value.actionRequests.find((j) => j.name === this.message.name)!,
+            reviewConfig: configOfHumanInTheLoop.value.reviewConfigs.find((j) => j.actionName === configOfHumanInTheLoop.value.actionRequests[actionRequestIndex].name)!,
+            actionRequest: configOfHumanInTheLoop.value.actionRequests[actionRequestIndex],
+            result: this.client.humanInTheLoop.result[toolActionRequestID],
         };
     }
     /** 发送恢复状态的数据 */
-    sendResumeData(response: InterruptResponse["decisions"][number]) {
+    sendResumeData(response: HumanInTheLoopDecision) {
         if (response.type === "edit") {
             /**@ts-ignore 修复 sb 的 langchain 官方的命名不统一，我们一致采用下划线版本，而非驼峰版本 */
             response.editedAction = response.edited_action;
         }
-        return this.client.doneFEToolWaiting(this.message.id!, { decisions: [response] });
+
+        return this.client.doneHumanInTheLoopWaiting(this.message.id!, this.getToolActionRequestID(), response);
     }
     get state() {
         if (this.message.type === "tool" && this.message?.additional_kwargs?.done) {
             return "done";
         }
-        if (this.client.status === "interrupted" && this.client.humanInTheLoop?.some((i) => i.value.reviewConfigs?.some((j) => j.actionName === this.message.name))) {
-            return "interrupted";
+        const humanInTheLoopData = this.getHumanInTheLoopData();
+        if (humanInTheLoopData?.result) {
+            return "done";
+        }
+        if (this.client.status === "interrupted" && humanInTheLoopData?.actionRequest) {
+            return "done";
         }
         if (this.message.tool_input) {
             return "loading";

package/src/tool/createTool.ts

@@ -1,9 +1,10 @@
 import { actionParametersToJsonSchema, convertJsonSchemaToZodRawShape } from "./utils.js";
-import { z, ZodRawShape } from "zod";
+import { uuidv4, z, ZodRawShape } from "zod";
 import { Action, Parameter } from "./copilotkit-actions.js";
 import { zodToJsonSchema } from "zod-to-json-schema";
 import { Message } from "@langchain/langgraph-sdk";
 import { ToolRenderData } from "./ToolUI.js";
+import { HumanInTheLoopDecision, InterruptResponse as HumanInTheLoopResponse } from "../humanInTheLoop.js";
 
 export interface UnionTool<Args extends ZodRawShape, Child extends Object = Object, ResponseType = any> {
     name: string;
@@ -25,23 +26,8 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     isPureParams?: boolean;
 }
 export type ToolCallback<Args extends ZodRawShape> = (args: z.infer<z.ZodObject<Args>>, context?: any) => CallToolResult | Promise<CallToolResult>;
-/**
- * HumanInTheLoop 的标准回复格式
- */
-export type InterruptResponse = {
-    decisions: (
-        | { type: "approve" }
-        | {
-              type: "edit";
-              edited_action: {
-                  name: string;
-                  args: Record<string, any>;
-              };
-          }
-        | { type: "reject"; message?: string }
-    )[];
-};
-export type CallToolResult = string | { type: "text"; text: string }[] | InterruptResponse;
+
+export type CallToolResult = string | { type: "text"; text: string }[] | HumanInTheLoopDecision | HumanInTheLoopResponse;
 
 /** 用于格式校验 */
 export const createTool = <Args extends ZodRawShape>(tool: UnionTool<Args>) => {