@langgraph-js/sdk 1.6.0 → 1.6.1

package/dist/LangGraphClient.d.ts

@@ -1,4 +1,4 @@
-import { Client, Thread, Message, Assistant, HumanMessage, ToolMessage, Command } from "@langchain/langgraph-sdk";
+import { Client, Thread, Message, Assistant, HumanMessage, AIMessage, ToolMessage, Command } from "@langchain/langgraph-sdk";
 import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
 interface AsyncCallerParams {
@@ -71,8 +71,8 @@ export interface LangGraphClientConfig {
 export declare class StreamingMessageType {
     static isUser(m: Message): m is HumanMessage;
     static isTool(m: Message): m is ToolMessage;
-    static isAssistant(m: Message): boolean;
-    static isToolAssistant(m: Message): any;
+    static isAssistant(m: Message): m is AIMessage;
+    static isToolAssistant(m: Message): m is AIMessage;
 }
 type StreamingUpdateEvent = {
     type: "message" | "value" | "update" | "error" | "thread" | "done" | "start";
@@ -120,7 +120,6 @@ export declare class LangGraphClient extends Client {
     graphMessages: RenderMessage[];
     cloneMessage(message: Message): Message;
     private updateStreamingMessage;
-    private replaceMessageWithValuesMessage;
     /** 将 graphMessages 和 streamingMessage 合并，并返回新的消息数组 */
     private combineGraphMessagesWithStreamingMessages;
     /**
@@ -167,6 +166,16 @@ export declare class LangGraphClient extends Client {
      * @en Sends a message to the LangGraph backend.
      */
     sendMessage(input: string | Message[], { extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
+    /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
+    private graphPosition;
+    getGraphPosition(): {
+        id: string;
+        name: string;
+    }[];
+    getGraphNodeNow(): {
+        id: string;
+        name: string;
+    };
     /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
     private mergeSubGraphMessagesToStreamingMessages;
     private runFETool;

package/dist/LangGraphClient.js

@@ -37,6 +37,8 @@ export class LangGraphClient extends Client {
         /** 图发过来的更新信息 */
         this.graphMessages = [];
         this.graphState = {};
+        /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
+        this.graphPosition = "";
     }
     listAssistants() {
         return this.assistants.search({
@@ -134,18 +136,6 @@ export class LangGraphClient extends Client {
         }
         this.streamingMessage[this.streamingMessage.length - 1] = message;
     }
-    replaceMessageWithValuesMessage(message, isTool = false) {
-        const key = (isTool ? "tool_call_id" : "id");
-        const valuesMessage = this.graphMessages.find((i) => i[key] === message[key]);
-        if (valuesMessage) {
-            return {
-                ...valuesMessage,
-                /** @ts-ignore */
-                tool_input: message.tool_input,
-            };
-        }
-        return message;
-    }
     /** 将 graphMessages 和 streamingMessage 合并，并返回新的消息数组 */
     combineGraphMessagesWithStreamingMessages() {
         const idMap = new Map(this.streamingMessage.map((i) => [i.id, i]));
@@ -171,6 +161,7 @@ export class LangGraphClient extends Client {
         const closedToolCallIds = new Set();
         const result = [];
         const inputMessages = this.combineGraphMessagesWithStreamingMessages();
+        console.log(inputMessages);
         // 从后往前遍历，这样可以保证最新的消息在前面
         for (let i = inputMessages.length - 1; i >= 0; i--) {
             const message = this.cloneMessage(inputMessages[i]);
@@ -178,8 +169,13 @@ export class LangGraphClient extends Client {
                 result.unshift(message);
                 continue;
             }
+            if (message.type === "ai") {
+                /** @ts-ignore */
+                if (!message.name)
+                    message.name = this.getGraphNodeNow().name;
+            }
             if (StreamingMessageType.isToolAssistant(message)) {
-                const m = this.replaceMessageWithValuesMessage(message);
+                const m = message;
                 // 记录这个 id 的消息，并添加到结果中
                 previousMessage.set(message.id, m);
                 /** @ts-ignore */
@@ -190,7 +186,7 @@ export class LangGraphClient extends Client {
                 })
                     .map((tool, index) => {
                     var _a, _b, _c, _d;
-                    return this.replaceMessageWithValuesMessage({
+                    return {
                         type: "tool",
                         additional_kwargs: {},
                         /** @ts-ignore */
@@ -199,7 +195,8 @@ export class LangGraphClient extends Client {
                         name: tool.name,
                         response_metadata: {},
                         tool_call_id: tool.id,
-                    }, true);
+                        content: "",
+                    };
                 });
                 for (const tool of new_tool_calls) {
                     if (!previousMessage.has(tool.id)) {
@@ -213,10 +210,8 @@ export class LangGraphClient extends Client {
                 if (message.type === "tool" && message.tool_call_id) {
                     closedToolCallIds.add(message.tool_call_id);
                 }
-                // 记录这个 id 的消息，并添加到结果中
-                const m = this.replaceMessageWithValuesMessage(message);
-                previousMessage.set(message.id, m);
-                result.unshift(m);
+                previousMessage.set(message.id, message);
+                result.unshift(message);
             }
         }
         return this.attachInfoForMessage(this.composeToolMessages(result));
@@ -255,7 +250,7 @@ export class LangGraphClient extends Client {
      * @en Composes tool messages, associating AI tool calls with tool execution results.
      */
     composeToolMessages(messages) {
-        var _a;
+        var _a, _b;
         const result = [];
         const assistantToolMessages = new Map();
         const toolParentMessage = new Map();
@@ -288,6 +283,10 @@ export class LangGraphClient extends Client {
                 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;
+                    }
                 }
             }
             result.push(message);
@@ -427,6 +426,7 @@ export class LangGraphClient extends Client {
                 if (chunk.data.messages) {
                     this.mergeSubGraphMessagesToStreamingMessages(chunk.data.messages);
                 }
+                this.graphPosition = chunk.event.split("|")[1];
             }
         }
         const data = await this.runFETool();
@@ -441,6 +441,19 @@ export class LangGraphClient extends Client {
         this.streamingMessage = [];
         return streamRecord;
     }
+    getGraphPosition() {
+        return this.graphPosition.split("|").map((i) => {
+            const [name, id] = i.split(":");
+            return {
+                id,
+                name,
+            };
+        });
+    }
+    getGraphNodeNow() {
+        const position = this.getGraphPosition();
+        return position[position.length - 1];
+    }
     /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
     mergeSubGraphMessagesToStreamingMessages(messages) {
         const map = new Map(messages.filter((i) => i.id).map((i) => [i.id, i]));

package/package.json

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

package/src/LangGraphClient.ts

@@ -68,16 +68,16 @@ export interface LangGraphClientConfig {
  * @en The StreamingMessageType class is used to determine the type of a message.
  */
 export class StreamingMessageType {
-    static isUser(m: Message) {
+    static isUser(m: Message): m is HumanMessage {
         return m.type === "human";
     }
-    static isTool(m: Message) {
+    static isTool(m: Message): m is ToolMessage {
         return m.type === "tool";
     }
-    static isAssistant(m: Message) {
+    static isAssistant(m: Message): m is AIMessage {
         return m.type === "ai" && !this.isToolAssistant(m);
     }
-    static isToolAssistant(m: Message) {
+    static isToolAssistant(m: Message): m is AIMessage {
         /** @ts-ignore */
         return m.type === "ai" && (m.tool_calls?.length || m.tool_call_chunks?.length);
     }
@@ -207,18 +207,6 @@ export class LangGraphClient extends Client {
         }
         this.streamingMessage[this.streamingMessage.length - 1] = message;
     }
-    private replaceMessageWithValuesMessage(message: AIMessage | ToolMessage, isTool = false): Message {
-        const key = (isTool ? "tool_call_id" : "id") as any as "id";
-        const valuesMessage = this.graphMessages.find((i) => i[key] === message[key]);
-        if (valuesMessage) {
-            return {
-                ...valuesMessage,
-                /** @ts-ignore */
-                tool_input: message.tool_input,
-            };
-        }
-        return message;
-    }
     /** 将 graphMessages 和 streamingMessage 合并，并返回新的消息数组 */
     private combineGraphMessagesWithStreamingMessages() {
         const idMap = new Map<string, RenderMessage>(this.streamingMessage.map((i) => [i.id!, i]));
@@ -243,6 +231,7 @@ export class LangGraphClient extends Client {
         const closedToolCallIds = new Set<string>();
         const result: Message[] = [];
         const inputMessages = this.combineGraphMessagesWithStreamingMessages();
+        console.log(inputMessages);
         // 从后往前遍历，这样可以保证最新的消息在前面
         for (let i = inputMessages.length - 1; i >= 0; i--) {
             const message = this.cloneMessage(inputMessages[i]);
@@ -251,9 +240,12 @@ export class LangGraphClient extends Client {
                 result.unshift(message);
                 continue;
             }
-
+            if (message.type === "ai") {
+                /** @ts-ignore */
+                if (!message.name) message.name = this.getGraphNodeNow().name;
+            }
             if (StreamingMessageType.isToolAssistant(message)) {
-                const m = this.replaceMessageWithValuesMessage(message as AIMessage);
+                const m = message;
                 // 记录这个 id 的消息，并添加到结果中
                 previousMessage.set(message.id, m);
 
@@ -264,19 +256,17 @@ export class LangGraphClient extends Client {
                         return !closedToolCallIds.has(i.id!);
                     })!
                     .map((tool, index) => {
-                        return this.replaceMessageWithValuesMessage(
-                            {
-                                type: "tool",
-                                additional_kwargs: {},
-                                /** @ts-ignore */
-                                tool_input: m.additional_kwargs?.tool_calls?.[index]?.function?.arguments,
-                                id: tool.id,
-                                name: tool.name,
-                                response_metadata: {},
-                                tool_call_id: tool.id!,
-                            },
-                            true
-                        );
+                        return {
+                            type: "tool",
+                            additional_kwargs: {},
+                            /** @ts-ignore */
+                            tool_input: m.additional_kwargs?.tool_calls?.[index]?.function?.arguments,
+                            id: tool.id,
+                            name: tool.name,
+                            response_metadata: {},
+                            tool_call_id: tool.id!,
+                            content: "",
+                        } as ToolMessage;
                     });
                 for (const tool of new_tool_calls) {
                     if (!previousMessage.has(tool.id!)) {
@@ -289,10 +279,9 @@ export class LangGraphClient extends Client {
                 if (message.type === "tool" && message.tool_call_id) {
                     closedToolCallIds.add(message.tool_call_id);
                 }
-                // 记录这个 id 的消息，并添加到结果中
-                const m = this.replaceMessageWithValuesMessage(message as AIMessage);
-                previousMessage.set(message.id, m);
-                result.unshift(m);
+
+                previousMessage.set(message.id, message);
+                result.unshift(message);
             }
         }
 
@@ -364,6 +353,10 @@ export class LangGraphClient extends Client {
                 if (parentMessage) {
                     message.usage_metadata = parentMessage.usage_metadata;
                     message.node_name = parentMessage.name;
+                    // 修补特殊情况下，tool name 丢失的问题
+                    if (!message.name) {
+                        message.name = (parentMessage as AIMessage).tool_calls!.find((i) => i.id === message.tool_call_id)?.name;
+                    }
                 }
             }
             result.push(message);
@@ -510,6 +503,7 @@ export class LangGraphClient extends Client {
                 if (chunk.data.messages) {
                     this.mergeSubGraphMessagesToStreamingMessages(chunk.data.messages);
                 }
+                this.graphPosition = chunk.event.split("|")[1];
             }
         }
         const data = await this.runFETool();
@@ -523,6 +517,21 @@ export class LangGraphClient extends Client {
         this.streamingMessage = [];
         return streamRecord;
     }
+    /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
+    private graphPosition = "";
+    getGraphPosition() {
+        return this.graphPosition.split("|").map((i) => {
+            const [name, id] = i.split(":");
+            return {
+                id,
+                name,
+            };
+        });
+    }
+    getGraphNodeNow() {
+        const position = this.getGraphPosition();
+        return position[position.length - 1];
+    }
     /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
     private mergeSubGraphMessagesToStreamingMessages(messages: Message[]) {
         const map = new Map(messages.filter((i) => i.id).map((i) => [i.id!, i]));