langchain_agentx_stream_ui 0.3.5 → 0.4.0

package/dist/{chunk-6PWGGGUK.js → chunk-AY7QLXFB.js}

@@ -15,7 +15,7 @@ import {
   extractLastToolInfoFromProgress,
   formatExploreSummaryText,
   formatThoughtDurationPrefix
-} from "./chunk-FL2G7SV3.js";
+} from "./chunk-QL3DIGFI.js";
 
 // src/types/tree.ts
 function createEmptyTree() {
@@ -77,6 +77,8 @@ var DEFAULT_EVENT_TIERS = {
   "subagent-model-step": "collapsed",
   "hook-progress": "debug",
   "hook-finished": "debug",
+  "api-retry": "debug",
+  "assistant-turn-superseded": "main",
   "compact-start": "debug",
   "compact-end": "debug",
   "task-started": "debug",
@@ -483,23 +485,31 @@ function parseSnapshotTasks(rawTasks) {
 }
 
 // src/types/modelRetry.ts
-function parseModelRetryHookData(data, timestampMs) {
-  if (data.name !== "model_retry") return null;
+function parseModelRetryFields(data, timestampMs) {
   const attempt = typeof data.attempt === "number" ? data.attempt : Number(data.attempt);
   const maxRetries = typeof data.max_retries === "number" ? data.max_retries : Number(data.max_retries);
   const retryDelayMs = typeof data.retry_delay_ms === "number" ? data.retry_delay_ms : Number(data.retry_delay_ms ?? 0);
   if (!Number.isFinite(attempt) || !Number.isFinite(maxRetries)) return null;
+  const errorCategory = typeof data.error_category === "string" ? data.error_category : typeof data.error === "string" ? data.error : "api_error";
   return {
     attempt: Math.max(1, Math.floor(attempt)),
     maxRetries: Math.max(1, Math.floor(maxRetries)),
     retryDelayMs: Number.isFinite(retryDelayMs) ? Math.max(0, retryDelayMs) : 0,
-    errorCategory: typeof data.error_category === "string" ? data.error_category : "api_error",
+    errorCategory,
     errorMessage: typeof data.error_message === "string" ? data.error_message : "",
     errorStatus: typeof data.error_status === "number" ? data.error_status : data.error_status != null ? Number(data.error_status) : null,
     phase: typeof data.phase === "string" ? data.phase : "scheduled",
     updatedAt: timestampMs
   };
 }
+function parseApiRetryEventData(data, timestampMs) {
+  if (data.subtype !== "api_retry") return null;
+  return parseModelRetryFields(data, timestampMs);
+}
+function parseModelRetryHookData(data, timestampMs) {
+  if (data.name !== "model_retry") return null;
+  return parseModelRetryFields(data, timestampMs);
+}
 function formatModelRetryCategoryLabel(category) {
   switch (category) {
     case "server_overload":
@@ -579,6 +589,38 @@ function agentToolNode(tree, parentNodeId) {
   const node = tree.byId[parentNodeId];
   return node?.kind === "tool_call" ? node : void 0;
 }
+function warnSubagentOrphanParent(eventType, data) {
+  if (import.meta.env?.DEV) {
+    console.warn("[langchain_agentx_stream_ui] subagent orphan", {
+      eventType,
+      tool_call_id: data.tool_call_id,
+      agent_id: data.agent_id
+    });
+  }
+}
+function ensureSyntheticAgentToolParent(tree, parentNodeId, event, data) {
+  const existing = tree.byId[parentNodeId];
+  if (existing?.kind === "tool_call") return;
+  const toolCallId = parentNodeId.startsWith("tool:") ? parentNodeId.slice("tool:".length) : "";
+  if (!toolCallId) return;
+  const node = {
+    kind: "tool_call",
+    id: parentNodeId,
+    toolCallId,
+    toolName: "Agent",
+    input: data.subagent_type ? { subagent_type: data.subagent_type } : null,
+    status: "running",
+    startedAt: eventTimeToMs(event.timestamp),
+    endedAt: null,
+    result: null,
+    progress: null,
+    inputDelta: "",
+    onMainStage: true,
+    syntheticParent: true
+  };
+  tree.byId[parentNodeId] = node;
+  appendToActiveParent(tree, parentNodeId);
+}
 function stepNameKey(event) {
   const step = event.step_index ?? 0;
   const name = event.tool_name ?? "unknown";
@@ -717,6 +759,7 @@ function finalizeStreamingNodes(tree) {
   }
 }
 function handleFinish(tree, event, seq) {
+  flushDeferredMainTextToTree(tree, event, seq);
   clearModelRetry(tree);
   const data = event.data;
   const endedAt = eventTimeToMs(event.timestamp);
@@ -787,6 +830,43 @@ function appendTaskDebugEvent(tree, event, eventIndex) {
     data: event.data
   });
 }
+function findLastMainStageTextNodeId(tree) {
+  const childIds = getActiveChildIds(tree);
+  for (let i = childIds.length - 1; i >= 0; i -= 1) {
+    const node = tree.byId[childIds[i]];
+    if (node?.kind === "text") {
+      return node.id;
+    }
+  }
+  return void 0;
+}
+function removeMainStageTextNode(tree, nodeId) {
+  delete tree.byId[nodeId];
+  tree.rootChildren = tree.rootChildren.filter((id) => id !== nodeId);
+}
+function handleAssistantTurnSuperseded(tree) {
+  clearModelRetry(tree);
+  const hadDeferredPartial = Boolean(tree.deferredMainText);
+  tree.deferredMainText = "";
+  const streaming = findStreamingNode(tree, "text");
+  if (streaming) {
+    removeMainStageTextNode(tree, streaming.id);
+    return;
+  }
+  if (!hadDeferredPartial) {
+    const lastTextId = findLastMainStageTextNodeId(tree);
+    if (lastTextId) {
+      removeMainStageTextNode(tree, lastTextId);
+    }
+  }
+}
+function handleApiRetry(tree, event) {
+  const data = event.data;
+  const modelRetry = parseApiRetryEventData(data, eventTimeToMs(event.timestamp));
+  if (modelRetry) {
+    tree.modelRetry = modelRetry;
+  }
+}
 function handleHookProgress(tree, event) {
   const data = event.data;
   const modelRetry = parseModelRetryHookData(data, eventTimeToMs(event.timestamp));
@@ -1184,11 +1264,19 @@ function handleToolProgress(tree, event, parentNodeId, _seq) {
     }
   });
 }
-function handleSubagentBoundaryStart(tree, event, parentNodeId) {
-  const tool = agentToolNode(tree, parentNodeId);
-  if (!tool || !parentNodeId) return;
+function handleSubagentBoundaryStart(tree, event, parentNodeId, options) {
   const data = event.data;
-  patchToolNode(tree, parentNodeId, {
+  let parentId = parentNodeId;
+  if (options?.syntheticAgentToolOnSubagentStart && data.tool_call_id && (!parentId || !agentToolNode(tree, parentId))) {
+    parentId = `tool:${data.tool_call_id}`;
+    ensureSyntheticAgentToolParent(tree, parentId, event, data);
+  }
+  const tool = agentToolNode(tree, parentId);
+  if (!tool || !parentId) {
+    warnSubagentOrphanParent("subagent-start", event.data);
+    return;
+  }
+  patchToolNode(tree, parentId, {
     subagentSessions: upsertSubagentSession(tool.subagentSessions ?? {}, {
       agentId: data.agent_id,
       subagentType: data.subagent_type,
@@ -1200,7 +1288,10 @@ function handleSubagentBoundaryStart(tree, event, parentNodeId) {
 }
 function handleSubagentBoundaryEnd(tree, event, parentNodeId) {
   const tool = agentToolNode(tree, parentNodeId);
-  if (!tool || !parentNodeId) return;
+  if (!tool || !parentNodeId) {
+    warnSubagentOrphanParent("subagent-end", event.data);
+    return;
+  }
   const data = event.data;
   const existing = tool.subagentSessions?.[data.agent_id];
   patchToolNode(tree, parentNodeId, {
@@ -1215,7 +1306,10 @@ function handleSubagentBoundaryEnd(tree, event, parentNodeId) {
 }
 function handleSubagentToolCall(tree, _event, parentNodeId, data) {
   const tool = agentToolNode(tree, parentNodeId);
-  if (!tool || !parentNodeId) return;
+  if (!tool || !parentNodeId) {
+    warnSubagentOrphanParent("subagent-tool-call", data);
+    return;
+  }
   const toolInput = data.tool_input ?? (typeof data.input === "object" ? data.input : void 0);
   const nestedInput = toolInput && typeof toolInput === "object" && !Array.isArray(toolInput) ? toolInput : null;
   const childToolCallId = typeof data.child_tool_call_id === "string" && data.child_tool_call_id || (nestedInput && typeof nestedInput.tool_call_id === "string" ? nestedInput.tool_call_id : void 0);
@@ -1233,7 +1327,10 @@ function handleSubagentToolCall(tree, _event, parentNodeId, data) {
 }
 function handleSubagentToolResult(tree, _event, parentNodeId, data) {
   const tool = agentToolNode(tree, parentNodeId);
-  if (!tool || !parentNodeId) return;
+  if (!tool || !parentNodeId) {
+    warnSubagentOrphanParent("subagent-tool-result", data);
+    return;
+  }
   patchToolNode(tree, parentNodeId, {
     subagentProgress: applySubagentToolResult(tool.subagentProgress ?? [], {
       agentId: data.agent_id,
@@ -1346,6 +1443,9 @@ function handleDebug(tree, event, eventIndex, parentNodeId, collectDebug) {
     appendTaskDebugEvent(tree, event, eventIndex);
   }
   switch (event.event_type) {
+    case "api-retry":
+      handleApiRetry(tree, event);
+      break;
     case "hook-progress":
       handleHookProgress(tree, event);
       break;
@@ -1479,7 +1579,7 @@ function applyClassified(tree, classified, eventIndex, options) {
         handleToolProgress(tree, event, parentNodeId, seq);
         break;
       case "subagent-start":
-        handleSubagentBoundaryStart(tree, event, parentNodeId);
+        handleSubagentBoundaryStart(tree, event, parentNodeId, options);
         break;
       case "subagent-end":
         handleSubagentBoundaryEnd(tree, event, parentNodeId);
@@ -1542,6 +1642,9 @@ function applyClassified(tree, classified, eventIndex, options) {
       handleTextEnd(tree, event);
       break;
     }
+    case "assistant-turn-superseded":
+      handleAssistantTurnSuperseded(tree);
+      break;
     case "reasoning-start":
       if (isDroppedByEmitterRoute(event.data)) break;
       if (isSubagentScopedReasoningEvent(event.data)) break;
@@ -1669,6 +1772,78 @@ function replayTree(initialTree, events, options) {
   );
 }
 
+// src/core/workflow/workflowAuthoritySelectors.ts
+var AUTHORITY_TERMINAL = /* @__PURE__ */ new Set([
+  "done",
+  "error",
+  "canceled",
+  "partial_success"
+]);
+function isAuthorityTerminalStatus(status) {
+  return status != null && AUTHORITY_TERMINAL.has(status);
+}
+function isAuthorityTerminal(state) {
+  return isAuthorityTerminalStatus(state.authority?.status);
+}
+function shouldShowGlobalSpinner(state) {
+  const auth = state.authority?.status;
+  if (auth === "canceled") return false;
+  if (isAuthorityTerminalStatus(auth)) return false;
+  if (state.display.awaitingAuthority) return false;
+  if (auth == null || auth === "pending" || auth === "running") {
+    return true;
+  }
+  const proj = state.display.projectionStatus;
+  return proj === "running" || proj === "connecting";
+}
+function resolveWorkflowBanner(state) {
+  if (state.display.authoritySubscribeFailed) {
+    return { kind: "sync_failed", message: "\u72B6\u6001\u540C\u6B65\u5931\u8D25\uFF0C\u8BF7\u91CD\u8BD5" };
+  }
+  if (state.display.authoritySyncFailed) {
+    return { kind: "sync_failed", message: "\u540C\u6B65\u72B6\u6001\u5931\u8D25\uFF0C\u8BF7\u5237\u65B0" };
+  }
+  const auth = state.authority?.status;
+  if (auth === "error") {
+    return { kind: "authority_error", message: "Workflow \u6267\u884C\u5931\u8D25" };
+  }
+  if (state.snapshotHydrateFailed) {
+    return {
+      kind: "snapshot_unavailable",
+      message: "\u65E0\u6CD5\u52A0\u8F7D projection snapshot\uFF0C\u5C55\u793A\u53EF\u80FD\u4E0D\u5B8C\u6574"
+    };
+  }
+  if (state.display.awaitingAuthority) {
+    return { kind: "awaiting_authority", message: "\u6536\u5C3E\u4E2D\u2026" };
+  }
+  if (auth === "partial_success") {
+    const forced = state.display.terminalKind === "forced_closeout";
+    return {
+      kind: "partial_success",
+      message: forced ? "\u90E8\u5206\u5B8C\u6210\uFF08\u90E8\u5206\u5BB9\u5668\u672A\u6B63\u5E38\u7ED3\u675F\uFF09" : "\u90E8\u5206\u5B8C\u6210\uFF08\u6240\u6709\u5BB9\u5668\u6B63\u5E38\u7ED3\u675F\uFF09"
+    };
+  }
+  const meta = state.authority?.metadata;
+  if (auth === "done" && meta?.projection_status === "degraded") {
+    const dropped = meta.projection_dropped_count;
+    const suffix = dropped != null && dropped > 0 ? `\uFF08\u7EA6 ${dropped} \u6761\u4E8B\u4EF6\u672A\u5B8C\u6574\u5C55\u793A\uFF09` : "";
+    return {
+      kind: "projection_degraded",
+      message: `\u5C55\u793A\u53EF\u80FD\u4E0D\u5B8C\u6574${suffix}`
+    };
+  }
+  if (state.display.terminalKind === "forced_closeout" && auth === "done") {
+    return { kind: "forced_closeout", message: "Workflow \u5DF2\u7ED3\u675F\uFF08\u90E8\u5206\u5BB9\u5668\u672A\u6B63\u5E38\u5173\u95ED\uFF09" };
+  }
+  return { kind: "none", message: "" };
+}
+function getDisplayProjectionStatus(state) {
+  return state.display.projectionStatus;
+}
+function applyAuthoritySnapshotToState(state, snapshot) {
+  return { ...state, authority: snapshot };
+}
+
 // src/core/projection/renderModel.ts
 var ProjectionContext = class _ProjectionContext {
   displayMode;
@@ -3726,6 +3901,14 @@ function useSessionMeta() {
   const store = useSessionStoreApi();
   return useStore5(store, (state) => state.tree.meta);
 }
+function useAgentAuthority() {
+  const store = useSessionStoreApi();
+  return useStore5(store, (state) => state.authority);
+}
+function useAgentAuthoritySubscribeFailed() {
+  const store = useSessionStoreApi();
+  return useStore5(store, (state) => state.authoritySubscribeFailed);
+}
 
 // src/view/hooks/useMinDisplayTime.ts
 import { useEffect, useRef, useState } from "react";
@@ -3891,6 +4074,18 @@ function SystemSummaryNode({ summaryText }) {
   return /* @__PURE__ */ jsx3("div", { className: "lax-system-summary", "data-kind": "system_summary", children: summaryText });
 }
 
+// src/core/agent/agentAuthoritySelectors.ts
+function shouldShowAgentSpinner(state) {
+  if (!state.authorityBindingActive) {
+    const status = state.tree.status;
+    return status === "running" || status === "connecting";
+  }
+  const auth = state.authority?.status;
+  if (auth === "canceled") return false;
+  if (isAuthorityTerminalStatus(auth)) return false;
+  return true;
+}
+
 // src/view/TaskList.tsx
 import { useEffect as useEffect3, useMemo as useMemo4, useRef as useRef2, useState as useState3 } from "react";
 
@@ -4769,7 +4964,11 @@ function Spinner({
 }) {
   const status = useSessionStatus();
   const store = useSessionStoreApi();
-  const isRunning = status === "running" || status === "connecting";
+  const showSpinner = useSyncExternalStore(
+    (callback) => store.subscribe(callback),
+    () => shouldShowAgentSpinner(store.getState())
+  );
+  const isRunning = showSpinner;
   const activeRoundId = useSyncExternalStore(
     (callback) => store.subscribe(callback),
     () => store.getState().tree.activeRoundId
@@ -5438,6 +5637,12 @@ export {
   getChildIds,
   replayEvents,
   replayTree,
+  isAuthorityTerminalStatus,
+  isAuthorityTerminal,
+  shouldShowGlobalSpinner,
+  resolveWorkflowBanner,
+  getDisplayProjectionStatus,
+  applyAuthoritySnapshotToState,
   ProjectionContext,
   createProjectionContext,
   DEFAULT_SESSION_VIEW_OPTIONS,
@@ -5489,11 +5694,14 @@ export {
   useSessionStatus,
   useInternalErrors,
   useSessionMeta,
+  useAgentAuthority,
+  useAgentAuthoritySubscribeFailed,
   MIN_HINT_DISPLAY_MS,
   useMinDisplayTime,
   TimelineItem,
   CollapsedExploreNode,
   SystemSummaryNode,
+  shouldShowAgentSpinner,
   TaskList,
   Spinner,
   formatTaskFooterLabel,
@@ -5507,4 +5715,4 @@ export {
   VirtualTimeline,
   SessionTimeline
 };
-//# sourceMappingURL=chunk-6PWGGGUK.js.map
+//# sourceMappingURL=chunk-AY7QLXFB.js.map