@langgraph-js/sdk/dist/tool/ToolUI.js

The UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces

import { getMessageContent } from "../ui-store/createChatStore.js";
import { jsonrepair } from "jsonrepair";
import { createActionRequestID } from "../humanInTheLoop.js";
export class ToolRenderData {
    message;
    client;
    constructor(message, client) {
        this.message = message;
        this.client = client;
    }
    getToolActionRequestID() {
        return createActionRequestID({
            name: this.message.name,
            args: this.getInputRepaired(),
        });
    }
    /**
     * 获取人机交互数据
     * 直接使用 reviewConfig 获取可以显示的按钮
     * actionRequest 获取当前工具的入参
     */
    getHumanInTheLoopData() {
        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 === configOfHumanInTheLoop.value.actionRequests[actionRequestIndex].name),
            actionRequest: configOfHumanInTheLoop.value.actionRequests[actionRequestIndex],
            result: this.client.humanInTheLoop.result[toolActionRequestID],
        };
    }
    /** 发送恢复状态的数据 */
    sendResumeData(response) {
        if (response.type === "edit") {
            /**@ts-ignore 修复 sb 的 langchain 官方的命名不统一，我们一致采用下划线版本，而非驼峰版本 */
            response.editedAction = response.edited_action;
        }
        return this.client.doneHumanInTheLoopWaiting(this.message.id, this.getToolActionRequestID(), response);
    }
    get state() {
        if (this.message.type === "tool" && this.message?.additional_kwargs?.done) {
            return "done";
        }
        const humanInTheLoopData = this.getHumanInTheLoopData();
        if (humanInTheLoopData?.result) {
            return "done";
        }
        if (this.client.status === "interrupted" && humanInTheLoopData?.actionRequest) {
            return "interrupted";
        }
        if (this.message.tool_input) {
            return "loading";
        }
        return "idle";
    }
    get input() {
        try {
            return JSON.parse(this.message.tool_input);
        }
        catch (e) {
            return null;
        }
    }
    get output() {
        return getMessageContent(this.message.content);
    }
    getJSONOutput() {
        return JSON.parse(this.output);
    }
    /** 如果解析失败，则返回 null */
    getJSONOutputSafe() {
        try {
            return JSON.parse(this.output);
        }
        catch (e) {
            return null;
        }
    }
    getInputRepaired() {
        try {
            return JSON.parse(jsonrepair(this.message.tool_input || ""));
        }
        catch (e) {
            return {};
        }
    }
    response(data) {
        this.client.doneFEToolWaiting(this.message.id, JSON.stringify(data));
    }
}