@langgraph-js/sdk 1.12.0 → 2.0.0

package/dist/LangGraphClient.js

@@ -1,47 +1,38 @@
-import { Client } from "@langchain/langgraph-sdk";
+import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
-/**
- * @zh StreamingMessageType 类用于判断消息的类型。
- * @en The StreamingMessageType class is used to determine the type of a message.
- */
-export class StreamingMessageType {
-    static isUser(m) {
-        return m.type === "human";
-    }
-    static isTool(m) {
-        return m.type === "tool";
-    }
-    static isAssistant(m) {
-        return m.type === "ai" && !this.isToolAssistant(m);
-    }
-    static isToolAssistant(m) {
-        var _a, _b;
-        /** @ts-ignore */
-        return m.type === "ai" && (((_a = m.tool_calls) === null || _a === void 0 ? void 0 : _a.length) || ((_b = m.tool_call_chunks) === null || _b === void 0 ? void 0 : _b.length));
-    }
-}
+import { MessageProcessor } from "./MessageProcessor.js";
 /**
  * @zh LangGraphClient 类是与 LangGraph 后端交互的主要客户端。
  * @en The LangGraphClient class is the main client for interacting with the LangGraph backend.
  */
-export class LangGraphClient extends Client {
+export class LangGraphClient extends EventEmitter {
     constructor(config) {
-        super(config);
+        super();
         this.currentAssistant = null;
         this.currentThread = null;
-        this.streamingCallbacks = new Set();
         this.tools = new ToolManager();
         this.stopController = null;
         /** 用于存储 subAgent 状态数据的键 */
         this.subAgentsKey = "task_store";
         this.availableAssistants = [];
-        this.streamingMessage = [];
-        /** 图发过来的更新信息 */
-        this.graphMessages = [];
         this.graphState = {};
         /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
         this.graphPosition = "";
         this.extraParams = {};
+        this.client = config.client;
+        this.messageProcessor = new MessageProcessor(this.subAgentsKey);
+    }
+    /** 代理 assistants 属性到内部 client */
+    get assistants() {
+        return this.client.assistants;
+    }
+    /** 代理 threads 属性到内部 client */
+    get threads() {
+        return this.client.threads;
+    }
+    /** 代理 runs 属性到内部 client */
+    get runs() {
+        return this.client.runs;
     }
     listAssistants() {
         return this.assistants.search({
@@ -109,6 +100,9 @@ export class LangGraphClient extends Client {
             sortOrder: "desc",
         });
     }
+    async deleteThread(threadId) {
+        return this.threads.delete(threadId);
+    }
     /**
      * @zh 从历史中恢复 Thread 数据。
      * @en Resets the Thread data from history.
@@ -118,14 +112,12 @@ export class LangGraphClient extends Client {
         await this.initAssistant(agent);
         this.currentThread = await this.threads.get(threadId);
         this.graphState = this.currentThread.values;
-        this.graphMessages = ((_a = this.graphState) === null || _a === void 0 ? void 0 : _a.messages) || [];
-        this.emitStreamingUpdate({
-            type: "value",
+        const graphMessages = ((_a = this.graphState) === null || _a === void 0 ? void 0 : _a.messages) || [];
+        this.messageProcessor.setGraphMessages(graphMessages);
+        this.emit("value", {
+            event: "messages/partial",
             data: {
-                event: "messages/partial",
-                data: {
-                    messages: this.graphMessages,
-                },
+                messages: this.messageProcessor.getGraphMessages(),
             },
         });
         return this.currentThread;
@@ -139,224 +131,21 @@ export class LangGraphClient extends Client {
         }
     }
     cloneMessage(message) {
-        return JSON.parse(JSON.stringify(message));
-    }
-    updateStreamingMessage(message) {
-        const lastMessage = this.streamingMessage[this.streamingMessage.length - 1];
-        if (!(lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.id) || message.id !== lastMessage.id) {
-            this.streamingMessage.push(message);
-            return;
-        }
-        this.streamingMessage[this.streamingMessage.length - 1] = message;
-    }
-    /** 将 graphMessages 和 streamingMessage 合并，并返回新的消息数组 */
-    combineGraphMessagesWithStreamingMessages() {
-        const idMap = new Map(this.streamingMessage.map((i) => [i.id, i]));
-        return [
-            ...this.graphMessages.map((i) => {
-                if (idMap.has(i.id)) {
-                    const newValue = idMap.get(i.id);
-                    idMap.delete(i.id);
-                    return newValue;
-                }
-                return i;
-            }),
-            ...idMap.values(),
-        ];
+        return this.messageProcessor.cloneMessage(message);
     }
     /**
      * @zh 用于 UI 中的流式渲染中的消息。
      * @en Messages used for streaming rendering in the UI.
      */
     get renderMessage() {
-        var _a;
-        const previousMessage = new Map();
-        const closedToolCallIds = new Set();
-        const result = [];
-        const inputMessages = this.combineGraphMessagesWithStreamingMessages();
-        // console.log(inputMessages);
-        // 从后往前遍历，这样可以保证最新的消息在前面
-        for (let i = inputMessages.length - 1; i >= 0; i--) {
-            const message = this.cloneMessage(inputMessages[i]);
-            if (!message.id) {
-                result.unshift(message);
-                continue;
-            }
-            if (message.type === "ai") {
-                /** @ts-ignore */
-                if (!message.name)
-                    message.name = this.getGraphNodeNow().name;
-            }
-            if (StreamingMessageType.isToolAssistant(message)) {
-                const m = message;
-                // 记录这个 id 的消息，并添加到结果中
-                previousMessage.set(message.id, m);
-                /** @ts-ignore */
-                const tool_calls = ((_a = m.tool_calls) === null || _a === void 0 ? void 0 : _a.length) ? m.tool_calls : m.tool_call_chunks;
-                const new_tool_calls = tool_calls
-                    .filter((i) => {
-                    return !closedToolCallIds.has(i.id);
-                })
-                    .map((tool, index) => {
-                    var _a, _b, _c, _d;
-                    return {
-                        type: "tool",
-                        additional_kwargs: {},
-                        /** @ts-ignore */
-                        tool_input: (_d = (_c = (_b = (_a = m.additional_kwargs) === null || _a === void 0 ? void 0 : _a.tool_calls) === null || _b === void 0 ? void 0 : _b[index]) === null || _c === void 0 ? void 0 : _c.function) === null || _d === void 0 ? void 0 : _d.arguments,
-                        id: tool.id,
-                        name: tool.name,
-                        response_metadata: {},
-                        tool_call_id: tool.id,
-                        content: "",
-                    };
-                });
-                for (const tool of new_tool_calls) {
-                    if (!previousMessage.has(tool.id)) {
-                        result.unshift(tool);
-                        previousMessage.set(tool.id, tool);
-                    }
-                }
-                result.unshift(m);
-            }
-            else {
-                if (message.type === "tool" && message.tool_call_id) {
-                    closedToolCallIds.add(message.tool_call_id);
-                }
-                previousMessage.set(message.id, message);
-                result.unshift(message);
-            }
-        }
-        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 等。
-     * @en Attaches additional information to messages, such as spend time, unique ID, etc.
-     */
-    attachInfoForMessage(result) {
-        var _a, _b, _c;
-        let lastMessage = null;
-        for (const message of result) {
-            const createTime = ((_a = message.response_metadata) === null || _a === void 0 ? void 0 : _a.create_time) || "";
-            // 工具必须要使用 tool_call_id 来保证一致性
-            message.unique_id = message.tool_call_id || message.id;
-            message.spend_time = new Date(createTime).getTime() - new Date(((_b = lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.response_metadata) === null || _b === void 0 ? void 0 : _b.create_time) || createTime).getTime();
-            if (!message.usage_metadata && ((_c = message.response_metadata) === null || _c === void 0 ? void 0 : _c.usage)) {
-                const usage = message.response_metadata.usage;
-                message.usage_metadata = {
-                    ...usage,
-                    input_tokens: usage.prompt_tokens,
-                    output_tokens: usage.completion_tokens,
-                    total_tokens: usage.total_tokens,
-                };
-            }
-            lastMessage = message;
-        }
-        return result;
-    }
-    /**
-     * @zh 组合工具消息，将 AI 的工具调用和工具的执行结果关联起来。
-     * @en Composes tool messages, associating AI tool calls with tool execution results.
-     */
-    composeToolMessages(messages) {
-        var _a, _b;
-        const result = [];
-        const assistantToolMessages = new Map();
-        const toolParentMessage = new Map();
-        for (const message of messages) {
-            if (StreamingMessageType.isToolAssistant(message)) {
-                /** @ts-ignore 只有 tool_call_chunks 的 args 才是文本 */
-                (_a = (message.tool_calls || message.tool_call_chunks)) === null || _a === void 0 ? void 0 : _a.forEach((element) => {
-                    assistantToolMessages.set(element.id, element);
-                    toolParentMessage.set(element.id, message);
-                });
-                if (!message.content)
-                    continue;
-            }
-            if (StreamingMessageType.isTool(message) && !message.tool_input) {
-                const assistantToolMessage = assistantToolMessages.get(message.tool_call_id);
-                const parentMessage = toolParentMessage.get(message.tool_call_id);
-                if (assistantToolMessage) {
-                    message.tool_input = typeof assistantToolMessage.args !== "string" ? JSON.stringify(assistantToolMessage.args) : assistantToolMessage.args;
-                    if (message.additional_kwargs) {
-                        message.additional_kwargs.done = true;
-                        message.done = true;
-                    }
-                    else {
-                        message.done = true;
-                        message.additional_kwargs = {
-                            done: true,
-                        };
-                    }
-                }
-                if (parentMessage) {
-                    message.usage_metadata = parentMessage.usage_metadata;
-                    message.node_name = parentMessage.name;
-                    // 修补特殊情况下，tool name 丢失的问题
-                    if (!message.name) {
-                        message.name = (_b = parentMessage.tool_calls.find((i) => i.id === message.tool_call_id)) === null || _b === void 0 ? void 0 : _b.name;
-                    }
-                }
-            }
-            result.push(message);
-        }
-        return result;
+        return this.messageProcessor.renderMessages(this.graphState, () => this.getGraphNodeNow());
     }
     /**
      * @zh 获取 Token 计数器信息。
      * @en Gets the Token counter information.
      */
     get tokenCounter() {
-        return this.graphMessages.reduce((acc, message) => {
+        return this.messageProcessor.getGraphMessages().reduce((acc, message) => {
             var _a, _b, _c, _d, _e;
             if (message.usage_metadata) {
                 acc.total_tokens += ((_a = message.usage_metadata) === null || _a === void 0 ? void 0 : _a.total_tokens) || 0;
@@ -376,19 +165,6 @@ export class LangGraphClient extends Client {
             output_tokens: 0,
         });
     }
-    /**
-     * @zh 注册流式更新的回调函数。
-     * @en Registers a callback function for streaming updates.
-     */
-    onStreamingUpdate(callback) {
-        this.streamingCallbacks.add(callback);
-        return () => {
-            this.streamingCallbacks.delete(callback);
-        };
-    }
-    emitStreamingUpdate(event) {
-        this.streamingCallbacks.forEach((callback) => callback(event));
-    }
     /** 前端工具人机交互时，锁住面板 */
     isFELocking(messages) {
         var _a;
@@ -420,13 +196,10 @@ export class LangGraphClient extends Client {
         }
         if (!this.currentThread) {
             await this.createThread();
-            this.emitStreamingUpdate({
-                type: "thread",
+            this.emit("thread", {
+                event: "thread/create",
                 data: {
-                    event: "thread/create",
-                    data: {
-                        thread: this.currentThread,
-                    },
+                    thread: this.currentThread,
                 },
             });
         }
@@ -460,11 +233,8 @@ export class LangGraphClient extends Client {
         };
         const streamResponse = await createStreamResponse();
         const streamRecord = [];
-        this.emitStreamingUpdate({
-            type: "start",
-            data: {
-                event: "start",
-            },
+        this.emit("start", {
+            event: "start",
         });
         for await (const chunk of streamResponse) {
             streamRecord.push(chunk);
@@ -472,33 +242,24 @@ export class LangGraphClient extends Client {
                 this.currentRun = chunk.data;
             }
             else if (chunk.event === "error") {
-                this.emitStreamingUpdate({
-                    type: "error",
-                    data: chunk,
-                });
+                this.emit("error", chunk);
             }
             else if (chunk.event === "messages/partial") {
                 for (const message of chunk.data) {
-                    this.updateStreamingMessage(message);
+                    this.messageProcessor.updateStreamingMessage(message);
                 }
-                this.emitStreamingUpdate({
-                    type: "message",
-                    data: chunk,
-                });
+                this.emit("message", chunk);
                 continue;
             }
             else if (chunk.event === "values") {
                 const data = chunk.data;
                 if (data.messages) {
                     const isResume = !!(command === null || command === void 0 ? void 0 : command.resume);
-                    const isLongerThanLocal = data.messages.length >= this.graphMessages.length;
+                    const isLongerThanLocal = data.messages.length >= this.messageProcessor.getGraphMessages().length;
                     // resume 情况下，长度低于前端 message 的统统不接受
                     if (!isResume || (isResume && isLongerThanLocal)) {
-                        this.graphMessages = data.messages;
-                        this.emitStreamingUpdate({
-                            type: "value",
-                            data: chunk,
-                        });
+                        this.messageProcessor.setGraphMessages(data.messages);
+                        this.emit("value", chunk);
                     }
                     this.graphState = chunk.data;
                 }
@@ -507,7 +268,7 @@ export class LangGraphClient extends Client {
             else if (chunk.event.startsWith("values|")) {
                 // 这个 values 必然是子 values
                 if ((_a = chunk.data) === null || _a === void 0 ? void 0 : _a.messages) {
-                    this.mergeSubGraphMessagesToStreamingMessages(chunk.data.messages);
+                    this.messageProcessor.mergeSubGraphMessagesToStreamingMessages(chunk.data.messages);
                 }
                 this.graphPosition = chunk.event.split("|")[1];
             }
@@ -515,13 +276,10 @@ export class LangGraphClient extends Client {
         const data = await this.runFETool();
         if (data)
             streamRecord.push(...data);
-        this.emitStreamingUpdate({
-            type: "done",
-            data: {
-                event: "done",
-            },
+        this.emit("done", {
+            event: "done",
         });
-        this.streamingMessage = [];
+        this.messageProcessor.clearStreamingMessages();
         return streamRecord;
     }
     getGraphPosition() {
@@ -537,26 +295,9 @@ export class LangGraphClient extends Client {
         const position = this.getGraphPosition();
         return position[position.length - 1];
     }
-    /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
-    mergeSubGraphMessagesToStreamingMessages(messages) {
-        const map = new Map(messages.filter((i) => i.id).map((i) => [i.id, i]));
-        this.streamingMessage.forEach((i) => {
-            if (map.has(i.id)) {
-                const newValue = map.get(i.id);
-                Object.assign(i, newValue);
-                map.delete(i.id);
-            }
-        });
-        // 剩余的 message 一定不在 streamMessage 中
-        map.forEach((i) => {
-            if (i.type === "tool" && i.tool_call_id) {
-                this.streamingMessage.push(i);
-            }
-        });
-    }
     runFETool() {
         var _a;
-        const data = this.streamingMessage; // 需要保证不被清理
+        const data = this.messageProcessor.getStreamingMessages(); // 需要保证不被清理
         const lastMessage = data[data.length - 1];
         if (!lastMessage)
             return;
@@ -632,14 +373,14 @@ export class LangGraphClient extends Client {
         await this.initAssistant((_a = this.currentAssistant) === null || _a === void 0 ? void 0 : _a.graph_id);
         this.currentThread = null;
         this.graphState = {};
-        this.graphMessages = [];
-        this.streamingMessage = [];
+        this.messageProcessor.setGraphMessages([]);
+        this.messageProcessor.clearStreamingMessages();
         this.currentRun = undefined;
         this.tools.clearWaiting();
-        this.emitStreamingUpdate({
-            type: "value",
+        this.emit("value", {
+            event: "messages/partial",
             data: {
-                event: "messages/partial",
+                messages: [],
             },
         });
     }

package/dist/MessageProcessor.d.ts

@@ -0,0 +1,94 @@
+import { Message, AIMessage, ToolMessage } from "@langchain/langgraph-sdk";
+import { RenderMessage } from "./LangGraphClient.js";
+/**
+ * @zh StreamingMessageType 类用于判断消息的类型。
+ * @en The StreamingMessageType class is used to determine the type of a message.
+ */
+export declare class StreamingMessageType {
+    static isTool(m: Message): m is ToolMessage;
+    static isToolAssistant(m: Message): m is AIMessage;
+}
+/**
+ * @zh MessageProcessor 类用于统一处理 Message 相关的逻辑，避免重复处理。
+ * @en The MessageProcessor class is used to uniformly handle Message-related logic and avoid duplicate processing.
+ */
+export declare class MessageProcessor {
+    private subAgentsKey;
+    /** 流式消息缓存 */
+    private streamingMessage;
+    /** 图发过来的更新信息 */
+    private graphMessages;
+    constructor(subAgentsKey?: string);
+    /**
+     * @zh 获取流式消息
+     * @en Get streaming messages
+     */
+    getStreamingMessages(): RenderMessage[];
+    /**
+     * @zh 设置流式消息
+     * @en Set streaming messages
+     */
+    setStreamingMessages(messages: RenderMessage[]): void;
+    /**
+     * @zh 清空流式消息
+     * @en Clear streaming messages
+     */
+    clearStreamingMessages(): void;
+    /**
+     * @zh 获取图消息
+     * @en Get graph messages
+     */
+    getGraphMessages(): RenderMessage[];
+    /**
+     * @zh 设置图消息
+     * @en Set graph messages
+     */
+    setGraphMessages(messages: RenderMessage[]): void;
+    /**
+     * @zh 更新流式消息
+     * @en Update streaming message
+     */
+    updateStreamingMessage(message: RenderMessage): void;
+    /**
+     * @zh 将 graphMessages 和 streamingMessage 合并，并返回新的消息数组
+     * @en Combine graphMessages and streamingMessage and return a new message array
+     */
+    combineGraphMessagesWithStreamingMessages(): RenderMessage[];
+    /**
+     * @zh 子图的数据需要通过 merge 的方式重新进行合并更新
+     * @en Subgraph data needs to be merged and updated through merge method
+     */
+    mergeSubGraphMessagesToStreamingMessages(messages: Message[]): void;
+    /**
+     * @zh 克隆消息对象
+     * @en Clone message object
+     */
+    cloneMessage(message: Message): Message;
+    /**
+     * @zh 为消息附加额外的信息，如耗时、唯一 ID 等。
+     * @en Attaches additional information to messages, such as spend time, unique ID, etc.
+     */
+    attachInfoForMessage(messages: RenderMessage[]): RenderMessage[];
+    /**
+     * @zh 组合工具消息，将 AI 的工具调用和工具的执行结果关联起来。
+     * @en Composes tool messages, associating AI tool calls with tool execution results.
+     */
+    composeToolMessages(messages: RenderMessage[]): RenderMessage[];
+    /**
+     * @zh 转换 subAgent 消息为工具的子消息
+     * @en Convert subAgent messages to tool sub-messages
+     */
+    convertSubAgentMessages(messages: RenderMessage[], graphState: any): RenderMessage[];
+    /**
+     * @zh 生成用于 UI 中的流式渲染的消息
+     * @en Generate messages used for streaming rendering in the UI
+     */
+    renderMessages(graphState: any, getGraphNodeNow: () => {
+        name: string;
+    }): RenderMessage[];
+    /**
+     * @zh 统一的消息处理入口，按顺序执行所有处理步骤
+     * @en Unified message processing entry point, executing all processing steps in order
+     */
+    processMessages(messages: RenderMessage[], graphState?: any): RenderMessage[];
+}