@langgraph-js/sdk 2.0.0 → 2.2.0

package/dist/LangGraphClient.d.ts

@@ -68,7 +68,7 @@ export interface LangGraphClientConfig {
     timeoutMs?: number;
     defaultHeaders?: Record<string, string | null | undefined>;
     /** 自定义客户端实现，如果不提供则使用官方 Client */
-    client: ILangGraphClient;
+    client: ILangGraphClient<any, any>;
 }
 export interface LangGraphEvents {
     /** 流开始事件 */
@@ -113,6 +113,11 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
     private currentAssistant;
     private currentThread;
     tools: ToolManager;
+    availableAssistants: Assistant[];
+    graphState: any;
+    currentRun?: {
+        run_id: string;
+    };
     stopController: AbortController | null;
     /** 用于存储 subAgent 状态数据的键 */
     subAgentsKey: string;
@@ -135,7 +140,7 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
     };
     /** 代理 threads 属性到内部 client */
     get threads(): {
-        create<ValuesType = TStateType>(payload?: {
+        create(payload?: {
             metadata?: import("@langchain/langgraph-sdk").Metadata;
             threadId?: string;
             ifExists?: import("@langchain/langgraph-sdk").OnConflictBehavior;
@@ -147,16 +152,16 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
                     asNode: string;
                 }>;
             }>;
-        }): Promise<Thread<ValuesType>>;
-        search<ValuesType = TStateType>(query?: {
+        }): Promise<Thread<TStateType>>;
+        search(query?: {
             metadata?: import("@langchain/langgraph-sdk").Metadata;
             limit?: number;
             offset?: number;
             status?: import("@langchain/langgraph-sdk").ThreadStatus;
             sortBy?: import("./types.js").ThreadSortBy;
             sortOrder?: import("./types.js").SortOrder;
-        }): Promise<Thread<ValuesType>[]>;
-        get<ValuesType = TStateType>(threadId: string): Promise<Thread<ValuesType>>;
+        }): Promise<Thread<TStateType>[]>;
+        get(threadId: string): Promise<Thread<TStateType>>;
         delete(threadId: string): Promise<void>;
     };
     /** 代理 runs 属性到内部 client */
@@ -166,31 +171,7 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
             offset?: number;
             status?: import("./types.js").RunStatus;
         }): Promise<import("@langchain/langgraph-sdk").Run[]>;
-        stream<TStreamMode extends import("@langchain/langgraph-sdk").StreamMode | import("@langchain/langgraph-sdk").StreamMode[] = import("@langchain/langgraph-sdk").StreamMode, TSubgraphs extends boolean = false>(threadId: null, assistantId: string, payload?: {
-            input?: Record<string, unknown> | null;
-            metadata?: import("@langchain/langgraph-sdk").Metadata;
-            config?: import("@langchain/langgraph-sdk").Config;
-            checkpointId?: string;
-            checkpoint?: Omit<import("@langchain/langgraph-sdk").Checkpoint, "thread_id">;
-            checkpointDuring?: boolean;
-            interruptBefore?: "*" | string[];
-            interruptAfter?: "*" | string[];
-            signal?: AbortController["signal"];
-            webhook?: string;
-            onDisconnect?: import("./types.js").DisconnectMode;
-            afterSeconds?: number;
-            ifNotExists?: "create" | "reject";
-            command?: Command;
-            onRunCreated?: (params: {
-                run_id: string;
-                thread_id?: string;
-            }) => void;
-            streamMode?: TStreamMode | undefined;
-            streamSubgraphs?: TSubgraphs | undefined;
-            streamResumable?: boolean;
-            feedbackKeys?: string[];
-        } | undefined): import("./types.js").TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType, unknown>;
-        stream<TStreamMode extends import("@langchain/langgraph-sdk").StreamMode | import("@langchain/langgraph-sdk").StreamMode[] = import("@langchain/langgraph-sdk").StreamMode, TSubgraphs extends boolean = false>(threadId: string, assistantId: string, payload?: {
+        stream<TSubgraphs extends boolean = false>(threadId: string, assistantId: string, payload?: {
             input?: Record<string, unknown> | null;
             metadata?: import("@langchain/langgraph-sdk").Metadata;
             config?: import("@langchain/langgraph-sdk").Config;
@@ -211,12 +192,12 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
                 run_id: string;
                 thread_id?: string;
             }) => void;
-            streamMode?: TStreamMode | undefined;
+            streamMode?: import("@langchain/langgraph-sdk").StreamMode[];
             streamSubgraphs?: TSubgraphs | undefined;
             streamResumable?: boolean;
             feedbackKeys?: string[];
-        } | undefined): import("./types.js").TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType, unknown>;
-        joinStream(threadId: string | undefined | null, runId: string, options?: {
+        } | undefined): import("./types.js").TypedAsyncGenerator<TSubgraphs, TStateType, TUpdateType>;
+        joinStream(threadId: string, runId: string, options?: {
             signal?: AbortSignal;
             cancelOnDisconnect?: boolean;
             lastEventId?: string;
@@ -228,7 +209,6 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
         }>;
         cancel(threadId: string, runId: string, wait?: boolean, action?: import("./types.js").CancelAction): Promise<void>;
     };
-    availableAssistants: Assistant[];
     private listAssistants;
     /**
      * @zh 初始化 Assistant。
@@ -239,8 +219,9 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
      * @zh 创建一个新的 Thread。
      * @en Creates a new Thread.
      */
-    createThread({ threadId, }?: {
+    createThread({ threadId, graphId, }?: {
         threadId?: string;
+        graphId?: string;
     }): Promise<Thread<TStateType>>;
     graphVisualize(): Promise<import("@langchain/langgraph-sdk").AssistantGraph>;
     /**
@@ -272,15 +253,18 @@ export declare class LangGraphClient<TStateType = unknown, TUpdateType = TStateT
     };
     /** 前端工具人机交互时，锁住面板 */
     isFELocking(messages: RenderMessage[]): boolean | undefined;
-    graphState: any;
-    currentRun?: {
-        run_id: string;
-    };
     /**
      * @zh 取消当前的 Run。
      * @en Cancels the current Run.
      */
     cancelRun(): void;
+    /**
+     * @zh 回滚到指定的消息。但是不会触发数据的重新更新
+     * @en Reverts to the specified message.
+     */
+    revertChatTo(messageId: string): Promise<{
+        messages: Message[];
+    }>;
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.

package/dist/LangGraphClient.js

@@ -1,6 +1,7 @@
 import { EventEmitter } from "eventemitter3";
 import { ToolManager } from "./ToolManager.js";
 import { MessageProcessor } from "./MessageProcessor.js";
+import { revertChatTo } from "./time-travel/index.js";
 /**
  * @zh LangGraphClient 类是与 LangGraph 后端交互的主要客户端。
  * @en The LangGraphClient class is the main client for interacting with the LangGraph backend.
@@ -11,11 +12,11 @@ export class LangGraphClient extends EventEmitter {
         this.currentAssistant = null;
         this.currentThread = null;
         this.tools = new ToolManager();
+        this.availableAssistants = [];
+        this.graphState = {};
         this.stopController = null;
         /** 用于存储 subAgent 状态数据的键 */
         this.subAgentsKey = "task_store";
-        this.availableAssistants = [];
-        this.graphState = {};
         /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
         this.graphPosition = "";
         this.extraParams = {};
@@ -73,10 +74,11 @@ export class LangGraphClient extends EventEmitter {
      * @zh 创建一个新的 Thread。
      * @en Creates a new Thread.
      */
-    async createThread({ threadId, } = {}) {
+    async createThread({ threadId, graphId, } = {}) {
         try {
             this.currentThread = await this.threads.create({
                 threadId,
+                graphId,
             });
             return this.currentThread;
         }
@@ -185,6 +187,17 @@ export class LangGraphClient extends EventEmitter {
             this.runs.cancel(this.currentThread.thread_id, this.currentRun.run_id);
         }
     }
+    /**
+     * @zh 回滚到指定的消息。但是不会触发数据的重新更新
+     * @en Reverts to the specified message.
+     */
+    async revertChatTo(messageId) {
+        const { state, checkpoint } = await revertChatTo(this.client, this.currentThread.thread_id, messageId);
+        this.graphState = state;
+        this.messageProcessor.clearStreamingMessages();
+        this.messageProcessor.setGraphMessages(state.messages);
+        return state;
+    }
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -195,7 +208,7 @@ export class LangGraphClient extends EventEmitter {
             throw new Error("Thread or Assistant not initialized");
         }
         if (!this.currentThread) {
-            await this.createThread();
+            await this.createThread({ graphId: this.currentAssistant.graph_id });
             this.emit("thread", {
                 event: "thread/create",
                 data: {

package/dist/client/LanggraphServer.d.ts

@@ -1,2 +1,3 @@
+import { LangGraphClientConfig } from "../LangGraphClient.js";
 import { ILangGraphClient } from "../types.js";
-export declare const createLangGraphServerClient: () => Promise<ILangGraphClient>;
+export declare const createLangGraphServerClient: (config: LangGraphClientConfig) => Promise<ILangGraphClient>;

package/dist/client/LanggraphServer.js

@@ -1,4 +1,4 @@
-export const createLangGraphServerClient = async () => {
+export const createLangGraphServerClient = async (config) => {
     const { Client } = await import("@langchain/langgraph-sdk");
-    return new Client();
+    return new Client(config);
 };

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

@@ -0,0 +1,11 @@
+import { Client, Message } from "@langchain/langgraph-sdk";
+export declare function revertChatTo(client: Client<{
+    messages: Message[];
+}, {
+    messages: Message[];
+}, unknown>, threadId: string, messageId: string): Promise<{
+    state: {
+        messages: Message[];
+    };
+    checkpoint: Pick<import("@langchain/langgraph-sdk").Config, "configurable">;
+}>;

package/dist/time-travel/index.js

@@ -0,0 +1,30 @@
+export async function revertChatTo(client, threadId, messageId) {
+    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 state = {
+        ...thread.values,
+        messages: removeMessages.map((i) => {
+            // sb langgraph 官方实现都不能正常工作
+            // return new RemoveMessage({ id: i.id! }).toJSON();
+            return {
+                type: "remove",
+                id: i.id,
+            };
+        }),
+    };
+    const res = await client.threads.updateState(threadId, {
+        values: state,
+    });
+    // client.runs.wait();
+    return {
+        state: {
+            messages: messages.slice(0, idx + 1),
+        },
+        checkpoint: res,
+    };
+}

package/dist/types.d.ts

@@ -8,7 +8,7 @@ export type MultitaskStrategy = "reject" | "interrupt" | "rollback" | "enqueue";
 export type DisconnectMode = "cancel" | "continue";
 export type OnCompletionBehavior = "complete" | "continue";
 export type CancelAction = "interrupt" | "rollback";
-export type TypedAsyncGenerator<TStreamMode extends StreamMode | StreamMode[] = [], TSubgraphs extends boolean = false, TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = AsyncGenerator<{
+export type TypedAsyncGenerator<TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = AsyncGenerator<{
     values: ValuesStreamEvent<TStateType>;
     updates: UpdatesStreamEvent<TUpdateType>;
     custom: CustomStreamEvent<TCustomType>;
@@ -16,11 +16,11 @@ export type TypedAsyncGenerator<TStreamMode extends StreamMode | StreamMode[] =
     messages: MessagesStreamEvent;
     "messages-tuple": MessagesTupleStreamEvent;
     events: EventsStreamEvent;
-}[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode] | ErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent>;
+}[StreamMode] | ErrorStreamEvent | MetadataStreamEvent | FeedbackStreamEvent>;
 /**
  * 兼容 LangGraph SDK 的接口定义，方便进行无侵入式的扩展
  */
-export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType> {
+export interface ILangGraphClient<TStateType = {}, TUpdateType = TStateType> {
     assistants: {
         search(query?: {
             graphId?: string;
@@ -35,7 +35,7 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
         }): Promise<AssistantGraph>;
     };
     threads: {
-        create<ValuesType = TStateType>(payload?: {
+        create(payload?: {
             metadata?: Metadata;
             threadId?: string;
             ifExists?: OnConflictBehavior;
@@ -47,16 +47,16 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
                     asNode: string;
                 }>;
             }>;
-        }): Promise<Thread<ValuesType>>;
-        search<ValuesType = TStateType>(query?: {
+        }): Promise<Thread<TStateType>>;
+        search(query?: {
             metadata?: Metadata;
             limit?: number;
             offset?: number;
             status?: ThreadStatus;
             sortBy?: ThreadSortBy;
             sortOrder?: SortOrder;
-        }): Promise<Thread<ValuesType>[]>;
-        get<ValuesType = TStateType>(threadId: string): Promise<Thread<ValuesType>>;
+        }): Promise<Thread<TStateType>[]>;
+        get(threadId: string): Promise<Thread<TStateType>>;
         delete(threadId: string): Promise<void>;
     };
     runs: {
@@ -65,31 +65,7 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
             offset?: number;
             status?: RunStatus;
         }): Promise<Run[]>;
-        stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(threadId: null, assistantId: string, payload?: {
-            input?: Record<string, unknown> | null;
-            metadata?: Metadata;
-            config?: Config;
-            checkpointId?: string;
-            checkpoint?: Omit<Checkpoint, "thread_id">;
-            checkpointDuring?: boolean;
-            interruptBefore?: "*" | string[];
-            interruptAfter?: "*" | string[];
-            signal?: AbortController["signal"];
-            webhook?: string;
-            onDisconnect?: DisconnectMode;
-            afterSeconds?: number;
-            ifNotExists?: "create" | "reject";
-            command?: Command;
-            onRunCreated?: (params: {
-                run_id: string;
-                thread_id?: string;
-            }) => void;
-            streamMode?: TStreamMode;
-            streamSubgraphs?: TSubgraphs;
-            streamResumable?: boolean;
-            feedbackKeys?: string[];
-        }): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType>;
-        stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(threadId: string, assistantId: string, payload?: {
+        stream<TSubgraphs extends boolean = false>(threadId: string, assistantId: string, payload?: {
             input?: Record<string, unknown> | null;
             metadata?: Metadata;
             config?: Config;
@@ -110,12 +86,12 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
                 run_id: string;
                 thread_id?: string;
             }) => void;
-            streamMode?: TStreamMode;
+            streamMode?: StreamMode[];
             streamSubgraphs?: TSubgraphs;
             streamResumable?: boolean;
             feedbackKeys?: string[];
-        }): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType>;
-        joinStream(threadId: string | undefined | null, runId: string, options?: {
+        }): TypedAsyncGenerator<TSubgraphs, TStateType, TUpdateType>;
+        joinStream(threadId: string, runId: string, options?: {
             signal?: AbortSignal;
             cancelOnDisconnect?: boolean;
             lastEventId?: string;

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

@@ -25,7 +25,7 @@ export declare const getHistoryContent: (thread: Thread) => string | any[];
  * @zh 创建一个用于聊天界面的状态管理器 (store)。
  * @en Creates a state manager (store) for the chat interface.
  */
-export declare const createChatStore: (initClientName: string, config: LangGraphClientConfig, context?: {
+export declare const createChatStore: (initClientName: string, config: Partial<LangGraphClientConfig>, context?: {
     showHistory?: boolean;
     showGraph?: boolean;
     onInit?: (client: LangGraphClient) => void;
@@ -53,7 +53,7 @@ export declare const createChatStore: (initClientName: string, config: LangGraph
         setTools(new_tools: UnionTool<any>[]): void;
         isFELocking(): boolean | undefined;
         initClient: () => Promise<LangGraphClient<unknown, unknown>>;
-        sendMessage: (message?: Message[], extraData?: SendMessageOptions) => Promise<void>;
+        sendMessage: (message?: Message[], extraData?: SendMessageOptions, withoutCheck?: boolean) => Promise<void>;
         stopGeneration: () => void;
         toggleToolCollapse: (toolId: string) => void;
         toggleHistoryVisible: () => void;
@@ -61,6 +61,11 @@ export declare const createChatStore: (initClientName: string, config: LangGraph
         addToHistory: (thread: Thread<{
             messages: Message[];
         }>) => void;
+        /**
+         * @zh 回滚到指定的消息。
+         * @en Reverts to the specified message.
+         */
+        revertChatTo(messageId: string, resend?: boolean, sendOptions?: SendMessageOptions): Promise<void>;
         /**
          * @zh 设置用户输入内容。
          * @en Sets the user input content.

package/dist/ui-store/createChatStore.js

@@ -106,7 +106,7 @@ export const createChatStore = (initClientName, config, context = {}) => {
         var _a, _b;
         const newClient = new LangGraphClient({
             ...config,
-            client: (_a = config.client) !== null && _a !== void 0 ? _a : (await createLangGraphServerClient()),
+            client: (_a = config.client) !== null && _a !== void 0 ? _a : (await createLangGraphServerClient(config)),
         });
         await newClient.initAssistant(currentAgent.get());
         currentAgent.set(newClient.getCurrentAssistant().graph_id);
@@ -158,15 +158,25 @@ export const createChatStore = (initClientName, config, context = {}) => {
      * @zh 发送消息。
      * @en Sends a message.
      */
-    const sendMessage = async (message, extraData) => {
-        var _a;
-        if ((!userInput.get().trim() && !(message === null || message === void 0 ? void 0 : message.length)) || loading.get() || !client.get())
+    const sendMessage = async (message, extraData, withoutCheck = false) => {
+        var _a, _b;
+        if ((!withoutCheck && !userInput.get().trim() && !(message === null || message === void 0 ? void 0 : message.length)) || loading.get() || !client.get())
             return;
         loading.set(true);
         inChatError.set(null);
-        await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.sendMessage(message || userInput.get(), extraData));
-        userInput.set("");
-        loading.set(false);
+        try {
+            await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.sendMessage(message || userInput.get(), extraData));
+        }
+        catch (e) {
+            const isThreadRunning = e.message.includes("422");
+            if (isThreadRunning) {
+                await ((_b = client.get()) === null || _b === void 0 ? void 0 : _b.resetStream());
+            }
+        }
+        finally {
+            userInput.set("");
+            loading.set(false);
+        }
     };
     /**
      * @zh 停止当前的消息生成。
@@ -259,6 +269,20 @@ export const createChatStore = (initClientName, config, context = {}) => {
             toggleHistoryVisible,
             refreshHistoryList,
             addToHistory,
+            /**
+             * @zh 回滚到指定的消息。
+             * @en Reverts to the specified message.
+             */
+            async revertChatTo(messageId, resend = false, sendOptions) {
+                var _a;
+                await ((_a = client.get()) === null || _a === void 0 ? void 0 : _a.revertChatTo(messageId));
+                if (resend) {
+                    return sendMessage([], sendOptions, true);
+                }
+                else {
+                    updateUI(client.get());
+                }
+            },
             /**
              * @zh 设置用户输入内容。
              * @en Sets the user input content.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "2.0.0",
+  "version": "2.2.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,6 +4,7 @@ import { ToolManager } from "./ToolManager.js";
 import { CallToolResult } from "./tool/createTool.js";
 import { ILangGraphClient } from "./types.js";
 import { MessageProcessor } from "./MessageProcessor.js";
+import { revertChatTo } from "./time-travel/index.js";
 
 export type RenderMessage = Message & {
     /** 对于 AIMessage 来说是节点名称，对于工具节点来说是工具名称 */
@@ -68,7 +69,7 @@ export interface LangGraphClientConfig {
     timeoutMs?: number;
     defaultHeaders?: Record<string, string | null | undefined>;
     /** 自定义客户端实现，如果不提供则使用官方 Client */
-    client: ILangGraphClient;
+    client: ILangGraphClient<any, any>;
 }
 
 // 定义事件数据类型
@@ -96,6 +97,9 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
     private currentAssistant: Assistant | null = null;
     private currentThread: Thread<TStateType> | null = null;
     tools: ToolManager = new ToolManager();
+    availableAssistants: Assistant[] = [];
+    graphState: any = {};
+    currentRun?: { run_id: string };
     stopController: AbortController | null = null;
     /** 用于存储 subAgent 状态数据的键 */
     subAgentsKey = "task_store";
@@ -122,7 +126,6 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
     get runs() {
         return this.client.runs;
     }
-    availableAssistants: Assistant[] = [];
     private listAssistants() {
         return this.assistants.search({
             metadata: null,
@@ -162,12 +165,15 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
      */
     async createThread({
         threadId,
+        graphId,
     }: {
         threadId?: string;
+        graphId?: string;
     } = {}) {
         try {
             this.currentThread = await this.threads.create({
                 threadId,
+                graphId,
             });
             return this.currentThread;
         } catch (error) {
@@ -272,8 +278,7 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
         const tool = this.tools.getTool(lastMessage?.name!);
         return tool && tool.render && lastMessage?.type === "tool" && !lastMessage?.additional_kwargs?.done;
     }
-    graphState: any = {};
-    currentRun?: { run_id: string };
+
     /**
      * @zh 取消当前的 Run。
      * @en Cancels the current Run.
@@ -283,6 +288,17 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
             this.runs.cancel((this.currentThread as any)!.thread_id, this.currentRun.run_id);
         }
     }
+    /**
+     * @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);
+        this.graphState = state;
+        this.messageProcessor.clearStreamingMessages();
+        this.messageProcessor.setGraphMessages(state.messages! as RenderMessage[]);
+        return state;
+    }
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -292,7 +308,7 @@ export class LangGraphClient<TStateType = unknown, TUpdateType = TStateType> ext
             throw new Error("Thread or Assistant not initialized");
         }
         if (!this.currentThread) {
-            await this.createThread();
+            await this.createThread({ graphId: this.currentAssistant!.graph_id });
             this.emit("thread", {
                 event: "thread/create",
                 data: {

package/src/client/LanggraphServer.ts

@@ -1,6 +1,7 @@
+import { LangGraphClientConfig } from "../LangGraphClient.js";
 import { ILangGraphClient } from "../types.js";
 
-export const createLangGraphServerClient = async (): Promise<ILangGraphClient> => {
+export const createLangGraphServerClient = async (config: LangGraphClientConfig): Promise<ILangGraphClient> => {
     const { Client } = await import("@langchain/langgraph-sdk");
-    return new Client() as unknown as ILangGraphClient;
+    return new Client(config) as unknown as ILangGraphClient;
 };

package/src/time-travel/index.ts

@@ -0,0 +1,33 @@
+import { RemoveMessage } from "@langchain/core/messages";
+import { Client, Message } from "@langchain/langgraph-sdk";
+
+export async function revertChatTo(client: Client<{ messages: Message[] }, { messages: Message[] }, unknown>, threadId: string, messageId: string) {
+    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 state = {
+        ...thread.values,
+        messages: removeMessages.map((i) => {
+            // sb langgraph 官方实现都不能正常工作
+            // return new RemoveMessage({ id: i.id! }).toJSON();
+            return {
+                type: "remove",
+                id: i.id!,
+            };
+        }),
+    };
+    const res = await client.threads.updateState(threadId, {
+        values: state,
+    });
+    // client.runs.wait();
+    return {
+        state: {
+            messages: messages.slice(0, idx + 1),
+        },
+        checkpoint: res,
+    };
+}

package/src/types.ts

@@ -34,13 +34,7 @@ export type OnCompletionBehavior = "complete" | "continue";
 export type CancelAction = "interrupt" | "rollback";
 
 // 流式异步生成器类型
-export type TypedAsyncGenerator<
-    TStreamMode extends StreamMode | StreamMode[] = [],
-    TSubgraphs extends boolean = false,
-    TStateType = unknown,
-    TUpdateType = TStateType,
-    TCustomType = unknown,
-> = AsyncGenerator<
+export type TypedAsyncGenerator<TStateType = unknown, TUpdateType = TStateType, TCustomType = unknown> = AsyncGenerator<
     | {
           values: ValuesStreamEvent<TStateType>;
           updates: UpdatesStreamEvent<TUpdateType>;
@@ -49,7 +43,7 @@ export type TypedAsyncGenerator<
           messages: MessagesStreamEvent;
           "messages-tuple": MessagesTupleStreamEvent;
           events: EventsStreamEvent;
-      }[TStreamMode extends StreamMode[] ? TStreamMode[number] : TStreamMode]
+      }[StreamMode]
     | ErrorStreamEvent
     | MetadataStreamEvent
     | FeedbackStreamEvent
@@ -58,13 +52,13 @@ export type TypedAsyncGenerator<
 /**
  * 兼容 LangGraph SDK 的接口定义，方便进行无侵入式的扩展
  */
-export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType> {
+export interface ILangGraphClient<TStateType = {}, TUpdateType = TStateType> {
     assistants: {
         search(query?: { graphId?: string; metadata?: Metadata; limit?: number; offset?: number; sortBy?: AssistantSortBy; sortOrder?: SortOrder }): Promise<Assistant[]>;
         getGraph(assistantId: string, options?: { xray?: boolean | number }): Promise<AssistantGraph>;
     };
     threads: {
-        create<ValuesType = TStateType>(payload?: {
+        create(payload?: {
             metadata?: Metadata;
             threadId?: string;
             ifExists?: OnConflictBehavior;
@@ -76,16 +70,9 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
                     asNode: string;
                 }>;
             }>;
-        }): Promise<Thread<ValuesType>>;
-        search<ValuesType = TStateType>(query?: {
-            metadata?: Metadata;
-            limit?: number;
-            offset?: number;
-            status?: ThreadStatus;
-            sortBy?: ThreadSortBy;
-            sortOrder?: SortOrder;
-        }): Promise<Thread<ValuesType>[]>;
-        get<ValuesType = TStateType>(threadId: string): Promise<Thread<ValuesType>>;
+        }): Promise<Thread<TStateType>>;
+        search(query?: { metadata?: Metadata; limit?: number; offset?: number; status?: ThreadStatus; sortBy?: ThreadSortBy; sortOrder?: SortOrder }): Promise<Thread<TStateType>[]>;
+        get(threadId: string): Promise<Thread<TStateType>>;
         delete(threadId: string): Promise<void>;
     };
     runs: {
@@ -97,32 +84,8 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
                 status?: RunStatus;
             }
         ): Promise<Run[]>;
-        stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(
-            threadId: null,
-            assistantId: string,
-            payload?: {
-                input?: Record<string, unknown> | null;
-                metadata?: Metadata;
-                config?: Config;
-                checkpointId?: string;
-                checkpoint?: Omit<Checkpoint, "thread_id">;
-                checkpointDuring?: boolean;
-                interruptBefore?: "*" | string[];
-                interruptAfter?: "*" | string[];
-                signal?: AbortController["signal"];
-                webhook?: string;
-                onDisconnect?: DisconnectMode;
-                afterSeconds?: number;
-                ifNotExists?: "create" | "reject";
-                command?: Command;
-                onRunCreated?: (params: { run_id: string; thread_id?: string }) => void;
-                streamMode?: TStreamMode;
-                streamSubgraphs?: TSubgraphs;
-                streamResumable?: boolean;
-                feedbackKeys?: string[];
-            }
-        ): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType>;
-        stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(
+
+        stream<TSubgraphs extends boolean = false>(
             threadId: string,
             assistantId: string,
             payload?: {
@@ -143,14 +106,14 @@ export interface ILangGraphClient<TStateType = unknown, TUpdateType = TStateType
                 ifNotExists?: "create" | "reject";
                 command?: Command;
                 onRunCreated?: (params: { run_id: string; thread_id?: string }) => void;
-                streamMode?: TStreamMode;
+                streamMode?: StreamMode[];
                 streamSubgraphs?: TSubgraphs;
                 streamResumable?: boolean;
                 feedbackKeys?: string[];
             }
-        ): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType>;
+        ): TypedAsyncGenerator<TSubgraphs, TStateType, TUpdateType>;
         joinStream(
-            threadId: string | undefined | null,
+            threadId: string,
             runId: string,
             options?:
                 | {

package/src/ui-store/createChatStore.ts

@@ -66,7 +66,7 @@ export const getHistoryContent = (thread: Thread) => {
  */
 export const createChatStore = (
     initClientName: string,
-    config: LangGraphClientConfig,
+    config: Partial<LangGraphClientConfig>,
     context: {
         showHistory?: boolean;
         showGraph?: boolean;
@@ -113,7 +113,7 @@ export const createChatStore = (
     async function initClient() {
         const newClient = new LangGraphClient({
             ...config,
-            client: config.client ?? (await createLangGraphServerClient()),
+            client: config.client ?? (await createLangGraphServerClient(config as LangGraphClientConfig)),
         });
         await newClient.initAssistant(currentAgent.get());
         currentAgent.set(newClient.getCurrentAssistant()!.graph_id);
@@ -167,16 +167,22 @@ export const createChatStore = (
      * @zh 发送消息。
      * @en Sends a message.
      */
-    const sendMessage = async (message?: Message[], extraData?: SendMessageOptions) => {
-        if ((!userInput.get().trim() && !message?.length) || loading.get() || !client.get()) return;
+    const sendMessage = async (message?: Message[], extraData?: SendMessageOptions, withoutCheck = false) => {
+        if ((!withoutCheck && !userInput.get().trim() && !message?.length) || loading.get() || !client.get()) return;
 
         loading.set(true);
         inChatError.set(null);
-
-        await client.get()?.sendMessage(message || userInput.get(), extraData);
-
-        userInput.set("");
-        loading.set(false);
+        try {
+            await client.get()?.sendMessage(message || userInput.get(), extraData);
+        } catch (e) {
+            const isThreadRunning = (e as Error).message.includes("422");
+            if (isThreadRunning) {
+                await client.get()?.resetStream();
+            }
+        } finally {
+            userInput.set("");
+            loading.set(false);
+        }
     };
 
     /**
@@ -270,6 +276,18 @@ export const createChatStore = (
             toggleHistoryVisible,
             refreshHistoryList,
             addToHistory,
+            /**
+             * @zh 回滚到指定的消息。
+             * @en Reverts to the specified message.
+             */
+            async revertChatTo(messageId: string, resend = false, sendOptions?: SendMessageOptions) {
+                await client.get()?.revertChatTo(messageId);
+                if (resend) {
+                    return sendMessage([], sendOptions, true);
+                } else {
+                    updateUI(client.get()!);
+                }
+            },
             /**
              * @zh 设置用户输入内容。
              * @en Sets the user input content.