langchain_agentx_stream_ui 0.2.0 → 0.2.2

package/dist/index.js

@@ -23,7 +23,7 @@ import {
   fixDiagramSvg,
   markEventIdSeen,
   shouldSkipDuplicateEvent
-} from "./chunk-2T26L3VT.js";
+} from "./chunk-DGC43A5L.js";
 import {
   DefaultToolBody,
   InteractionBusContext,
@@ -32,8 +32,8 @@ import {
   getNoopInteractionBus,
   resolveToolBody,
   useInteractionBus
-} from "./chunk-6DYISADG.js";
-import "./chunk-DP7V33X7.js";
+} from "./chunk-T626YBJX.js";
+import "./chunk-6Z4ZF36Z.js";
 import {
   CollapseKind,
   CollapsedExploreNode,
@@ -111,7 +111,7 @@ import {
   useSessionStatus,
   useSessionViewOptions,
   useTimeline
-} from "./chunk-ZNXAQ4ZZ.js";
+} from "./chunk-OFXC2GZI.js";
 import {
   Agent,
   AgentToolBody,
@@ -138,7 +138,7 @@ import {
   WriteToolBody,
   createDefaultToolRegistry,
   formatAgentTitle
-} from "./chunk-7CLAU74Y.js";
+} from "./chunk-YUZXTGJJ.js";
 import {
   BodyBlockList,
   DiffView,
@@ -154,7 +154,7 @@ import {
   formatTimeoutFooter,
   parseDiffText,
   truncateCommand
-} from "./chunk-4RIOBLGB.js";
+} from "./chunk-GBY5DJ7L.js";
 
 // src/view/AgentSession.tsx
 import { useEffect as useEffect2, useMemo, useRef as useRef2 } from "react";
@@ -275,6 +275,14 @@ function transportLog(message, level = "warn") {
     console.warn(line);
   }
 }
+function isSseStreamTerminalEvent(agentEvent, expectedApplicationKind) {
+  const eventType = agentEvent.event_type;
+  if (eventType === "error") return true;
+  if (expectedApplicationKind === "workflow") {
+    return eventType === "workflow-end" || eventType === "workflow-failed";
+  }
+  return eventType === "finish";
+}
 function buildResumeStreamUrl(baseUrl, lastEventId, queryParam = "last_event_id") {
   if (!lastEventId) return baseUrl;
   const sep = baseUrl.includes("?") ? "&" : "?";
@@ -420,7 +428,7 @@ function createAgentxSseSource(options) {
               } else {
                 flushBufferedDelta();
                 handler(agentEvent, ctx);
-                if (agentEvent.event_type === "finish" || agentEvent.event_type === "error") {
+                if (isSseStreamTerminalEvent(agentEvent, expectedApplicationKind)) {
                   streamTerminal = true;
                 }
               }
@@ -626,6 +634,8 @@ function AgentSession({
       workspaceRoot
     ]
   );
+  const onErrorRef = useRef2(onError);
+  onErrorRef.current = onError;
   useEffect2(() => {
     const store = storeRef.current;
     const controller = new AbortController();
@@ -637,12 +647,15 @@ function AgentSession({
         if (tree.status === "running" || tree.status === "connecting") {
           store.setState({ tree: { ...tree, status: "error" } });
         }
+        onErrorRef.current?.(err);
+      } else if (err instanceof Error) {
+        onErrorRef.current?.(err);
       }
     });
     return () => {
       controller.abort();
     };
-  }, [source, onError]);
+  }, [source]);
   return /* @__PURE__ */ jsx3(SessionStoreContext.Provider, { value: storeRef.current, children: /* @__PURE__ */ jsx3(InteractionBusContext.Provider, { value: busRef, children: /* @__PURE__ */ jsx3(NodeRegistryContext.Provider, { value: registryRef, children: /* @__PURE__ */ jsx3(MarkdownRendererContext.Provider, { value: markdownRenderer, children: /* @__PURE__ */ jsx3(ToolDisplayOptionsContext.Provider, { value: toolDisplayRef, children: /* @__PURE__ */ jsx3(SessionViewOptionsContext.Provider, { value: sessionViewRef, children: /* @__PURE__ */ jsxs("div", { className: "lax-agent-session", "data-testid": "lax-agent-session", children: [
     children ?? /* @__PURE__ */ jsx3(
       AgentLoopView,
@@ -762,6 +775,9 @@ function createEmptyWorkflowSessionState() {
 }
 
 // src/core/workflow/workflowLoopTreeUtils.ts
+function isWorkflowLoopTreeActive(tree) {
+  return tree?.status === "running" || tree?.status === "connecting";
+}
 function createWorkflowRunningLoopTree(seedMs = Date.now()) {
   const tree = createEmptyTree();
   tree.status = "running";
@@ -793,6 +809,168 @@ function finalizeWorkflowLoopTree(tree, terminalStatus = "done") {
   };
 }
 
+// src/core/workflow/stageDoneSummary.ts
+function countToolUses(tree) {
+  return Object.values(tree.byId).filter((node) => node.kind === "tool_call").length;
+}
+function computeDurationSeconds(tree) {
+  const started = tree.meta.startedAt;
+  if (started == null) return null;
+  let maxTs = started;
+  for (const node of Object.values(tree.byId)) {
+    if ("endedAt" in node && typeof node.endedAt === "number") {
+      maxTs = Math.max(maxTs, node.endedAt);
+    }
+    if ("startedAt" in node && typeof node.startedAt === "number") {
+      maxTs = Math.max(maxTs, node.startedAt);
+    }
+  }
+  const seconds = Math.round((maxTs - started) / 1e3);
+  return seconds > 0 ? seconds : null;
+}
+function lastTextContent(tree) {
+  const textNodes = Object.values(tree.byId).filter((node) => node.kind === "text");
+  if (textNodes.length === 0) return "";
+  return textNodes[textNodes.length - 1].accumulated.trim();
+}
+function buildStageDoneSummary(tree) {
+  if (tree.status !== "done" && tree.status !== "error") return null;
+  const toolUses = countToolUses(tree);
+  const duration = computeDurationSeconds(tree);
+  const durationSuffix = duration != null ? ` \xB7 ${duration}s` : "";
+  const label = toolUses > 0 ? `Done (${toolUses} tool use${toolUses === 1 ? "" : "s"}${durationSuffix})` : `Done (1 agent turn${durationSuffix})`;
+  const fullText = lastTextContent(tree);
+  const lines = fullText.split("\n").map((line) => line.trim()).filter(Boolean);
+  const teaser = lines[0] ?? "";
+  const extraLines = Math.max(0, lines.length - 1);
+  return { label, teaser, extraLines };
+}
+function formatTeaserExpandHint(extraLines, verbose) {
+  if (extraLines <= 0) return null;
+  const shown = verbose ? Math.min(extraLines, 2) : 0;
+  const hidden = extraLines - shown;
+  if (hidden <= 0) return null;
+  return `\u2026 +${hidden} lines (ctrl+o to expand)`;
+}
+
+// src/core/workflow/workflowLoopEviction.ts
+function containersByLoopSessionId(tree) {
+  const map = /* @__PURE__ */ new Map();
+  for (const node of Object.values(tree.containersById)) {
+    if (node.loopSessionId) map.set(node.loopSessionId, node);
+  }
+  return map;
+}
+function hasRunningParallelSibling(node, tree) {
+  if (node.scope !== "item" || !node.parent_container_id) return false;
+  for (const sibling of Object.values(tree.containersById)) {
+    if (sibling.container_id === node.container_id) continue;
+    if (sibling.scope !== "item") continue;
+    if (sibling.parent_container_id !== node.parent_container_id) continue;
+    if (sibling.status === "running" || sibling.status === "retrying") return true;
+  }
+  return false;
+}
+function collectProtectedLoopSessionIds(state, pinnedLoopSessionIds) {
+  const protectedIds = new Set(Object.keys(pinnedLoopSessionIds));
+  const byLoop = containersByLoopSessionId(state.containerTree);
+  if (state.activeLoopSessionId) {
+    const activeContainer = byLoop.get(state.activeLoopSessionId);
+    if (activeContainer && (activeContainer.status === "running" || activeContainer.status === "retrying")) {
+      protectedIds.add(state.activeLoopSessionId);
+    }
+  }
+  for (const containerId of state.containerTree.activeContainerIds) {
+    const node = state.containerTree.containersById[containerId];
+    if (!node?.loopSessionId) continue;
+    if (node.status === "running" || node.status === "retrying") {
+      protectedIds.add(node.loopSessionId);
+    }
+  }
+  for (const node of Object.values(state.containerTree.containersById)) {
+    if (!node.loopSessionId) continue;
+    if (node.status === "running" || node.status === "retrying") {
+      protectedIds.add(node.loopSessionId);
+    }
+    if ((node.status === "completed" || node.status === "failed") && hasRunningParallelSibling(node, state.containerTree)) {
+      protectedIds.add(node.loopSessionId);
+    }
+  }
+  return protectedIds;
+}
+function evictInactiveLoopTrees(state, pinnedLoopSessionIds) {
+  const protectedIds = collectProtectedLoopSessionIds(state, pinnedLoopSessionIds);
+  for (const id of state.hotCompletedLoopSessionIds) {
+    protectedIds.add(id);
+  }
+  const byLoop = containersByLoopSessionId(state.containerTree);
+  const nextLoops = { ...state.loopTreesBySessionId };
+  const nextSummaries = {
+    ...state.stageDoneSummariesByContainerId
+  };
+  for (const [loopSessionId, tree] of Object.entries(state.loopTreesBySessionId)) {
+    if (protectedIds.has(loopSessionId)) continue;
+    const container = byLoop.get(loopSessionId);
+    if (!container) continue;
+    if (container.status !== "completed" && container.status !== "failed") continue;
+    const summary = buildStageDoneSummary(tree);
+    if (summary) {
+      nextSummaries[container.container_id] = summary;
+    }
+    delete nextLoops[loopSessionId];
+  }
+  return {
+    ...state,
+    loopTreesBySessionId: nextLoops,
+    stageDoneSummariesByContainerId: nextSummaries
+  };
+}
+function touchCompletedLoopSession(state, loopSessionId) {
+  const ids = state.hotCompletedLoopSessionIds.filter((id) => id !== loopSessionId);
+  ids.push(loopSessionId);
+  return {
+    ...state,
+    hotCompletedLoopSessionIds: ids
+  };
+}
+function evictCompletedLoopOverflow(state, maxHydratedCompletedLoops) {
+  if (state.hotCompletedLoopSessionIds.length <= maxHydratedCompletedLoops) {
+    return state;
+  }
+  const overflow = state.hotCompletedLoopSessionIds.length - maxHydratedCompletedLoops;
+  const candidateEvictIds = state.hotCompletedLoopSessionIds.slice(0, overflow);
+  let trimmedIds = state.hotCompletedLoopSessionIds.slice(overflow);
+  const byLoop = containersByLoopSessionId(state.containerTree);
+  const nextLoops = { ...state.loopTreesBySessionId };
+  const nextSummaries = {
+    ...state.stageDoneSummariesByContainerId
+  };
+  for (const id of candidateEvictIds) {
+    const tree = nextLoops[id];
+    if (!tree) continue;
+    const container = byLoop.get(id);
+    if (container && hasRunningParallelSibling(container, state.containerTree)) {
+      if (!trimmedIds.includes(id)) {
+        trimmedIds = [id, ...trimmedIds];
+      }
+      continue;
+    }
+    if (container) {
+      const summary = buildStageDoneSummary(tree);
+      if (summary) {
+        nextSummaries[container.container_id] = summary;
+      }
+    }
+    delete nextLoops[id];
+  }
+  return {
+    ...state,
+    loopTreesBySessionId: nextLoops,
+    stageDoneSummariesByContainerId: nextSummaries,
+    hotCompletedLoopSessionIds: trimmedIds
+  };
+}
+
 // src/types/workflowDisplay.ts
 function readDisplayPayload(data) {
   const display = data.display;
@@ -1727,6 +1905,9 @@ function settleCompletedStageSubtree(tree, stageContainerId) {
     }
     if (node.container_id === stageContainerId) continue;
     if (node.status === "completed" || node.status === "failed") continue;
+    if (node.status === "running" || node.status === "retrying" || node.status === "pending" || node.status === "blocked") {
+      continue;
+    }
     next = upsertRuntimeNode(next, {
       ...node,
       status: "completed",
@@ -1908,17 +2089,26 @@ function findContainerByLoopSession(tree, loopSessionId) {
   return findContainerIdByLoopSession(tree, loopSessionId);
 }
 var WORKFLOW_DEFAULT_LOOP_SESSION_ID = "workflow-loop-default";
-function resolveAgentLoopSessionId(event, activeLoopSessionId) {
+function resolveAgentLoopSessionId(event, activeLoopSessionId, containerTree) {
   if (event.session_id) return event.session_id;
   if (isWorkflowStructureEventType(event.event_type)) return null;
   if (event.event_type === "task-list-snapshot") return null;
   const derived = deriveLoopSessionIdFromAgentEvent(event);
   if (derived) return derived;
+  const byLoop = containersByLoopSessionId(containerTree);
+  if (activeLoopSessionId) {
+    const activeContainer = byLoop.get(activeLoopSessionId);
+    if (activeContainer && (activeContainer.status === "running" || activeContainer.status === "retrying")) {
+      return activeLoopSessionId;
+    }
+  }
+  const runningLoopIds = [...byLoop.entries()].filter(([, node]) => node.status === "running" || node.status === "retrying").map(([loopSessionId]) => loopSessionId);
+  if (runningLoopIds.length === 1) return runningLoopIds[0];
   if (activeLoopSessionId) return activeLoopSessionId;
   return WORKFLOW_DEFAULT_LOOP_SESSION_ID;
 }
 function bindLoopSessionToContainerTree(tree, event, activeLoopSessionId) {
-  const sessionId = resolveAgentLoopSessionId(event, activeLoopSessionId);
+  const sessionId = resolveAgentLoopSessionId(event, activeLoopSessionId, tree);
   if (!sessionId) return tree;
   let nextTree = ensureEmbeddedSubworkflowContainers(tree, sessionId);
   if (findContainerByLoopSession(nextTree, sessionId)) return nextTree;
@@ -2157,7 +2347,7 @@ function reduceWorkflowSession(state, event, eventIndex, options) {
   if (isWorkflowStructureEventType(event.event_type)) {
     return applyStructureEvent(state, event);
   }
-  const loopSessionId = resolveAgentLoopSessionId(event, state.activeLoopSessionId) ?? WORKFLOW_DEFAULT_LOOP_SESSION_ID;
+  const loopSessionId = resolveAgentLoopSessionId(event, state.activeLoopSessionId, state.containerTree) ?? WORKFLOW_DEFAULT_LOOP_SESSION_ID;
   let containerTree = bindLoopSessionToContainerTree(
     state.containerTree,
     event,
@@ -2176,6 +2366,11 @@ function reduceWorkflowSession(state, event, eventIndex, options) {
     next.containerTree,
     next.workflowProgress
   );
+  let status = next.status;
+  if (workflowProgress.status === "done") status = "done";
+  else if (workflowProgress.status === "error") status = "error";
+  else if (reducedLoop.status === "error") status = "error";
+  else if (status === "connecting") status = "running";
   return {
     ...next,
     workflowProgress,
@@ -2184,7 +2379,85 @@ function reduceWorkflowSession(state, event, eventIndex, options) {
       [loopSessionId]: reducedLoop
     },
     activeLoopSessionId: loopSessionId,
-    status: reducedLoop.status === "error" ? "error" : next.status === "connecting" ? "running" : next.status
+    status
+  };
+}
+function finalizeWorkflowSessionOnStreamComplete(state) {
+  if (state.status === "done" || state.status === "error") {
+    return { ...state, activeLoopSessionId: null };
+  }
+  const tree = state.containerTree;
+  const rootId = tree.rootContainerIds[0];
+  if (!rootId) {
+    return { ...state, status: "done", activeLoopSessionId: null };
+  }
+  const root = tree.containersById[rootId];
+  if (!root) {
+    return { ...state, status: "done", activeLoopSessionId: null };
+  }
+  const topLevelStages = Object.values(tree.containersById).filter(
+    (node) => node.scope === "stage" && node.workflow_path === root.workflow_path && node.workflow_depth === root.workflow_depth
+  );
+  const progressTotal = root.display?.progress_total ?? (topLevelStages.reduce(
+    (max, s) => Math.max(max, s.display?.progress_total ?? 0),
+    0
+  ) || topLevelStages.length);
+  const settledStageCount = topLevelStages.filter(
+    (s) => s.status === "completed" || s.status === "failed" || s.status === "skipped"
+  ).length;
+  const progressCurrent = Math.max(
+    root.display?.progress_current ?? 0,
+    settledStageCount
+  );
+  const rootParallelItems = Object.values(tree.containersById).filter(
+    (node) => node.scope === "item" && node.workflow_path === root.workflow_path && node.workflow_depth === root.workflow_depth
+  );
+  const rootParallelSettled = topLevelStages.length === 0 && rootParallelItems.length > 0 && rootParallelItems.every(
+    (item) => item.status === "completed" || item.status === "failed" || item.status === "skipped"
+  ) && !Object.values(tree.containersById).some(
+    (node) => node.container_id !== rootId && node.workflow_path === root.workflow_path && node.workflow_depth === root.workflow_depth && (node.status === "running" || node.status === "retrying" || node.status === "pending")
+  );
+  const workflowVisuallyComplete = root.status === "completed" || root.status === "failed" || progressTotal > 0 && progressCurrent >= progressTotal || rootParallelSettled;
+  if (!workflowVisuallyComplete) {
+    return state;
+  }
+  let containersById = { ...tree.containersById };
+  for (const [id, node] of Object.entries(containersById)) {
+    if (node.status === "running" || node.status === "pending") {
+      containersById[id] = { ...node, status: "completed", is_open: false };
+    }
+  }
+  const settledRoot = containersById[rootId] ?? root;
+  if (settledRoot.status !== "completed" && settledRoot.status !== "failed") {
+    containersById = {
+      ...containersById,
+      [rootId]: { ...settledRoot, status: "completed", is_open: false }
+    };
+  }
+  const loopTreesBySessionId = { ...state.loopTreesBySessionId };
+  for (const [loopSessionId, loopTree] of Object.entries(loopTreesBySessionId)) {
+    if (loopTree.status === "running" || loopTree.status === "connecting") {
+      loopTreesBySessionId[loopSessionId] = finalizeWorkflowLoopTree(loopTree, "done");
+    }
+  }
+  const containerTree = {
+    ...tree,
+    containersById,
+    activeContainerIds: tree.activeContainerIds.filter(
+      (id) => containersById[id]?.is_open
+    )
+  };
+  const workflowProgress = syncWorkflowProgressFromContainerTree(
+    containerTree,
+    state.workflowProgress
+  );
+  return {
+    ...state,
+    status: workflowProgress.status === "error" ? "error" : "done",
+    containerTree,
+    workflowProgress,
+    loopTreesBySessionId,
+    activeLoopSessionId: null
   };
 }
 function reduceWorkflowEvents(events, options) {
@@ -2194,180 +2467,37 @@ function reduceWorkflowEvents(events, options) {
   );
 }
 
-// src/core/workflow/stageDoneSummary.ts
-function countToolUses(tree) {
-  return Object.values(tree.byId).filter((node) => node.kind === "tool_call").length;
+// src/core/workflowSessionStore.ts
+var DEFAULT_HYDRATE_CHUNK_SIZE = 50;
+function nextAnimationFrame() {
+  return new Promise((resolve) => {
+    requestAnimationFrame(() => resolve());
+  });
 }
-function computeDurationSeconds(tree) {
-  const started = tree.meta.startedAt;
-  if (started == null) return null;
-  let maxTs = started;
-  for (const node of Object.values(tree.byId)) {
-    if ("endedAt" in node && typeof node.endedAt === "number") {
-      maxTs = Math.max(maxTs, node.endedAt);
-    }
-    if ("startedAt" in node && typeof node.startedAt === "number") {
-      maxTs = Math.max(maxTs, node.startedAt);
+function withoutHydrating(ids, loopSessionId) {
+  if (!ids[loopSessionId]) return ids;
+  const next = { ...ids };
+  delete next[loopSessionId];
+  return next;
+}
+function safeReduceWorkflow(state, event, eventIndex, options) {
+  try {
+    return { state: reduceWorkflowSession(state, event, eventIndex, options) };
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    const stack = err instanceof Error ? err.stack : void 0;
+    if (import.meta.env?.DEV) {
+      console.error("[langchain_agentx_stream_ui] workflow reducer error:", err);
     }
-  }
-  const seconds = Math.round((maxTs - started) / 1e3);
-  return seconds > 0 ? seconds : null;
-}
-function lastTextContent(tree) {
-  const textNodes = Object.values(tree.byId).filter((node) => node.kind === "text");
-  if (textNodes.length === 0) return "";
-  return textNodes[textNodes.length - 1].accumulated.trim();
-}
-function buildStageDoneSummary(tree) {
-  if (tree.status !== "done" && tree.status !== "error") return null;
-  const toolUses = countToolUses(tree);
-  const duration = computeDurationSeconds(tree);
-  const durationSuffix = duration != null ? ` \xB7 ${duration}s` : "";
-  const label = toolUses > 0 ? `Done (${toolUses} tool use${toolUses === 1 ? "" : "s"}${durationSuffix})` : `Done (1 agent turn${durationSuffix})`;
-  const fullText = lastTextContent(tree);
-  const lines = fullText.split("\n").map((line) => line.trim()).filter(Boolean);
-  const teaser = lines[0] ?? "";
-  const extraLines = Math.max(0, lines.length - 1);
-  return { label, teaser, extraLines };
-}
-function formatTeaserExpandHint(extraLines, verbose) {
-  if (extraLines <= 0) return null;
-  const shown = verbose ? Math.min(extraLines, 2) : 0;
-  const hidden = extraLines - shown;
-  if (hidden <= 0) return null;
-  return `\u2026 +${hidden} lines (ctrl+o to expand)`;
-}
-
-// src/core/workflow/workflowLoopEviction.ts
-function containersByLoopSessionId(tree) {
-  const map = /* @__PURE__ */ new Map();
-  for (const node of Object.values(tree.containersById)) {
-    if (node.loopSessionId) map.set(node.loopSessionId, node);
-  }
-  return map;
-}
-function collectProtectedLoopSessionIds(state, pinnedLoopSessionIds) {
-  const protectedIds = new Set(Object.keys(pinnedLoopSessionIds));
-  const byLoop = containersByLoopSessionId(state.containerTree);
-  if (state.activeLoopSessionId) {
-    const activeContainer = byLoop.get(state.activeLoopSessionId);
-    if (activeContainer && (activeContainer.status === "running" || activeContainer.status === "retrying")) {
-      protectedIds.add(state.activeLoopSessionId);
-    }
-  }
-  for (const containerId of state.containerTree.activeContainerIds) {
-    const node = state.containerTree.containersById[containerId];
-    if (!node?.loopSessionId) continue;
-    if (node.status === "running" || node.status === "retrying") {
-      protectedIds.add(node.loopSessionId);
-    }
-  }
-  for (const node of Object.values(state.containerTree.containersById)) {
-    if (!node.loopSessionId) continue;
-    if (node.status === "running" || node.status === "retrying") {
-      protectedIds.add(node.loopSessionId);
-    }
-  }
-  return protectedIds;
-}
-function evictInactiveLoopTrees(state, pinnedLoopSessionIds) {
-  const protectedIds = collectProtectedLoopSessionIds(state, pinnedLoopSessionIds);
-  for (const id of state.hotCompletedLoopSessionIds) {
-    protectedIds.add(id);
-  }
-  const byLoop = containersByLoopSessionId(state.containerTree);
-  const nextLoops = { ...state.loopTreesBySessionId };
-  const nextSummaries = {
-    ...state.stageDoneSummariesByContainerId
-  };
-  for (const [loopSessionId, tree] of Object.entries(state.loopTreesBySessionId)) {
-    if (protectedIds.has(loopSessionId)) continue;
-    const container = byLoop.get(loopSessionId);
-    if (!container) continue;
-    if (container.status !== "completed" && container.status !== "failed") continue;
-    const summary = buildStageDoneSummary(tree);
-    if (summary) {
-      nextSummaries[container.container_id] = summary;
-    }
-    delete nextLoops[loopSessionId];
-  }
-  return {
-    ...state,
-    loopTreesBySessionId: nextLoops,
-    stageDoneSummariesByContainerId: nextSummaries
-  };
-}
-function touchCompletedLoopSession(state, loopSessionId) {
-  const ids = state.hotCompletedLoopSessionIds.filter((id) => id !== loopSessionId);
-  ids.push(loopSessionId);
-  return {
-    ...state,
-    hotCompletedLoopSessionIds: ids
-  };
-}
-function evictCompletedLoopOverflow(state, maxHydratedCompletedLoops) {
-  if (state.hotCompletedLoopSessionIds.length <= maxHydratedCompletedLoops) {
-    return state;
-  }
-  const overflow = state.hotCompletedLoopSessionIds.length - maxHydratedCompletedLoops;
-  const evictedIds = new Set(state.hotCompletedLoopSessionIds.slice(0, overflow));
-  const trimmedIds = state.hotCompletedLoopSessionIds.slice(overflow);
-  const byLoop = containersByLoopSessionId(state.containerTree);
-  const nextLoops = { ...state.loopTreesBySessionId };
-  const nextSummaries = {
-    ...state.stageDoneSummariesByContainerId
-  };
-  for (const id of evictedIds) {
-    const tree = nextLoops[id];
-    if (!tree) continue;
-    const container = byLoop.get(id);
-    if (container) {
-      const summary = buildStageDoneSummary(tree);
-      if (summary) {
-        nextSummaries[container.container_id] = summary;
-      }
-    }
-    delete nextLoops[id];
-  }
-  return {
-    ...state,
-    loopTreesBySessionId: nextLoops,
-    stageDoneSummariesByContainerId: nextSummaries,
-    hotCompletedLoopSessionIds: trimmedIds
-  };
-}
-
-// src/core/workflowSessionStore.ts
-var DEFAULT_HYDRATE_CHUNK_SIZE = 50;
-function nextAnimationFrame() {
-  return new Promise((resolve) => {
-    requestAnimationFrame(() => resolve());
-  });
-}
-function withoutHydrating(ids, loopSessionId) {
-  if (!ids[loopSessionId]) return ids;
-  const next = { ...ids };
-  delete next[loopSessionId];
-  return next;
-}
-function safeReduceWorkflow(state, event, eventIndex, options) {
-  try {
-    return { state: reduceWorkflowSession(state, event, eventIndex, options) };
-  } catch (err) {
-    const message = err instanceof Error ? err.message : String(err);
-    const stack = err instanceof Error ? err.stack : void 0;
-    if (import.meta.env?.DEV) {
-      console.error("[langchain_agentx_stream_ui] workflow reducer error:", err);
-    }
-    return {
-      state: {
-        ...state,
-        internalErrors: [
-          ...state.internalErrors,
-          { eventIndex, eventType: event.event_type, message, stack }
-        ]
-      }
-    };
+    return {
+      state: {
+        ...state,
+        internalErrors: [
+          ...state.internalErrors,
+          { eventIndex, eventType: event.event_type, message, stack }
+        ]
+      }
+    };
   }
 }
 function finalizeWorkflowSessionOnStreamError(state, message, eventIndex, stack) {
@@ -2462,6 +2592,13 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
         )
       });
     },
+    markStreamComplete() {
+      set((current) => ({
+        state: finalizeState(
+          finalizeWorkflowSessionOnStreamComplete(current.state)
+        )
+      }));
+    },
     /**
      * P2：将 completed loop 加入 LRU 热缓存并淘汰溢出。
      * running 保护由 collectProtectedLoopSessionIds 按状态自动覆盖，不再依赖永久 pin。
@@ -2716,6 +2853,76 @@ function WorkflowMarkdownPreview({
   return /* @__PURE__ */ jsx7("div", { className, "data-testid": testId, children: /* @__PURE__ */ jsx7(RichMarkdown, { content, deferDiagrams: true }) });
 }
 
+// src/view/workflow/workflowStageListUtils.ts
+var ACTIVE_CONTAINER_STATUSES = /* @__PURE__ */ new Set([
+  "running",
+  "retrying",
+  "pending",
+  "blocked"
+]);
+function isDescendantContainer(containersById, ancestorId, nodeId) {
+  const seen = /* @__PURE__ */ new Set();
+  let current = containersById[nodeId]?.parent_container_id ?? null;
+  while (current) {
+    if (current === ancestorId) return true;
+    if (seen.has(current)) return false;
+    seen.add(current);
+    current = containersById[current]?.parent_container_id ?? null;
+  }
+  return false;
+}
+function subtreeHasActiveWork(containersById, rootContainerId) {
+  for (const node of Object.values(containersById)) {
+    if (node.container_id === rootContainerId) continue;
+    if (!isDescendantContainer(containersById, rootContainerId, node.container_id)) continue;
+    if (ACTIVE_CONTAINER_STATUSES.has(node.status)) return true;
+  }
+  return false;
+}
+function stageSortKey(node) {
+  return node.display?.progress_current ?? 0;
+}
+function partitionWorkflowStages(stages, maxVisibleDoneCount, containersById) {
+  const sorted = [...stages].sort(
+    (a, b) => stageSortKey(a) - stageSortKey(b) || a.scope_key.localeCompare(b.scope_key)
+  );
+  const always = [];
+  const done = [];
+  for (const stage of sorted) {
+    if (containersById && subtreeHasActiveWork(containersById, stage.container_id)) {
+      always.push(stage);
+      continue;
+    }
+    if (stage.status === "running" || stage.status === "retrying" || stage.status === "failed" || stage.status === "pending" || stage.status === "blocked") {
+      always.push(stage);
+    } else if (stage.status === "completed" || stage.status === "skipped") {
+      done.push(stage);
+    } else {
+      always.push(stage);
+    }
+  }
+  const visibleDone = done.slice(Math.max(0, done.length - maxVisibleDoneCount));
+  const visible = [...always, ...visibleDone].sort(
+    (a, b) => stageSortKey(a) - stageSortKey(b) || a.scope_key.localeCompare(b.scope_key)
+  );
+  const hiddenCount = stages.length - visible.length;
+  return {
+    visible,
+    hiddenCount,
+    total: stages.length,
+    visibleDoneCount: visibleDone.length,
+    totalDoneCount: done.length
+  };
+}
+function isActiveWorkflowContainer(node, activeLoopSessionId, activeContainerIds) {
+  if (activeContainerIds.includes(node.container_id)) return true;
+  if (node.status === "completed" || node.status === "failed" || node.status === "skipped") {
+    return false;
+  }
+  if (node.loopSessionId && node.loopSessionId === activeLoopSessionId) return true;
+  return false;
+}
+
 // src/view/workflow/workflowContainerLineUtils.ts
 function formatContainerScopeMetaLabel(scope) {
   return scope;
@@ -2785,6 +2992,27 @@ function buildWorkflowRootMeta(node) {
   }
   return withScopeMeta(node, meta);
 }
+function buildEffectiveWorkflowRootMeta(node, containersById) {
+  const meta = [];
+  if (node.pattern) meta.push(node.pattern);
+  const topStages = Object.values(containersById).filter(
+    (n) => n.scope === "stage" && n.parent_container_id === node.container_id
+  );
+  const total = node.display?.progress_total ?? Math.max(topStages.length, 1);
+  let settled = 0;
+  for (const stage of topStages) {
+    if (subtreeHasActiveWork(containersById, stage.container_id)) continue;
+    if (stage.status === "completed" || stage.status === "failed" || stage.status === "skipped") {
+      settled += 1;
+    }
+  }
+  if (total > 0) meta.push(`${settled}/${total}`);
+  return withScopeMeta(node, meta);
+}
+function isWorkflowRootVisuallyActive(node, containersById) {
+  if (node.status === "running" || node.status === "retrying") return true;
+  return subtreeHasActiveWork(containersById, node.container_id);
+}
 function buildStageLineMeta(node) {
   const progress = buildStageProgressMeta(node);
   return withScopeMeta(node, progress ? [progress] : []);
@@ -3008,9 +3236,41 @@ var WorkflowAggregateContainer = memo(
 
 // src/view/workflow/WorkflowParallelGroup.tsx
 import { useCallback as useCallback3, useEffect as useEffect5, useMemo as useMemo4, useState as useState3, memo as memo3 } from "react";
-import { useStore as useStore3 } from "zustand";
 
 // src/core/workflow/workflowParallelGroupMetrics.ts
+var ACTIVE_ITEM_STATUSES = /* @__PURE__ */ new Set([
+  "running",
+  "retrying",
+  "pending",
+  "blocked"
+]);
+function countMetricsFromContainerStatus(items) {
+  let running = 0;
+  let completed = 0;
+  let failed = 0;
+  for (const item of items) {
+    if (ACTIVE_ITEM_STATUSES.has(item.status)) {
+      running += 1;
+      continue;
+    }
+    switch (item.status) {
+      case "completed":
+        completed += 1;
+        break;
+      case "failed":
+        failed += 1;
+        break;
+      default:
+        break;
+    }
+  }
+  return {
+    total: items.length,
+    running,
+    completed,
+    failed
+  };
+}
 function readDisplayMetrics(items) {
   let hasMetrics = false;
   let running = 0;
@@ -3031,34 +3291,29 @@ function readDisplayMetrics(items) {
   if (total <= 0) total = items.length;
   return { total, running, completed, failed };
 }
+function mergeParallelGroupMetrics(fromDisplay, fromStatus, itemCount) {
+  return {
+    total: Math.max(fromDisplay.total, fromStatus.total, itemCount),
+    running: Math.max(fromDisplay.running, fromStatus.running),
+    completed: Math.max(fromDisplay.completed, fromStatus.completed),
+    failed: Math.max(fromDisplay.failed, fromStatus.failed)
+  };
+}
 function computeParallelGroupMetrics(items) {
+  const fromStatus = countMetricsFromContainerStatus(items);
   const fromDisplay = readDisplayMetrics(items);
-  if (fromDisplay) return fromDisplay;
-  let running = 0;
-  let completed = 0;
-  let failed = 0;
+  if (!fromDisplay) return fromStatus;
+  return mergeParallelGroupMetrics(fromDisplay, fromStatus, items.length);
+}
+function parallelGroupHasActiveWork(items, loopTreesBySessionId = {}) {
+  const metrics = computeParallelGroupMetrics(items);
+  if (metrics.running > 0) return true;
   for (const item of items) {
-    switch (item.status) {
-      case "running":
-      case "retrying":
-        running += 1;
-        break;
-      case "completed":
-        completed += 1;
-        break;
-      case "failed":
-        failed += 1;
-        break;
-      default:
-        break;
-    }
+    const loopSessionId = item.loopSessionId;
+    if (!loopSessionId) continue;
+    if (isWorkflowLoopTreeActive(loopTreesBySessionId[loopSessionId])) return true;
   }
-  return {
-    total: items.length,
-    running,
-    completed,
-    failed
-  };
+  return false;
 }
 function formatParallelGroupSummary(metrics) {
   if (metrics.running > 0) {
@@ -3115,47 +3370,6 @@ function WorkflowStageDoneBody({
   );
 }
 
-// src/view/workflow/workflowStageListUtils.ts
-function stageSortKey(node) {
-  return node.display?.progress_current ?? 0;
-}
-function partitionWorkflowStages(stages, maxVisibleDoneCount) {
-  const sorted = [...stages].sort(
-    (a, b) => stageSortKey(a) - stageSortKey(b) || a.scope_key.localeCompare(b.scope_key)
-  );
-  const always = [];
-  const done = [];
-  for (const stage of sorted) {
-    if (stage.status === "running" || stage.status === "retrying" || stage.status === "failed" || stage.status === "pending" || stage.status === "blocked") {
-      always.push(stage);
-    } else if (stage.status === "completed" || stage.status === "skipped") {
-      done.push(stage);
-    } else {
-      always.push(stage);
-    }
-  }
-  const visibleDone = done.slice(Math.max(0, done.length - maxVisibleDoneCount));
-  const visible = [...always, ...visibleDone].sort(
-    (a, b) => stageSortKey(a) - stageSortKey(b) || a.scope_key.localeCompare(b.scope_key)
-  );
-  const hiddenCount = stages.length - visible.length;
-  return {
-    visible,
-    hiddenCount,
-    total: stages.length,
-    visibleDoneCount: visibleDone.length,
-    totalDoneCount: done.length
-  };
-}
-function isActiveWorkflowContainer(node, activeLoopSessionId, activeContainerIds) {
-  if (activeContainerIds.includes(node.container_id)) return true;
-  if (node.status === "completed" || node.status === "failed" || node.status === "skipped") {
-    return false;
-  }
-  if (node.loopSessionId && node.loopSessionId === activeLoopSessionId) return true;
-  return false;
-}
-
 // src/view/workflow/WorkflowStageContainer.tsx
 import { jsx as jsx10, jsxs as jsxs5 } from "react/jsx-runtime";
 function containerTestId(node) {
@@ -3174,10 +3388,10 @@ function WorkflowStageContainerInner({
   virtualized,
   virtualizeThreshold,
   groupParallelTools,
-  suppressHostScopedLoop = false
+  suppressHostScopedLoop = false,
+  subtreeHasActiveWork: subtreeHasActiveWork2 = false
 }) {
   const { verbose } = useSessionViewOptions();
-  const scale = useWorkflowScaleOptions();
   const store = useWorkflowSessionStoreApi();
   const replayHandler = useWorkflowLoopReplayHandler();
   const activeLoopSessionId = useStore2(store, (s) => s.state.activeLoopSessionId);
@@ -3187,31 +3401,30 @@ function WorkflowStageContainerInner({
     (s) => s.state.stageDoneSummariesByContainerId[node.container_id]
   );
   const pinLoopSession = useStore2(store, (s) => s.pinLoopSession);
-  const uiStatus = mapContainerUiStatus(node.status);
   const isRunning = node.status === "running" || node.status === "retrying";
   const isDone = node.status === "completed" || node.status === "failed";
   const isSkipped = node.status === "skipped";
+  const loopStillActive = isWorkflowLoopTreeActive(loopTree);
+  const effectivelyRunning = isRunning || loopStillActive || subtreeHasActiveWork2 && isDone;
+  const effectivelyDone = isDone && !subtreeHasActiveWork2 && !loopStillActive;
+  const uiStatus = mapContainerUiStatus(effectivelyRunning ? "running" : node.status);
   const isActive = isActiveWorkflowContainer(node, activeLoopSessionId, activeContainerIds);
   const doneSummaryFromTree = useMemo3(
     () => loopTree && isDone ? buildStageDoneSummary(loopTree) : null,
     [loopTree, isDone]
   );
   const doneSummary = doneSummaryFromTree ?? (isDone ? cachedSummary : null);
-  const shouldAutoExpandRunning = isRunning && !isSkipped && (!scale.onlyExpandActiveRunning || isActive);
+  const shouldAutoExpandRunning = effectivelyRunning && !isSkipped;
   const [expanded, setExpanded] = useState2(shouldAutoExpandRunning);
   useEffect4(() => {
     if (shouldAutoExpandRunning) {
       setExpanded(true);
       return;
     }
-    if (isRunning && !isSkipped && scale.onlyExpandActiveRunning) {
+    if (effectivelyDone || isSkipped) {
       setExpanded(false);
-      return;
     }
-    if (isDone || isSkipped) {
-      setExpanded(false);
-    }
-  }, [shouldAutoExpandRunning, isRunning, isDone, isSkipped, scale.onlyExpandActiveRunning]);
+  }, [shouldAutoExpandRunning, effectivelyDone, isSkipped]);
   const requestLoopHydration = useCallback2(() => {
     const loopSessionId = node.loopSessionId;
     if (!loopSessionId || loopTree) return;
@@ -3236,17 +3449,18 @@ function WorkflowStageContainerInner({
     window.addEventListener("keydown", onKeyDown);
     return () => window.removeEventListener("keydown", onKeyDown);
   }, [requestLoopHydration]);
+  const testId = containerTestId(node);
+  const isItem = node.scope === "item";
   const displayTitle = node.title;
   const usesScopedLoop = Boolean(node.loopSessionId) && !suppressHostScopedLoop;
-  const runningLoopTree = loopTree ?? (usesScopedLoop && isRunning && isActive ? createWorkflowRunningLoopTree() : void 0);
+  const runningLoopTree = loopTree ?? (usesScopedLoop && effectivelyRunning ? createWorkflowRunningLoopTree() : void 0);
+  const itemLoopVirtualized = isItem ? false : virtualized;
   const staticContentTeaser = !usesScopedLoop && !loopTree && node.content_blocks.length > 0 ? node.content_blocks[node.content_blocks.length - 1].preview : "";
   const teaserText = doneSummary?.teaser || staticContentTeaser;
   const expandHint = doneSummary ? formatTeaserExpandHint(doneSummary.extraLines, verbose) : null;
   const doneLabel = doneSummary?.label ?? "done";
-  const statusLabel = isDone ? doneLabel : suppressHostScopedLoop && isRunning ? "running" : mapContainerStatusLabel(node.status);
-  const testId = containerTestId(node);
-  const isItem = node.scope === "item";
-  const canToggleShell = isRunning && !isSkipped || isDone;
+  const statusLabel = effectivelyDone ? doneLabel : suppressHostScopedLoop && effectivelyRunning ? "running" : mapContainerStatusLabel(effectivelyRunning ? "running" : node.status);
+  const canToggleShell = effectivelyRunning && !isSkipped || effectivelyDone;
   const shellToggle = canToggleShell ? toggleExpanded : void 0;
   return /* @__PURE__ */ jsxs5(
     "div",
@@ -3272,13 +3486,13 @@ function WorkflowStageContainerInner({
             onToggle: shellToggle,
             interactive: canToggleShell,
             expanded,
-            headerTestId: isDone ? `${testId}-done` : void 0,
+            headerTestId: effectivelyDone ? `${testId}-done` : void 0,
             titleTestId: `${testId}-title`,
             descriptionTestId: `${testId}-desc`,
             className: isItem ? "lax-workflow-parallel-item__header" : void 0
           }
         ),
-        isDone && !expanded && (teaserText || expandHint) ? /* @__PURE__ */ jsxs5("div", { className: "lax-workflow-stage-container__done-teaser", children: [
+        effectivelyDone && !expanded && (teaserText || expandHint) ? /* @__PURE__ */ jsxs5("div", { className: "lax-workflow-stage-container__done-teaser", children: [
           teaserText ? /* @__PURE__ */ jsx10(
             "span",
             {
@@ -3304,7 +3518,7 @@ function WorkflowStageContainerInner({
             ))
           }
         ) : null,
-        expanded && isRunning && runningLoopTree ? /* @__PURE__ */ jsx10(
+        expanded && effectivelyRunning && runningLoopTree ? /* @__PURE__ */ jsx10(
           "div",
           {
             className: isItem ? "lax-workflow-parallel-item__body lax-workflow-stage-container__body" : "lax-workflow-stage-container__body",
@@ -3314,20 +3528,20 @@ function WorkflowStageContainerInner({
               {
                 tree: runningLoopTree,
                 showTaskListFooter: false,
-                virtualized,
+                virtualized: itemLoopVirtualized,
                 virtualizeThreshold,
                 groupParallelTools
               }
             )
           }
         ) : null,
-        isDone && expanded ? loopTree ? /* @__PURE__ */ jsx10(
+        effectivelyDone && expanded ? loopTree ? /* @__PURE__ */ jsx10(
           WorkflowStageDoneBody,
           {
             node,
             loopTree,
             testId: bodyTestId(node),
-            virtualized,
+            virtualized: itemLoopVirtualized,
             virtualizeThreshold,
             groupParallelTools,
             bodyClassName: isItem ? "lax-workflow-parallel-item__body lax-workflow-stage-container__body" : "lax-workflow-stage-container__body"
@@ -3349,6 +3563,7 @@ function WorkflowStageContainerInner({
 }
 function workflowStageContainerPropsEqual(prev, next) {
   if (prev.suppressHostScopedLoop !== next.suppressHostScopedLoop) return false;
+  if (prev.subtreeHasActiveWork !== next.subtreeHasActiveWork) return false;
   if (prev.loopTree !== next.loopTree) return false;
   if (!workflowContainerViewPropsEqual(prev, next)) return false;
   return workflowContainerNodeVisualEqual(prev.node, next.node);
@@ -3370,34 +3585,23 @@ function WorkflowParallelGroupInner({
   virtualizeThreshold,
   groupParallelTools
 }) {
-  const scale = useWorkflowScaleOptions();
-  const store = useWorkflowSessionStoreApi();
-  const activeLoopSessionId = useStore3(store, (s) => s.state.activeLoopSessionId);
-  const activeContainerIds = useStore3(store, (s) => s.state.containerTree.activeContainerIds);
   const metrics = useMemo4(() => computeParallelGroupMetrics(items), [items]);
   const summary = formatParallelGroupSummary(metrics);
-  const hasRunning = metrics.running > 0;
-  const isNestedGroup = groupKey.includes(">");
-  const hasActiveItem = useMemo4(
-    () => items.some((item) => isActiveWorkflowContainer(item, activeLoopSessionId, activeContainerIds)),
-    [items, activeLoopSessionId, activeContainerIds]
+  const hasActiveWork = useMemo4(
+    () => parallelGroupHasActiveWork(items, loopTreesBySessionId),
+    [items, loopTreesBySessionId]
   );
   const allSettled = metrics.total > 0 && metrics.running === 0 && metrics.completed + metrics.failed >= metrics.total;
-  const shouldAutoExpandGroup = hasRunning && (isNestedGroup || !scale.onlyExpandActiveRunning || hasActiveItem);
-  const [expanded, setExpanded] = useState3(shouldAutoExpandGroup);
+  const [expanded, setExpanded] = useState3(hasActiveWork);
   useEffect5(() => {
-    if (shouldAutoExpandGroup) {
+    if (hasActiveWork) {
       setExpanded(true);
       return;
     }
-    if (hasRunning && scale.onlyExpandActiveRunning) {
-      setExpanded(false);
-      return;
-    }
     if (allSettled) {
       setExpanded(false);
     }
-  }, [shouldAutoExpandGroup, hasRunning, allSettled, scale.onlyExpandActiveRunning]);
+  }, [hasActiveWork, allSettled]);
   const toggleExpanded = useCallback3(() => {
     setExpanded((prev) => !prev);
   }, []);
@@ -3434,7 +3638,7 @@ function WorkflowParallelGroupInner({
           WorkflowExpandTrigger,
           {
             summary,
-            expandHint: !expanded && hasRunning ? "(click to expand agents)" : !expanded && !hasRunning ? "(ctrl+o to expand)" : null,
+            expandHint: !expanded && hasActiveWork ? "(click to expand agents)" : !expanded && !hasActiveWork ? "(ctrl+o to expand)" : null,
             onToggle: toggleExpanded,
             expanded,
             testId: `lax-workflow-parallel-group-summary-${groupKey}`,
@@ -3469,11 +3673,14 @@ var WorkflowParallelGroup = memo3(WorkflowParallelGroupInner, workflowParallelGr
 import { jsx as jsx12, jsxs as jsxs7 } from "react/jsx-runtime";
 function WorkflowRootContainer({
   node,
-  children
+  children,
+  containersById
 }) {
   const { verbose } = useSessionViewOptions();
-  const uiStatus = mapContainerUiStatus(node.status);
-  const statusLabel = mapWorkflowRootStatusLabel(node.status);
+  const visuallyActive = containersById ? isWorkflowRootVisuallyActive(node, containersById) : node.status === "running" || node.status === "retrying";
+  const uiStatus = mapContainerUiStatus(visuallyActive ? "running" : node.status);
+  const statusLabel = mapWorkflowRootStatusLabel(visuallyActive ? "running" : node.status);
+  const meta = containersById ? buildEffectiveWorkflowRootMeta(node, containersById) : buildWorkflowRootMeta(node);
   return /* @__PURE__ */ jsxs7("div", { className: "lax-workflow-root", "data-testid": "lax-workflow-root", children: [
     /* @__PURE__ */ jsx12(
       WorkflowContainerLine,
@@ -3482,7 +3689,7 @@ function WorkflowRootContainer({
         title: node.title,
         description: node.subtitle || void 0,
         uiStatus,
-        meta: buildWorkflowRootMeta(node),
+        meta,
         statusLabel,
         titleTestId: "lax-workflow-root-title",
         descriptionTestId: "lax-workflow-root-desc",
@@ -3610,31 +3817,36 @@ function buildDoneSummary(node) {
 function WorkflowSubworkflowContainer({
   node,
   depth = 1,
-  children
+  children,
+  subtreeHasActiveWork: subtreeHasActiveWork2 = false
 }) {
   const { verbose } = useSessionViewOptions();
-  const uiStatus = mapContainerUiStatus(node.status);
   const isRunning = node.status === "running" || node.status === "retrying";
   const isDone = node.status === "completed" || node.status === "failed";
+  const effectivelyRunning = isRunning || subtreeHasActiveWork2;
+  const effectivelyDone = isDone && !subtreeHasActiveWork2;
+  const uiStatus = mapContainerUiStatus(
+    effectivelyRunning ? "running" : node.status
+  );
   const testId = `lax-workflow-subworkflow-${node.scope_key}`;
   const teaser = useMemo6(() => {
     if (node.content_blocks.length === 0) return "";
     return node.content_blocks[node.content_blocks.length - 1]?.preview ?? "";
   }, [node.content_blocks]);
-  const [expanded, setExpanded] = useState5(isRunning);
+  const [expanded, setExpanded] = useState5(effectivelyRunning);
   useEffect7(() => {
-    if (isRunning) setExpanded(true);
-    if (isDone) setExpanded(false);
-  }, [isRunning, isDone]);
+    if (effectivelyRunning) setExpanded(true);
+    else if (effectivelyDone) setExpanded(false);
+  }, [effectivelyRunning, effectivelyDone]);
   const toggleExpanded = useCallback5(() => {
     setExpanded((prev) => !prev);
   }, []);
-  const shellToggle = isRunning ? toggleExpanded : isDone && expanded ? toggleExpanded : void 0;
+  const shellToggle = effectivelyRunning ? toggleExpanded : effectivelyDone && expanded ? toggleExpanded : void 0;
   const depthStyle = {
     "--lax-workflow-depth": depth,
     "--lax-workflow-indent-unit": workflowIndentUnit(depth)
   };
-  const showShellLine = !(isDone && !expanded);
+  const showShellLine = !(effectivelyDone && !expanded);
   return /* @__PURE__ */ jsxs9(
     "div",
     {
@@ -3651,9 +3863,11 @@ function WorkflowSubworkflowContainer({
             description: node.subtitle || void 0,
             uiStatus,
             meta: buildWorkflowRootMeta(node),
-            statusLabel: mapWorkflowRootStatusLabel(node.status),
+            statusLabel: mapWorkflowRootStatusLabel(
+              effectivelyRunning ? "running" : node.status
+            ),
             onToggle: shellToggle,
-            interactive: isRunning,
+            interactive: effectivelyRunning,
             expanded,
             titleTestId: `${testId}-title`,
             descriptionTestId: `${testId}-desc`,
@@ -3673,13 +3887,13 @@ function WorkflowSubworkflowContainer({
         !expanded && children ? /* @__PURE__ */ jsx14(
           WorkflowExpandTrigger,
           {
-            summary: `${node.title} \xB7 ${isDone ? buildDoneSummary(node) : "running"}`,
+            summary: `${node.title} \xB7 ${effectivelyDone ? buildDoneSummary(node) : "running"}`,
             summaryTestId: `${testId}-title`,
             teaser: teaser || void 0,
-            expandHint: isRunning ? "(click to expand subworkflow)" : void 0,
+            expandHint: effectivelyRunning ? "(click to expand subworkflow)" : void 0,
             onToggle: () => setExpanded(true),
             expanded: false,
-            testId: isDone ? `${testId}-done` : `${testId}-running-collapsed`,
+            testId: effectivelyDone ? `${testId}-done` : `${testId}-running-collapsed`,
             teaserTestId: teaser ? `${testId}-teaser` : void 0
           }
         ) : null
@@ -3705,6 +3919,7 @@ function collectSiblingItems(node, containersById) {
 }
 function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, viewProps, options) {
   const loopTree = resolveLoopTree2(node, loopTreesBySessionId);
+  const subtreeActive = options?.subtreeHasActiveWork ?? subtreeHasActiveWork(containerTree.containersById, node.container_id);
   switch (node.scope) {
     case "aggregate":
       return /* @__PURE__ */ jsx15(
@@ -3737,6 +3952,7 @@ function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, v
           loopTree,
           depth,
           suppressHostScopedLoop: options?.suppressHostScopedLoop,
+          subtreeHasActiveWork: subtreeActive,
           ...viewProps
         },
         node.container_id
@@ -3790,6 +4006,10 @@ function renderContainerNode(node, containerTree, loopTreesBySessionId, viewProp
       {
         node,
         depth,
+        subtreeHasActiveWork: subtreeHasActiveWork(
+          containerTree.containersById,
+          node.container_id
+        ),
         children: hasNested ? nested : null
       },
       node.container_id
@@ -3851,8 +4071,8 @@ function WorkflowStageListPanel({
   const [extraCompletedPages, setExtraCompletedPages] = useState6(0);
   const maxVisibleDoneCount = scale.completedStagePageSize + extraCompletedPages * scale.completedStagePageSize;
   const partition = useMemo7(
-    () => partitionWorkflowStages(stages, maxVisibleDoneCount),
-    [stages, maxVisibleDoneCount]
+    () => partitionWorkflowStages(stages, maxVisibleDoneCount, containerTree.containersById),
+    [stages, maxVisibleDoneCount, containerTree.containersById]
   );
   if (partition.visible.length === 0) return null;
   const showMoreCount = Math.min(
@@ -3950,17 +4170,25 @@ function WorkflowChrome({
       }
     ) });
   }
-  return /* @__PURE__ */ jsx15("div", { className: "lax-workflow-chrome", "data-testid": "lax-workflow-chrome", children: rootNodes.map((workflowNode) => /* @__PURE__ */ jsx15(WorkflowRootContainer, { node: workflowNode, children: renderWorkflowChildren(
-    workflowNode,
-    containerTree,
-    loopTreesBySessionId,
-    viewProps
-  ) }, workflowNode.container_id)) });
+  return /* @__PURE__ */ jsx15("div", { className: "lax-workflow-chrome", "data-testid": "lax-workflow-chrome", children: rootNodes.map((workflowNode) => /* @__PURE__ */ jsx15(
+    WorkflowRootContainer,
+    {
+      node: workflowNode,
+      containersById: containerTree.containersById,
+      children: renderWorkflowChildren(
+        workflowNode,
+        containerTree,
+        loopTreesBySessionId,
+        viewProps
+      )
+    },
+    workflowNode.container_id
+  )) });
 }
 
 // src/view/workflow/WorkflowTaskListFooter.tsx
 import { useMemo as useMemo8, useState as useState7 } from "react";
-import { useStore as useStore4 } from "zustand";
+import { useStore as useStore3 } from "zustand";
 
 // src/view/workflow/workflowTaskListMerge.ts
 var STATUS_ORDER = {
@@ -4040,7 +4268,7 @@ function MessageResponse({ children }) {
 }
 function WorkflowTaskListFooter() {
   const store = useWorkflowSessionStoreApi();
-  const state = useStore4(store, (s) => s.state);
+  const state = useStore3(store, (s) => s.state);
   const [expanded, setExpanded] = useState7(false);
   const footerState = useMemo8(() => deriveWorkflowTaskFooterState(state), [state]);
   const visible = shouldShowWorkflowTaskListFooter(state);
@@ -4080,7 +4308,7 @@ function WorkflowTaskListFooter() {
 }
 
 // src/view/workflow/WorkflowSession.tsx
-import { useStore as useStore5 } from "zustand";
+import { useStore as useStore4 } from "zustand";
 import { jsx as jsx17, jsxs as jsxs12 } from "react/jsx-runtime";
 function WorkflowSession({
   source,
@@ -4103,6 +4331,7 @@ function WorkflowSession({
   groupParallelTools = false,
   workflowScale,
   onRequestLoopReplay,
+  onStreamComplete,
   children
 }) {
   const storeRef = useRef3(null);
@@ -4152,39 +4381,46 @@ function WorkflowSession({
       workspaceRoot
     ]
   );
-  const containerTree = useStore5(
+  const containerTree = useStore4(
     storeRef.current,
     (s) => s.state.containerTree
   );
-  const loopTreesBySessionId = useStore5(
+  const loopTreesBySessionId = useStore4(
     storeRef.current,
     (s) => s.state.loopTreesBySessionId
   );
-  const sessionStatus = useStore5(
+  const sessionStatus = useStore4(
     storeRef.current,
     (s) => s.state.status
   );
-  const internalErrors = useStore5(
+  const internalErrors = useStore4(
     storeRef.current,
     (s) => s.state.internalErrors
   );
+  const onErrorRef = useRef3(onError);
+  const onStreamCompleteRef = useRef3(onStreamComplete);
+  onErrorRef.current = onError;
+  onStreamCompleteRef.current = onStreamComplete;
   useEffect8(() => {
     const store = storeRef.current;
     const controller = new AbortController();
     void source.start((event, ctx) => {
       store.getState().applyEvent(event, ctx);
-    }, controller.signal).catch((err) => {
+    }, controller.signal).then(() => {
+      store.getState().markStreamComplete();
+      onStreamCompleteRef.current?.();
+    }).catch((err) => {
       if (err instanceof SseTransportTerminalError) {
         store.getState().markAsError(err);
-        onError?.(err);
+        onErrorRef.current?.(err);
       } else if (err instanceof Error) {
-        onError?.(err);
+        onErrorRef.current?.(err);
       }
     });
     return () => {
       controller.abort();
     };
-  }, [source, onError]);
+  }, [source]);
   return /* @__PURE__ */ jsx17(WorkflowSessionStoreContext.Provider, { value: storeRef.current, children: /* @__PURE__ */ jsx17(WorkflowScaleContext.Provider, { value: scaleOptions, children: /* @__PURE__ */ jsx17(WorkflowLoopReplayProvider, { onRequestLoopReplay, children: /* @__PURE__ */ jsx17(InteractionBusContext.Provider, { value: busRef, children: /* @__PURE__ */ jsx17(NodeRegistryContext.Provider, { value: registryRef, children: /* @__PURE__ */ jsx17(MarkdownRendererContext.Provider, { value: markdownRenderer, children: /* @__PURE__ */ jsx17(ToolDisplayOptionsContext.Provider, { value: toolDisplayRef, children: /* @__PURE__ */ jsx17(SessionViewOptionsContext.Provider, { value: sessionViewRef, children: /* @__PURE__ */ jsxs12(
     "div",
     {
@@ -4407,6 +4643,7 @@ export {
   createReplaySource,
   createSessionStore,
   createWorkflowSessionStore,
+  deriveLoopSessionIdFromAgentEvent,
   displayPath,
   entryKey,
   filterAgentLoopReplayEvents,