@langgraph-js/sdk 1.3.1 → 1.4.0

package/README.md

@@ -1,163 +1,163 @@
-# @langgraph-js/sdk
-
-![npm version](https://img.shields.io/npm/v/@langgraph-js/sdk)
-![license](https://img.shields.io/npm/l/@langgraph-js/sdk)
-
-> The missing UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces
-
-## Why @langgraph-js/sdk?
-
-Building AI agent applications is complex, especially when you need to bridge the gap between LangGraph agents and interactive user interfaces. This SDK solves the critical challenges of frontend integration:
-
-- **Provides a complete UI integration layer** - no more complex custom code to handle tools, streaming, and state management
-- **Simplifies human-in-the-loop interactions** - easily incorporate user feedback within agent workflows
-- **Handles edge cases automatically** - interruptions, errors, token management and more
-- **Offers a rich set of UI components** - ready-to-use elements to display agent interactions
-
-[DOCS](https://langgraph-js.netlify.app)
-
-## Installation
-
-```bash
-# Using npm
-npm install @langgraph-js/sdk
-
-# Using yarn
-yarn add @langgraph-js/sdk
-
-# Using pnpm
-pnpm add @langgraph-js/sdk
-```
-
-## Key Features
-
-### Generative UI
-
-- ✅ Custom Tool Messages
-- ✅ Token Counter
-- ✅ Stop Graph Progress
-- ✅ Interrupt Handling
-- ✅ Error Handling
-- ✅ Spend Time Tracking
-- ✅ Time Persistence
-
-### Frontend Actions
-
-- ✅ Definition of Union Tools
-- ✅ Frontend Functions As Tools
-- ✅ Human-in-the-Loop Interaction
-- ✅ Interrupt Mode
-
-### Authorization
-
-- ✅ Cookie-Based Authentication
-- ✅ Custom Token Authentication
-
-### Persistence
-
-- ✅ Read History from LangGraph
-
-## Advanced Usage
-
-### Creating a Chat Store
-
-You can easily create a reactive store for your LangGraph client:
-
-```typescript
-import { createChatStore } from "@langgraph-js/sdk";
-
-export const globalChatStore = createChatStore(
-    "agent",
-    {
-        // Custom LangGraph backend interaction
-        apiUrl: "http://localhost:8123",
-        // Custom headers for authentication
-        defaultHeaders: JSON.parse(localStorage.getItem("code") || "{}"),
-        callerOptions: {
-            // Example for including cookies
-            // fetch(url: string, options: RequestInit) {
-            //     options.credentials = "include";
-            //     return fetch(url, options);
-            // },
-        },
-    },
-    {
-        onInit(client) {
-            client.tools.bindTools([]);
-        },
-    }
-);
-```
-
-### React Integration
-
-First, install the nanostores React integration:
-
-```bash
-pnpm i @nanostores/react
-```
-
-Then create a context provider for your chat:
-
-```tsx
-import React, { createContext, useContext, useEffect } from "react";
-import { globalChatStore } from "../store"; // Import your store
-import { UnionStore, useUnionStore } from "@langgraph-js/sdk";
-import { useStore } from "@nanostores/react";
-
-type ChatContextType = UnionStore<typeof globalChatStore>;
-
-const ChatContext = createContext<ChatContextType | undefined>(undefined);
-
-export const useChat = () => {
-    const context = useContext(ChatContext);
-    if (!context) {
-        throw new Error("useChat must be used within a ChatProvider");
-    }
-    return context;
-};
-
-export const ChatProvider = ({ children }) => {
-    // Use store to ensure React gets reactive state updates
-    const store = useUnionStore(globalChatStore, useStore);
-    
-    useEffect(() => {
-        // Initialize client
-        store.initClient().then(() => {
-            // Initialize conversation history
-            store.refreshHistoryList();
-        });
-    }, [store.currentAgent]);
-
-    return <ChatContext.Provider value={store}>{children}</ChatContext.Provider>;
-};
-```
-
-Use it in your components:
-
-```tsx
-export const MyChat = () => {
-    return (
-        <ChatProvider>
-            <ChatComp></ChatComp>
-        </ChatProvider>
-    );
-};
-
-function ChatComp() {
-    const chat = useChat();
-    // Use chat store methods and state here
-}
-```
-
-## Documentation
-
-For complete documentation, visit our [official docs](https://langgraph-js.netlify.app).
-
-## Contributing
-
-Contributions are welcome! Please feel free to submit a Pull Request.
-
-## License
-
-This project is licensed under the Apache-2.0 License.
+# @langgraph-js/sdk
+
+![npm version](https://img.shields.io/npm/v/@langgraph-js/sdk)
+![license](https://img.shields.io/npm/l/@langgraph-js/sdk)
+
+> The missing UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces
+
+## Why @langgraph-js/sdk?
+
+Building AI agent applications is complex, especially when you need to bridge the gap between LangGraph agents and interactive user interfaces. This SDK solves the critical challenges of frontend integration:
+
+- **Provides a complete UI integration layer** - no more complex custom code to handle tools, streaming, and state management
+- **Simplifies human-in-the-loop interactions** - easily incorporate user feedback within agent workflows
+- **Handles edge cases automatically** - interruptions, errors, token management and more
+- **Offers a rich set of UI components** - ready-to-use elements to display agent interactions
+
+[DOCS](https://langgraph-js.netlify.app)
+
+## Installation
+
+```bash
+# Using npm
+npm install @langgraph-js/sdk
+
+# Using yarn
+yarn add @langgraph-js/sdk
+
+# Using pnpm
+pnpm add @langgraph-js/sdk
+```
+
+## Key Features
+
+### Generative UI
+
+- ✅ Custom Tool Messages
+- ✅ Token Counter
+- ✅ Stop Graph Progress
+- ✅ Interrupt Handling
+- ✅ Error Handling
+- ✅ Spend Time Tracking
+- ✅ Time Persistence
+
+### Frontend Actions
+
+- ✅ Definition of Union Tools
+- ✅ Frontend Functions As Tools
+- ✅ Human-in-the-Loop Interaction
+- ✅ Interrupt Mode
+
+### Authorization
+
+- ✅ Cookie-Based Authentication
+- ✅ Custom Token Authentication
+
+### Persistence
+
+- ✅ Read History from LangGraph
+
+## Advanced Usage
+
+### Creating a Chat Store
+
+You can easily create a reactive store for your LangGraph client:
+
+```typescript
+import { createChatStore } from "@langgraph-js/sdk";
+
+export const globalChatStore = createChatStore(
+    "agent",
+    {
+        // Custom LangGraph backend interaction
+        apiUrl: "http://localhost:8123",
+        // Custom headers for authentication
+        defaultHeaders: JSON.parse(localStorage.getItem("code") || "{}"),
+        callerOptions: {
+            // Example for including cookies
+            // fetch(url: string, options: RequestInit) {
+            //     options.credentials = "include";
+            //     return fetch(url, options);
+            // },
+        },
+    },
+    {
+        onInit(client) {
+            client.tools.bindTools([]);
+        },
+    }
+);
+```
+
+### React Integration
+
+First, install the nanostores React integration:
+
+```bash
+pnpm i @nanostores/react
+```
+
+Then create a context provider for your chat:
+
+```tsx
+import React, { createContext, useContext, useEffect } from "react";
+import { globalChatStore } from "../store"; // Import your store
+import { UnionStore, useUnionStore } from "@langgraph-js/sdk";
+import { useStore } from "@nanostores/react";
+
+type ChatContextType = UnionStore<typeof globalChatStore>;
+
+const ChatContext = createContext<ChatContextType | undefined>(undefined);
+
+export const useChat = () => {
+    const context = useContext(ChatContext);
+    if (!context) {
+        throw new Error("useChat must be used within a ChatProvider");
+    }
+    return context;
+};
+
+export const ChatProvider = ({ children }) => {
+    // Use store to ensure React gets reactive state updates
+    const store = useUnionStore(globalChatStore, useStore);
+    
+    useEffect(() => {
+        // Initialize client
+        store.initClient().then(() => {
+            // Initialize conversation history
+            store.refreshHistoryList();
+        });
+    }, [store.currentAgent]);
+
+    return <ChatContext.Provider value={store}>{children}</ChatContext.Provider>;
+};
+```
+
+Use it in your components:
+
+```tsx
+export const MyChat = () => {
+    return (
+        <ChatProvider>
+            <ChatComp></ChatComp>
+        </ChatProvider>
+    );
+};
+
+function ChatComp() {
+    const chat = useChat();
+    // Use chat store methods and state here
+}
+```
+
+## Documentation
+
+For complete documentation, visit our [official docs](https://langgraph-js.netlify.app).
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+This project is licensed under the Apache-2.0 License.

package/dist/LangGraphClient.d.ts

@@ -119,6 +119,7 @@ export declare class LangGraphClient extends Client {
     /** 图发过来的更新信息 */
     graphMessages: RenderMessage[];
     cloneMessage(message: Message): Message;
+    private updateStreamingMessage;
     private replaceMessageWithValuesMessage;
     /**
      * @zh 用于 UI 中的流式渲染中的消息。
@@ -164,6 +165,8 @@ export declare class LangGraphClient extends Client {
      * @en Sends a message to the LangGraph backend.
      */
     sendMessage(input: string | Message[], { extraParams, _debug, command }?: SendMessageOptions): Promise<any[]>;
+    /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
+    private mergeSubGraphMessagesToStreamingMessages;
     private runFETool;
     private callFETool;
     /**

package/dist/LangGraphClient.js

@@ -126,6 +126,14 @@ export class LangGraphClient extends Client {
     cloneMessage(message) {
         return JSON.parse(JSON.stringify(message));
     }
+    updateStreamingMessage(message) {
+        const lastMessage = this.streamingMessage[this.streamingMessage.length - 1];
+        if (!(lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.id) || message.id !== lastMessage.id) {
+            this.streamingMessage.push(message);
+            return;
+        }
+        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]);
@@ -145,8 +153,10 @@ export class LangGraphClient extends Client {
     get renderMessage() {
         var _a;
         const previousMessage = new Map();
+        const closedToolCallIds = new Set();
         const result = [];
         const inputMessages = [...this.graphMessages, ...this.streamingMessage];
+        console.log(inputMessages);
         // 从后往前遍历，这样可以保证最新的消息在前面
         for (let i = inputMessages.length - 1; i >= 0; i--) {
             const message = this.cloneMessage(inputMessages[i]);
@@ -164,7 +174,11 @@ export class LangGraphClient extends Client {
                 previousMessage.set(message.id, m);
                 /** @ts-ignore */
                 const tool_calls = ((_a = m.tool_calls) === null || _a === void 0 ? void 0 : _a.length) ? m.tool_calls : m.tool_call_chunks;
-                const new_tool_calls = tool_calls.map((tool, index) => {
+                const new_tool_calls = tool_calls
+                    .filter((i) => {
+                    return !closedToolCallIds.has(i.id);
+                })
+                    .map((tool, index) => {
                     var _a, _b, _c, _d;
                     return this.replaceMessageWithValuesMessage({
                         type: "tool",
@@ -186,6 +200,9 @@ export class LangGraphClient extends Client {
                 result.unshift(m);
             }
             else {
+                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);
@@ -364,7 +381,7 @@ export class LangGraphClient extends Client {
             }
             else if (chunk.event === "messages/partial") {
                 for (const message of chunk.data) {
-                    this.streamingMessage.push(message);
+                    this.updateStreamingMessage(message);
                 }
                 this.emitStreamingUpdate({
                     type: "message",
@@ -372,7 +389,7 @@ export class LangGraphClient extends Client {
                 });
                 continue;
             }
-            else if (chunk.event.startsWith("values")) {
+            else if (chunk.event === "values") {
                 const data = chunk.data;
                 if (data.messages) {
                     const isResume = !!(command === null || command === void 0 ? void 0 : command.resume);
@@ -390,6 +407,12 @@ export class LangGraphClient extends Client {
                 this.streamingMessage = [];
                 continue;
             }
+            else if (chunk.event.startsWith("values|")) {
+                // 这个 values 必然是子 values
+                if (chunk.data.messages) {
+                    this.mergeSubGraphMessagesToStreamingMessages(chunk.data.messages);
+                }
+            }
         }
         this.streamingMessage = [];
         const data = await this.runFETool();
@@ -403,6 +426,23 @@ export class LangGraphClient extends Client {
         });
         return streamRecord;
     }
+    /** 子图的数据需要通过 merge 的方式重新进行合并更新 */
+    mergeSubGraphMessagesToStreamingMessages(messages) {
+        const map = new Map(messages.filter((i) => i.id).map((i) => [i.id, i]));
+        this.streamingMessage.forEach((i) => {
+            if (map.has(i.id)) {
+                const newValue = map.get(i.id);
+                Object.assign(i, newValue);
+                map.delete(i.id);
+            }
+        });
+        // 剩余的 message 一定不在 streamMessage 中
+        map.forEach((i) => {
+            if (i.type === "tool" && i.tool_call_id) {
+                this.streamingMessage.push(i);
+            }
+        });
+    }
     runFETool() {
         var _a;
         const data = this.graphMessages;

package/dist/ui-store/createChatStore.js

@@ -105,6 +105,8 @@ export const createChatStore = (initClientName, config, context = {}) => {
                 // console.log(event.data);
                 // 创建新流程时，默认为 __start__
                 currentNodeName.set("__start__");
+                if (event.type === "done")
+                    loading.set(false);
                 // 创建新会话时，需要自动刷新历史面板
                 return refreshHistoryList();
             }

package/package.json

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