@langgraph-js/sdk 3.5.2 → 3.7.0

package/dist/LangGraphClient.d.ts

@@ -3,6 +3,7 @@ import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
 import { type ILangGraphClient } from "@langgraph-js/pure-graph/dist/types.js";
+import { RevertChatToOptions } from "./time-travel/index.js";
 export type RenderMessage = Message & {
     /** 对于 AIMessage 来说是节点名称，对于工具节点来说是工具名称 */
     name?: string;
@@ -143,10 +144,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
             metadata?: import("@langchain/langgraph-sdk").Metadata;
             limit?: number;
             offset?: number;
-            sortBy? /**
-             * The maximum number of retries that can be made for a single call,
-             * with an exponential backoff between each attempt. Defaults to 6.
-             */: import("@langgraph-js/pure-graph/dist/types.js").AssistantSortBy;
+            sortBy?: import("@langgraph-js/pure-graph/dist/types.js").AssistantSortBy;
             sortOrder?: import("@langgraph-js/pure-graph/dist/types.js").SortOrder;
         }): Promise<Assistant[]>;
         getGraph(assistantId: string, options?: {
@@ -230,7 +228,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
      * @zh 回滚到指定的消息。但是不会触发数据的重新更新
      * @en Reverts to the specified message.
      */
-    revertChatTo(messageId: string): Promise<{
+    revertChatTo(messageId: string, options: RevertChatToOptions): Promise<{
         messages: Message[];
     }>;
     messagesMetadata: {};

package/dist/LangGraphClient.js

@@ -197,8 +197,8 @@ export class LangGraphClient extends EventEmitter {
      * @zh 回滚到指定的消息。但是不会触发数据的重新更新
      * @en Reverts to the specified message.
      */
-    async revertChatTo(messageId) {
-        const { state, checkpoint } = await revertChatTo(this.client, this.currentThread.thread_id, messageId);
+    async revertChatTo(messageId, options) {
+        const { state, checkpoint } = await revertChatTo(this.client, this.currentThread.thread_id, messageId, options);
         this.graphState = state;
         this.messageProcessor.clearStreamingMessages();
         this.messageProcessor.setGraphMessages(state.messages);

package/dist/MessageProcessor.js

@@ -145,16 +145,13 @@ export class MessageProcessor {
                 const parentMessage = toolParentMessage.get(message.tool_call_id);
                 if (assistantToolMessage) {
                     message.tool_input = typeof assistantToolMessage.args !== "string" ? JSON.stringify(assistantToolMessage.args) : assistantToolMessage.args;
-                    if (message.additional_kwargs) {
-                        message.additional_kwargs.done = true;
-                        message.done = true;
-                    }
-                    else {
-                        message.done = true;
-                        message.additional_kwargs = {
-                            done: true,
-                        };
-                    }
+                    const isDone = !!message.content;
+                    message.done = isDone;
+                    message.additional_kwargs = {
+                        ...((parentMessage === null || parentMessage === void 0 ? void 0 : parentMessage.additional_kwargs) || {}),
+                        ...(message.additional_kwargs || {}),
+                        done: isDone,
+                    };
                 }
                 if (parentMessage) {
                     message.usage_metadata = parentMessage.usage_metadata;

package/dist/react/ChatContext.d.ts

@@ -38,7 +38,7 @@ export declare const useChat: () => UnionStore<{
         addToHistory: (thread: import("@langchain/langgraph-sdk").Thread<{
             messages: import("@langchain/langgraph-sdk").Message[];
         }>) => void;
-        revertChatTo(messageId: string, resend?: boolean, sendOptions?: import("../LangGraphClient.js").SendMessageOptions): Promise<void>;
+        revertChatTo(messageId: string, resend?: boolean, sendOptions?: import("../LangGraphClient.js").SendMessageOptions & import("../time-travel/index.js").RevertChatToOptions): Promise<void>;
         setUserInput(input: string): void;
         setCurrentAgent(agent: string): Promise<void>;
         toggleGraphVisible(): void;

package/dist/time-travel/index.d.ts

@@ -1,9 +1,12 @@
 import { Client, Message } from "@langchain/langgraph-sdk";
+export interface RevertChatToOptions {
+    includeMessageId?: boolean;
+}
 export declare function revertChatTo(client: Client<{
     messages: Message[];
 }, {
     messages: Message[];
-}, unknown>, threadId: string, messageId: string): Promise<{
+}, unknown>, threadId: string, messageId: string, options: RevertChatToOptions): Promise<{
     state: {
         messages: Message[];
     };

package/dist/time-travel/index.js

@@ -1,11 +1,12 @@
-export async function revertChatTo(client, threadId, messageId) {
+export async function revertChatTo(client, threadId, messageId, options) {
     const thread = await client.threads.get(threadId);
     const messages = thread.values.messages;
     const idx = messages.findIndex((message) => message.id === messageId);
     if (idx === -1) {
         throw new Error(`Message id ${messageId} not found`);
     }
-    const removeMessages = messages.slice(idx + 1);
+    const removeStartIdx = options.includeMessageId ? idx : idx + 1;
+    const removeMessages = messages.slice(removeStartIdx);
     const state = {
         ...thread.values,
         messages: removeMessages.map((i) => {
@@ -23,7 +24,7 @@ export async function revertChatTo(client, threadId, messageId) {
     // client.runs.wait();
     return {
         state: {
-            messages: messages.slice(0, idx + 1),
+            messages: messages.slice(0, removeStartIdx),
         },
         checkpoint: res,
     };

package/dist/ui-store/createChatStore.d.ts

@@ -1,6 +1,7 @@
 import { LangGraphClient, LangGraphClientConfig, RenderMessage, SendMessageOptions } from "../LangGraphClient.js";
 import { AssistantGraph, Message, Thread } from "@langchain/langgraph-sdk";
 import { UnionTool } from "../tool/createTool.js";
+import { RevertChatToOptions } from "../time-travel/index.js";
 /**
  * @zh 格式化日期对象为时间字符串。
  * @en Formats a Date object into a time string.
@@ -72,7 +73,7 @@ export declare const createChatStore: (initClientName: string, config: Partial<L
          * @zh 回滚到指定的消息。
          * @en Reverts to the specified message.
          */
-        revertChatTo(messageId: string, resend?: boolean, sendOptions?: SendMessageOptions): Promise<void>;
+        revertChatTo(messageId: string, resend?: boolean, sendOptions?: SendMessageOptions & RevertChatToOptions): Promise<void>;
         /**
          * @zh 设置用户输入内容。
          * @en Sets the user input content.

package/dist/ui-store/createChatStore.js

@@ -285,7 +285,7 @@ export const createChatStore = (initClientName, config, context = {}) => {
              */
             async revertChatTo(messageId, resend = false, sendOptions) {
                 var _a;
-                await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.revertChatTo(messageId));
+                await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.revertChatTo(messageId, sendOptions || {}));
                 if (resend) {
                     return sendMessage([], sendOptions, true);
                 }

package/dist/vue/ChatContext.d.ts

@@ -77,7 +77,7 @@ export declare const useChatProvider: (props: ChatProviderProps) => {
             addToHistory: (thread: import("@langchain/langgraph-sdk").Thread<{
                 messages: import("@langchain/langgraph-sdk").Message[];
             }>) => void;
-            revertChatTo(messageId: string, resend?: boolean, sendOptions?: import("../LangGraphClient.js").SendMessageOptions): Promise<void>;
+            revertChatTo(messageId: string, resend?: boolean, sendOptions?: import("../LangGraphClient.js").SendMessageOptions & import("../time-travel/index.js").RevertChatToOptions): Promise<void>;
             setUserInput(input: string): void;
             setCurrentAgent(agent: string): Promise<void>;
             toggleGraphVisible(): void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "3.5.2",
+  "version": "3.7.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

@@ -4,7 +4,7 @@ import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
 import { type ILangGraphClient } from "@langgraph-js/pure-graph/dist/types.js";
 import { MessageProcessor } from "./MessageProcessor.js";
-import { revertChatTo } from "./time-travel/index.js";
+import { revertChatTo, RevertChatToOptions } from "./time-travel/index.js";
 
 export type RenderMessage = Message & {
     /** 对于 AIMessage 来说是节点名称，对于工具节点来说是工具名称 */
@@ -310,8 +310,8 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
      * @zh 回滚到指定的消息。但是不会触发数据的重新更新
      * @en Reverts to the specified message.
      */
-    async revertChatTo(messageId: string) {
-        const { state, checkpoint } = await revertChatTo(this.client as any, this.currentThread!.thread_id, messageId);
+    async revertChatTo(messageId: string, options: RevertChatToOptions) {
+        const { state, checkpoint } = await revertChatTo(this.client as any, this.currentThread!.thread_id, messageId, options);
         this.graphState = state;
         this.messageProcessor.clearStreamingMessages();
         this.messageProcessor.setGraphMessages(state.messages! as RenderMessage[]);

package/src/MessageProcessor.ts

@@ -164,15 +164,13 @@ export class MessageProcessor {
                 const parentMessage = toolParentMessage.get(message.tool_call_id!);
                 if (assistantToolMessage) {
                     message.tool_input = typeof assistantToolMessage.args !== "string" ? JSON.stringify(assistantToolMessage.args) : assistantToolMessage.args;
-                    if (message.additional_kwargs) {
-                        message.additional_kwargs.done = true;
-                        message.done = true;
-                    } else {
-                        message.done = true;
-                        message.additional_kwargs = {
-                            done: true,
-                        };
-                    }
+                    const isDone = !!message.content;
+                    message.done = isDone;
+                    message.additional_kwargs = {
+                        ...(parentMessage?.additional_kwargs || {}),
+                        ...(message.additional_kwargs || {}),
+                        done: isDone,
+                    };
                 }
                 if (parentMessage) {
                     message.usage_metadata = parentMessage.usage_metadata;

package/src/time-travel/index.ts

@@ -1,13 +1,18 @@
 import { Client, Message } from "@langchain/langgraph-sdk";
 
-export async function revertChatTo(client: Client<{ messages: Message[] }, { messages: Message[] }, unknown>, threadId: string, messageId: string) {
+export interface RevertChatToOptions {
+    includeMessageId?: boolean;
+}
+
+export async function revertChatTo(client: Client<{ messages: Message[] }, { messages: Message[] }, unknown>, threadId: string, messageId: string, options: RevertChatToOptions) {
     const thread = await client.threads.get(threadId);
     const messages = thread.values.messages;
     const idx = messages.findIndex((message) => message.id === messageId);
     if (idx === -1) {
         throw new Error(`Message id ${messageId} not found`);
     }
-    const removeMessages = messages.slice(idx + 1);
+    const removeStartIdx = options.includeMessageId ? idx : idx + 1;
+    const removeMessages = messages.slice(removeStartIdx);
     const state = {
         ...thread.values,
         messages: removeMessages.map((i) => {
@@ -25,7 +30,7 @@ export async function revertChatTo(client: Client<{ messages: Message[] }, { mes
     // client.runs.wait();
     return {
         state: {
-            messages: messages.slice(0, idx + 1),
+            messages: messages.slice(0, removeStartIdx),
         },
         checkpoint: res,
     };

package/src/ui-store/createChatStore.ts

@@ -6,6 +6,7 @@ import { ToolRenderData } from "../tool/ToolUI.js";
 import { UnionTool } from "../tool/createTool.js";
 import { createLangGraphServerClient } from "../client/LanggraphServer.js";
 import { useArtifacts } from "../artifacts/index.js";
+import { RevertChatToOptions } from "../time-travel/index.js";
 
 /**
  * @zh 格式化日期对象为时间字符串。
@@ -290,8 +291,8 @@ export const createChatStore = (
              * @zh 回滚到指定的消息。
              * @en Reverts to the specified message.
              */
-            async revertChatTo(messageId: string, resend = false, sendOptions?: SendMessageOptions) {
-                await client.get()?.revertChatTo(messageId);
+            async revertChatTo(messageId: string, resend = false, sendOptions?: SendMessageOptions & RevertChatToOptions) {
+                await client.get()?.revertChatTo(messageId, sendOptions || {});
                 if (resend) {
                     return sendMessage([], sendOptions, true);
                 } else {