@langgraph-js/sdk 1.10.3 → 1.11.0

package/dist/LangGraphClient.d.ts

@@ -57,6 +57,7 @@ export type SendMessageOptions = {
         streamResponse?: any;
     };
     command?: Command;
+    joinRunId?: string;
 };
 export interface LangGraphClientConfig {
     apiUrl?: string;
@@ -115,7 +116,8 @@ export declare class LangGraphClient extends Client {
      * @zh 从历史中恢复 Thread 数据。
      * @en Resets the Thread data from history.
      */
-    resetThread(agent: string, threadId: string): Promise<void>;
+    resetThread(agent: string, threadId: string): Promise<Thread<import("@langchain/langgraph-sdk").DefaultValues>>;
+    resetStream(): Promise<void>;
     streamingMessage: RenderMessage[];
     /** 图发过来的更新信息 */
     graphMessages: RenderMessage[];
@@ -168,7 +170,7 @@ export declare class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    sendMessage(input: string | Message[], { extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
+    sendMessage(input: string | Message[], { joinRunId, extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
     /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
     private graphPosition;
     getGraphPosition(): {

package/dist/LangGraphClient.js

@@ -112,10 +112,11 @@ export class LangGraphClient extends Client {
      * @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;
-        this.graphMessages = this.graphState.messages;
+        this.graphMessages = ((_a = this.graphState) === null || _a === void 0 ? void 0 : _a.messages) || [];
         this.emitStreamingUpdate({
             type: "value",
             data: {
@@ -125,6 +126,15 @@ export class LangGraphClient extends Client {
                 },
             },
         });
+        return this.currentThread;
+    }
+    // 从历史中恢复时，应该恢复流式状态
+    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");
+        if (runningRun) {
+            await this.sendMessage([], { joinRunId: runningRun.run_id });
+        }
     }
     cloneMessage(message) {
         return JSON.parse(JSON.stringify(message));
@@ -352,7 +362,7 @@ export class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    async sendMessage(input, { extraParams, _debug, command } = {}) {
+    async sendMessage(input, { joinRunId, extraParams, _debug, command } = {}) {
         var _a;
         if (!this.currentAssistant) {
             throw new Error("Thread or Assistant not initialized");
@@ -377,8 +387,14 @@ export class LangGraphClient extends Client {
                     content: input,
                 },
             ];
-        const streamResponse = (_debug === null || _debug === void 0 ? void 0 : _debug.streamResponse) ||
-            this.runs.stream(this.currentThread.thread_id, this.currentAssistant.assistant_id, {
+        const createStreamResponse = async () => {
+            if (_debug === null || _debug === void 0 ? void 0 : _debug.streamResponse) {
+                return _debug.streamResponse;
+            }
+            if (joinRunId) {
+                return this.runs.joinStream(this.currentThread.thread_id, joinRunId);
+            }
+            return this.runs.stream(this.currentThread.thread_id, this.currentAssistant.assistant_id, {
                 input: {
                     ...this.graphState,
                     ...this.extraParams,
@@ -390,6 +406,8 @@ export class LangGraphClient extends Client {
                 streamSubgraphs: true,
                 command,
             });
+        };
+        const streamResponse = await createStreamResponse();
         const streamRecord = [];
         this.emitStreamingUpdate({
             type: "start",

package/dist/TestKit.d.ts

@@ -16,7 +16,7 @@ interface TestTask {
 }
 export declare class TestLogger {
     info(message: string): void;
-    logMessage(message: RenderMessage): string | undefined;
+    logMessage(message: RenderMessage): void;
 }
 /**
  * @zh LangGraph 测试工具，可以配合 vitest 等常用框架进行测试

package/dist/TestKit.js

@@ -8,10 +8,10 @@ export class TestLogger {
         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}
+            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}
@@ -83,7 +83,7 @@ export class TestLangGraphChat {
                         this.checkAllTask(this.getMessages(), {
                             skipLengthCheck: true,
                         });
-                    }, 100);
+                    }, 10);
                     return oldExecute(...args);
                 };
             }
@@ -166,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.d.ts

@@ -84,7 +84,7 @@ export declare const createChatStore: (initClientName: string, config: LangGraph
          */
         toHistoryChat(thread: Thread<{
             messages: Message[];
-        }>): void;
+        }>): Promise<Thread<import("@langchain/langgraph-sdk").DefaultValues> | undefined>;
         /**
          * @zh 删除指定的历史聊天会话。
          * @en Deletes the specified historical chat session.

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") {
@@ -281,11 +283,15 @@ export const createChatStore = (initClientName, config, context = {}) => {
              * @zh 切换到指定的历史聊天会话。
              * @en Switches to the specified historical chat session.
              */
-            toHistoryChat(thread) {
-                var _a, _b;
+            async toHistoryChat(thread) {
+                var _a, _b, _c;
                 inChatError.set(null);
                 loading.set(false);
-                (_a = client.get()) === null || _a === void 0 ? void 0 : _a.resetThread((_b = thread.metadata) === null || _b === void 0 ? void 0 : _b.graph_id, thread.thread_id);
+                const nowThread = await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.resetThread((_b = thread.metadata) === null || _b === void 0 ? void 0 : _b.graph_id, thread.thread_id));
+                if (nowThread) {
+                    (_c = client.get()) === null || _c === void 0 ? void 0 : _c.resetStream();
+                }
+                return nowThread;
             },
             /**
              * @zh 删除指定的历史聊天会话。

package/package.json

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

package/src/LangGraphClient.ts

@@ -55,6 +55,7 @@ export type SendMessageOptions = {
     extraParams?: Record<string, any>;
     _debug?: { streamResponse?: any };
     command?: Command;
+    joinRunId?: string;
 };
 export interface LangGraphClientConfig {
     apiUrl?: string;
@@ -182,7 +183,7 @@ export class LangGraphClient extends Client {
         await this.initAssistant(agent);
         this.currentThread = await this.threads.get(threadId);
         this.graphState = this.currentThread.values;
-        this.graphMessages = this.graphState.messages;
+        this.graphMessages = this.graphState?.messages || [];
         this.emitStreamingUpdate({
             type: "value",
             data: {
@@ -192,6 +193,15 @@ export class LangGraphClient extends Client {
                 },
             },
         });
+        return this.currentThread;
+    }
+    // 从历史中恢复时，应该恢复流式状态
+    async resetStream() {
+        const runs = await this.runs.list(this.currentThread!.thread_id);
+        const runningRun = runs?.find((run) => run.status === "running" || run.status === "pending");
+        if (runningRun) {
+            await this.sendMessage([], { joinRunId: runningRun.run_id });
+        }
     }
 
     streamingMessage: RenderMessage[] = [];
@@ -432,7 +442,7 @@ export class LangGraphClient extends Client {
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
      */
-    async sendMessage(input: string | Message[], { extraParams, _debug, command }: SendMessageOptions = {}) {
+    async sendMessage(input: string | Message[], { joinRunId, extraParams, _debug, command }: SendMessageOptions = {}) {
         if (!this.currentAssistant) {
             throw new Error("Thread or Assistant not initialized");
         }
@@ -457,20 +467,29 @@ export class LangGraphClient extends Client {
                       content: input,
                   } as HumanMessage,
               ];
-        const streamResponse =
-            _debug?.streamResponse ||
-            this.runs.stream(this.currentThread!.thread_id, this.currentAssistant.assistant_id, {
+        const createStreamResponse = async () => {
+            if (_debug?.streamResponse) {
+                return _debug.streamResponse;
+            }
+            if (joinRunId) {
+                return this.runs.joinStream(this.currentThread!.thread_id, joinRunId);
+            }
+
+            return this.runs.stream(this.currentThread!.thread_id, this.currentAssistant!.assistant_id, {
                 input: {
                     ...this.graphState,
                     ...this.extraParams,
                     ...(extraParams || {}),
                     messages: messagesToSend,
-                    fe_tools: await this.tools.toJSON(this.currentAssistant.graph_id),
+                    fe_tools: await this.tools.toJSON(this.currentAssistant!.graph_id),
                 },
                 streamMode: ["messages", "values"],
                 streamSubgraphs: true,
                 command,
             });
+        };
+        const streamResponse = await createStreamResponse();
+
         const streamRecord: any[] = [];
         this.emitStreamingUpdate({
             type: "start",

package/src/TestKit.ts

@@ -25,10 +25,10 @@ export class TestLogger {
 
         const header = `${emoji} ${message.type} | ${message.name ?? "null"} | ${message.id}`;
         if (message.type === "tool") {
-            return `${header}
+            return console.log(`${header}
 🔧 Input: ${message.tool_input?.slice(0, 100) ?? ""}
 💬 Output: ${message.content?.slice(0, 100) ?? ""}
-`;
+`);
         }
 
         console.log(`---
@@ -110,7 +110,7 @@ export class TestLangGraphChat {
                         this.checkAllTask(this.getMessages(), {
                             skipLengthCheck: true,
                         });
-                    }, 100);
+                    }, 10);
                     return oldExecute!(...args);
                 };
             }
@@ -199,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);
@@ -288,14 +289,18 @@ export const createChatStore = (
              * @zh 切换到指定的历史聊天会话。
              * @en Switches to the specified historical chat session.
              */
-            toHistoryChat(
+            async toHistoryChat(
                 thread: Thread<{
                     messages: Message[];
                 }>
             ) {
                 inChatError.set(null);
                 loading.set(false);
-                client.get()?.resetThread(thread.metadata?.graph_id as string, thread.thread_id);
+                const nowThread = await client.get()?.resetThread(thread.metadata?.graph_id as string, thread.thread_id);
+                if (nowThread) {
+                    client.get()?.resetStream();
+                }
+                return nowThread;
             },
             /**
              * @zh 删除指定的历史聊天会话。