@dudousxd/nestjs-agent-react 0.12.0 → 0.13.0

package/dist/index.d.cts

@@ -673,6 +673,8 @@ interface TranscriptUiBlock {
     props: Record<string, unknown>;
     /** Schema version of `props`; `null` when the server did not stamp one. */
     version: number | null;
+    /** The tool call that pushed it (`ctx.emitUi`); `null` for a component pushed outside a tool. */
+    toolCallId: string | null;
 }
 /** One choice a question offers, with its live selection state. */
 interface TranscriptQuestionOption {

package/dist/index.d.ts

@@ -673,6 +673,8 @@ interface TranscriptUiBlock {
     props: Record<string, unknown>;
     /** Schema version of `props`; `null` when the server did not stamp one. */
     version: number | null;
+    /** The tool call that pushed it (`ctx.emitUi`); `null` for a component pushed outside a tool. */
+    toolCallId: string | null;
 }
 /** One choice a question offers, with its live selection state. */
 interface TranscriptQuestionOption {

package/dist/index.js

@@ -460,6 +460,9 @@ var AgentChatTransport = class {
                   props: event.props,
                   ...event.version !== void 0 ? {
                     version: event.version
+                  } : {},
+                  ...event.toolCallId !== void 0 ? {
+                    toolCallId: event.toolCallId
                   } : {}
                 }
               });
@@ -1816,7 +1819,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");
@@ -3254,8 +3258,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: {
@@ -3264,9 +3270,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({
@@ -3337,6 +3353,7 @@ function storedMessageToUiMessage(message) {
         input: call.input
       }
     });
+    parts.push(...pushedByCall.get(call.id) ?? []);
   }
   return {
     id: message.id,