langchain_agentx_stream_ui 0.3.0 → 0.3.3

package/dist/index.js

@@ -23,7 +23,7 @@ import {
   fixDiagramSvg,
   markEventIdSeen,
   shouldSkipDuplicateEvent
-} from "./chunk-CUKNUCGU.js";
+} from "./chunk-ISQ7JWGD.js";
 import {
   DefaultToolBody,
   InteractionBusContext,
@@ -32,8 +32,8 @@ import {
   getNoopInteractionBus,
   resolveToolBody,
   useInteractionBus
-} from "./chunk-6DYISADG.js";
-import "./chunk-DP7V33X7.js";
+} from "./chunk-TCKOXX7O.js";
+import "./chunk-6Z4ZF36Z.js";
 import {
   CollapseKind,
   CollapsedExploreNode,
@@ -72,7 +72,6 @@ import {
   createEmptyTree,
   createProjectionContext,
   entryKey,
-  formatExploreSummaryText,
   formatHookFinishedSummary,
   formatShellProgressSuffix,
   formatTaskFooterLabel,
@@ -111,7 +110,7 @@ import {
   useSessionStatus,
   useSessionViewOptions,
   useTimeline
-} from "./chunk-SUEMZC4U.js";
+} from "./chunk-6ORA5RMV.js";
 import {
   Agent,
   AgentToolBody,
@@ -137,8 +136,9 @@ import {
   Write,
   WriteToolBody,
   createDefaultToolRegistry,
-  formatAgentTitle
-} from "./chunk-7CLAU74Y.js";
+  formatAgentTitle,
+  formatExploreSummaryText
+} from "./chunk-FL2G7SV3.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";
@@ -184,7 +184,8 @@ function safeReduce(tree, event, eventIndex, options) {
 function createSessionStore(initialTree = createEmptyTree(), storeOptions) {
   const reduceOpts = {
     tierOverrides: storeOptions?.tierOverrides,
-    collectDebug: storeOptions?.debug === true
+    collectDebug: storeOptions?.debug === true,
+    subagentLeakGate: storeOptions?.subagentLeakGate === true
   };
   const initialEventCount = storeOptions?.initialEventCount ?? 0;
   const seenEventIds = new Set(storeOptions?.initialSeenEventIds);
@@ -317,6 +318,15 @@ var PROFILE_PRESETS = {
       ...BOUNDED_DEFAULTS
     },
     terminal: WORKFLOW_TERMINAL_WITH_FALLBACK
+  },
+  "workflow-tail": {
+    profile: "workflow-tail",
+    replay: "none",
+    reconnect: {
+      mode: "bounded",
+      ...BOUNDED_DEFAULTS
+    },
+    terminal: WORKFLOW_TERMINAL_WITH_FALLBACK
   }
 };
 function legacyTerminal(expectedApplicationKind) {
@@ -778,19 +788,22 @@ function AgentSession({
   exploreFullscreenBash = true,
   memoryDir = null,
   workspaceRoot = null,
+  subagentLeakGate = false,
   children
 }) {
   const storeRef = useRef2(null);
   if (storeRef.current === null) {
     const replayOpts = {
       tierOverrides: _tierOverrides,
-      collectDebug: debug
+      collectDebug: debug,
+      subagentLeakGate
     };
     const initialTree = initialEvents && initialEvents.length > 0 ? replayEvents(initialEvents, replayOpts) : void 0;
     storeRef.current = createSessionStore(initialTree, {
       tierOverrides: _tierOverrides,
       debug,
-      initialEventCount: initialEvents?.length ?? 0
+      initialEventCount: initialEvents?.length ?? 0,
+      subagentLeakGate
     });
   }
   const registryRef = useMemo(() => registry ?? createDefaultRegistry(), [registry]);
@@ -815,7 +828,8 @@ function AgentSession({
       verbose,
       exploreFullscreenBash,
       memoryDir,
-      workspaceRoot
+      workspaceRoot,
+      subagentLeakGate
     }),
     [
       groupParallelTools,
@@ -825,7 +839,8 @@ function AgentSession({
       verbose,
       exploreFullscreenBash,
       memoryDir,
-      workspaceRoot
+      workspaceRoot,
+      subagentLeakGate
     ]
   );
   const onErrorRef = useRef2(onError);
@@ -901,10 +916,19 @@ function resolveWorkflowBanner(state) {
   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" };
   }
-  const auth = state.authority?.status;
   if (auth === "partial_success") {
     const forced = state.display.terminalKind === "forced_closeout";
     return {
@@ -912,6 +936,15 @@ function resolveWorkflowBanner(state) {
       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" };
   }
@@ -1031,6 +1064,10 @@ function createEmptyWorkflowSessionState() {
     stageDoneSummariesByContainerId: {},
     hotCompletedLoopSessionIds: [],
     activeLoopSessionId: null,
+    projectionCursor: null,
+    snapshotVersion: null,
+    reconnectSuppressedReason: null,
+    snapshotHydrateFailed: false,
     openScopeStack: [],
     internalErrors: []
   };
@@ -1389,6 +1426,35 @@ function contentBlockFromEvent(event) {
   }
   return null;
 }
+function wouldAssignParentCreateCycle(containersById, containerId, parentId) {
+  if (!parentId) return false;
+  if (parentId === containerId) return true;
+  const seen = /* @__PURE__ */ new Set();
+  let current = parentId;
+  while (current) {
+    if (current === containerId) return true;
+    if (seen.has(current)) return true;
+    seen.add(current);
+    current = containersById[current]?.parent_container_id ?? null;
+  }
+  return false;
+}
+function sanitizeContainerTreeParentLinks(containersById) {
+  const next = { ...containersById };
+  for (const [containerId, node] of Object.entries(next)) {
+    const seen = /* @__PURE__ */ new Set([containerId]);
+    let current = node.parent_container_id;
+    while (current) {
+      if (seen.has(current)) {
+        next[containerId] = { ...node, parent_container_id: null };
+        break;
+      }
+      seen.add(current);
+      current = next[current]?.parent_container_id ?? null;
+    }
+  }
+  return next;
+}
 function resolveContentScopeKey(data) {
   for (const keyName of ["stage_key", "item_key", "branch_key", "aggregate_key", "debug_key"]) {
     const value = readString(data, keyName);
@@ -1466,7 +1532,11 @@ var WorkflowContainerActiveSet = class {
         );
       }
     }
-    return this.deepestActive();
+    const openNonWorkflow = this.openOrder.map((cid) => this.nodes[cid]).filter((node) => node.is_open && node.scope !== "workflow");
+    if (openNonWorkflow.length === 1) {
+      return openNonWorkflow[0].container_id;
+    }
+    return null;
   }
   closeDescendantsOfPath(workflowPath, status) {
     const toClose = this.openOrder.filter((cid) => {
@@ -1544,7 +1614,12 @@ var WorkflowContainerStateMachine = class {
     const subtitle = display?.description ?? "";
     const pattern = display?.pattern ?? "";
     const status = displayStatus(data, "running");
-    const parentId = this.resolveParentId(workflowPath, scope);
+    const existing = this.active.getNode(containerId);
+    const computedParentId = this.resolveParentId(workflowPath, scope);
+    let parentId = existing != null ? existing.parent_container_id : computedParentId;
+    if (wouldAssignParentCreateCycle(this.active.nodesById, containerId, parentId)) {
+      parentId = existing?.parent_container_id ?? null;
+    }
     const node = {
       container_id: containerId,
       scope,
@@ -1557,7 +1632,7 @@ var WorkflowContainerStateMachine = class {
       pattern,
       parent_container_id: parentId,
       display,
-      content_blocks: [],
+      content_blocks: existing?.content_blocks ?? [],
       is_open: true
     };
     this.active.open(node);
@@ -1635,9 +1710,7 @@ var WorkflowContainerStateMachine = class {
         (best, node) => this.active.compareContainerRank(node.container_id, best.container_id) > 0 ? node : best
       ).container_id;
     }
-    return openNodes.reduce(
-      (best, node) => this.active.compareContainerRank(node.container_id, best.container_id) > 0 ? node : best
-    ).container_id;
+    return null;
   }
 };
 var WorkflowContainerTreeProjector = class {
@@ -2753,6 +2826,45 @@ function reduceWorkflowEvents(events, options) {
   );
 }
 
+// src/core/workflow/hydrateWorkflowSnapshot.ts
+function isAtOrBeforeProjectionCursor(projectionCursor, eventId) {
+  if (!projectionCursor || !eventId) return false;
+  const cursorNum = Number.parseInt(projectionCursor, 10);
+  const eventNum = Number.parseInt(eventId, 10);
+  if (!Number.isNaN(cursorNum) && !Number.isNaN(eventNum)) {
+    return eventNum <= cursorNum;
+  }
+  return eventId <= projectionCursor;
+}
+function hydrateWorkflowFromSnapshot(state, snapshot, options = {}) {
+  const authority = options.preserveAuthority ? state.authority : null;
+  const containerTree = structuredClone(snapshot.containerTree);
+  containerTree.containersById = sanitizeContainerTreeParentLinks(
+    containerTree.containersById
+  );
+  return {
+    ...createEmptyWorkflowSessionState(),
+    authority,
+    display: {
+      ...snapshot.display,
+      streamEnded: false
+    },
+    containerTree,
+    projectionCursor: snapshot.cursor,
+    snapshotVersion: snapshot.snapshotVersion,
+    meta: {
+      ...state.meta,
+      workflowRunId: snapshot.taskId,
+      lastEventId: snapshot.cursor,
+      startedAt: state.meta.startedAt ?? snapshot.capturedAt
+    },
+    loopTreesBySessionId: {},
+    internalErrors: [],
+    reconnectSuppressedReason: null,
+    snapshotHydrateFailed: false
+  };
+}
+
 // src/core/workflowSessionStore.ts
 var DEFAULT_HYDRATE_CHUNK_SIZE = 50;
 function nextAnimationFrame() {
@@ -2841,8 +2953,12 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
     hydratingLoopSessionIds: {},
     applyEvent(event, ctx) {
       const sseEventId = ctx?.sseEventId;
-      if (shouldSkipDuplicateEvent(seenEventIds, sseEventId)) return;
       const { state, eventCount } = get();
+      if (isAtOrBeforeProjectionCursor(state.projectionCursor, sseEventId)) {
+        markEventIdSeen(seenEventIds, sseEventId);
+        return;
+      }
+      if (shouldSkipDuplicateEvent(seenEventIds, sseEventId)) return;
       const { state: reduced } = safeReduceWorkflow(state, event, eventCount, reduceOpts);
       const evicted = finalizeState(reduced);
       const nextMeta = sseEventId ? { ...evicted.meta, lastEventId: sseEventId } : evicted.meta;
@@ -3039,6 +3155,23 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
           loopSessionId
         )
       }));
+    },
+    hydrateFromSnapshot(snapshot, options) {
+      const { state } = get();
+      const next = finalizeState(
+        hydrateWorkflowFromSnapshot(state, snapshot, options)
+      );
+      markEventIdSeen(seenEventIds, snapshot.cursor);
+      set({ state: next });
+    },
+    markSnapshotHydrateFailed() {
+      set((current) => ({
+        state: {
+          ...current.state,
+          snapshotHydrateFailed: true,
+          reconnectSuppressedReason: "snapshot_unavailable"
+        }
+      }));
     }
   }));
 }
@@ -4409,22 +4542,30 @@ function stageHasSubworkflowChild(stageId, containerTree) {
     (node) => node.scope === "subworkflow"
   );
 }
-function renderContainerChildren(parentId, containerTree, loopTreesBySessionId, viewProps, depth) {
+function renderContainerChildren(parentId, containerTree, loopTreesBySessionId, viewProps, depth, ancestorVisited = /* @__PURE__ */ new Set()) {
   const children = collectChildContainers(containerTree.containersById, parentId);
   const items = children.filter((node) => node.scope === "item");
   const rest = children.filter((node) => node.scope !== "item");
   return /* @__PURE__ */ jsxs10(Fragment2, { children: [
     renderItemSiblings(items, parentId, depth, containerTree, loopTreesBySessionId, viewProps),
-    rest.map((node) => renderContainerNode(node, containerTree, loopTreesBySessionId, viewProps, depth))
+    rest.map(
+      (node) => renderContainerNode(node, containerTree, loopTreesBySessionId, viewProps, depth, ancestorVisited)
+    )
   ] });
 }
-function renderContainerNode(node, containerTree, loopTreesBySessionId, viewProps, depth) {
+function renderContainerNode(node, containerTree, loopTreesBySessionId, viewProps, depth, ancestorVisited = /* @__PURE__ */ new Set()) {
+  if (ancestorVisited.has(node.container_id)) {
+    return null;
+  }
+  const pathVisited = new Set(ancestorVisited);
+  pathVisited.add(node.container_id);
   const nested = renderContainerChildren(
     node.container_id,
     containerTree,
     loopTreesBySessionId,
     viewProps,
-    depth + 1
+    depth + 1,
+    pathVisited
   );
   const hasNested = collectChildContainers(containerTree.containersById, node.container_id).length > 0;
   if (node.scope === "subworkflow") {
@@ -4642,7 +4783,35 @@ function WorkflowSessionBanners({ state }) {
   if (banner.kind === "none" && display.projectionStatus !== "error" && display.projectionStatus !== "done_with_warning" && state.internalErrors.length === 0) {
     return null;
   }
+  const showTransportError = (display.projectionStatus === "error" || state.internalErrors.length > 0) && banner.kind !== "projection_degraded" && banner.kind !== "authority_error";
   return /* @__PURE__ */ jsxs12(Fragment3, { children: [
+    banner.kind === "authority_error" ? /* @__PURE__ */ jsx17(
+      "div",
+      {
+        className: "lax-workflow-banner lax-workflow-banner--authority-error",
+        "data-testid": "lax-workflow-authority-error-banner",
+        role: "alert",
+        children: banner.message
+      }
+    ) : null,
+    banner.kind === "projection_degraded" ? /* @__PURE__ */ jsx17(
+      "div",
+      {
+        className: "lax-workflow-banner lax-workflow-banner--projection-degraded",
+        "data-testid": "lax-workflow-projection-degraded-banner",
+        role: "status",
+        children: banner.message
+      }
+    ) : null,
+    banner.kind === "snapshot_unavailable" ? /* @__PURE__ */ jsx17(
+      "div",
+      {
+        className: "lax-workflow-banner lax-workflow-banner--snapshot-unavailable",
+        "data-testid": "lax-workflow-snapshot-unavailable-banner",
+        role: "status",
+        children: banner.message
+      }
+    ) : null,
     banner.kind === "partial_success" ? /* @__PURE__ */ jsx17(
       "div",
       {
@@ -4686,7 +4855,7 @@ function WorkflowSessionBanners({ state }) {
         children: banner.message
       }
     ) : null,
-    display.projectionStatus === "error" || state.internalErrors.length > 0 ? /* @__PURE__ */ jsx17(
+    showTransportError ? /* @__PURE__ */ jsx17(
       "div",
       {
         className: "lax-workflow-error",
@@ -4825,6 +4994,8 @@ import { jsx as jsx19, jsxs as jsxs14 } from "react/jsx-runtime";
 function WorkflowSession({
   source,
   authoritySource,
+  snapshotBootstrap,
+  snapshotHydrateFailed = false,
   initialEvents,
   registry,
   tierOverrides,
@@ -4863,6 +5034,16 @@ function WorkflowSession({
   const streamEnded = useStore5(store, (s) => s.state.display.streamEnded);
   const authorityStatus = useStore5(store, (s) => s.state.authority?.status);
   useWorkflowAwaitingAuthorityTimers(store, streamEnded, authorityStatus);
+  useEffect10(() => {
+    if (snapshotBootstrap) {
+      store.getState().hydrateFromSnapshot(snapshotBootstrap, { preserveAuthority: true });
+    }
+  }, [snapshotBootstrap, store]);
+  useEffect10(() => {
+    if (snapshotHydrateFailed) {
+      store.getState().markSnapshotHydrateFailed();
+    }
+  }, [snapshotHydrateFailed, store]);
   const scaleOptions = useMemo9(
     () => ({ ...DEFAULT_WORKFLOW_SCALE_OPTIONS, ...workflowScale }),
     [workflowScale]
@@ -4905,10 +5086,13 @@ function WorkflowSession({
   const loopTreesBySessionId = sessionState.loopTreesBySessionId;
   const onErrorRef = useRef4(onError);
   const onDisplayStreamEndedRef = useRef4(onDisplayStreamEnded);
+  const streamAbortRef = useRef4(null);
   onErrorRef.current = onError;
   onDisplayStreamEndedRef.current = onDisplayStreamEnded;
   useEffect10(() => {
+    if (!source) return void 0;
     const controller = new AbortController();
+    streamAbortRef.current = controller;
     void source.start((event, ctx) => {
       store.getState().applyEvent(event, ctx);
     }, controller.signal).then(() => {
@@ -4924,8 +5108,14 @@ function WorkflowSession({
     });
     return () => {
       controller.abort();
+      streamAbortRef.current = null;
     };
   }, [source, store]);
+  useEffect10(() => {
+    if (isAuthorityTerminalStatus(authorityStatus)) {
+      streamAbortRef.current?.abort();
+    }
+  }, [authorityStatus]);
   return /* @__PURE__ */ jsx19(WorkflowSessionStoreContext.Provider, { value: store, children: /* @__PURE__ */ jsx19(WorkflowScaleContext.Provider, { value: scaleOptions, children: /* @__PURE__ */ jsx19(WorkflowLoopReplayProvider, { onRequestLoopReplay, children: /* @__PURE__ */ jsx19(InteractionBusContext.Provider, { value: busRef, children: /* @__PURE__ */ jsx19(NodeRegistryContext.Provider, { value: registryRef, children: /* @__PURE__ */ jsx19(MarkdownRendererContext.Provider, { value: markdownRenderer, children: /* @__PURE__ */ jsx19(ToolDisplayOptionsContext.Provider, { value: toolDisplayRef, children: /* @__PURE__ */ jsx19(SessionViewOptionsContext.Provider, { value: sessionViewRef, children: /* @__PURE__ */ jsxs14(
     "div",
     {
@@ -5315,7 +5505,9 @@ export {
   getDisplayProjectionStatus,
   getMainStageIds,
   getNoopInteractionBus,
+  hydrateWorkflowFromSnapshot,
   isAgentLoopReplayEventType,
+  isAtOrBeforeProjectionCursor,
   isAuthorityTerminal,
   isAuthorityTerminalStatus,
   isGitOperationCommand,