@dudousxd/nestjs-agent-react 0.12.0 → 0.13.0

package/dist/index.cjs

@@ -542,6 +542,9 @@ var AgentChatTransport = class {
                   props: event.props,
                   ...event.version !== void 0 ? {
                     version: event.version
+                  } : {},
+                  ...event.toolCallId !== void 0 ? {
+                    toolCallId: event.toolCallId
                   } : {}
                 }
               });
@@ -1898,7 +1901,8 @@ function readUiComponent(part) {
     id,
     component: data.component,
     props: isRecord2(data.props) ? data.props : {},
-    version: typeof data.version === "number" ? data.version : null
+    version: typeof data.version === "number" ? data.version : null,
+    toolCallId: typeof data.toolCallId === "string" ? data.toolCallId : null
   };
 }
 __name(readUiComponent, "readUiComponent");
@@ -3336,8 +3340,10 @@ function storedMessageToUiMessage(message) {
       url: attachment.url
     });
   }
+  const toolCallIds = new Set((message.toolCalls ?? []).map((call) => call.id));
+  const pushedByCall = /* @__PURE__ */ new Map();
   for (const component of message.ui ?? []) {
-    parts.push({
+    const part = {
       type: "data-ui",
       id: component.id,
       data: {
@@ -3346,9 +3352,19 @@ function storedMessageToUiMessage(message) {
         props: component.props,
         ...component.version !== void 0 ? {
           version: component.version
+        } : {},
+        ...component.toolCallId !== void 0 ? {
+          toolCallId: component.toolCallId
         } : {}
       }
-    });
+    };
+    if (component.toolCallId !== void 0 && toolCallIds.has(component.toolCallId)) {
+      const list = pushedByCall.get(component.toolCallId) ?? [];
+      list.push(part);
+      pushedByCall.set(component.toolCallId, list);
+    } else {
+      parts.push(part);
+    }
   }
   for (const approval of message.approvals ?? []) {
     parts.push({
@@ -3419,6 +3435,7 @@ function storedMessageToUiMessage(message) {
         input: call.input
       }
     });
+    parts.push(...pushedByCall.get(call.id) ?? []);
   }
   return {
     id: message.id,