@langgraph-js/sdk 4.6.1 → 4.6.2

package/dist/LangGraphClient.d.ts

@@ -35,6 +35,8 @@ export type RenderMessage = Message & {
     unique_id?: string;
     /** 工具调用是否完成 */
     done?: boolean;
+    /** 标记工具消息的原始 ai 消息 */
+    source_ai_message_id?: string;
 };
 export type SendMessageOptions = {
     extraParams?: Record<string, any>;
@@ -157,7 +159,7 @@ export declare class LangGraphClient<TStateType = unknown> extends EventEmitter<
         }): Promise<Thread<TStateType, unknown>>;
         search(query?: {
             metadata?: import("@langchain/langgraph-sdk").Metadata;
-            limit? /** 自定义客户端实现，如果不提供则使用官方 Client */: number;
+            limit?: number;
             offset?: number;
             status?: import("@langchain/langgraph-sdk").ThreadStatus;
             sortBy?: import("@langgraph-js/pure-graph/dist/types.js").ThreadSortBy;

package/dist/MessageProcessor.js

@@ -162,6 +162,8 @@ export class MessageProcessor {
                 if (parentMessage) {
                     message.usage_metadata = parentMessage.usage_metadata;
                     message.node_name = parentMessage.name;
+                    /** @ts-ignore 用于标记原始的 AI message 的id */
+                    message.source_ai_message_id = parentMessage.id;
                     // 修补特殊情况下，tool name 丢失的问题
                     if (!message.name) {
                         message.name = parentMessage.tool_calls.find((i) => i.id === message.tool_call_id)?.name;
@@ -251,10 +253,11 @@ export class MessageProcessor {
         const childrenMap = state_sub_messages_map;
         const rootMessages = [];
         for (const message of messages) {
-            const isRoot = !nonRootMessageId.has(message.id);
+            const sourceId = message.source_ai_message_id || message.id;
+            const isRoot = !nonRootMessageId.has(sourceId);
             if (!isRoot) {
                 // 处理子消息
-                const parentId = parentPointer.get(message.id);
+                const parentId = parentPointer.get(sourceId);
                 const children = childrenMap.get(parentId);
                 if (children) {
                     children.push(message);

package/package.json

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

package/src/LangGraphClient.ts

@@ -40,6 +40,8 @@ export type RenderMessage = Message & {
     unique_id?: string;
     /** 工具调用是否完成 */
     done?: boolean;
+    /** 标记工具消息的原始 ai 消息 */
+    source_ai_message_id?: string;
 };
 export type SendMessageOptions = {
     extraParams?: Record<string, any>;
@@ -370,7 +372,7 @@ export class LangGraphClient<TStateType = unknown> extends EventEmitter<LangGrap
             throw new Error("Thread or Assistant not initialized");
         }
         if (!this.currentThread) {
-            await this.createThread({ 
+            await this.createThread({
                 graphId: this.currentAssistant!.graph_id!,
                 metadata,
             });

package/src/MessageProcessor.ts

@@ -184,6 +184,8 @@ export class MessageProcessor {
                 if (parentMessage) {
                     message.usage_metadata = parentMessage.usage_metadata;
                     message.node_name = parentMessage.name;
+                    /** @ts-ignore 用于标记原始的 AI message 的id */
+                    message.source_ai_message_id = parentMessage.id;
                     // 修补特殊情况下，tool name 丢失的问题
                     if (!message.name) {
                         message.name = (parentMessage as AIMessage).tool_calls!.find((i) => i.id === message.tool_call_id)?.name;
@@ -294,10 +296,11 @@ export class MessageProcessor {
         const rootMessages: RenderMessage[] = [];
 
         for (const message of messages) {
-            const isRoot = !nonRootMessageId.has(message.id!);
+            const sourceId = message.source_ai_message_id || message.id!;
+            const isRoot = !nonRootMessageId.has(sourceId!);
             if (!isRoot) {
                 // 处理子消息
-                const parentId = parentPointer.get(message.id!)!;
+                const parentId = parentPointer.get(sourceId)!;
                 const children = childrenMap.get(parentId);
                 if (children) {
                     children.push(message);