@langgraph-js/sdk 1.10.2 → 1.10.4

package/dist/TestKit.d.ts

@@ -14,6 +14,10 @@ interface TestTask {
     /** 任务失败时的回调函数 */
     fail: () => void;
 }
+export declare class TestLogger {
+    info(message: string): void;
+    logMessage(message: RenderMessage): void;
+}
 /**
  * @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

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 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 : ""}
+`);
+        }
+        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;
@@ -28,6 +48,7 @@ export class TestLangGraphChat {
         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;
         // 订阅消息变化，自动检查任务完成状态
@@ -62,13 +83,29 @@ export class TestLangGraphChat {
                         this.checkAllTask(this.getMessages(), {
                             skipLengthCheck: true,
                         });
-                    }, 100);
+                    }, 10);
                     return oldExecute(...args);
                 };
             }
         });
         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
@@ -84,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);
         }
     }
     /**
@@ -128,7 +166,7 @@ export class TestLangGraphChat {
             ])
                 .then(async () => {
                 // messages 有 10 ms 的 debounce，我们需要稍等一下
-                await new Promise((resolve) => setTimeout(resolve, 20));
+                await new Promise((resolve) => setTimeout(resolve, 100));
                 this.checkAllTask(this.getMessages(), {
                     skipLengthCheck: true,
                 });

package/dist/ui-store/createChatStore.js

@@ -110,6 +110,8 @@ export const createChatStore = (initClientName, config, context = {}) => {
         // await newClient.createThread();
         inChatError.set(null);
         newClient.onStreamingUpdate((event) => {
+            var _a;
+            currentChatId.set(((_a = newClient.getCurrentThread()) === null || _a === void 0 ? void 0 : _a.thread_id) || null);
             if (event.type === "start")
                 loading.set(true);
             if (event.type === "thread" || event.type === "done") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "1.10.2",
+  "version": "1.10.4",
   "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 console.log(`${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;
 
@@ -87,7 +110,7 @@ export class TestLangGraphChat {
                         this.checkAllTask(this.getMessages(), {
                             skipLengthCheck: true,
                         });
-                    }, 100);
+                    }, 10);
                     return oldExecute!(...args);
                 };
             }
@@ -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,21 +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
+    private readited = false;
     /**
      * @zh 准备测试环境，初始化客户端连接
      * @en Prepare test environment, initialize client connection
      */
     ready() {
         if (this.readited) {
-            return
+            return;
         }
-        this.readited = true
+        this.readited = true;
         return this.store.mutations.initClient();
     }
 
@@ -159,7 +199,7 @@ export class TestLangGraphChat {
                 ])
                 .then(async () => {
                     // messages 有 10 ms 的 debounce，我们需要稍等一下
-                    await new Promise((resolve) => setTimeout(resolve, 20));
+                    await new Promise((resolve) => setTimeout(resolve, 100));
                     this.checkAllTask(this.getMessages(), {
                         skipLengthCheck: true,
                     });

package/src/ui-store/createChatStore.ts

@@ -117,6 +117,7 @@ export const createChatStore = (
         // await newClient.createThread();
         inChatError.set(null);
         newClient.onStreamingUpdate((event) => {
+            currentChatId.set(newClient.getCurrentThread()?.thread_id || null);
             if (event.type === "start") loading.set(true);
             if (event.type === "thread" || event.type === "done") {
                 // console.log(event.data);

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: {},
+};