langchain_agentx_stream_ui 0.1.1 → 0.1.3

package/dist/index.js

@@ -19,7 +19,7 @@ import {
   createMultiSessionStore,
   markEventIdSeen,
   shouldSkipDuplicateEvent
-} from "./chunk-R35BQYAY.js";
+} from "./chunk-A4W6B7C3.js";
 import {
   DefaultToolBody,
   InteractionBusContext,
@@ -107,7 +107,7 @@ import {
   useSessionStatus,
   useSessionViewOptions,
   useTimeline
-} from "./chunk-Q4HRD7EM.js";
+} from "./chunk-3B4WC67E.js";
 import {
   Agent,
   AgentToolBody,
@@ -653,7 +653,7 @@ function AgentSession({
 }
 
 // src/view/workflow/WorkflowSession.tsx
-import { useEffect as useEffect8, useMemo as useMemo9, useRef as useRef4 } from "react";
+import { useEffect as useEffect8, useMemo as useMemo9, useRef as useRef3 } from "react";
 
 // src/core/context/WorkflowSessionContext.tsx
 import { createContext, useContext } from "react";
@@ -693,10 +693,11 @@ import { createContext as createContext3, useContext as useContext3 } from "reac
 // src/view/workflow/workflowScaleOptions.ts
 var DEFAULT_WORKFLOW_SCALE_OPTIONS = {
   stageVirtualizeThreshold: 40,
-  stagePageSize: 25,
+  completedStagePageSize: 25,
   evictCompletedLoops: true,
   onlyExpandActiveRunning: true,
-  parallelItemVirtualizeThreshold: 40
+  parallelItemVirtualizeThreshold: 40,
+  maxHydratedCompletedLoops: 4
 };
 
 // src/core/context/WorkflowScaleContext.tsx
@@ -749,6 +750,7 @@ function createEmptyWorkflowSessionState() {
     workflowProgress: createEmptyWorkflowProgress(),
     loopTreesBySessionId: {},
     stageDoneSummariesByContainerId: {},
+    hotCompletedLoopSessionIds: [],
     activeLoopSessionId: null,
     openScopeStack: [],
     internalErrors: []
@@ -1247,6 +1249,13 @@ function loopTaskScopeForStructureOpen(eventType) {
 }
 
 // src/core/workflowStructureEvents.ts
+var WORKFLOW_STRUCTURE_EVENT_PREFIXES = [
+  "workflow-",
+  "stage-",
+  "parallel-",
+  "route-",
+  "subworkflow-"
+];
 var WORKFLOW_STRUCTURE_EVENT_TYPES = /* @__PURE__ */ new Set([
   "workflow-start",
   "workflow-end",
@@ -1266,8 +1275,20 @@ var WORKFLOW_STRUCTURE_EVENT_TYPES = /* @__PURE__ */ new Set([
   "route-branch-done",
   "route-branch-failed"
 ]);
+var AGENT_LOOP_REPLAY_EVENT_TYPES = new Set(
+  Object.keys(DEFAULT_EVENT_TIERS)
+);
 function isWorkflowStructureEventType(eventType) {
-  return WORKFLOW_STRUCTURE_EVENT_TYPES.has(eventType);
+  if (WORKFLOW_STRUCTURE_EVENT_TYPES.has(eventType)) return true;
+  return WORKFLOW_STRUCTURE_EVENT_PREFIXES.some((prefix) => eventType.startsWith(prefix));
+}
+function isAgentLoopReplayEventType(eventType) {
+  return AGENT_LOOP_REPLAY_EVENT_TYPES.has(eventType);
+}
+function filterAgentLoopReplayEvents(events) {
+  return events.filter(
+    (event) => isAgentLoopReplayEventType(event.event_type) && !isWorkflowStructureEventType(event.event_type)
+  );
 }
 
 // src/core/workflowReducer.ts
@@ -1419,16 +1440,17 @@ function findContainerByLoopSession(tree, loopSessionId) {
   return null;
 }
 var WORKFLOW_DEFAULT_LOOP_SESSION_ID = "workflow-loop-default";
-function resolveAgentLoopSessionId(event) {
+function resolveAgentLoopSessionId(event, activeLoopSessionId) {
   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;
+  if (activeLoopSessionId) return activeLoopSessionId;
   return WORKFLOW_DEFAULT_LOOP_SESSION_ID;
 }
-function bindLoopSessionToContainerTree(tree, event) {
-  const sessionId = resolveAgentLoopSessionId(event);
+function bindLoopSessionToContainerTree(tree, event, activeLoopSessionId) {
+  const sessionId = resolveAgentLoopSessionId(event, activeLoopSessionId);
   if (!sessionId) return tree;
   if (findContainerByLoopSession(tree, sessionId)) return tree;
   const data = event.data ?? {};
@@ -1534,8 +1556,12 @@ function reduceWorkflowSession(state, event, eventIndex, options) {
   if (isWorkflowStructureEventType(event.event_type)) {
     return applyStructureEvent(state, event);
   }
-  const loopSessionId = resolveAgentLoopSessionId(event) ?? WORKFLOW_DEFAULT_LOOP_SESSION_ID;
-  let containerTree = bindLoopSessionToContainerTree(state.containerTree, event);
+  const loopSessionId = resolveAgentLoopSessionId(event, state.activeLoopSessionId) ?? WORKFLOW_DEFAULT_LOOP_SESSION_ID;
+  let containerTree = bindLoopSessionToContainerTree(
+    state.containerTree,
+    event,
+    state.activeLoopSessionId
+  );
   containerTree = applyStructureToContainerTree(containerTree, event);
   let next = ensureLoopTree({ ...state, containerTree }, loopSessionId);
   const loopTree = next.loopTreesBySessionId[loopSessionId];
@@ -1640,6 +1666,9 @@ function collectProtectedLoopSessionIds(state, pinnedLoopSessionIds) {
 }
 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 = {
@@ -1662,6 +1691,45 @@ function evictInactiveLoopTrees(state, pinnedLoopSessionIds) {
     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
 function safeReduceWorkflow(state, event, eventIndex, options) {
@@ -1691,6 +1759,7 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
   const initialEventCount = storeOptions?.initialEventCount ?? 0;
   const seenEventIds = new Set(storeOptions?.initialSeenEventIds);
   const evictCompletedLoops = storeOptions?.evictCompletedLoops ?? true;
+  const maxHydratedCompletedLoops = storeOptions?.maxHydratedCompletedLoops ?? 4;
   const pinnedLoopSessionIds = {};
   const finalizeState = (reduced) => {
     if (!evictCompletedLoops) return reduced;
@@ -1747,29 +1816,42 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
         }
       });
     },
+    /**
+     * P2：将 completed loop 加入 LRU 热缓存并淘汰溢出。
+     * running 保护由 collectProtectedLoopSessionIds 按状态自动覆盖，不再依赖永久 pin。
+     */
     pinLoopSession(loopSessionId) {
-      pinnedLoopSessionIds[loopSessionId] = true;
-      set((current) => ({
-        state: finalizeState(current.state)
-      }));
+      set((current) => {
+        let next = touchCompletedLoopSession(current.state, loopSessionId);
+        next = evictCompletedLoopOverflow(next, maxHydratedCompletedLoops);
+        next = finalizeState(next);
+        return { state: next };
+      });
     },
+    /**
+     * P2：宿主 replay 后灌入 loop 树，touch LRU 并淘汰溢出。
+     * completed loop 受 hotCompletedLoopSessionIds 保护，不再永久 pin。
+     */
     hydrateLoopSession(loopSessionId, events) {
-      pinnedLoopSessionIds[loopSessionId] = true;
       const { state, eventCount } = get();
       let tree = state.loopTreesBySessionId[loopSessionId] ?? createEmptyTree();
       let idx = eventCount;
-      for (const event of events) {
+      for (const event of filterAgentLoopReplayEvents(events)) {
         tree = reduceTree(tree, event, idx, reduceOpts);
         idx += 1;
       }
-      set({
-        state: finalizeState({
-          ...state,
+      set((current) => {
+        let next = {
+          ...current.state,
           loopTreesBySessionId: {
-            ...state.loopTreesBySessionId,
+            ...current.state.loopTreesBySessionId,
             [loopSessionId]: tree
           }
-        })
+        };
+        next = touchCompletedLoopSession(next, loopSessionId);
+        next = evictCompletedLoopOverflow(next, maxHydratedCompletedLoops);
+        next = finalizeState(next);
+        return { state: next };
       });
     }
   }));
@@ -2246,10 +2328,7 @@ function WorkflowStageDoneBody({
 function stageSortKey(node) {
   return node.display?.progress_current ?? 0;
 }
-function partitionWorkflowStages(stages, pageSize, showAll) {
-  if (showAll || stages.length <= pageSize) {
-    return { visible: stages, hiddenCount: 0, total: stages.length };
-  }
+function partitionWorkflowStages(stages, maxVisibleDoneCount) {
   const sorted = [...stages].sort(
     (a, b) => stageSortKey(a) - stageSortKey(b) || a.scope_key.localeCompare(b.scope_key)
   );
@@ -2264,12 +2343,18 @@ function partitionWorkflowStages(stages, pageSize, showAll) {
       always.push(stage);
     }
   }
-  const recentDone = done.slice(Math.max(0, done.length - pageSize));
-  const visible = [...always, ...recentDone].sort(
+  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 };
+  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;
@@ -2468,66 +2553,8 @@ function WorkflowStageContainer({
   );
 }
 
-// src/view/workflow/WorkflowVirtualRowList.tsx
-import { useRef as useRef3 } from "react";
-import { useVirtualizer } from "@tanstack/react-virtual";
-import { jsx as jsx10 } from "react/jsx-runtime";
-var DEFAULT_STAGE_ROW_ESTIMATE = 56;
-function WorkflowVirtualRowList({
-  count,
-  estimateSize = DEFAULT_STAGE_ROW_ESTIMATE,
-  testId,
-  className,
-  renderRow
-}) {
-  const parentRef = useRef3(null);
-  const virtualizer = useVirtualizer({
-    count,
-    getScrollElement: () => parentRef.current,
-    estimateSize: () => estimateSize,
-    overscan: 8
-  });
-  if (count === 0) return null;
-  return /* @__PURE__ */ jsx10(
-    "div",
-    {
-      ref: parentRef,
-      className: className ? `lax-workflow-virtual-list ${className}` : "lax-workflow-virtual-list",
-      "data-testid": testId,
-      children: /* @__PURE__ */ jsx10(
-        "div",
-        {
-          className: "lax-workflow-virtual-list__inner",
-          style: {
-            height: `${virtualizer.getTotalSize()}px`,
-            position: "relative",
-            width: "100%"
-          },
-          children: virtualizer.getVirtualItems().map((virtualRow) => /* @__PURE__ */ jsx10(
-            "div",
-            {
-              className: "lax-workflow-virtual-list__item",
-              "data-index": virtualRow.index,
-              ref: virtualizer.measureElement,
-              style: {
-                position: "absolute",
-                top: 0,
-                left: 0,
-                width: "100%",
-                transform: `translateY(${virtualRow.start}px)`
-              },
-              children: renderRow(virtualRow.index)
-            },
-            virtualRow.key
-          ))
-        }
-      )
-    }
-  );
-}
-
 // src/view/workflow/WorkflowParallelGroup.tsx
-import { jsx as jsx11, jsxs as jsxs6 } from "react/jsx-runtime";
+import { jsx as jsx10, jsxs as jsxs6 } from "react/jsx-runtime";
 function resolveLoopTree(node, loopTreesBySessionId) {
   if (!node.loopSessionId) return void 0;
   return loopTreesBySessionId[node.loopSessionId];
@@ -2579,10 +2606,9 @@ function WorkflowParallelGroup({
     window.addEventListener("keydown", onKeyDown);
     return () => window.removeEventListener("keydown", onKeyDown);
   }, []);
-  const useVirtualItems = items.length >= scale.parallelItemVirtualizeThreshold;
   const renderItem = (index) => {
     const item = items[index];
-    return /* @__PURE__ */ jsx11(
+    return /* @__PURE__ */ jsx10(
       WorkflowStageContainer,
       {
         node: item,
@@ -2601,7 +2627,7 @@ function WorkflowParallelGroup({
       className: "lax-workflow-parallel-group",
       "data-testid": `lax-workflow-parallel-group-${groupKey}`,
       children: [
-        /* @__PURE__ */ jsx11(
+        /* @__PURE__ */ jsx10(
           WorkflowExpandTrigger,
           {
             summary,
@@ -2612,22 +2638,14 @@ function WorkflowParallelGroup({
             className: "lax-workflow-parallel-group__summary"
           }
         ),
-        expanded ? useVirtualItems ? /* @__PURE__ */ jsx11(
-          WorkflowVirtualRowList,
-          {
-            count: items.length,
-            testId: "lax-workflow-parallel-virtual-list",
-            className: "lax-workflow-parallel-list lax-workflow-parallel-list--virtual",
-            renderRow: renderItem
-          }
-        ) : /* @__PURE__ */ jsx11("div", { className: "lax-workflow-parallel-list", "data-testid": "lax-workflow-parallel-list", children: items.map((_, index) => renderItem(index)) }) : null
+        expanded ? /* @__PURE__ */ jsx10("div", { className: "lax-workflow-parallel-list", "data-testid": "lax-workflow-parallel-list", children: items.map((_, index) => renderItem(index)) }) : null
       ]
     }
   );
 }
 
 // src/view/workflow/WorkflowRootContainer.tsx
-import { jsx as jsx12, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx11, jsxs as jsxs7 } from "react/jsx-runtime";
 function WorkflowRootContainer({
   node,
   children
@@ -2636,7 +2654,7 @@ function WorkflowRootContainer({
   const uiStatus = mapContainerUiStatus(node.status);
   const statusLabel = mapWorkflowRootStatusLabel(node.status);
   return /* @__PURE__ */ jsxs7("div", { className: "lax-workflow-root", "data-testid": "lax-workflow-root", children: [
-    /* @__PURE__ */ jsx12(
+    /* @__PURE__ */ jsx11(
       WorkflowContainerLine,
       {
         variant: "root",
@@ -2651,13 +2669,13 @@ function WorkflowRootContainer({
         verbose
       }
     ),
-    children ? /* @__PURE__ */ jsx12("div", { className: "lax-workflow-container-tree", children }) : null
+    children ? /* @__PURE__ */ jsx11("div", { className: "lax-workflow-container-tree", children }) : null
   ] });
 }
 
 // src/view/workflow/WorkflowRouteContainer.tsx
 import { useCallback as useCallback4, useEffect as useEffect6, useMemo as useMemo5, useState as useState4 } from "react";
-import { jsx as jsx13, jsxs as jsxs8 } from "react/jsx-runtime";
+import { jsx as jsx12, jsxs as jsxs8 } from "react/jsx-runtime";
 function WorkflowRouteContainer({
   node,
   loopTree,
@@ -2703,7 +2721,7 @@ function WorkflowRouteContainer({
       "data-container-scope": "branch",
       style: depthStyle,
       children: [
-        /* @__PURE__ */ jsx13(
+        /* @__PURE__ */ jsx12(
           WorkflowContainerLine,
           {
             title: node.title,
@@ -2718,7 +2736,7 @@ function WorkflowRouteContainer({
             descriptionTestId: `${testId}-desc`
           }
         ),
-        !isSkipped && expanded && isRunning && loopTree ? /* @__PURE__ */ jsx13("div", { className: "lax-workflow-stage-container__body", "data-testid": `${testId}-body`, children: /* @__PURE__ */ jsx13(
+        !isSkipped && expanded && isRunning && loopTree ? /* @__PURE__ */ jsx12("div", { className: "lax-workflow-stage-container__body", "data-testid": `${testId}-body`, children: /* @__PURE__ */ jsx12(
           AgentLoopView,
           {
             tree: loopTree,
@@ -2728,7 +2746,7 @@ function WorkflowRouteContainer({
             groupParallelTools
           }
         ) }) : null,
-        !isSkipped && isDone && !expanded ? /* @__PURE__ */ jsx13(
+        !isSkipped && isDone && !expanded ? /* @__PURE__ */ jsx12(
           WorkflowExpandTrigger,
           {
             summary: `${node.title} \xB7 ${doneLabel}`,
@@ -2740,7 +2758,7 @@ function WorkflowRouteContainer({
             teaserTestId: teaserText ? `${testId}-teaser` : void 0
           }
         ) : null,
-        !isSkipped && isDone && expanded ? /* @__PURE__ */ jsx13(
+        !isSkipped && isDone && expanded ? /* @__PURE__ */ jsx12(
           WorkflowStageDoneBody,
           {
             node,
@@ -2758,7 +2776,7 @@ function WorkflowRouteContainer({
 
 // src/view/workflow/WorkflowSubworkflowContainer.tsx
 import { useCallback as useCallback5, useEffect as useEffect7, useMemo as useMemo6, useState as useState5 } from "react";
-import { jsx as jsx14, jsxs as jsxs9 } from "react/jsx-runtime";
+import { jsx as jsx13, jsxs as jsxs9 } from "react/jsx-runtime";
 function buildDoneSummary(node) {
   const display = node.display;
   if (display?.progress_total != null) {
@@ -2804,7 +2822,7 @@ function WorkflowSubworkflowContainer({
       "data-container-scope": "subworkflow",
       style: depthStyle,
       children: [
-        showShellLine ? /* @__PURE__ */ jsx14(
+        showShellLine ? /* @__PURE__ */ jsx13(
           WorkflowContainerLine,
           {
             variant: "root",
@@ -2822,7 +2840,7 @@ function WorkflowSubworkflowContainer({
             verbose
           }
         ) : null,
-        expanded && children ? /* @__PURE__ */ jsx14(
+        expanded && children ? /* @__PURE__ */ jsx13(
           "div",
           {
             className: "lax-workflow-subworkflow-boundary",
@@ -2831,7 +2849,7 @@ function WorkflowSubworkflowContainer({
             children
           }
         ) : null,
-        isDone && !expanded ? /* @__PURE__ */ jsx14(
+        isDone && !expanded ? /* @__PURE__ */ jsx13(
           WorkflowExpandTrigger,
           {
             summary: `${node.title} \xB7 ${buildDoneSummary(node)}`,
@@ -2849,7 +2867,7 @@ function WorkflowSubworkflowContainer({
 }
 
 // src/view/workflow/WorkflowChrome.tsx
-import { Fragment as Fragment2, jsx as jsx15, jsxs as jsxs10 } from "react/jsx-runtime";
+import { Fragment as Fragment2, jsx as jsx14, jsxs as jsxs10 } from "react/jsx-runtime";
 function collectChildContainers(containersById, parentId) {
   return Object.values(containersById).filter((node) => node.parent_container_id === parentId).sort((a, b) => a.scope_key.localeCompare(b.scope_key));
 }
@@ -2867,7 +2885,7 @@ function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, v
   const loopTree = resolveLoopTree2(node, loopTreesBySessionId);
   switch (node.scope) {
     case "aggregate":
-      return /* @__PURE__ */ jsx15(
+      return /* @__PURE__ */ jsx14(
         WorkflowAggregateContainer,
         {
           node,
@@ -2879,7 +2897,7 @@ function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, v
         node.container_id
       );
     case "branch":
-      return /* @__PURE__ */ jsx15(
+      return /* @__PURE__ */ jsx14(
         WorkflowRouteContainer,
         {
           node,
@@ -2890,7 +2908,7 @@ function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, v
         node.container_id
       );
     default:
-      return /* @__PURE__ */ jsx15(
+      return /* @__PURE__ */ jsx14(
         WorkflowStageContainer,
         {
           node,
@@ -2904,7 +2922,7 @@ function renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, v
 }
 function renderItemSiblings(items, groupKey, depth, containerTree, loopTreesBySessionId, viewProps) {
   if (items.length >= 2) {
-    return /* @__PURE__ */ jsx15(
+    return /* @__PURE__ */ jsx14(
       WorkflowParallelGroup,
       {
         groupKey,
@@ -2939,7 +2957,7 @@ function renderContainerNode(node, containerTree, loopTreesBySessionId, viewProp
   );
   const hasNested = collectChildContainers(containerTree.containersById, node.container_id).length > 0;
   if (node.scope === "subworkflow") {
-    return /* @__PURE__ */ jsx15(
+    return /* @__PURE__ */ jsx14(
       WorkflowSubworkflowContainer,
       {
         node,
@@ -2957,7 +2975,7 @@ function renderContainerNode(node, containerTree, loopTreesBySessionId, viewProp
         "data-testid": `lax-workflow-nested-${node.scope}-${node.scope_key}`,
         children: [
           renderLeafContainer(node, depth, containerTree, loopTreesBySessionId, viewProps),
-          hasNested ? /* @__PURE__ */ jsx15("div", { className: "lax-workflow-container-children", children: nested }) : null
+          hasNested ? /* @__PURE__ */ jsx14("div", { className: "lax-workflow-container-children", children: nested }) : null
         ]
       },
       node.container_id
@@ -2983,9 +3001,9 @@ function renderWorkflowStageRow(node, containerTree, loopTreesBySessionId, viewP
     viewProps
   );
   if (as === "div") {
-    return /* @__PURE__ */ jsx15("div", { className, role: "listitem", children: content }, node.container_id);
+    return /* @__PURE__ */ jsx14("div", { className, role: "listitem", children: content }, node.container_id);
   }
-  return /* @__PURE__ */ jsx15("li", { className, children: content }, node.container_id);
+  return /* @__PURE__ */ jsx14("li", { className, children: content }, node.container_id);
 }
 function WorkflowStageListPanel({
   stages,
@@ -2994,30 +3012,19 @@ function WorkflowStageListPanel({
   viewProps
 }) {
   const scale = useWorkflowScaleOptions();
-  const [showAllStages, setShowAllStages] = useState6(false);
+  const [extraCompletedPages, setExtraCompletedPages] = useState6(0);
+  const maxVisibleDoneCount = scale.completedStagePageSize + extraCompletedPages * scale.completedStagePageSize;
   const partition = useMemo7(
-    () => partitionWorkflowStages(stages, scale.stagePageSize, showAllStages),
-    [stages, scale.stagePageSize, showAllStages]
+    () => partitionWorkflowStages(stages, maxVisibleDoneCount),
+    [stages, maxVisibleDoneCount]
   );
-  const useVirtual = partition.visible.length >= scale.stageVirtualizeThreshold;
   if (partition.visible.length === 0) return null;
-  const renderRow = (index) => renderWorkflowStageRow(
-    partition.visible[index],
-    containerTree,
-    loopTreesBySessionId,
-    viewProps,
-    "div"
+  const showMoreCount = Math.min(
+    scale.completedStagePageSize,
+    partition.totalDoneCount - partition.visibleDoneCount
   );
   return /* @__PURE__ */ jsxs10(Fragment2, { children: [
-    useVirtual ? /* @__PURE__ */ jsx15("div", { className: "lax-workflow-stage-list-virtual-host", role: "list", children: /* @__PURE__ */ jsx15(
-      WorkflowVirtualRowList,
-      {
-        count: partition.visible.length,
-        testId: "lax-workflow-stage-virtual-list",
-        className: "lax-workflow-stage-list lax-workflow-stage-list--virtual",
-        renderRow
-      }
-    ) }) : /* @__PURE__ */ jsx15("ol", { className: "lax-workflow-stage-list", "data-testid": "lax-workflow-stage-list", children: partition.visible.map(
+    /* @__PURE__ */ jsx14("ol", { className: "lax-workflow-stage-list", "data-testid": "lax-workflow-stage-list", children: partition.visible.map(
       (node) => renderWorkflowStageRow(node, containerTree, loopTreesBySessionId, viewProps)
     ) }),
     partition.hiddenCount > 0 ? /* @__PURE__ */ jsxs10(
@@ -3025,25 +3032,38 @@ function WorkflowStageListPanel({
       {
         type: "button",
         className: "lax-workflow-stage-list__show-all",
-        "data-testid": "lax-workflow-stage-show-all",
-        onClick: () => setShowAllStages(true),
+        "data-testid": "lax-workflow-stage-show-more",
+        onClick: () => setExtraCompletedPages((p) => p + 1),
         children: [
           "+",
           partition.hiddenCount,
-          " stages collapsed \xB7 Show all (",
-          partition.total,
-          ")"
+          " completed stages hidden \xB7 Show ",
+          showMoreCount,
+          " more"
         ]
       }
     ) : null,
-    showAllStages && partition.hiddenCount === 0 && stages.length > scale.stagePageSize ? /* @__PURE__ */ jsx15(
+    extraCompletedPages > 0 && partition.hiddenCount === 0 ? /* @__PURE__ */ jsx14(
       "button",
       {
         type: "button",
         className: "lax-workflow-stage-list__show-all",
         "data-testid": "lax-workflow-stage-show-less",
-        onClick: () => setShowAllStages(false),
-        children: "Show fewer stages"
+        onClick: () => setExtraCompletedPages(0),
+        children: "Show fewer completed stages"
+      }
+    ) : null,
+    partition.hiddenCount > 0 ? /* @__PURE__ */ jsxs10(
+      "p",
+      {
+        className: "lax-workflow-stage-list__hidden-hint",
+        "data-testid": "lax-workflow-stage-hidden-hint",
+        children: [
+          partition.hiddenCount,
+          " completed stage",
+          partition.hiddenCount !== 1 ? "s" : "",
+          " hidden \u2014 running, pending, and failed stages are always visible."
+        ]
       }
     ) : null
   ] });
@@ -3059,7 +3079,7 @@ function renderWorkflowChildren(workflowNode, containerTree, loopTreesBySessionI
     (node) => node.scope !== "stage" && node.scope !== "item"
   );
   return /* @__PURE__ */ jsxs10(Fragment2, { children: [
-    stages.length > 0 ? /* @__PURE__ */ jsx15(
+    stages.length > 0 ? /* @__PURE__ */ jsx14(
       WorkflowStageListPanel,
       {
         stages,
@@ -3084,7 +3104,7 @@ function WorkflowChrome({
   const viewProps = { virtualized, virtualizeThreshold, groupParallelTools };
   const rootNodes = containerTree.rootContainerIds.map((id) => containerTree.containersById[id]).filter((node) => node != null);
   if (rootNodes.length === 0) {
-    return /* @__PURE__ */ jsx15("div", { className: "lax-workflow-chrome", "data-testid": "lax-workflow-chrome", children: /* @__PURE__ */ jsx15(
+    return /* @__PURE__ */ jsx14("div", { className: "lax-workflow-chrome", "data-testid": "lax-workflow-chrome", children: /* @__PURE__ */ jsx14(
       WorkflowContainerLine,
       {
         variant: "root",
@@ -3094,7 +3114,7 @@ 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(
+  return /* @__PURE__ */ jsx14("div", { className: "lax-workflow-chrome", "data-testid": "lax-workflow-chrome", children: rootNodes.map((workflowNode) => /* @__PURE__ */ jsx14(WorkflowRootContainer, { node: workflowNode, children: renderWorkflowChildren(
     workflowNode,
     containerTree,
     loopTreesBySessionId,
@@ -3175,11 +3195,11 @@ function loopSessionDisplayLabel(loopSessionId) {
 }
 
 // src/view/workflow/WorkflowTaskListFooter.tsx
-import { jsx as jsx16, jsxs as jsxs11 } from "react/jsx-runtime";
+import { jsx as jsx15, jsxs as jsxs11 } from "react/jsx-runtime";
 function MessageResponse({ children }) {
   return /* @__PURE__ */ jsxs11("div", { className: "lax-message-response", children: [
-    /* @__PURE__ */ jsx16("span", { className: "lax-message-response__marker", children: "\u23BF " }),
-    /* @__PURE__ */ jsx16("span", { className: "lax-message-response__content", children })
+    /* @__PURE__ */ jsx15("span", { className: "lax-message-response__marker", children: "\u23BF " }),
+    /* @__PURE__ */ jsx15("span", { className: "lax-message-response__content", children })
   ] });
 }
 function WorkflowTaskListFooter() {
@@ -3204,18 +3224,18 @@ function WorkflowTaskListFooter() {
         onClick: () => setExpanded((v) => !v),
         children: [
           label,
-          /* @__PURE__ */ jsx16("span", { className: "lax-task-list-footer__hint", children: expanded ? " \xB7 \u2191 to hide" : " \xB7 \u2193 to view" })
+          /* @__PURE__ */ jsx15("span", { className: "lax-task-list-footer__hint", children: expanded ? " \xB7 \u2191 to hide" : " \xB7 \u2193 to view" })
         ]
       }
     ),
-    expanded ? /* @__PURE__ */ jsx16(MessageResponse, { children: /* @__PURE__ */ jsx16("div", { className: "lax-workflow-task-list-groups", children: groups.map((group) => /* @__PURE__ */ jsxs11(
+    expanded ? /* @__PURE__ */ jsx15(MessageResponse, { children: /* @__PURE__ */ jsx15("div", { className: "lax-workflow-task-list-groups", children: groups.map((group) => /* @__PURE__ */ jsxs11(
       "div",
       {
         className: "lax-workflow-task-list-group",
         "data-testid": `lax-workflow-task-group-${group.loopSessionId}`,
         children: [
-          /* @__PURE__ */ jsx16("div", { className: "lax-workflow-task-list-group-title", children: loopSessionDisplayLabel(group.loopSessionId) }),
-          /* @__PURE__ */ jsx16(TaskList, { tasks: group.tasks })
+          /* @__PURE__ */ jsx15("div", { className: "lax-workflow-task-list-group-title", children: loopSessionDisplayLabel(group.loopSessionId) }),
+          /* @__PURE__ */ jsx15(TaskList, { tasks: group.tasks })
         ]
       },
       group.loopSessionId
@@ -3225,7 +3245,7 @@ function WorkflowTaskListFooter() {
 
 // src/view/workflow/WorkflowSession.tsx
 import { useStore as useStore5 } from "zustand";
-import { jsx as jsx17, jsxs as jsxs12 } from "react/jsx-runtime";
+import { jsx as jsx16, jsxs as jsxs12 } from "react/jsx-runtime";
 function WorkflowSession({
   source,
   initialEvents,
@@ -3249,13 +3269,14 @@ function WorkflowSession({
   onRequestLoopReplay,
   children
 }) {
-  const storeRef = useRef4(null);
+  const storeRef = useRef3(null);
   if (storeRef.current === null) {
     const initialState = initialEvents && initialEvents.length > 0 ? reduceWorkflowEvents(initialEvents, { tierOverrides }) : void 0;
     storeRef.current = createWorkflowSessionStore(initialState, {
       tierOverrides,
       initialEventCount: initialEvents?.length ?? 0,
-      evictCompletedLoops: workflowScale?.evictCompletedLoops ?? DEFAULT_WORKFLOW_SCALE_OPTIONS.evictCompletedLoops
+      evictCompletedLoops: workflowScale?.evictCompletedLoops ?? DEFAULT_WORKFLOW_SCALE_OPTIONS.evictCompletedLoops,
+      maxHydratedCompletedLoops: workflowScale?.maxHydratedCompletedLoops ?? DEFAULT_WORKFLOW_SCALE_OPTIONS.maxHydratedCompletedLoops
     });
   }
   const scaleOptions = useMemo9(
@@ -3320,13 +3341,13 @@ function WorkflowSession({
       controller.abort();
     };
   }, [source, onError]);
-  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(
+  return /* @__PURE__ */ jsx16(WorkflowSessionStoreContext.Provider, { value: storeRef.current, children: /* @__PURE__ */ jsx16(WorkflowScaleContext.Provider, { value: scaleOptions, children: /* @__PURE__ */ jsx16(WorkflowLoopReplayProvider, { onRequestLoopReplay, children: /* @__PURE__ */ jsx16(InteractionBusContext.Provider, { value: busRef, children: /* @__PURE__ */ jsx16(NodeRegistryContext.Provider, { value: registryRef, children: /* @__PURE__ */ jsx16(MarkdownRendererContext.Provider, { value: markdownRenderer, children: /* @__PURE__ */ jsx16(ToolDisplayOptionsContext.Provider, { value: toolDisplayRef, children: /* @__PURE__ */ jsx16(SessionViewOptionsContext.Provider, { value: sessionViewRef, children: /* @__PURE__ */ jsxs12(
     "div",
     {
       className: "lax-agent-session lax-workflow-session",
       "data-testid": "lax-workflow-session",
       children: [
-        /* @__PURE__ */ jsx17(
+        /* @__PURE__ */ jsx16(
           WorkflowChrome,
           {
             containerTree,
@@ -3336,7 +3357,7 @@ function WorkflowSession({
             groupParallelTools
           }
         ),
-        /* @__PURE__ */ jsx17(WorkflowTaskListFooter, {}),
+        /* @__PURE__ */ jsx16(WorkflowTaskListFooter, {}),
         children
       ]
     }
@@ -3380,7 +3401,7 @@ function buildTimelineEntries(rootIds, byId, options = {}) {
 }
 
 // src/view/nodes/SubAgentNode.tsx
-import { jsx as jsx18, jsxs as jsxs13 } from "react/jsx-runtime";
+import { jsx as jsx17, jsxs as jsxs13 } from "react/jsx-runtime";
 function SubAgentBlock({ nodeId }) {
   const node = useNodeTyped(nodeId, "subagent");
   const childIds = useChildren(nodeId);
@@ -3391,7 +3412,7 @@ function SubAgentBlock({ nodeId }) {
       "SubAgent: ",
       node.subagentId
     ] }),
-    /* @__PURE__ */ jsx18("div", { className: "lax-subagent-block__children", children: childIds.map((childId) => /* @__PURE__ */ jsx18(TimelineItem, { nodeId: childId, registry }, childId)) })
+    /* @__PURE__ */ jsx17("div", { className: "lax-subagent-block__children", children: childIds.map((childId) => /* @__PURE__ */ jsx17(TimelineItem, { nodeId: childId, registry }, childId)) })
   ] });
 }
 
@@ -3492,6 +3513,7 @@ export {
   TruncatedContent,
   UnknownCard,
   VirtualTimeline,
+  WORKFLOW_STRUCTURE_EVENT_PREFIXES,
   WORKFLOW_STRUCTURE_EVENT_TYPES,
   WebFetch,
   WebFetchToolBody,
@@ -3531,6 +3553,7 @@ export {
   createWorkflowSessionStore,
   displayPath,
   entryKey,
+  filterAgentLoopReplayEvents,
   formatAgentTitle,
   formatDiffFromStrings,
   formatExploreSummaryText,
@@ -3544,6 +3567,7 @@ export {
   getChildIds,
   getMainStageIds,
   getNoopInteractionBus,
+  isAgentLoopReplayEventType,
   isGitOperationCommand,
   isGroupableToolName,
   isMcpToolName,