@dudousxd/nestjs-agent-react 0.7.5 → 0.8.0

package/dist/index.js

@@ -140,7 +140,8 @@ var AgentChatTransport = class {
   * Parse the backend's SSE byte stream and re-emit it as a valid v7 UI-message chunk stream.
   * Recognized frames:
   *  - `event: meta`  `data: {"runId","threadId"}`   → records identity
-  *  - `data: <AgentStreamEvent JSON>`                → mapped to UI chunks (text / reasoning / tool)
+  *  - `data: <AgentStreamEvent JSON>`                → mapped to UI chunks (text / reasoning / tool /
+  *                                                     `data-*`; see {@link AgentChatTransport} docs)
   *  - `event: done`  `data: {}`                      → terminates
   *  - `event: error` `data: {code,message}`          → error chunk
   *
@@ -157,7 +158,10 @@ var AgentChatTransport = class {
     let stepIndex = 0;
     let textId = null;
     let reasoningId = null;
+    let textSeq = 0;
+    let reasoningSeq = 0;
     const announced = /* @__PURE__ */ new Set();
+    const toolMetadata = /* @__PURE__ */ new Map();
     const record = /* @__PURE__ */ __name((meta) => this.recordMeta(meta), "record");
     const endAttempt = /* @__PURE__ */ __name(() => {
       this.attemptLive = false;
@@ -179,6 +183,8 @@ var AgentChatTransport = class {
           stepIndex += 1;
           textId = null;
           reasoningId = null;
+          textSeq = 0;
+          reasoningSeq = 0;
           controller.enqueue({
             type: "start-step"
           });
@@ -188,8 +194,7 @@ var AgentChatTransport = class {
           if (!stepOpen) openStep();
         }
         __name(ensureStep, "ensureStep");
-        function closeStep() {
-          if (!stepOpen) return;
+        function closeRuns() {
           if (textId !== null) {
             controller.enqueue({
               type: "text-end",
@@ -204,12 +209,43 @@ var AgentChatTransport = class {
             });
             reasoningId = null;
           }
+        }
+        __name(closeRuns, "closeRuns");
+        function closeStep() {
+          if (!stepOpen) return;
+          closeRuns();
           controller.enqueue({
             type: "finish-step"
           });
           stepOpen = false;
         }
         __name(closeStep, "closeStep");
+        function metadataFor(id, toolKind2, parentId) {
+          const merged = {
+            ...toolMetadata.get(id)
+          };
+          if (toolKind2 !== void 0) merged.toolKind = toolKind2;
+          if (parentId !== void 0 && merged.parentId === void 0) merged.parentId = parentId;
+          if (Object.keys(merged).length === 0) return void 0;
+          toolMetadata.set(id, merged);
+          return merged;
+        }
+        __name(metadataFor, "metadataFor");
+        function forwardAsData(event, transient) {
+          ensureStarted();
+          const { kind, ...data } = event;
+          controller.enqueue({
+            type: `data-${kind}`,
+            ...typeof data.id === "string" ? {
+              id: data.id
+            } : {},
+            data,
+            ...transient ? {
+              transient: true
+            } : {}
+          });
+        }
+        __name(forwardAsData, "forwardAsData");
         function emit(event) {
           switch (event.kind) {
             case "step-start":
@@ -229,7 +265,8 @@ var AgentChatTransport = class {
             case "text":
               ensureStep();
               if (textId === null) {
-                textId = `txt-${stepIndex}`;
+                textId = textSeq === 0 ? `txt-${stepIndex}` : `txt-${stepIndex}.${textSeq}`;
+                textSeq += 1;
                 controller.enqueue({
                   type: "text-start",
                   id: textId
@@ -244,7 +281,8 @@ var AgentChatTransport = class {
             case "reasoning":
               ensureStep();
               if (reasoningId === null) {
-                reasoningId = `rsn-${stepIndex}`;
+                reasoningId = reasoningSeq === 0 ? `rsn-${stepIndex}` : `rsn-${stepIndex}.${reasoningSeq}`;
+                reasoningSeq += 1;
                 controller.enqueue({
                   type: "reasoning-start",
                   id: reasoningId
@@ -256,23 +294,20 @@ var AgentChatTransport = class {
                 delta: event.text
               });
               break;
-            case "tool-input-start":
+            case "tool-input-start": {
               ensureStep();
               announced.add(event.id);
+              const metadata = metadataFor(event.id, event.toolKind, event.parentId);
               controller.enqueue({
                 type: "tool-input-start",
                 toolCallId: event.id,
                 toolName: event.name,
-                // `toolKind` is absent on older backends — omitted (not `undefined`-valued) so
-                // `toolMetadata` itself is only present when there's something to say, letting a
-                // UI gate approval affordances on `kind === 'action'` without hardcoding tool names.
-                ...event.toolKind !== void 0 ? {
-                  toolMetadata: {
-                    toolKind: event.toolKind
-                  }
+                ...metadata !== void 0 ? {
+                  toolMetadata: metadata
                 } : {}
               });
               break;
+            }
             case "tool-input-delta":
               ensureStep();
               controller.enqueue({
@@ -281,21 +316,21 @@ var AgentChatTransport = class {
                 inputTextDelta: event.delta
               });
               break;
-            case "tool-input-available":
+            case "tool-input-available": {
               ensureStep();
               announced.add(event.id);
+              const metadata = metadataFor(event.id, event.toolKind, event.parentId);
               controller.enqueue({
                 type: "tool-input-available",
                 toolCallId: event.id,
                 toolName: event.name,
                 input: event.input,
-                ...event.toolKind !== void 0 ? {
-                  toolMetadata: {
-                    toolKind: event.toolKind
-                  }
+                ...metadata !== void 0 ? {
+                  toolMetadata: metadata
                 } : {}
               });
               break;
+            }
             case "elicitation":
               ensureStep();
               if (!announced.has(event.id)) {
@@ -336,7 +371,39 @@ var AgentChatTransport = class {
                 toolCallId: event.id
               });
               break;
+            case "approval-requested":
+              ensureStep();
+              forwardAsData(event, false);
+              if (announced.has(event.id)) {
+                controller.enqueue({
+                  type: "tool-approval-request",
+                  approvalId: event.id,
+                  toolCallId: event.id
+                });
+              }
+              break;
+            case "ui":
+              ensureStep();
+              closeRuns();
+              controller.enqueue({
+                type: "data-ui",
+                id: event.id,
+                data: {
+                  id: event.id,
+                  component: event.component,
+                  props: event.props,
+                  ...event.version !== void 0 ? {
+                    version: event.version
+                  } : {}
+                }
+              });
+              break;
+            case "title":
+            case "cancelled":
+              forwardAsData(event, true);
+              break;
             default:
+              forwardAsData(event, false);
               break;
           }
         }
@@ -832,7 +899,7 @@ import React3 from "react";
 import { useCallback as useCallback3, useEffect as useEffect4, useRef as useRef3, useState as useState4 } from "react";
 
 // src/transcript/model.ts
-import { getToolName, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolUIPart } from "ai";
+import { getToolName, isDataUIPart, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolUIPart } from "ai";
 function buildTranscriptBlocks(message, options) {
   const blocks = [];
   let toolBuffer = [];
@@ -844,15 +911,18 @@ function buildTranscriptBlocks(message, options) {
   let reasoningCounter = 0;
   let sourcesCounter = 0;
   let elicitationCounter = 0;
+  const approvals = readApprovals(message.parts ?? []);
   function flushTools() {
     if (toolBuffer.length === 0) {
       return;
     }
+    const calls = toolBuffer.map((part) => buildToolCall(part, options.approval, approvals.get(part.toolCallId) ?? null));
     blocks.push({
       kind: "tools",
       key: `${message.id}-tools-${toolCounter++}`,
       parts: toolBuffer,
-      calls: toolBuffer.map((part) => buildToolCall(part, options.approval))
+      calls,
+      roots: nestToolCalls(calls)
     });
     toolBuffer = [];
   }
@@ -915,7 +985,21 @@ function buildTranscriptBlocks(message, options) {
       fileBuffer.push(part);
       continue;
     }
+    if (part.type === APPROVAL_PART) {
+      continue;
+    }
     flushAll();
+    if (isDataUIPart(part)) {
+      const ui = part.type === UI_PART ? readUiComponent(part) : null;
+      if (ui !== null) {
+        blocks.push({
+          kind: "ui",
+          key: `${message.id}-ui-${ui.id}`,
+          ...ui
+        });
+      }
+      continue;
+    }
     if (isTextUIPart(part)) {
       blocks.push({
         kind: "text",
@@ -999,7 +1083,42 @@ function isAwaitingApproval(part) {
   return toolKind(part) === "action" && readElicitationRequest(part) === null;
 }
 __name(isAwaitingApproval, "isAwaitingApproval");
-function buildToolCall(part, options) {
+function toolParentId(part) {
+  const metadata = part.toolMetadata;
+  return isRecord(metadata) && typeof metadata.parentId === "string" ? metadata.parentId : null;
+}
+__name(toolParentId, "toolParentId");
+function nestToolCalls(calls) {
+  const byId = new Map(calls.map((call) => [
+    call.toolCallId,
+    call
+  ]));
+  const roots = [];
+  for (const call of calls) {
+    const parent = call.parentId !== null ? byId.get(call.parentId) : void 0;
+    if (parent === void 0 || isAncestor(call, parent, byId)) {
+      roots.push(call);
+      continue;
+    }
+    parent.children.push(call);
+  }
+  return roots;
+}
+__name(nestToolCalls, "nestToolCalls");
+function isAncestor(call, candidate, byId) {
+  const seen = /* @__PURE__ */ new Set();
+  let current = candidate;
+  while (current !== void 0 && !seen.has(current.toolCallId)) {
+    if (current.toolCallId === call.toolCallId) {
+      return true;
+    }
+    seen.add(current.toolCallId);
+    current = current.parentId !== null ? byId.get(current.parentId) : void 0;
+  }
+  return false;
+}
+__name(isAncestor, "isAncestor");
+function buildToolCall(part, options, approval) {
   const toolCallId = part.toolCallId;
   const awaiting = isAwaitingApproval(part);
   const sending = awaiting ? options?.submitting(toolCallId) ?? null : null;
@@ -1007,6 +1126,10 @@ function buildToolCall(part, options) {
     part,
     toolCallId,
     name: getToolName(part),
+    toolKind: toolKind(part) ?? null,
+    parentId: toolParentId(part),
+    children: [],
+    approval,
     isAwaitingApproval: awaiting,
     approve: {
       available: awaiting && options?.canApprove === true,
@@ -1137,6 +1260,44 @@ function buildElicitationBlock(key, part, request, options) {
   };
 }
 __name(buildElicitationBlock, "buildElicitationBlock");
+var UI_PART = "data-ui";
+var APPROVAL_PART = "data-approval-requested";
+function readApprovals(parts) {
+  const out = /* @__PURE__ */ new Map();
+  for (const part of parts) {
+    if (part.type !== APPROVAL_PART) {
+      continue;
+    }
+    const data = part.data;
+    if (!isRecord(data) || typeof data.id !== "string" || typeof data.approver !== "string") {
+      continue;
+    }
+    out.set(data.id, {
+      approver: data.approver,
+      expiresAt: typeof data.expiresAt === "string" ? data.expiresAt : null,
+      reason: typeof data.reason === "string" ? data.reason : null
+    });
+  }
+  return out;
+}
+__name(readApprovals, "readApprovals");
+function readUiComponent(part) {
+  const data = part.data;
+  if (!isRecord(data) || typeof data.component !== "string") {
+    return null;
+  }
+  const id = typeof data.id === "string" ? data.id : part.id;
+  if (id === void 0) {
+    return null;
+  }
+  return {
+    id,
+    component: data.component,
+    props: isRecord(data.props) ? data.props : {},
+    version: typeof data.version === "number" ? data.version : null
+  };
+}
+__name(readUiComponent, "readUiComponent");
 function isRecord(value) {
   return typeof value === "object" && value !== null;
 }
@@ -1830,6 +1991,14 @@ function renderBlocks(item, slots) {
     }
     if (block.kind === "tools") {
       nodes.push(...renderToolBlock(item, block, slots));
+      continue;
+    }
+    if (block.kind === "ui" && slots.renderUi) {
+      nodes.push(/* @__PURE__ */ React3.createElement("div", {
+        key: block.key,
+        "data-slot": "ui",
+        "data-component": block.component
+      }, slots.renderUi(block)));
     }
   }
   return nodes;
@@ -1937,7 +2106,7 @@ function joinClasses(...entries) {
 __name(joinClasses, "joinClasses");
 
 // src/components/message-list.tsx
-function MessageList({ messages, status, renderToolPart, renderToolGroup, renderText, renderReasoning, renderFiles, reasoningLabel, getMeta, onFork, editable, onEditSubmit, regeneratable, onRegenerate, getUsage, getCreatedAt, emptyState, typingIndicator, followUps, onFollowUpClick, error, onRetry, classNames }) {
+function MessageList({ messages, status, renderToolPart, renderToolGroup, renderText, renderReasoning, renderFiles, renderUi, reasoningLabel, getMeta, onFork, editable, onEditSubmit, regeneratable, onRegenerate, getUsage, getCreatedAt, emptyState, typingIndicator, followUps, onFollowUpClick, error, onRetry, classNames }) {
   const transcript = useChatTranscript({
     messages,
     status,
@@ -1987,6 +2156,9 @@ function MessageList({ messages, status, renderToolPart, renderToolGroup, render
     ...renderFiles ? {
       renderFiles
     } : {},
+    ...renderUi ? {
+      renderUi
+    } : {},
     ...reasoningLabel !== void 0 ? {
       reasoningLabel
     } : {},
@@ -2641,6 +2813,13 @@ function useAgentChat(options) {
     ...options.initialMessages !== void 0 ? {
       messages: options.initialMessages
     } : {},
+    onData: /* @__PURE__ */ __name((part) => {
+      latest.current.onData?.(part);
+      if (part.type === "data-title") {
+        const title = part.data?.title;
+        if (typeof title === "string") latest.current.onTitle?.(title);
+      }
+    }, "onData"),
     onFinish: /* @__PURE__ */ __name(({ isError }) => {
       latest.current.onFinish?.();
       if (latest.current.background === true) {