@langgraph-js/sdk 1.10.1 → 1.10.3

package/dist/TestKit.d.ts

@@ -14,6 +14,10 @@ interface TestTask {
     /** 任务失败时的回调函数 */
     fail: () => void;
 }
+export declare class TestLogger {
+    info(message: string): void;
+    logMessage(message: RenderMessage): string | undefined;
+}
 /**
  * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
  * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
@@ -35,6 +39,8 @@ export declare class TestLangGraphChat {
     private lastLength;
     /** 待处理的测试任务列表 */
     protected processFunc: TestTask[];
+    /** 自定义日志器 */
+    private logger;
     /**
      * @zh 构造函数，初始化测试环境
      * @en Constructor, initialize test environment
@@ -42,6 +48,7 @@ export declare class TestLangGraphChat {
     constructor(store: ReturnType<typeof createChatStore>, options: {
         debug?: boolean;
         tools?: UnionTool<any, any, any>[];
+        logger?: TestLogger;
     });
     /**
      * @zh 获取当前所有渲染消息
@@ -59,6 +66,17 @@ export declare class TestLangGraphChat {
      * ```
      */
     addTools(tools: UnionTool<any, any, any>[]): void;
+    /**
+     * @zh 设置额外参数
+     * @en Set extra states to LangGraph
+     *
+     * @example
+     * ```typescript
+     * testChat.setExtraParams({
+     *     extraParam: "value",
+     * });
+     */
+    setExtraParams(extraParams: Record<string, any>): void;
     /**
      * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
      * @en Check all pending test tasks, only executes when message count changes
@@ -66,11 +84,12 @@ export declare class TestLangGraphChat {
     checkAllTask(messages: readonly RenderMessage[], options?: {
         skipLengthCheck?: boolean;
     }): void;
+    private readited;
     /**
      * @zh 准备测试环境，初始化客户端连接
      * @en Prepare test environment, initialize client connection
      */
-    ready(): Promise<import("./LangGraphClient.js").LangGraphClient>;
+    ready(): Promise<import("./LangGraphClient.js").LangGraphClient> | undefined;
     /**
      * @zh 模拟人类输入消息并等待测试任务完成，这是测试的核心方法
      * @en Simulate human input and wait for test tasks to complete, this is the core test method

package/dist/TestKit.js

@@ -1,4 +1,24 @@
 import { ToolRenderData } from "./tool/ToolUI.js";
+export class TestLogger {
+    info(message) {
+        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}`;
+        if (message.type === "tool") {
+            return `${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 : ""}
+`;
+        }
+        console.log(`---
+${header}
+💬 Output: ${(_g = (_f = message.content) === null || _f === void 0 ? void 0 : _f.slice(0, 100)) !== null && _g !== void 0 ? _g : ""}
+`);
+    }
+}
 /**
  * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
  * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
@@ -18,7 +38,7 @@ export class TestLangGraphChat {
      * @en Constructor, initialize test environment
      */
     constructor(store, options) {
-        var _a;
+        var _a, _b;
         this.store = store;
         /** 是否开启调试模式 */
         this.debug = false;
@@ -26,7 +46,9 @@ export class TestLangGraphChat {
         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();
         options.tools && this.addTools(options.tools);
         const renderMessages = this.store.data.renderMessages;
         // 订阅消息变化，自动检查任务完成状态
@@ -68,6 +90,22 @@ export class TestLangGraphChat {
         });
         this.store.mutations.setTools(tools);
     }
+    /**
+     * @zh 设置额外参数
+     * @en Set extra states to LangGraph
+     *
+     * @example
+     * ```typescript
+     * testChat.setExtraParams({
+     *     extraParam: "value",
+     * });
+     */
+    setExtraParams(extraParams) {
+        const client = this.store.data.client.get();
+        if (client) {
+            client.extraParams = extraParams;
+        }
+    }
     /**
      * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
      * @en Check all pending test tasks, only executes when message count changes
@@ -83,8 +121,9 @@ export class TestLangGraphChat {
             !task.success && task.runTask(options.skipLengthCheck ? messages : messages.slice(0, -1));
         }
         // 调试模式下打印最新消息
-        if (this.debug) {
-            console.log(messages[messages.length - (options.skipLengthCheck ? 1 : 2)]);
+        const item = messages[messages.length - (options.skipLengthCheck ? 1 : 2)];
+        if (this.debug && item) {
+            this.logger.logMessage(item);
         }
     }
     /**
@@ -92,6 +131,10 @@ export class TestLangGraphChat {
      * @en Prepare test environment, initialize client connection
      */
     ready() {
+        if (this.readited) {
+            return;
+        }
+        this.readited = true;
         return this.store.mutations.initClient();
     }
     /**
@@ -121,7 +164,9 @@ export class TestLangGraphChat {
                     content: text,
                 },
             ])
-                .then(() => {
+                .then(async () => {
+                // messages 有 10 ms 的 debounce，我们需要稍等一下
+                await new Promise((resolve) => setTimeout(resolve, 20));
                 this.checkAllTask(this.getMessages(), {
                     skipLengthCheck: true,
                 });

package/package.json

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

package/src/TestKit.ts

@@ -3,7 +3,6 @@ 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
@@ -17,6 +16,27 @@ interface TestTask {
     fail: () => void;
 }
 
+export class TestLogger {
+    info(message: string) {
+        console.log(message);
+    }
+    logMessage(message: RenderMessage) {
+        const emoji = message.type === "ai" ? "🤖" : message.type === "human" ? "👤" : "🔧";
+
+        const header = `${emoji} ${message.type} | ${message.name ?? "null"} | ${message.id}`;
+        if (message.type === "tool") {
+            return `${header}
+🔧 Input: ${message.tool_input?.slice(0, 100) ?? ""}
+💬 Output: ${message.content?.slice(0, 100) ?? ""}
+`;
+        }
+
+        console.log(`---
+${header}
+💬 Output: ${message.content?.slice(0, 100) ?? ""}
+`);
+    }
+}
 /**
  * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试
  * @en LangGraph test tool, can be used with vitest and other common frameworks for testing
@@ -37,7 +57,8 @@ export class TestLangGraphChat {
     private lastLength = 0;
     /** 待处理的测试任务列表 */
     protected processFunc: TestTask[] = [];
-
+    /** 自定义日志器 */
+    private logger: TestLogger;
     /**
      * @zh 构造函数，初始化测试环境
      * @en Constructor, initialize test environment
@@ -47,9 +68,11 @@ export class TestLangGraphChat {
         options: {
             debug?: boolean;
             tools?: UnionTool<any, any, any>[];
+            logger?: TestLogger;
         }
     ) {
         this.debug = options.debug ?? false;
+        this.logger = options.logger ?? new TestLogger();
         options.tools && this.addTools(options.tools);
         const renderMessages = this.store.data.renderMessages;
 
@@ -95,6 +118,22 @@ export class TestLangGraphChat {
         this.store.mutations.setTools(tools);
     }
 
+    /**
+     * @zh 设置额外参数
+     * @en Set extra states to LangGraph
+     *
+     * @example
+     * ```typescript
+     * testChat.setExtraParams({
+     *     extraParam: "value",
+     * });
+     */
+    setExtraParams(extraParams: Record<string, any>) {
+        const client = this.store.data.client.get();
+        if (client) {
+            client.extraParams = extraParams;
+        }
+    }
     /**
      * @zh 检查所有待处理的测试任务，只有在消息数量发生变化时才执行检查
      * @en Check all pending test tasks, only executes when message count changes
@@ -112,16 +151,22 @@ export class TestLangGraphChat {
         }
 
         // 调试模式下打印最新消息
-        if (this.debug) {
-            console.log(messages[messages.length - (options.skipLengthCheck ? 1 : 2)]);
+        const item = messages[messages.length - (options.skipLengthCheck ? 1 : 2)];
+        if (this.debug && item) {
+            this.logger.logMessage(item);
         }
     }
 
+    private readited = false;
     /**
      * @zh 准备测试环境，初始化客户端连接
      * @en Prepare test environment, initialize client connection
      */
     ready() {
+        if (this.readited) {
+            return;
+        }
+        this.readited = true;
         return this.store.mutations.initClient();
     }
 
@@ -152,7 +197,9 @@ export class TestLangGraphChat {
                         content: text,
                     },
                 ])
-                .then(() => {
+                .then(async () => {
+                    // messages 有 10 ms 的 debounce，我们需要稍等一下
+                    await new Promise((resolve) => setTimeout(resolve, 20));
                     this.checkAllTask(this.getMessages(), {
                         skipLengthCheck: true,
                     });

package/test/TestKit.test.ts

@@ -0,0 +1,24 @@
+import { expect, test } from "vitest";
+import { TestLangGraphChat, createChatStore } from "../src";
+
+test("test", async () => {
+    const testChat = new TestLangGraphChat(
+        createChatStore(
+            "agent",
+            {
+                defaultHeaders: {
+                    Authorization: "Bearer 123",
+                },
+            },
+            {}
+        ),
+        { debug: true }
+    );
+    await testChat.ready();
+
+    await testChat.humanInput("你好", async () => {
+        const aiMessage = await testChat.waitFor("ai");
+        expect(aiMessage.content).toBeDefined();
+    });
+    expect(1).toBe(1);
+});

package/vitest.confit.ts

@@ -0,0 +1,3 @@
+export default {
+    test: {},
+};