@langgraph-js/sdk 1.9.1 → 1.10.0

package/dist/TestKit.d.ts

@@ -0,0 +1,204 @@
+import { RenderMessage } from "./LangGraphClient.js";
+import { Message } from "@langchain/langgraph-sdk";
+import { CallToolResult, UnionTool } from "./tool/createTool.js";
+import { createChatStore } from "./ui-store/createChatStore.js";
+/**
+ * @zh 测试任务接口
+ * @en Test task interface
+ */
+interface TestTask {
+    /** 任务是否成功完成 */
+    success: boolean;
+    /** 执行任务的函数 */
+    runTask: (messages: readonly RenderMessage[]) => Promise<void>;
+    /** 任务失败时的回调函数 */
+    fail: () => void;
+}
+/**
+ * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
+ * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
+ *
+ * @example
+ * ```typescript
+ * const testChat = new TestLangGraphChat(createLangGraphClient(), { debug: true });
+ * await testChat.humanInput("你好", async () => {
+ *     const aiMessage = await testChat.waitFor("ai");
+ *     expect(aiMessage.content).toBeDefined();
+ * });
+ * ```
+ */
+export declare class TestLangGraphChat {
+    readonly store: ReturnType<typeof createChatStore>;
+    /** 是否开启调试模式 */
+    private debug;
+    /** 上次消息数量，用于检测消息变化 */
+    private lastLength;
+    /** 待处理的测试任务列表 */
+    protected processFunc: TestTask[];
+    /**
+     * @zh 构造函数，初始化测试环境
+     * @en Constructor, initialize test environment
+     */
+    constructor(store: ReturnType<typeof createChatStore>, options: {
+        debug?: boolean;
+        tools?: UnionTool<any, any, any>[];
+    });
+    /**
+     * @zh 获取当前所有渲染消息
+     * @en Get all current render messages
+     */
+    getMessages(): RenderMessage[];
+    /**
+     * @zh 添加工具到测试环境中，会自动包装工具的 execute 方法
+     * @en Add tools to test environment, automatically wraps tool execute methods
+     *
+     * @example
+     * ```typescript
+     * const tools = [createUITool({ name: "test_tool", ... })];
+     * testChat.addTools(tools);
+     * ```
+     */
+    addTools(tools: UnionTool<any, any, any>[]): void;
+    /**
+     * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
+     * @en Check all pending test tasks, only executes when message count changes
+     */
+    checkAllTask(messages: readonly RenderMessage[], options?: {
+        skipLengthCheck?: boolean;
+    }): void;
+    /**
+     * @zh 准备测试环境，初始化客户端连接
+     * @en Prepare test environment, initialize client connection
+     */
+    ready(): Promise<import("./LangGraphClient.js").LangGraphClient>;
+    /**
+     * @zh 模拟人类输入消息并等待测试任务完成，这是测试的核心方法
+     * @en Simulate human input and wait for test tasks to complete, this is the core test method
+     *
+     * @example
+     * ```typescript
+     * await testChat.humanInput("请帮我思考一下", async () => {
+     *     const toolMessage = await testChat.waitFor("tool", "thinking");
+     *     expect(toolMessage.tool_input).toBeDefined();
+     *
+     *     const aiMessage = await testChat.waitFor("ai");
+     *     expect(aiMessage.content).toContain("思考");
+     * });
+     * ```
+     */
+    humanInput(text: Message["content"], context: () => Promise<void>): Promise<[void, void]>;
+    /**
+     * @zh 等待特定类型的消息出现，创建异步等待任务
+     * @en Wait for specific type of message to appear, creates async waiting task
+     *
+     * @example
+     * ```typescript
+     * // 等待 AI 回复
+     * const aiMessage = await testChat.waitFor("ai");
+     *
+     * // 等待特定工具调用
+     * const toolMessage = await testChat.waitFor("tool", "sequential-thinking");
+     * ```
+     */
+    waitFor<D extends "tool" | "ai", T extends RenderMessage, N extends D extends "tool" ? string : undefined>(type: D, name?: N): Promise<T>;
+    /**
+     * @zh 响应前端工具调用，模拟用户对工具的响应
+     * @en Respond to frontend tool calls, simulates user response to tools
+     *
+     * @example
+     * ```typescript
+     * const toolMessage = await testChat.waitFor("tool", "ask_user_for_approve");
+     * await testChat.responseFeTool(toolMessage, "approved");
+     * ```
+     */
+    responseFeTool(message: RenderMessage, value: CallToolResult): Promise<RenderMessage>;
+    /**
+     * @zh 查找最后一条指定类型的消息，从消息数组末尾开始向前查找
+     * @en Find the last message of specified type, searches backwards from end of messages
+     *
+     * @example
+     * ```typescript
+     * // 查找最后一条 AI 消息
+     * const lastAI = testChat.findLast("ai");
+     *
+     * // 查找最后一条人类消息
+     * const lastHuman = testChat.findLast("human");
+     * ```
+     */
+    findLast(type: "human" | "ai" | "tool", options?: {
+        before?: (item: RenderMessage) => boolean;
+    }): (import("@langchain/langgraph-sdk").HumanMessage & {
+        name?: string;
+        node_name?: string;
+        tool_input?: string;
+        additional_kwargs?: {
+            done?: boolean;
+            tool_calls?: {
+                function: {
+                    arguments: string;
+                };
+            }[];
+        };
+        usage_metadata?: {
+            total_tokens: number;
+            input_tokens: number;
+            output_tokens: number;
+        };
+        tool_call_id?: string;
+        response_metadata?: {
+            create_time: string;
+        };
+        spend_time?: number;
+        unique_id?: string;
+        done?: boolean;
+    }) | (import("@langchain/langgraph-sdk").AIMessage & {
+        name?: string;
+        node_name?: string;
+        tool_input?: string;
+        additional_kwargs?: {
+            done?: boolean;
+            tool_calls?: {
+                function: {
+                    arguments: string;
+                };
+            }[];
+        };
+        usage_metadata?: {
+            total_tokens: number;
+            input_tokens: number;
+            output_tokens: number;
+        };
+        tool_call_id?: string;
+        response_metadata?: {
+            create_time: string;
+        };
+        spend_time?: number;
+        unique_id?: string;
+        done?: boolean;
+    }) | (import("@langchain/langgraph-sdk").ToolMessage & {
+        name?: string;
+        node_name?: string;
+        tool_input?: string;
+        additional_kwargs?: {
+            done?: boolean;
+            tool_calls?: {
+                function: {
+                    arguments: string;
+                };
+            }[];
+        };
+        usage_metadata?: {
+            total_tokens: number;
+            input_tokens: number;
+            output_tokens: number;
+        };
+        tool_call_id?: string;
+        response_metadata?: {
+            create_time: string;
+        };
+        spend_time?: number;
+        unique_id?: string;
+        done?: boolean;
+    });
+}
+export {};

package/dist/TestKit.js

@@ -0,0 +1,226 @@
+import { ToolRenderData } from "./tool/ToolUI.js";
+/**
+ * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
+ * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
+ *
+ * @example
+ * ```typescript
+ * const testChat = new TestLangGraphChat(createLangGraphClient(), { debug: true });
+ * await testChat.humanInput("你好", async () => {
+ *     const aiMessage = await testChat.waitFor("ai");
+ *     expect(aiMessage.content).toBeDefined();
+ * });
+ * ```
+ */
+export class TestLangGraphChat {
+    /**
+     * @zh 构造函数，初始化测试环境
+     * @en Constructor, initialize test environment
+     */
+    constructor(store, options) {
+        var _a;
+        this.store = store;
+        /** 是否开启调试模式 */
+        this.debug = false;
+        /** 上次消息数量，用于检测消息变化 */
+        this.lastLength = 0;
+        /** 待处理的测试任务列表 */
+        this.processFunc = [];
+        this.debug = (_a = options.debug) !== null && _a !== void 0 ? _a : false;
+        options.tools && this.addTools(options.tools);
+        const renderMessages = this.store.data.renderMessages;
+        // 订阅消息变化，自动检查任务完成状态
+        renderMessages.subscribe((messages) => {
+            this.checkAllTask(messages);
+        });
+    }
+    /**
+     * @zh 获取当前所有渲染消息
+     * @en Get all current render messages
+     */
+    getMessages() {
+        return this.store.data.renderMessages.get();
+    }
+    /**
+     * @zh 添加工具到测试环境中，会自动包装工具的 execute 方法
+     * @en Add tools to test environment, automatically wraps tool execute methods
+     *
+     * @example
+     * ```typescript
+     * const tools = [createUITool({ name: "test_tool", ... })];
+     * testChat.addTools(tools);
+     * ```
+     */
+    addTools(tools) {
+        tools.forEach((tool) => {
+            if (tool.execute) {
+                const oldExecute = tool.execute;
+                // 包装原始的 execute 方法，在执行后触发任务检查
+                tool.execute = (...args) => {
+                    setTimeout(() => {
+                        this.checkAllTask(this.getMessages(), {
+                            skipLengthCheck: true,
+                        });
+                    }, 100);
+                    return oldExecute(...args);
+                };
+            }
+        });
+        this.store.mutations.setTools(tools);
+    }
+    /**
+     * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
+     * @en Check all pending test tasks, only executes when message count changes
+     */
+    checkAllTask(messages, options = {}) {
+        // 只有 lastLength 发生变化时，才执行检查
+        if (!options.skipLengthCheck && this.lastLength === messages.length) {
+            return;
+        }
+        this.lastLength = messages.length;
+        // 执行所有未完成的任务
+        for (const task of this.processFunc) {
+            !task.success && task.runTask(options.skipLengthCheck ? messages : messages.slice(0, -1));
+        }
+        // 调试模式下打印最新消息
+        if (this.debug) {
+            console.log(messages[messages.length - (options.skipLengthCheck ? 1 : 2)]);
+        }
+    }
+    /**
+     * @zh 准备测试环境，初始化客户端连接
+     * @en Prepare test environment, initialize client connection
+     */
+    ready() {
+        return this.store.mutations.initClient();
+    }
+    /**
+     * @zh 模拟人类输入消息并等待测试任务完成，这是测试的核心方法
+     * @en Simulate human input and wait for test tasks to complete, this is the core test method
+     *
+     * @example
+     * ```typescript
+     * await testChat.humanInput("请帮我思考一下", async () => {
+     *     const toolMessage = await testChat.waitFor("tool", "thinking");
+     *     expect(toolMessage.tool_input).toBeDefined();
+     *
+     *     const aiMessage = await testChat.waitFor("ai");
+     *     expect(aiMessage.content).toContain("思考");
+     * });
+     * ```
+     */
+    async humanInput(text, context) {
+        await this.ready();
+        // console.log(text);
+        return Promise.all([
+            context(),
+            this.store.mutations
+                .sendMessage([
+                {
+                    type: "human",
+                    content: text,
+                },
+            ])
+                .then(() => {
+                this.checkAllTask(this.getMessages(), {
+                    skipLengthCheck: true,
+                });
+            })
+                .then(async (res) => {
+                // 检查是否还有未完成的任务
+                const tasks = this.processFunc.filter((i) => {
+                    return !i.success;
+                });
+                if (tasks.length) {
+                    console.warn("still have ", tasks.length, " tasks");
+                    await Promise.all(tasks.map((i) => i.fail()));
+                    throw new Error("test task failed");
+                }
+                this.processFunc = [];
+                return res;
+            }),
+        ]);
+    }
+    /**
+     * @zh 等待特定类型的消息出现，创建异步等待任务
+     * @en Wait for specific type of message to appear, creates async waiting task
+     *
+     * @example
+     * ```typescript
+     * // 等待 AI 回复
+     * const aiMessage = await testChat.waitFor("ai");
+     *
+     * // 等待特定工具调用
+     * const toolMessage = await testChat.waitFor("tool", "sequential-thinking");
+     * ```
+     */
+    waitFor(type, name) {
+        return new Promise((resolve, reject) => {
+            this.processFunc.push({
+                success: false,
+                async runTask(messages) {
+                    const lastMessage = messages[messages.length - 1];
+                    if (!lastMessage) {
+                        return;
+                    }
+                    // 检查消息类型和名称是否匹配
+                    if (lastMessage.type === type && (name ? lastMessage.name === name : true)) {
+                        resolve(lastMessage);
+                        this.success = true;
+                    }
+                },
+                fail() {
+                    reject(new Error(`wait for ${type} ${name} failed`));
+                },
+            });
+        });
+    }
+    /**
+     * @zh 响应前端工具调用，模拟用户对工具的响应
+     * @en Respond to frontend tool calls, simulates user response to tools
+     *
+     * @example
+     * ```typescript
+     * const toolMessage = await testChat.waitFor("tool", "ask_user_for_approve");
+     * await testChat.responseFeTool(toolMessage, "approved");
+     * ```
+     */
+    async responseFeTool(message, value) {
+        if (message.content) {
+            throw new Error(`message is Done. content: ${message.content}`);
+        }
+        const tool = new ToolRenderData(message, this.store.data.client.get());
+        tool.response(value);
+        const messages = await this.waitFor("tool", message.name);
+        if (messages.content) {
+            return messages;
+        }
+        throw new Error("tool response failed");
+    }
+    /**
+     * @zh 查找最后一条指定类型的消息，从消息数组末尾开始向前查找
+     * @en Find the last message of specified type, searches backwards from end of messages
+     *
+     * @example
+     * ```typescript
+     * // 查找最后一条 AI 消息
+     * const lastAI = testChat.findLast("ai");
+     *
+     * // 查找最后一条人类消息
+     * const lastHuman = testChat.findLast("human");
+     * ```
+     */
+    findLast(type, options = {}) {
+        const messages = this.getMessages();
+        for (let i = messages.length - 1; i >= 0; i--) {
+            const item = messages[i];
+            if (type === item.type) {
+                return item;
+            }
+            if (options.before && options.before(item)) {
+                throw new Error(`${type} not found; before specified`);
+            }
+        }
+        throw new Error(`${type} not found `);
+    }
+}

package/dist/ToolManager.d.ts

@@ -55,12 +55,7 @@ export declare class ToolManager {
     toJSON(graphId: string, remote?: boolean): Promise<{
         name: string;
         description: string;
-        parameters: import("zod-to-json-schema").JsonSchema7Type & {
-            $schema?: string | undefined;
-            definitions?: {
-                [key: string]: import("zod-to-json-schema").JsonSchema7Type;
-            } | undefined;
-        };
+        parameters: any;
     }[]>;
     /**
      * @zh 标记指定 ID 的工具等待已完成，并传递结果。

package/dist/index.d.ts

@@ -3,3 +3,4 @@ export * from "./tool/index.js";
 export * from "@langchain/langgraph-sdk";
 export * from "./ui-store/index.js";
 export * from "./ToolManager.js";
+export * from "./TestKit.js";

package/dist/index.js

@@ -3,3 +3,4 @@ export * from "./tool/index.js";
 export * from "@langchain/langgraph-sdk";
 export * from "./ui-store/index.js";
 export * from "./ToolManager.js";
+export * from "./TestKit.js";

package/dist/tool/createTool.d.ts

@@ -18,6 +18,8 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     allowAgent?: string[];
     /** 只允许指定的 Graph 使用该工具 */
     allowGraph?: string[];
+    /** 是否是纯净的 json schema 参数，而不是 zod 参数 */
+    isPureParams?: boolean;
 }
 export type ToolCallback<Args extends ZodRawShape> = (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => CallToolResult | Promise<CallToolResult>;
 export type CallToolResult = string | {
@@ -43,12 +45,17 @@ export declare const createFETool: <const T extends Parameter[], Args extends Zo
 export declare const createJSONDefineTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => {
     name: string;
     description: string;
-    parameters: import("zod-to-json-schema").JsonSchema7Type & {
+    parameters: Args | ({
+        title?: string;
+        default?: any;
+        description?: string;
+        markdownDescription?: string;
+    } & {
         $schema?: string | undefined;
         definitions?: {
             [key: string]: import("zod-to-json-schema").JsonSchema7Type;
         } | undefined;
-    };
+    });
 };
 export declare const createMCPTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => (string | Args | ((args: z.objectOutputType<Args, ZodTypeAny>) => Promise<{
     content: {

package/dist/tool/createTool.js

@@ -63,7 +63,7 @@ export const createJSONDefineTool = (tool) => {
     return {
         name: tool.name,
         description: tool.description,
-        parameters: zodToJsonSchema(z.object(tool.parameters)),
+        parameters: tool.isPureParams ? tool.parameters : zodToJsonSchema(z.object(tool.parameters)),
     };
 };
 export const createMCPTool = (tool) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "1.9.1",
+  "version": "1.10.0",
   "description": "The UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces",
   "main": "dist/index.js",
   "type": "module",

package/src/TestKit.ts

@@ -0,0 +1,262 @@
+import { RenderMessage } from "./LangGraphClient.js";
+import { Message } from "@langchain/langgraph-sdk";
+import { CallToolResult, UnionTool } from "./tool/createTool.js";
+import { ToolRenderData } from "./tool/ToolUI.js";
+import { createChatStore } from "./ui-store/createChatStore.js";
+
+/**
+ * @zh 测试任务接口
+ * @en Test task interface
+ */
+interface TestTask {
+    /** 任务是否成功完成 */
+    success: boolean;
+    /** 执行任务的函数 */
+    runTask: (messages: readonly RenderMessage[]) => Promise<void>;
+    /** 任务失败时的回调函数 */
+    fail: () => void;
+}
+
+/**
+ * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
+ * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
+ *
+ * @example
+ * ```typescript
+ * const testChat = new TestLangGraphChat(createLangGraphClient(), { debug: true });
+ * await testChat.humanInput("你好", async () => {
+ *     const aiMessage = await testChat.waitFor("ai");
+ *     expect(aiMessage.content).toBeDefined();
+ * });
+ * ```
+ */
+export class TestLangGraphChat {
+    /** 是否开启调试模式 */
+    private debug = false;
+    /** 上次消息数量，用于检测消息变化 */
+    private lastLength = 0;
+    /** 待处理的测试任务列表 */
+    protected processFunc: TestTask[] = [];
+
+    /**
+     * @zh 构造函数，初始化测试环境
+     * @en Constructor, initialize test environment
+     */
+    constructor(
+        readonly store: ReturnType<typeof createChatStore>,
+        options: {
+            debug?: boolean;
+            tools?: UnionTool<any, any, any>[];
+        }
+    ) {
+        this.debug = options.debug ?? false;
+        options.tools && this.addTools(options.tools);
+        const renderMessages = this.store.data.renderMessages;
+
+        // 订阅消息变化，自动检查任务完成状态
+        renderMessages.subscribe((messages) => {
+            this.checkAllTask(messages);
+        });
+    }
+
+    /**
+     * @zh 获取当前所有渲染消息
+     * @en Get all current render messages
+     */
+    getMessages() {
+        return this.store.data.renderMessages.get();
+    }
+
+    /**
+     * @zh 添加工具到测试环境中，会自动包装工具的 execute 方法
+     * @en Add tools to test environment, automatically wraps tool execute methods
+     *
+     * @example
+     * ```typescript
+     * const tools = [createUITool({ name: "test_tool", ... })];
+     * testChat.addTools(tools);
+     * ```
+     */
+    addTools(tools: UnionTool<any, any, any>[]) {
+        tools.forEach((tool) => {
+            if (tool.execute) {
+                const oldExecute = tool.execute;
+                // 包装原始的 execute 方法，在执行后触发任务检查
+                tool.execute = (...args) => {
+                    setTimeout(() => {
+                        this.checkAllTask(this.getMessages(), {
+                            skipLengthCheck: true,
+                        });
+                    }, 100);
+                    return oldExecute!(...args);
+                };
+            }
+        });
+        this.store.mutations.setTools(tools);
+    }
+
+    /**
+     * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
+     * @en Check all pending test tasks, only executes when message count changes
+     */
+    checkAllTask(messages: readonly RenderMessage[], options: { skipLengthCheck?: boolean } = {}) {
+        // 只有 lastLength 发生变化时，才执行检查
+        if (!options.skipLengthCheck && this.lastLength === messages.length) {
+            return;
+        }
+        this.lastLength = messages.length;
+
+        // 执行所有未完成的任务
+        for (const task of this.processFunc) {
+            !task.success && task.runTask(options.skipLengthCheck ? messages : messages.slice(0, -1));
+        }
+
+        // 调试模式下打印最新消息
+        if (this.debug) {
+            console.log(messages[messages.length - (options.skipLengthCheck ? 1 : 2)]);
+        }
+    }
+
+    /**
+     * @zh 准备测试环境，初始化客户端连接
+     * @en Prepare test environment, initialize client connection
+     */
+    ready() {
+        return this.store.mutations.initClient();
+    }
+
+    /**
+     * @zh 模拟人类输入消息并等待测试任务完成，这是测试的核心方法
+     * @en Simulate human input and wait for test tasks to complete, this is the core test method
+     *
+     * @example
+     * ```typescript
+     * await testChat.humanInput("请帮我思考一下", async () => {
+     *     const toolMessage = await testChat.waitFor("tool", "thinking");
+     *     expect(toolMessage.tool_input).toBeDefined();
+     *
+     *     const aiMessage = await testChat.waitFor("ai");
+     *     expect(aiMessage.content).toContain("思考");
+     * });
+     * ```
+     */
+    async humanInput(text: Message["content"], context: () => Promise<void>) {
+        await this.ready();
+        // console.log(text);
+        return Promise.all([
+            context(),
+            this.store.mutations
+                .sendMessage([
+                    {
+                        type: "human",
+                        content: text,
+                    },
+                ])
+                .then(() => {
+                    this.checkAllTask(this.getMessages(), {
+                        skipLengthCheck: true,
+                    });
+                })
+                .then(async (res) => {
+                    // 检查是否还有未完成的任务
+                    const tasks = this.processFunc.filter((i) => {
+                        return !i.success;
+                    });
+                    if (tasks.length) {
+                        console.warn("still have ", tasks.length, " tasks");
+                        await Promise.all(tasks.map((i) => i.fail()));
+                        throw new Error("test task failed");
+                    }
+                    this.processFunc = [];
+                    return res;
+                }),
+        ]);
+    }
+
+    /**
+     * @zh 等待特定类型的消息出现，创建异步等待任务
+     * @en Wait for specific type of message to appear, creates async waiting task
+     *
+     * @example
+     * ```typescript
+     * // 等待 AI 回复
+     * const aiMessage = await testChat.waitFor("ai");
+     *
+     * // 等待特定工具调用
+     * const toolMessage = await testChat.waitFor("tool", "sequential-thinking");
+     * ```
+     */
+    waitFor<D extends "tool" | "ai", T extends RenderMessage, N extends D extends "tool" ? string : undefined>(type: D, name?: N): Promise<T> {
+        return new Promise((resolve, reject) => {
+            this.processFunc.push({
+                success: false,
+                async runTask(messages) {
+                    const lastMessage = messages[messages.length - 1];
+                    if (!lastMessage) {
+                        return;
+                    }
+                    // 检查消息类型和名称是否匹配
+                    if (lastMessage.type === type && (name ? lastMessage.name === name : true)) {
+                        resolve(lastMessage as T);
+                        this.success = true;
+                    }
+                },
+                fail() {
+                    reject(new Error(`wait for ${type} ${name} failed`));
+                },
+            });
+        });
+    }
+
+    /**
+     * @zh 响应前端工具调用，模拟用户对工具的响应
+     * @en Respond to frontend tool calls, simulates user response to tools
+     *
+     * @example
+     * ```typescript
+     * const toolMessage = await testChat.waitFor("tool", "ask_user_for_approve");
+     * await testChat.responseFeTool(toolMessage, "approved");
+     * ```
+     */
+    async responseFeTool(message: RenderMessage, value: CallToolResult) {
+        if (message.content) {
+            throw new Error(`message is Done. content: ${message.content}`);
+        }
+        const tool = new ToolRenderData(message, this.store.data.client.get()!);
+        tool.response(value);
+        const messages = await this.waitFor("tool", message.name!);
+
+        if (messages.content) {
+            return messages;
+        }
+        throw new Error("tool response failed");
+    }
+
+    /**
+     * @zh 查找最后一条指定类型的消息，从消息数组末尾开始向前查找
+     * @en Find the last message of specified type, searches backwards from end of messages
+     *
+     * @example
+     * ```typescript
+     * // 查找最后一条 AI 消息
+     * const lastAI = testChat.findLast("ai");
+     *
+     * // 查找最后一条人类消息
+     * const lastHuman = testChat.findLast("human");
+     * ```
+     */
+    findLast(type: "human" | "ai" | "tool", options: { before?: (item: RenderMessage) => boolean } = {}) {
+        const messages = this.getMessages();
+
+        for (let i = messages.length - 1; i >= 0; i--) {
+            const item = messages[i];
+            if (type === item.type) {
+                return item;
+            }
+            if (options.before && options.before(item)) {
+                throw new Error(`${type} not found; before specified`);
+            }
+        }
+        throw new Error(`${type} not found `);
+    }
+}

package/src/index.ts

@@ -3,3 +3,4 @@ export * from "./tool/index.js";
 export * from "@langchain/langgraph-sdk";
 export * from "./ui-store/index.js";
 export * from "./ToolManager.js";
+export * from "./TestKit.js";

package/src/tool/createTool.ts

@@ -21,6 +21,8 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     allowAgent?: string[];
     /** 只允许指定的 Graph 使用该工具 */
     allowGraph?: string[];
+    /** 是否是纯净的 json schema 参数，而不是 zod 参数 */
+    isPureParams?: boolean;
 }
 export type ToolCallback<Args extends ZodRawShape> = (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => CallToolResult | Promise<CallToolResult>;
 
@@ -95,7 +97,7 @@ export const createJSONDefineTool = <Args extends ZodRawShape>(tool: UnionTool<A
     return {
         name: tool.name,
         description: tool.description,
-        parameters: zodToJsonSchema(z.object(tool.parameters)),
+        parameters: tool.isPureParams ? tool.parameters : zodToJsonSchema(z.object(tool.parameters)),
     };
 };