@langgraph-js/sdk 3.6.0 → 3.7.1

package/README.md

@@ -57,6 +57,35 @@ pnpm add @langgraph-js/sdk
 
 - ✅ Read History from LangGraph
 
+## Legacy Mode
+
+Legacy mode is designed to be compatible with environments that don't support `AsyncGeneratorFunction` (such as WeChat Mini Programs). In these environments, standard async iterators may not work properly.
+
+### Legacy Mode Example
+
+```typescript
+import { TestLangGraphChat, createChatStore, createLowerJSClient } from "@langgraph-js/sdk";
+
+const client = await createLowerJSClient({
+    apiUrl: "http://localhost:8123",
+    defaultHeaders: {
+        Authorization: "Bearer 123",
+    },
+});
+
+createChatStore(
+    "graph",
+    {
+        defaultHeaders: {
+            Authorization: "Bearer 123",
+        },
+        client,
+        legacyMode: true,
+    },
+    {}
+);
+```
+
 ## Advanced Usage
 
 ### Creating a Chat Store

package/dist/LangGraphClient.d.ts

@@ -84,6 +84,8 @@ export interface LangGraphClientConfig {
     defaultHeaders?: Record<string, string | null | undefined>;
     /** 自定义客户端实现，如果不提供则使用官方 Client */
     client: ILangGraphClient<any>;
+    /** 是否使用 legacy 模式，默认 false */
+    legacyMode?: boolean;
 }
 export interface LangGraphEvents {
     /** 流开始事件 */
@@ -136,11 +138,15 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
     stopController: AbortController | null;
     /** Message 处理器 */
     private messageProcessor;
+    private legacyMode;
     constructor(config: LangGraphClientConfig);
     /** 代理 assistants 属性到内部 client */
     get assistants(): {
         search(query?: {
-            graphId?: string;
+            graphId? /**
+             * The maximum number of concurrent calls that can be made.
+             * Defaults to `Infinity`, which means no limit.
+             */: string;
             metadata?: import("@langchain/langgraph-sdk").Metadata;
             limit?: number;
             offset?: number;
@@ -248,6 +254,12 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
         id: string;
         name: string;
     };
+    /**
+     * @zh 处理流式响应的单个 chunk。
+     * @en Processes a single chunk from the stream response.
+     * @returns 是否需要跳过后续处理 (continue)
+     */
+    private processStreamChunk;
     private runFETool;
     private callFETool;
     extraParams: Record<string, any>;

package/dist/LangGraphClient.js

@@ -2,25 +2,27 @@ import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
 import { MessageProcessor } from "./MessageProcessor.js";
 import { revertChatTo } from "./time-travel/index.js";
+import camelcaseKeys from "camelcase-keys";
 /**
  * @zh LangGraphClient 类是与 LangGraph 后端交互的主要客户端。
  * @en The LangGraphClient class is the main client for interacting with the LangGraph backend.
  */
 export class LangGraphClient extends EventEmitter {
+    client;
+    currentAssistant = null;
+    currentThread = null;
+    tools = new ToolManager();
+    availableAssistants = [];
+    graphState = {};
+    currentRun;
+    stopController = null;
+    /** Message 处理器 */
+    messageProcessor;
+    legacyMode;
     constructor(config) {
         super();
-        this.currentAssistant = null;
-        this.currentThread = null;
-        this.tools = new ToolManager();
-        this.availableAssistants = [];
-        this.graphState = {};
-        this.stopController = null;
-        this.messagesMetadata = {};
-        this.humanInTheLoop = null;
-        /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
-        this.graphPosition = "";
-        this.extraParams = {};
         this.client = config.client;
+        this.legacyMode = config.legacyMode ?? false;
         this.messageProcessor = new MessageProcessor();
     }
     /** 代理 assistants 属性到内部 client */
@@ -93,8 +95,7 @@ export class LangGraphClient extends EventEmitter {
         }
     }
     graphVisualize() {
-        var _a;
-        return this.assistants.getGraph((_a = this.currentAssistant) === null || _a === void 0 ? void 0 : _a.assistant_id, {
+        return this.assistants.getGraph(this.currentAssistant?.assistant_id, {
             xray: true,
         });
     }
@@ -116,11 +117,10 @@ export class LangGraphClient extends EventEmitter {
      * @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;
-        const graphMessages = ((_a = this.graphState) === null || _a === void 0 ? void 0 : _a.messages) || [];
+        const graphMessages = this.graphState?.messages || [];
         this.messageProcessor.setGraphMessages(graphMessages);
         this.emit("value", {
             event: "messages/partial",
@@ -133,7 +133,7 @@ export class LangGraphClient extends EventEmitter {
     // 从历史中恢复时，应该恢复流式状态
     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");
+        const runningRun = runs?.find((run) => run.status === "running" || run.status === "pending");
         if (runningRun) {
             await this.sendMessage([], { joinRunId: runningRun.run_id });
         }
@@ -154,14 +154,13 @@ export class LangGraphClient extends EventEmitter {
      */
     get tokenCounter() {
         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;
-                acc.input_tokens += ((_b = message.usage_metadata) === null || _b === void 0 ? void 0 : _b.input_tokens) || 0;
-                acc.output_tokens += ((_c = message.usage_metadata) === null || _c === void 0 ? void 0 : _c.output_tokens) || 0;
+                acc.total_tokens += message.usage_metadata?.total_tokens || 0;
+                acc.input_tokens += message.usage_metadata?.input_tokens || 0;
+                acc.output_tokens += message.usage_metadata?.output_tokens || 0;
             }
-            else if ((_d = message.response_metadata) === null || _d === void 0 ? void 0 : _d.usage) {
-                const usage = (_e = message.response_metadata) === null || _e === void 0 ? void 0 : _e.usage;
+            else if (message.response_metadata?.usage) {
+                const usage = message.response_metadata?.usage;
                 acc.total_tokens += usage.total_tokens || 0;
                 acc.input_tokens += usage.prompt_tokens || 0;
                 acc.output_tokens += usage.completion_tokens || 0;
@@ -175,21 +174,19 @@ export class LangGraphClient extends EventEmitter {
     }
     /** 前端工具人机交互时，锁住面板 */
     isFELocking(messages) {
-        var _a;
         const lastMessage = messages[messages.length - 1];
         if (!lastMessage) {
             return false;
         }
-        const tool = this.tools.getTool(lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.name);
-        return tool && tool.render && (lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.type) === "tool" && !((_a = lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.additional_kwargs) === null || _a === void 0 ? void 0 : _a.done);
+        const tool = this.tools.getTool(lastMessage?.name);
+        return tool && tool.render && lastMessage?.type === "tool" && !lastMessage?.additional_kwargs?.done;
     }
     /**
      * @zh 取消当前的 Run。
      * @en Cancels the current Run.
      */
     cancelRun() {
-        var _a, _b;
-        if (((_a = this.currentThread) === null || _a === void 0 ? void 0 : _a.thread_id) && ((_b = this.currentRun) === null || _b === void 0 ? void 0 : _b.run_id)) {
+        if (this.currentThread?.thread_id && this.currentRun?.run_id) {
             this.runs.cancel(this.currentThread.thread_id, this.currentRun.run_id);
         }
     }
@@ -204,6 +201,8 @@ export class LangGraphClient extends EventEmitter {
         this.messageProcessor.setGraphMessages(state.messages);
         return state;
     }
+    messagesMetadata = {};
+    humanInTheLoop = null;
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -229,12 +228,25 @@ export class LangGraphClient extends EventEmitter {
                     content: input,
                 },
             ];
+        const streamRecord = [];
+        this.emit("start", {
+            event: "start",
+        });
         const createStreamResponse = async () => {
-            if (_debug === null || _debug === void 0 ? void 0 : _debug.streamResponse) {
+            if (_debug?.streamResponse) {
                 return _debug.streamResponse;
             }
+            const onCallback = this.legacyMode
+                ? (chunk) => {
+                    streamRecord.push(chunk);
+                    this.processStreamChunk(chunk, command);
+                }
+                : undefined;
             if (joinRunId) {
-                return this.runs.joinStream(this.currentThread.thread_id, joinRunId);
+                return this.runs.joinStream(this.currentThread.thread_id, joinRunId, {
+                    /** @ts-ignore */
+                    onCallback,
+                });
             }
             return this.runs.stream(this.currentThread.thread_id, this.currentAssistant.assistant_id, {
                 input: {
@@ -247,51 +259,16 @@ export class LangGraphClient extends EventEmitter {
                 streamMode: ["messages", "values"],
                 streamSubgraphs: true,
                 command,
+                /** @ts-ignore 为兼容不支持 AsyncIterableFunction 的环境*/
+                onCallback,
             });
         };
         const streamResponse = await createStreamResponse();
-        const streamRecord = [];
-        this.emit("start", {
-            event: "start",
-        });
-        for await (const chunk of streamResponse) {
-            streamRecord.push(chunk);
-            if (chunk.event === "metadata") {
-                this.currentRun = chunk.data;
-            }
-            else if (chunk.event === "error" || chunk.event === "Error" || chunk.event === "__stream_error__") {
-                this.emit("error", chunk);
-            }
-            else if (chunk.event === "messages/metadata") {
-                Object.assign(this.messagesMetadata, chunk.data);
-                continue;
-            }
-            else if (chunk.event === "messages/partial" || chunk.event === "messages/complete") {
-                for (const message of chunk.data) {
-                    this.messageProcessor.updateStreamingMessage(message);
-                }
-                this.emit("message", chunk);
-                continue;
-            }
-            else if (chunk.event === "values") {
-                const data = chunk.data;
-                if (data.__interrupt__) {
-                    this.humanInTheLoop = data.__interrupt__;
-                }
-                else if (data.messages) {
-                    const isResume = !!(command === null || command === void 0 ? void 0 : command.resume);
-                    const isLongerThanLocal = data.messages.length >= this.messageProcessor.getGraphMessages().length;
-                    // resume 情况下，长度低于前端 message 的统统不接受
-                    if (!isResume || (isResume && isLongerThanLocal)) {
-                        this.messageProcessor.setGraphMessages(data.messages);
-                        this.emit("value", chunk);
-                    }
-                    this.graphState = chunk.data;
-                }
-                continue;
-            }
-            else if (chunk.event.startsWith("values|")) {
-                this.graphPosition = chunk.event.split("|")[1];
+        if (!this.legacyMode) {
+            // 正常的 JS 环境都可以执行，但是部分环境不支持 AsyncGeneratorFunction（比如 sb 的微信小程序）
+            for await (const chunk of streamResponse) {
+                streamRecord.push(chunk);
+                this.processStreamChunk(chunk, command);
             }
         }
         const data = await this.runFETool();
@@ -304,6 +281,8 @@ export class LangGraphClient extends EventEmitter {
         this.messageProcessor.clearStreamingMessages();
         return streamRecord;
     }
+    /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
+    graphPosition = "";
     getGraphPosition() {
         return this.graphPosition.split("|").map((i) => {
             const [name, id] = i.split(":");
@@ -317,14 +296,60 @@ export class LangGraphClient extends EventEmitter {
         const position = this.getGraphPosition();
         return position[position.length - 1];
     }
+    /**
+     * @zh 处理流式响应的单个 chunk。
+     * @en Processes a single chunk from the stream response.
+     * @returns 是否需要跳过后续处理 (continue)
+     */
+    processStreamChunk(chunk, command) {
+        if (chunk.event === "metadata") {
+            this.currentRun = chunk.data;
+        }
+        else if (chunk.event === "error" || chunk.event === "Error" || chunk.event === "__stream_error__") {
+            this.emit("error", chunk);
+        }
+        else if (chunk.event === "messages/metadata") {
+            Object.assign(this.messagesMetadata, chunk.data);
+            return true;
+        }
+        else if (chunk.event === "messages/partial" || chunk.event === "messages/complete") {
+            for (const message of chunk.data) {
+                this.messageProcessor.updateStreamingMessage(message);
+            }
+            this.emit("message", chunk);
+            return true;
+        }
+        else if (chunk.event === "values") {
+            const data = chunk.data;
+            if (data.__interrupt__) {
+                this.humanInTheLoop = camelcaseKeys(data.__interrupt__, {
+                    deep: true,
+                });
+            }
+            else if (data.messages) {
+                const isResume = !!command?.resume;
+                const isLongerThanLocal = data.messages.length >= this.messageProcessor.getGraphMessages().length;
+                // resume 情况下，长度低于前端 message 的统统不接受
+                if (!isResume || (isResume && isLongerThanLocal)) {
+                    this.messageProcessor.setGraphMessages(data.messages);
+                    this.emit("value", chunk);
+                }
+                this.graphState = chunk.data;
+            }
+            return true;
+        }
+        else if (chunk.event.startsWith("values|")) {
+            this.graphPosition = chunk.event.split("|")[1];
+        }
+        return false;
+    }
     runFETool() {
-        var _a;
         const data = this.messageProcessor.getStreamingMessages(); // 需要保证不被清理
         const lastMessage = data[data.length - 1];
         if (!lastMessage)
             return;
         // 如果最后一条消息是前端工具消息，则调用工具
-        if (lastMessage.type === "ai" && ((_a = lastMessage.tool_calls) === null || _a === void 0 ? void 0 : _a.length)) {
+        if (lastMessage.type === "ai" && lastMessage.tool_calls?.length) {
             const result = lastMessage.tool_calls.map((tool) => {
                 const toolMessage = {
                     ...tool,
@@ -348,6 +373,7 @@ export class LangGraphClient extends EventEmitter {
         }
         return this.resume(result);
     }
+    extraParams = {};
     /**
      * @zh 继续被前端工具中断的流程。
      * @en Resumes a process interrupted by a frontend tool.
@@ -364,9 +390,8 @@ export class LangGraphClient extends EventEmitter {
      * @en Marks the frontend tool waiting as completed.
      */
     doneFEToolWaiting(id, result) {
-        var _a;
         const done = this.tools.doneWaiting(id, result);
-        if (!done && ((_a = this.currentThread) === null || _a === void 0 ? void 0 : _a.status) === "interrupted") {
+        if (!done && this.currentThread?.status === "interrupted") {
             this.resume(result);
         }
     }
@@ -389,8 +414,7 @@ export class LangGraphClient extends EventEmitter {
      * @en Resets the client state.
      */
     async reset() {
-        var _a;
-        await this.initAssistant((_a = this.currentAssistant) === null || _a === void 0 ? void 0 : _a.graph_id);
+        await this.initAssistant(this.currentAssistant?.graph_id);
         this.currentThread = null;
         this.graphState = {};
         this.messageProcessor.setGraphMessages([]);

package/dist/MessageProcessor.js

@@ -7,9 +7,8 @@ export class StreamingMessageType {
         return m.type === "tool";
     }
     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));
+        return m.type === "ai" && (m.tool_calls?.length || m.tool_call_chunks?.length);
     }
 }
 /**
@@ -17,12 +16,11 @@ export class StreamingMessageType {
  * @en The MessageProcessor class is used to uniformly handle Message-related logic and avoid duplicate processing.
  */
 export class MessageProcessor {
-    constructor() {
-        /** 流式消息缓存 */
-        this.streamingMessage = [];
-        /** 图发过来的更新信息 */
-        this.graphMessages = [];
-    }
+    /** 流式消息缓存 */
+    streamingMessage = [];
+    /** 图发过来的更新信息 */
+    graphMessages = [];
+    constructor() { }
     /**
      * @zh 获取流式消息
      * @en Get streaming messages
@@ -64,7 +62,7 @@ export class MessageProcessor {
      */
     updateStreamingMessage(message) {
         const lastMessage = this.streamingMessage[this.streamingMessage.length - 1];
-        if (!(lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.id) || message.id !== lastMessage.id) {
+        if (!lastMessage?.id || message.id !== lastMessage.id) {
             this.streamingMessage.push(message);
             return;
         }
@@ -100,15 +98,14 @@ export class MessageProcessor {
      * @en Attaches additional information to messages, such as spend time, unique ID, etc.
      */
     attachInfoForMessage(messages) {
-        var _a, _b, _c;
         let lastMessage = null;
         const result = [...messages]; // 创建副本避免修改原数组
         for (const message of result) {
-            const createTime = ((_a = message.response_metadata) === null || _a === void 0 ? void 0 : _a.create_time) || "";
+            const createTime = message.response_metadata?.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)) {
+            message.spend_time = new Date(createTime).getTime() - new Date(lastMessage?.response_metadata?.create_time || createTime).getTime();
+            if (!message.usage_metadata && message.response_metadata?.usage) {
                 const usage = message.response_metadata.usage;
                 message.usage_metadata = {
                     ...usage,
@@ -126,14 +123,13 @@ export class MessageProcessor {
      * @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) => {
+                (message.tool_calls || message.tool_call_chunks)?.forEach((element) => {
                     assistantToolMessages.set(element.id, element);
                     toolParentMessage.set(element.id, message);
                 });
@@ -145,23 +141,20 @@ export class MessageProcessor {
                 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,
-                        };
-                    }
+                    const isDone = !!message.content;
+                    message.done = isDone;
+                    message.additional_kwargs = {
+                        ...(parentMessage?.additional_kwargs || {}),
+                        ...(message.additional_kwargs || {}),
+                        done: isDone,
+                    };
                 }
                 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;
+                        message.name = parentMessage.tool_calls.find((i) => i.id === message.tool_call_id)?.name;
                     }
                 }
             }
@@ -174,7 +167,6 @@ export class MessageProcessor {
      * @en Generate messages used for streaming rendering in the UI
      */
     renderMessages(graphState, getGraphNodeNow, messagesMetadata) {
-        var _a;
         const previousMessage = new Map();
         const closedToolCallIds = new Set();
         const result = [];
@@ -196,18 +188,17 @@ export class MessageProcessor {
                 // 记录这个 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 tool_calls = m.tool_calls?.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,
+                        tool_input: m.additional_kwargs?.tool_calls?.[index]?.function?.arguments,
                         id: tool.id,
                         name: tool.name,
                         response_metadata: {},
@@ -234,14 +225,14 @@ export class MessageProcessor {
         return this.processMessages(result, graphState, messagesMetadata);
     }
     foldTreeMessages(messages, graphState, messagesMetadata) {
-        const state_sub_messages = Object.entries((graphState === null || graphState === void 0 ? void 0 : graphState.task_store) || {}).map(([key, value]) => [key, value.messages]);
+        const state_sub_messages = Object.entries(graphState?.task_store || {}).map(([key, value]) => [key, value.messages]);
         const state_sub_messages_map = new Map(state_sub_messages);
         const nonRootMessageId = new Set();
         const parentPointer = new Map(Object.entries(messagesMetadata || {})
             .map(([childId, metadata]) => {
-            if (metadata === null || metadata === void 0 ? void 0 : metadata.parent_id) {
+            if (metadata?.parent_id) {
                 nonRootMessageId.add(childId);
-                return [childId, metadata === null || metadata === void 0 ? void 0 : metadata.parent_id];
+                return [childId, metadata?.parent_id];
             }
             return;
         })

package/dist/SpendTime.js

@@ -3,9 +3,7 @@
  * @en The SpendTime class is used to calculate and record the time spent on operations.
  */
 export class SpendTime {
-    constructor() {
-        this.timeCounter = new Map();
-    }
+    timeCounter = new Map();
     /**
      * @zh 开始计时。
      * @en Starts timing.
@@ -18,8 +16,7 @@ export class SpendTime {
      * @en Ends timing.
      */
     end(key) {
-        var _a;
-        this.timeCounter.set(key, [((_a = this.timeCounter.get(key)) === null || _a === void 0 ? void 0 : _a[0]) || new Date(), new Date()]);
+        this.timeCounter.set(key, [this.timeCounter.get(key)?.[0] || new Date(), new Date()]);
     }
     /**
      * @zh 设置或更新指定键的耗时记录。如果键已存在，则更新结束时间；否则，开始新的计时。
@@ -38,16 +35,14 @@ export class SpendTime {
      * @en Gets the start time for the specified key.
      */
     getStartTime(key) {
-        var _a;
-        return ((_a = this.timeCounter.get(key)) === null || _a === void 0 ? void 0 : _a[0]) || new Date();
+        return this.timeCounter.get(key)?.[0] || new Date();
     }
     /**
      * @zh 获取指定键的结束时间。
      * @en Gets the end time for the specified key.
      */
     getEndTime(key) {
-        var _a;
-        return ((_a = this.timeCounter.get(key)) === null || _a === void 0 ? void 0 : _a[1]) || new Date();
+        return this.timeCounter.get(key)?.[1] || new Date();
     }
     /**
      * @zh 获取指定键的耗时（毫秒）。

package/dist/TestKit.js

@@ -4,18 +4,17 @@ export class TestLogger {
         console.log(message);
     }
     logMessage(message) {
-        var _a, _b, _c, _d, _e, _f, _g;
         const emoji = message.type === "ai" ? "🤖" : message.type === "human" ? "👤" : "🔧";
-        const header = `${emoji} ${message.type} | ${(_a = message.name) !== null && _a !== void 0 ? _a : "null"} | ${message.id}`;
+        const header = `${emoji} ${message.type} | ${message.name ?? "null"} | ${message.id}`;
         if (message.type === "tool") {
             return console.log(`${header}
-🔧 Input: ${(_c = (_b = message.tool_input) === null || _b === void 0 ? void 0 : _b.slice(0, 100)) !== null && _c !== void 0 ? _c : ""}
-💬 Output: ${(_e = (_d = message.content) === null || _d === void 0 ? void 0 : _d.slice(0, 100)) !== null && _e !== void 0 ? _e : ""}
+🔧 Input: ${message.tool_input?.slice(0, 100) ?? ""}
+💬 Output: ${message.content?.slice(0, 100) ?? ""}
 `);
         }
         console.log(`---
 ${header}
-💬 Output: ${(_g = (_f = message.content) === null || _f === void 0 ? void 0 : _f.slice(0, 100)) !== null && _g !== void 0 ? _g : ""}
+💬 Output: ${message.content?.slice(0, 100) ?? ""}
 `);
     }
 }
@@ -33,22 +32,23 @@ ${header}
  * ```
  */
 export class TestLangGraphChat {
+    store;
+    /** 是否开启调试模式 */
+    debug = false;
+    /** 上次消息数量，用于检测消息变化 */
+    lastLength = 0;
+    /** 待处理的测试任务列表 */
+    processFunc = [];
+    /** 自定义日志器 */
+    logger;
     /**
      * @zh 构造函数，初始化测试环境
      * @en Constructor, initialize test environment
      */
     constructor(store, options) {
-        var _a, _b;
         this.store = store;
-        /** 是否开启调试模式 */
-        this.debug = false;
-        /** 上次消息数量，用于检测消息变化 */
-        this.lastLength = 0;
-        /** 待处理的测试任务列表 */
-        this.processFunc = [];
-        this.readited = false;
-        this.debug = (_a = options.debug) !== null && _a !== void 0 ? _a : false;
-        this.logger = (_b = options.logger) !== null && _b !== void 0 ? _b : new TestLogger();
+        this.debug = options.debug ?? false;
+        this.logger = options.logger ?? new TestLogger();
         options.tools && this.addTools(options.tools);
         const renderMessages = this.store.data.renderMessages;
         // 订阅消息变化，自动检查任务完成状态
@@ -126,6 +126,7 @@ export class TestLangGraphChat {
             this.logger.logMessage(item);
         }
     }
+    readited = false;
     /**
      * @zh 准备测试环境，初始化客户端连接
      * @en Prepare test environment, initialize client connection

package/dist/ToolManager.js

@@ -4,11 +4,9 @@ import { createJSONDefineTool } from "./tool/createTool.js";
  * @en The ToolManager class is used to manage and execute tools.
  */
 export class ToolManager {
-    constructor() {
-        this.tools = new Map();
-        // === 专门为前端设计的异步触发结构
-        this.waitingMap = new Map();
-    }
+    tools = new Map();
+    // === 专门为前端设计的异步触发结构
+    waitingMap = new Map();
     /**
      * @zh 注册一个工具。
      * @en Registers a tool.
@@ -66,9 +64,8 @@ export class ToolManager {
      * @en Calls the tool with the specified name.
      */
     async callTool(name, args, context) {
-        var _a;
         const tool = this.getTool(name) || this.getTool("__default__");
-        return await ((_a = tool.execute) === null || _a === void 0 ? void 0 : _a.call(tool, args, context));
+        return await tool.execute?.(args, context);
     }
     /**
      * @zh 将所有工具转换为 JSON 定义格式。