@langgraph-js/sdk 3.3.1 → 3.4.1

package/dist/LangGraphClient.d.ts

@@ -222,6 +222,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
     revertChatTo(messageId: string): Promise<{
         messages: Message[];
     }>;
+    messagesMetadata: {};
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.

package/dist/LangGraphClient.js

@@ -17,6 +17,7 @@ export class LangGraphClient extends EventEmitter {
         this.stopController = null;
         /** 用于存储 subAgent 状态数据的键 */
         this.subAgentsKey = "task_store";
+        this.messagesMetadata = {};
         /** 当前子图位置，但是依赖 stream，不太适合稳定使用*/
         this.graphPosition = "";
         this.extraParams = {};
@@ -146,7 +147,7 @@ export class LangGraphClient extends EventEmitter {
      * @en Messages used for streaming rendering in the UI.
      */
     get renderMessage() {
-        return this.messageProcessor.renderMessages(this.graphState, () => this.getGraphNodeNow());
+        return this.messageProcessor.renderMessages(this.graphState, () => this.getGraphNodeNow(), this.messagesMetadata);
     }
     /**
      * @zh 获取 Token 计数器信息。
@@ -263,6 +264,10 @@ export class LangGraphClient extends EventEmitter {
             else if (chunk.event === "error" || chunk.event === "Error" || chunk.event === "__stream_error__") {
                 this.emit("error", chunk);
             }
+            else if (chunk.event === "messages/metadata") {
+                Object.assign(this.messagesMetadata, chunk.data);
+                continue;
+            }
             else if (chunk.event === "messages/partial") {
                 for (const message of chunk.data) {
                     this.messageProcessor.updateStreamingMessage(message);

package/dist/MessageProcessor.d.ts

@@ -78,17 +78,23 @@ export declare class MessageProcessor {
      * @zh 转换 subAgent 消息为工具的子消息
      * @en Convert subAgent messages to tool sub-messages
      */
-    convertSubAgentMessages(messages: RenderMessage[], graphState: any): RenderMessage[];
+    convertSubAgentMessages(messages: RenderMessage[], graphState: any, messagesMetadata: Record<string, {
+        subagent_id?: string;
+    }>): RenderMessage[];
     /**
      * @zh 生成用于 UI 中的流式渲染的消息
      * @en Generate messages used for streaming rendering in the UI
      */
     renderMessages(graphState: any, getGraphNodeNow: () => {
         name: string;
-    }): RenderMessage[];
+    }, messagesMetadata: Record<string, {
+        subagent_id?: string;
+    }>): RenderMessage[];
     /**
      * @zh 统一的消息处理入口，按顺序执行所有处理步骤
      * @en Unified message processing entry point, executing all processing steps in order
      */
-    processMessages(messages: RenderMessage[], graphState?: any): RenderMessage[];
+    processMessages(messages: RenderMessage[], graphState: any, messagesMetadata: Record<string, {
+        subagent_id?: string;
+    }>): RenderMessage[];
 }

package/dist/MessageProcessor.js

@@ -194,7 +194,7 @@ export class MessageProcessor {
      * @zh 转换 subAgent 消息为工具的子消息
      * @en Convert subAgent messages to tool sub-messages
      */
-    convertSubAgentMessages(messages, graphState) {
+    convertSubAgentMessages(messages, graphState, messagesMetadata) {
         const origin_task_store = graphState[this.subAgentsKey];
         if (!origin_task_store)
             return messages;
@@ -202,11 +202,12 @@ export class MessageProcessor {
         /** 获取 subAgent 消息的 id，用于流式过程中对数据进行标记 */
         messages
             .filter((i) => {
-            var _a;
-            return (_a = i.node_name) === null || _a === void 0 ? void 0 : _a.startsWith("subagent_");
+            var _a, _b;
+            return ((_a = messagesMetadata[i.id]) === null || _a === void 0 ? void 0 : _a.subagent_id) || ((_b = i.node_name) === null || _b === void 0 ? void 0 : _b.startsWith("subagent_"));
         })
             .forEach((i) => {
-            const tool_call_id = i.node_name.replace("subagent_", "");
+            var _a;
+            const tool_call_id = ((_a = messagesMetadata[i.id]) === null || _a === void 0 ? void 0 : _a.subagent_id) || i.node_name.replace("subagent_", "");
             const store = task_store[tool_call_id];
             if (store) {
                 // 根据 id 进行去重
@@ -233,7 +234,7 @@ export class MessageProcessor {
                 const task = task_store[message.tool_call_id];
                 if (task) {
                     // 递归处理子消息，但避免重复处理
-                    message.sub_agent_messages = this.processMessages(task.messages);
+                    message.sub_agent_messages = this.processMessages(task.messages, task, messagesMetadata);
                 }
             }
             if (message.id && ignoreIds.has(message.id))
@@ -246,7 +247,7 @@ export class MessageProcessor {
      * @zh 生成用于 UI 中的流式渲染的消息
      * @en Generate messages used for streaming rendering in the UI
      */
-    renderMessages(graphState, getGraphNodeNow) {
+    renderMessages(graphState, getGraphNodeNow, messagesMetadata) {
         var _a;
         const previousMessage = new Map();
         const closedToolCallIds = new Set();
@@ -304,20 +305,20 @@ export class MessageProcessor {
                 result.unshift(message);
             }
         }
-        return this.processMessages(result, graphState);
+        return this.processMessages(result, graphState, messagesMetadata);
     }
     /**
      * @zh 统一的消息处理入口，按顺序执行所有处理步骤
      * @en Unified message processing entry point, executing all processing steps in order
      */
-    processMessages(messages, graphState) {
+    processMessages(messages, graphState, messagesMetadata) {
         // 1. 组合工具消息
         const composedMessages = this.composeToolMessages(messages);
         // 2. 附加信息
         const messagesWithInfo = this.attachInfoForMessage(composedMessages);
         // 3. 转换子代理消息（如果提供了 graphState）
         if (graphState) {
-            return this.convertSubAgentMessages(messagesWithInfo, graphState);
+            return this.convertSubAgentMessages(messagesWithInfo, graphState, messagesMetadata);
         }
         return messagesWithInfo;
     }

package/dist/artifacts/types.d.ts

@@ -1,6 +1,10 @@
 import { z } from "zod";
 export declare const ArtifactCommandSchema: {
-    command: z.ZodEnum<["create", "update", "rewrite"]>;
+    command: z.ZodEnum<{
+        create: "create";
+        update: "update";
+        rewrite: "rewrite";
+    }>;
     id: z.ZodString;
     title: z.ZodString;
     type: z.ZodString;

package/dist/tool/createTool.d.ts

@@ -1,4 +1,4 @@
-import { z, ZodRawShape, ZodTypeAny } from "zod";
+import { z, ZodRawShape } from "zod";
 import { Action, Parameter } from "./copilotkit-actions.js";
 import { Message } from "@langchain/langgraph-sdk";
 import { ToolRenderData } from "./ToolUI.js";
@@ -11,8 +11,8 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     execute?: ToolCallback<Args>;
     /** 工具执行成功后触发的附加消息 */
     callbackMessage?: (result: CallToolResult) => Message[];
-    handler?: (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => ResponseType | Promise<ResponseType>;
-    render?: (tool: ToolRenderData<z.objectOutputType<Args, ZodTypeAny>, ResponseType>) => Child;
+    handler?: (args: z.infer<z.ZodObject<Args>>, context?: any) => ResponseType | Promise<ResponseType>;
+    render?: (tool: ToolRenderData<z.infer<z.ZodObject<Args>>, ResponseType>) => Child;
     onlyRender?: boolean;
     /** 只允许指定的 agent 使用该工具，如果未指定，则所有 agent 都可以使用 */
     allowAgent?: string[];
@@ -21,7 +21,7 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     /** 是否是纯净的 json schema 参数，而不是 zod 参数 */
     isPureParams?: boolean;
 }
-export type ToolCallback<Args extends ZodRawShape> = (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => CallToolResult | Promise<CallToolResult>;
+export type ToolCallback<Args extends ZodRawShape> = (args: z.infer<z.ZodObject<Args>>, context?: any) => CallToolResult | Promise<CallToolResult>;
 export type CallToolResult = string | {
     type: "text";
     text: string;
@@ -70,7 +70,7 @@ export declare const createJSONDefineTool: <Args extends ZodRawShape>(tool: Unio
             [key: string]: import("zod-to-json-schema").JsonSchema7Type;
         } | undefined;
     }) | ({
-        type: ("string" | "number" | "boolean" | "integer" | "null") | ("string" | "number" | "boolean" | "integer" | "null")[];
+        type: ("string" | "number" | "boolean" | "null" | "integer") | ("string" | "number" | "boolean" | "null" | "integer")[];
     } & {
         title?: string;
         default?: any;
@@ -149,7 +149,7 @@ export declare const createJSONDefineTool: <Args extends ZodRawShape>(tool: Unio
         } | undefined;
     });
 };
-export declare const createMCPTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => (string | Args | ((args: z.objectOutputType<Args, ZodTypeAny>) => Promise<{
+export declare const createMCPTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => (string | Args | ((args: z.infer<z.ZodObject<Args>>) => Promise<{
     content: {
         type: string;
         text: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/sdk",
-  "version": "3.3.1",
+  "version": "3.4.1",
   "description": "The UI SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces",
   "main": "dist/index.js",
   "type": "module",
@@ -49,7 +49,7 @@
     "jsonrepair": "^3.12.0",
     "nanostores": "^1.0.1",
     "ts-debounce": "^4.0.0",
-    "zod": "^3.25.17",
+    "zod": "^4",
     "zod-to-json-schema": "^3.24.5"
   },
   "peerDependencies": {

package/src/LangGraphClient.ts

@@ -240,7 +240,7 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
      * @en Messages used for streaming rendering in the UI.
      */
     get renderMessage() {
-        return this.messageProcessor.renderMessages(this.graphState, () => this.getGraphNodeNow());
+        return this.messageProcessor.renderMessages(this.graphState, () => this.getGraphNodeNow(), this.messagesMetadata);
     }
     /**
      * @zh 获取 Token 计数器信息。
@@ -304,6 +304,7 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
         this.messageProcessor.setGraphMessages(state.messages! as RenderMessage[]);
         return state;
     }
+    public messagesMetadata = {};
     /**
      * @zh 发送消息到 LangGraph 后端。
      * @en Sends a message to the LangGraph backend.
@@ -363,6 +364,9 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
                 this.currentRun = chunk.data;
             } else if (chunk.event === "error" || chunk.event === "Error" || chunk.event === "__stream_error__") {
                 this.emit("error", chunk);
+            } else if (chunk.event === "messages/metadata") {
+                Object.assign(this.messagesMetadata, chunk.data);
+                continue;
             } else if (chunk.event === "messages/partial") {
                 for (const message of chunk.data) {
                     this.messageProcessor.updateStreamingMessage(message);

package/src/MessageProcessor.ts

@@ -216,7 +216,16 @@ export class MessageProcessor {
      * @zh 转换 subAgent 消息为工具的子消息
      * @en Convert subAgent messages to tool sub-messages
      */
-    convertSubAgentMessages(messages: RenderMessage[], graphState: any): RenderMessage[] {
+    convertSubAgentMessages(
+        messages: RenderMessage[],
+        graphState: any,
+        messagesMetadata: Record<
+            string,
+            {
+                subagent_id?: string;
+            }
+        >
+    ): RenderMessage[] {
         const origin_task_store = graphState[this.subAgentsKey];
         if (!origin_task_store) return messages;
 
@@ -225,10 +234,10 @@ export class MessageProcessor {
         /** 获取 subAgent 消息的 id，用于流式过程中对数据进行标记 */
         messages
             .filter((i) => {
-                return i.node_name?.startsWith("subagent_");
+                return messagesMetadata[i.id!]?.subagent_id || i.node_name?.startsWith("subagent_");
             })
             .forEach((i) => {
-                const tool_call_id = i.node_name!.replace("subagent_", "");
+                const tool_call_id = messagesMetadata[i.id!]?.subagent_id || i.node_name!.replace("subagent_", "");
                 const store = task_store[tool_call_id];
                 if (store) {
                     // 根据 id 进行去重
@@ -256,7 +265,7 @@ export class MessageProcessor {
                 const task = task_store[message.tool_call_id];
                 if (task) {
                     // 递归处理子消息，但避免重复处理
-                    message.sub_agent_messages = this.processMessages(task.messages);
+                    message.sub_agent_messages = this.processMessages(task.messages, task, messagesMetadata);
                 }
             }
             if (message.id && ignoreIds.has(message.id)) continue;
@@ -269,7 +278,16 @@ export class MessageProcessor {
      * @zh 生成用于 UI 中的流式渲染的消息
      * @en Generate messages used for streaming rendering in the UI
      */
-    renderMessages(graphState: any, getGraphNodeNow: () => { name: string }): RenderMessage[] {
+    renderMessages(
+        graphState: any,
+        getGraphNodeNow: () => { name: string },
+        messagesMetadata: Record<
+            string,
+            {
+                subagent_id?: string;
+            }
+        >
+    ): RenderMessage[] {
         const previousMessage = new Map<string, Message>();
         const closedToolCallIds = new Set<string>();
         const result: Message[] = [];
@@ -328,14 +346,23 @@ export class MessageProcessor {
             }
         }
 
-        return this.processMessages(result as RenderMessage[], graphState);
+        return this.processMessages(result as RenderMessage[], graphState, messagesMetadata);
     }
 
     /**
      * @zh 统一的消息处理入口，按顺序执行所有处理步骤
      * @en Unified message processing entry point, executing all processing steps in order
      */
-    processMessages(messages: RenderMessage[], graphState?: any): RenderMessage[] {
+    processMessages(
+        messages: RenderMessage[],
+        graphState: any,
+        messagesMetadata: Record<
+            string,
+            {
+                subagent_id?: string;
+            }
+        >
+    ): RenderMessage[] {
         // 1. 组合工具消息
         const composedMessages = this.composeToolMessages(messages);
 
@@ -344,7 +371,7 @@ export class MessageProcessor {
 
         // 3. 转换子代理消息（如果提供了 graphState）
         if (graphState) {
-            return this.convertSubAgentMessages(messagesWithInfo, graphState);
+            return this.convertSubAgentMessages(messagesWithInfo, graphState, messagesMetadata);
         }
 
         return messagesWithInfo;

package/src/tool/createTool.ts

@@ -1,5 +1,5 @@
 import { actionParametersToJsonSchema, convertJsonSchemaToZodRawShape } from "./utils.js";
-import { z, ZodRawShape, ZodTypeAny } from "zod";
+import { z, ZodRawShape } from "zod";
 import { Action, Parameter } from "./copilotkit-actions.js";
 import { zodToJsonSchema } from "zod-to-json-schema";
 import { Message } from "@langchain/langgraph-sdk";
@@ -14,8 +14,8 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     execute?: ToolCallback<Args>;
     /** 工具执行成功后触发的附加消息 */
     callbackMessage?: (result: CallToolResult) => Message[];
-    handler?: (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => ResponseType | Promise<ResponseType>;
-    render?: (tool: ToolRenderData<z.objectOutputType<Args, ZodTypeAny>, ResponseType>) => Child;
+    handler?: (args: z.infer<z.ZodObject<Args>>, context?: any) => ResponseType | Promise<ResponseType>;
+    render?: (tool: ToolRenderData<z.infer<z.ZodObject<Args>>, ResponseType>) => Child;
     onlyRender?: boolean;
     /** 只允许指定的 agent 使用该工具，如果未指定，则所有 agent 都可以使用 */
     allowAgent?: string[];
@@ -24,7 +24,7 @@ export interface UnionTool<Args extends ZodRawShape, Child extends Object = Obje
     /** 是否是纯净的 json schema 参数，而不是 zod 参数 */
     isPureParams?: boolean;
 }
-export type ToolCallback<Args extends ZodRawShape> = (args: z.objectOutputType<Args, ZodTypeAny>, context?: any) => CallToolResult | Promise<CallToolResult>;
+export type ToolCallback<Args extends ZodRawShape> = (args: z.infer<z.ZodObject<Args>>, context?: any) => CallToolResult | Promise<CallToolResult>;
 
 export type CallToolResult = string | { type: "text"; text: string }[];
 
@@ -80,7 +80,7 @@ export const createFETool = <const T extends Parameter[], Args extends ZodRawSha
         allowGraph: tool.allowGraph,
         async execute(args, context) {
             try {
-                const result = await tool.handler?.(args, context);
+                const result = await tool.handler?.(args as any, context);
                 if (typeof result === "string") {
                     return [{ type: "text", text: result }];
                 }
@@ -106,7 +106,7 @@ export const createMCPTool = <Args extends ZodRawShape>(tool: UnionTool<Args>) =
         tool.name,
         tool.description,
         tool.parameters,
-        async (args: z.objectOutputType<Args, ZodTypeAny>) => {
+        async (args: z.infer<z.ZodObject<Args>>) => {
             try {
                 const result = await tool.execute?.(args);
                 if (typeof result === "string") {