langchain_agentx_stream_ui 0.1.3 → 0.1.9

package/dist/{chunk-3B4WC67E.js → chunk-ZNXAQ4ZZ.js}

@@ -1,5 +1,6 @@
 import {
   Agent,
+  AgentProgressLines,
   Bash,
   Edit,
   Glob,
@@ -7,9 +8,12 @@ import {
   INTERNAL_TOOL_NAMES,
   Read,
   SILENT_TASK_TOOL_NAMES,
+  SubagentTranscriptBlock,
   Write,
-  createDefaultToolRegistry
-} from "./chunk-MHK53ZHC.js";
+  coerceSubagentProgressEntries,
+  createDefaultToolRegistry,
+  extractLastToolInfoFromProgress
+} from "./chunk-7CLAU74Y.js";
 
 // src/types/tree.ts
 function createEmptyTree() {
@@ -37,7 +41,9 @@ function createEmptyTree() {
       completed: 0,
       customMessage: null
     },
-    modelRetry: null
+    modelRetry: null,
+    parallelAgentTextGateCallIds: null,
+    deferredMainText: ""
   };
 }
 
@@ -65,6 +71,7 @@ var DEFAULT_EVENT_TIERS = {
   "subagent-end": "collapsed",
   "subagent-tool-call": "collapsed",
   "subagent-tool-result": "collapsed",
+  "subagent-transcript-message": "collapsed",
   "subagent-model-step": "collapsed",
   "hook-progress": "debug",
   "hook-finished": "debug",
@@ -102,7 +109,7 @@ function resolveToolParentId(event, ctx, eventIndex) {
   return toolNodeIdFromEvent(event, eventIndex);
 }
 function isSubagentFoldEvent(event) {
-  if (event.event_type === "subagent-start" || event.event_type === "subagent-end" || event.event_type === "subagent-tool-call" || event.event_type === "subagent-tool-result" || event.event_type === "subagent-model-step") {
+  if (event.event_type === "subagent-start" || event.event_type === "subagent-end" || event.event_type === "subagent-tool-call" || event.event_type === "subagent-tool-result" || event.event_type === "subagent-transcript-message" || event.event_type === "subagent-model-step") {
     return true;
   }
   if (event.event_type === "tool-progress") {
@@ -132,11 +139,30 @@ function classifyEvent(event, ctx, overrides, eventIndex = 0) {
   return { event, tier, parentNodeId };
 }
 
+// src/core/mainTextDefer.ts
+function deferMainText(buffer, text) {
+  if (!text) return buffer ?? "";
+  return (buffer ?? "") + text;
+}
+function flushDeferredMainText(buffer) {
+  return buffer ?? "";
+}
+
 // src/core/eventVisibility.ts
 var SUBAGENT_LEAK_FOLD_TOOLS = /* @__PURE__ */ new Set(["Bash", "Read", "Glob", "Grep"]);
 function isSubagentLeakFoldToolName(toolName) {
   return SUBAGENT_LEAK_FOLD_TOOLS.has(toolName);
 }
+function findAgentToolCallNodeByCallId(tree, callId) {
+  const trimmed = callId.trim();
+  if (!trimmed) return null;
+  for (const node of Object.values(tree.byId)) {
+    if (node.kind !== "tool_call" || node.toolName !== "Agent") continue;
+    const id = (node.toolCallId ?? node.id.replace(/^tool:/, "")).trim();
+    if (id === trimmed) return node;
+  }
+  return null;
+}
 function getRunningAgentParentCallIds(tree) {
   const ids = [];
   const seen = /* @__PURE__ */ new Set();
@@ -217,6 +243,65 @@ function isSubagentScopedTextEvent(data) {
 function isSubagentScopedReasoningEvent(data) {
   return isSubagentScopedTextEvent(data);
 }
+var EXPLICIT_MAIN_QUERY_SOURCES = /* @__PURE__ */ new Set([
+  "repl_main",
+  "repl_main_thread",
+  "main_thread"
+]);
+function isExplicitMainOrchestratorText(data) {
+  if (!data) return false;
+  const meta = bridgeEventDisplayMetadata(data);
+  const qs = meta?.query_source ?? data.query_source;
+  if (typeof qs === "string") {
+    const trimmed = qs.trim();
+    if (trimmed && !isAgentSubagentQuerySource(trimmed)) {
+      if (EXPLICIT_MAIN_QUERY_SOURCES.has(trimmed)) return true;
+      if (trimmed.startsWith("repl:")) return true;
+    }
+  }
+  if (data.emitter_route === "main") return true;
+  if (meta?.emitter_route === "main") return true;
+  return false;
+}
+function isParallelAgentTextGateActive(tree) {
+  const latched = tree.parallelAgentTextGateCallIds;
+  if (latched && latched.length >= 2) {
+    return latched.some((id) => {
+      const node = findAgentToolCallNodeByCallId(tree, id);
+      return !node || node.status === "running";
+    });
+  }
+  return getRunningAgentParentCallIds(tree).length >= 2;
+}
+function updateParallelAgentTextGate(tree) {
+  const running = getRunningAgentParentCallIds(tree);
+  let ids = tree.parallelAgentTextGateCallIds ?? [];
+  if (running.length >= 2) {
+    ids = [.../* @__PURE__ */ new Set([...ids, ...running])];
+  }
+  if (ids.length < 2) {
+    return tree.parallelAgentTextGateCallIds === null ? tree : { ...tree, parallelAgentTextGateCallIds: null };
+  }
+  const allTerminal = ids.every((id) => {
+    const node = findAgentToolCallNodeByCallId(tree, id);
+    return node != null && (node.status === "done" || node.status === "failed");
+  });
+  if (allTerminal) {
+    return { ...tree, parallelAgentTextGateCallIds: null };
+  }
+  const nextIds = [...ids];
+  if (tree.parallelAgentTextGateCallIds?.length === nextIds.length && tree.parallelAgentTextGateCallIds.every((id, i) => id === nextIds[i])) {
+    return tree;
+  }
+  return { ...tree, parallelAgentTextGateCallIds: nextIds };
+}
+function shouldDropParallelSubagentText(tree, data) {
+  if (!isParallelAgentTextGateActive(tree)) return false;
+  if (!data) return false;
+  if (isSubagentScopedTextEvent(data)) return true;
+  if (isExplicitMainOrchestratorText(data)) return false;
+  return true;
+}
 function isDroppedByEmitterRoute(data) {
   if (!data) return false;
   if (isDropRouteValue(data.emitter_route)) return true;
@@ -296,6 +381,56 @@ function upsertSubagentSession(sessions, marker) {
   return { ...sessions, [marker.agentId]: marker };
 }
 
+// src/core/subagentTranscriptLane.ts
+function readField(row, camel, snake) {
+  const camelValue = row[camel];
+  if (camelValue !== void 0 && camelValue !== null) return camelValue;
+  const snakeValue = row[snake];
+  if (snakeValue !== void 0 && snakeValue !== null) return snakeValue;
+  return void 0;
+}
+function appendSubagentTranscriptMessage(entries, payload) {
+  const agentId = payload.agentId.trim();
+  if (!agentId) return entries;
+  const messageKind = payload.messageKind;
+  if (messageKind !== "assistant" && messageKind !== "user") return entries;
+  if (!payload.message || typeof payload.message !== "object") return entries;
+  return [
+    ...entries,
+    {
+      agentId,
+      messageKind,
+      message: payload.message,
+      step: payload.step,
+      toolName: payload.toolName,
+      subagentType: payload.subagentType
+    }
+  ];
+}
+function coerceSubagentTranscriptEntries(raw) {
+  if (!Array.isArray(raw)) return [];
+  const entries = [];
+  for (const item of raw) {
+    if (item == null || typeof item !== "object" || Array.isArray(item)) continue;
+    const row = item;
+    const agentId = readField(row, "agentId", "agent_id");
+    const kindRaw = readField(row, "messageKind", "message_kind");
+    const message = readField(row, "message", "message");
+    if (!agentId || !message) continue;
+    const messageKind = kindRaw === "user" ? "user" : kindRaw === "assistant" ? "assistant" : null;
+    if (!messageKind) continue;
+    entries.push({
+      agentId,
+      messageKind,
+      message,
+      step: readField(row, "step", "step"),
+      toolName: readField(row, "toolName", "tool_name"),
+      subagentType: readField(row, "subagentType", "subagent_type")
+    });
+  }
+  return entries;
+}
+
 // src/core/taskSnapshot.ts
 var OPEN_TASK_STATUSES = /* @__PURE__ */ new Set(["pending", "in_progress"]);
 function normalizeTaskStatus(raw) {
@@ -514,6 +649,9 @@ function appendStreamChunk(current, data) {
   return (current ?? "") + streamChunk(data);
 }
 function handleStart(tree, event) {
+  if (tree.status === "done" || tree.status === "error") {
+    return;
+  }
   tree.status = "running";
   tree.meta.startedAt = eventTimeToMs(event.timestamp);
   if (event.session_id) tree.meta.sessionId = event.session_id;
@@ -578,10 +716,12 @@ function handleFinish(tree, event, seq) {
     appendFinishErrorNode(tree, event, data, seq);
     finalizeRunningToolCalls(tree, endedAt, "failed");
     finalizeStreamingNodes(tree);
+    tree.activeRoundId = null;
     return;
   }
   finalizeRunningToolCalls(tree, endedAt, "done");
   finalizeStreamingNodes(tree);
+  tree.activeRoundId = null;
   tree.status = "done";
 }
 var TASK_DEBUG_EVENT_TYPES = /* @__PURE__ */ new Set([
@@ -685,6 +825,26 @@ function handleHookFinished(tree, event) {
     };
   }
 }
+function extractDeferText(data) {
+  if (typeof data.text === "string") return data.text;
+  if (typeof data.delta === "string") return data.delta;
+  return "";
+}
+function flushDeferredMainTextToTree(tree, event, seq) {
+  const text = flushDeferredMainText(tree.deferredMainText);
+  tree.deferredMainText = "";
+  if (!text) return;
+  const id = `text:defer-${seq}`;
+  const node = {
+    kind: "text",
+    id,
+    accumulated: text,
+    status: "done",
+    startedAt: eventTimeToMs(event.timestamp)
+  };
+  tree.byId[id] = node;
+  appendMainStageChild(tree, id, event, seq);
+}
 function handleTextStart(tree, event, seq) {
   clearModelRetry(tree);
   const id = `text:${seq}`;
@@ -768,6 +928,7 @@ function handleReasoningEnd(tree, event) {
     elapsedSeconds,
     isCollapsible: true
   };
+  tree.activeRoundId = null;
 }
 function upsertToolNode(tree, id, event, init, seq, mountMain) {
   const data = event.data;
@@ -1039,6 +1200,24 @@ function handleSubagentToolResult(tree, _event, parentNodeId, data) {
     })
   });
 }
+function handleSubagentTranscriptMessage(tree, parentNodeId, data) {
+  const tool = agentToolNode(tree, parentNodeId);
+  if (!tool || !parentNodeId) return;
+  const message = data.message;
+  if (message == null || typeof message !== "object" || Array.isArray(message)) return;
+  const messageKind = data.message_kind;
+  if (messageKind !== "assistant" && messageKind !== "user") return;
+  patchToolNode(tree, parentNodeId, {
+    subagentTranscript: appendSubagentTranscriptMessage(tool.subagentTranscript ?? [], {
+      agentId: data.agent_id,
+      messageKind,
+      message,
+      step: data.step,
+      toolName: data.tool_name,
+      subagentType: data.subagent_type
+    })
+  });
+}
 function handleStepStart(tree, event, parentNodeId, seq) {
   const toolId = parentNodeId;
   if (toolId && tree.byId[toolId]?.kind === "tool_call") {
@@ -1268,6 +1447,13 @@ function applyClassified(tree, classified, eventIndex, options) {
       case "subagent-tool-result":
         handleSubagentToolResult(tree, event, parentNodeId, event.data);
         break;
+      case "subagent-transcript-message":
+        handleSubagentTranscriptMessage(
+          tree,
+          parentNodeId,
+          event.data
+        );
+        break;
       case "subagent-model-step":
         break;
       case "start-step":
@@ -1283,21 +1469,36 @@ function applyClassified(tree, classified, eventIndex, options) {
     return;
   }
   switch (event.event_type) {
-    case "text-start":
-      if (isDroppedByEmitterRoute(event.data)) break;
-      if (isSubagentScopedTextEvent(event.data)) break;
+    case "text-start": {
+      const textData = event.data;
+      if (isDroppedByEmitterRoute(textData)) break;
+      if (isSubagentScopedTextEvent(textData)) break;
+      if (shouldDropParallelSubagentText(tree, textData)) break;
       handleTextStart(tree, event, seq);
       break;
-    case "text-delta":
-      if (isDroppedByEmitterRoute(event.data)) break;
-      if (isSubagentScopedTextEvent(event.data)) break;
+    }
+    case "text-delta": {
+      const textData = event.data;
+      if (isDroppedByEmitterRoute(textData)) break;
+      if (isSubagentScopedTextEvent(textData)) break;
+      if (shouldDropParallelSubagentText(tree, textData)) {
+        tree.deferredMainText = deferMainText(
+          tree.deferredMainText,
+          extractDeferText(textData)
+        );
+        break;
+      }
       handleTextDelta(tree, event, seq);
       break;
-    case "text-end":
-      if (isDroppedByEmitterRoute(event.data)) break;
-      if (isSubagentScopedTextEvent(event.data)) break;
+    }
+    case "text-end": {
+      const textData = event.data;
+      if (isDroppedByEmitterRoute(textData)) break;
+      if (isSubagentScopedTextEvent(textData)) break;
+      if (shouldDropParallelSubagentText(tree, textData)) break;
       handleTextEnd(tree, event);
       break;
+    }
     case "reasoning-start":
       if (isDroppedByEmitterRoute(event.data)) break;
       if (isSubagentScopedReasoningEvent(event.data)) break;
@@ -1369,7 +1570,12 @@ function reduceTree(tree, event, eventIndex, options) {
   if (event.session_id && !next.meta.sessionId) {
     next.meta.sessionId = event.session_id;
   }
-  return next;
+  const latchedBeforeGateUpdate = next.parallelAgentTextGateCallIds;
+  const updated = updateParallelAgentTextGate(next);
+  if (latchedBeforeGateUpdate !== null && updated.parallelAgentTextGateCallIds === null) {
+    flushDeferredMainTextToTree(updated, event, eventIndex);
+  }
+  return updated;
 }
 var reduceAgentLoopEvent = reduceTree;
 function reduceEvents(events, options) {
@@ -2793,7 +2999,7 @@ var GroupedToolUseAccumulator = class {
     for (const item of bucket.toolUses) {
       const payload = item.entry.payload;
       nodeIds.push(item.nodeId);
-      toolUses.push({
+      const toolUse = {
         tool_call_id: item.toolCallId,
         tool_name: item.toolName,
         tool_input: payload.tool_input ?? {},
@@ -2802,7 +3008,14 @@ var GroupedToolUseAccumulator = class {
         is_error: payload.is_error,
         display: payload.display,
         meta: payload.meta
-      });
+      };
+      if (Array.isArray(payload.subagent_progress) && payload.subagent_progress.length > 0) {
+        toolUse.subagent_progress = payload.subagent_progress;
+      }
+      if (Array.isArray(payload.subagent_transcript) && payload.subagent_transcript.length > 0) {
+        toolUse.subagent_transcript = payload.subagent_transcript;
+      }
+      toolUses.push(toolUse);
       if (isResolved(payload)) {
         resolvedCallIds.push(item.toolCallId);
         toolResults.push({
@@ -3315,6 +3528,9 @@ function mapToolCall(node, pendingPermissionTools) {
   if (node.subagentProgress?.length) {
     payload.subagent_progress = node.subagentProgress;
   }
+  if (node.subagentTranscript?.length) {
+    payload.subagent_transcript = node.subagentTranscript;
+  }
   if (responseId) {
     payload.message_id = responseId;
   }
@@ -4298,8 +4514,7 @@ function SpinnerGlimmerVerb({
   mode,
   reducedMotion,
   stalledIntensity,
-  flashOpacity,
-  shimmerActive
+  flashOpacity
 }) {
   const classes = ["lax-spinner-verb"];
   if (reducedMotion) {
@@ -4308,11 +4523,8 @@ function SpinnerGlimmerVerb({
     classes.push("lax-spinner-verb--stalled");
   } else if (mode === "tool-use") {
     classes.push("lax-spinner-verb--tool-use");
-  } else if (shimmerActive) {
-    classes.push("lax-spinner-verb--shimmer");
-    if (mode === "requesting") {
-      classes.push("lax-spinner-verb--shimmer-fast");
-    }
+  } else {
+    classes.push("lax-spinner-verb--static");
   }
   const style = {
     ...stalledIntensity > 0 && {
@@ -4496,7 +4708,6 @@ function SpinnerAnimationRow({
   );
   const frameIndex = getSpinnerFrameIndex(time, reducedMotion);
   const frame = SPINNER_FRAMES[frameIndex];
-  const shimmerActive = !reducedMotion && stalledIntensity === 0 && (mode === "requesting" || mode === "thinking");
   const flashOpacity = reducedMotion || mode !== "tool-use" ? 0 : (Math.sin(time / 1e3 * Math.PI) + 1) / 2;
   if (reducedMotion) {
     tokenCounterRef.current = targetTokenCount;
@@ -4539,8 +4750,7 @@ function SpinnerAnimationRow({
         mode,
         reducedMotion,
         stalledIntensity,
-        flashOpacity,
-        shimmerActive
+        flashOpacity
       }
     ),
     statusParts.length > 0 && /* @__PURE__ */ jsxs5("span", { className: "lax-spinner-status", children: [
@@ -4817,6 +5027,7 @@ function TaskListFooter() {
 }
 
 // src/view/GroupedToolCallRow.tsx
+import { useCallback, useState as useState9 } from "react";
 import { jsx as jsx9, jsxs as jsxs8 } from "react/jsx-runtime";
 var INITIALIZING = "Initializing\u2026";
 function readGroupMeta(payload) {
@@ -4831,7 +5042,7 @@ function readToolUses(payload) {
     (item) => item != null && typeof item === "object" && !Array.isArray(item)
   );
 }
-function sublineStatus(callId, payload) {
+function sublineStatus(callId, entry, payload) {
   const pending = new Set(
     Array.isArray(payload.pendingCallIds) ? payload.pendingCallIds : []
   );
@@ -4841,7 +5052,9 @@ function sublineStatus(callId, payload) {
   const resolved = new Set(
     Array.isArray(payload.resolvedCallIds) ? payload.resolvedCallIds : []
   );
-  if (pending.has(callId)) return INITIALIZING;
+  if (pending.has(callId)) {
+    return extractLastToolInfoFromProgress(entry.subagent_progress) ?? INITIALIZING;
+  }
   if (errored.has(callId)) return "Error";
   if (resolved.has(callId)) {
     const results = Array.isArray(payload.toolResults) ? payload.toolResults : [];
@@ -4863,6 +5076,67 @@ function sublineStatus(callId, payload) {
   }
   return null;
 }
+function GroupedSubline({
+  callId,
+  entry,
+  payload,
+  hideType,
+  treeChar,
+  index,
+  enableTranscript
+}) {
+  const [expanded, setExpanded] = useState9(false);
+  const subTitle = formatAgentSublineTitle(entry, hideType);
+  const status = sublineStatus(callId, entry, payload);
+  const transcript = coerceSubagentTranscriptEntries(entry.subagent_transcript);
+  const progress = coerceSubagentProgressEntries(entry.subagent_progress);
+  const hasTranscriptLane = transcript.length > 0;
+  const canExpand = enableTranscript && (hasTranscriptLane || progress.length > 0);
+  const toggleExpanded = useCallback(() => {
+    setExpanded((prev) => !prev);
+  }, []);
+  return /* @__PURE__ */ jsxs8(
+    "div",
+    {
+      className: "lax-tool-group__item lax-tool-group__subline-block",
+      "data-group-index": index + 1,
+      "data-tool-call-id": callId,
+      children: [
+        /* @__PURE__ */ jsxs8("div", { className: "lax-tool-group__subline", children: [
+          /* @__PURE__ */ jsx9("span", { className: "lax-tool-group__tree", "aria-hidden": true, children: treeChar }),
+          /* @__PURE__ */ jsx9("span", { className: "lax-tool-group__subline-title", children: subTitle }),
+          status && /* @__PURE__ */ jsxs8("span", { className: "lax-tool-group__subline-status", children: [
+            " \xB7 ",
+            status
+          ] })
+        ] }),
+        canExpand && !expanded ? /* @__PURE__ */ jsx9(
+          "button",
+          {
+            type: "button",
+            className: "lax-tool-group__transcript-link",
+            onClick: toggleExpanded,
+            "aria-expanded": false,
+            children: "Show transcript"
+          }
+        ) : null,
+        canExpand && expanded ? /* @__PURE__ */ jsxs8("div", { className: "lax-tool-group__transcript-panel", "data-testid": "lax-grouped-transcript", children: [
+          hasTranscriptLane ? /* @__PURE__ */ jsx9(SubagentTranscriptBlock, { entries: transcript, dataSource: "transcript-lane" }) : /* @__PURE__ */ jsx9(AgentProgressLines, { entries: progress, mode: "full" }),
+          /* @__PURE__ */ jsx9(
+            "button",
+            {
+              type: "button",
+              className: "lax-tool-group__transcript-link",
+              onClick: toggleExpanded,
+              "aria-expanded": true,
+              children: "Hide transcript"
+            }
+          )
+        ] }) : null
+      ]
+    }
+  );
+}
 function GroupedToolCallRow({ payload }) {
   const toolName = String(payload.toolName ?? payload.tool_name ?? "Agent");
   const groupMeta = readGroupMeta(payload);
@@ -4871,6 +5145,7 @@ function GroupedToolCallRow({ payload }) {
   const title = groupMeta?.title ?? `${toolName} \xD7 ${toolUses.length}`;
   const anyPending = Array.isArray(payload.pendingCallIds) && payload.pendingCallIds.length > 0;
   const anyError = Array.isArray(payload.erroredCallIds) && payload.erroredCallIds.length > 0;
+  const enableTranscript = toolUses.length >= 2;
   return /* @__PURE__ */ jsxs8("div", { className: "lax-tool-group", "data-tool-name": toolName, "data-kind": "grouped_tool_use", children: [
     /* @__PURE__ */ jsxs8(
       "div",
@@ -4888,24 +5163,18 @@ function GroupedToolCallRow({ payload }) {
     ),
     /* @__PURE__ */ jsx9("div", { className: "lax-tool-group__items", children: toolUses.map((entry, index) => {
       const callId = String(entry.tool_call_id ?? "");
-      const subTitle = formatAgentSublineTitle(entry, hideType);
-      const status = sublineStatus(callId, payload);
       const isLast = index === toolUses.length - 1;
       const treeChar = isLast ? "\u2514\u2500" : "\u251C\u2500";
-      return /* @__PURE__ */ jsxs8(
-        "div",
+      return /* @__PURE__ */ jsx9(
+        GroupedSubline,
         {
-          className: "lax-tool-group__item lax-tool-group__subline",
-          "data-group-index": index + 1,
-          "data-tool-call-id": callId,
-          children: [
-            /* @__PURE__ */ jsx9("span", { className: "lax-tool-group__tree", "aria-hidden": true, children: treeChar }),
-            /* @__PURE__ */ jsx9("span", { className: "lax-tool-group__subline-title", children: subTitle }),
-            status && /* @__PURE__ */ jsxs8("span", { className: "lax-tool-group__subline-status", children: [
-              " \xB7 ",
-              status
-            ] })
-          ]
+          callId,
+          entry,
+          payload,
+          hideType,
+          treeChar,
+          index,
+          enableTranscript
         },
         callId || `group-item-${index}`
       );
@@ -5294,4 +5563,4 @@ export {
   VirtualTimeline,
   SessionTimeline
 };
-//# sourceMappingURL=chunk-3B4WC67E.js.map
+//# sourceMappingURL=chunk-ZNXAQ4ZZ.js.map