langchain_agentx_stream_ui 0.6.4 → 0.6.8

package/dist/index.js

@@ -23,7 +23,7 @@ import {
   fixDiagramSvg,
   markEventIdSeen,
   shouldSkipDuplicateEvent
-} from "./chunk-YHCK76OU.js";
+} from "./chunk-VBUKRPEO.js";
 import {
   DefaultToolBody,
   InteractionBusContext,
@@ -32,7 +32,7 @@ import {
   getNoopInteractionBus,
   resolveToolBody,
   useInteractionBus
-} from "./chunk-5XELRP3S.js";
+} from "./chunk-B2NG5KGH.js";
 import "./chunk-6Z4ZF36Z.js";
 import {
   CollapseKind,
@@ -79,6 +79,7 @@ import {
   getChildIds,
   getDisplayProjectionStatus,
   getMainStageIds,
+  isAgentInputLocked,
   isAuthorityTerminal,
   isAuthorityTerminalStatus,
   isGitOperationCommand,
@@ -119,7 +120,7 @@ import {
   useSessionStatus,
   useSessionViewOptions,
   useTimeline
-} from "./chunk-O5GERXTY.js";
+} from "./chunk-UBYOUDQE.js";
 import {
   AGENT_PROGRESS_GUTTER,
   AGENT_PROGRESS_INDENT,
@@ -149,7 +150,7 @@ import {
   createDefaultToolRegistry,
   formatAgentTitle,
   formatExploreSummaryText
-} from "./chunk-GYFC2DI7.js";
+} from "./chunk-7XSQYG3X.js";
 import {
   BodyBlockList,
   DiffView,
@@ -990,6 +991,70 @@ function trimWorkflowLoopTreesSoft(state, options) {
   };
 }
 
+// src/core/sessionTreeAuthorityFinalize.ts
+function finalizeSessionTreeOnAuthorityCanceled(tree, endedAtMs = Date.now()) {
+  const byId = { ...tree.byId };
+  let changed = false;
+  for (const [id, node] of Object.entries(byId)) {
+    if (!node) continue;
+    if ((node.kind === "text" || node.kind === "reasoning") && node.status === "streaming") {
+      byId[id] = { ...node, status: "done" };
+      changed = true;
+      continue;
+    }
+    if (node.kind !== "tool_call") continue;
+    let next = node;
+    if (node.status === "running") {
+      next = {
+        ...next,
+        status: "failed",
+        endedAt: node.endedAt ?? endedAtMs
+      };
+      changed = true;
+    }
+    if (next.subagentSessions) {
+      const sessions = { ...next.subagentSessions };
+      let sessChanged = false;
+      for (const [agentId, marker] of Object.entries(sessions)) {
+        if (marker.status === "running") {
+          sessions[agentId] = {
+            ...marker,
+            status: "done",
+            endedAt: endedAtMs
+          };
+          sessChanged = true;
+        }
+      }
+      if (sessChanged) {
+        next = { ...next, subagentSessions: sessions };
+        changed = true;
+      }
+    }
+    if (next.subagentProgress?.some((e) => e.status === "running")) {
+      next = {
+        ...next,
+        subagentProgress: next.subagentProgress.map(
+          (entry) => entry.status === "running" ? { ...entry, status: "error" } : entry
+        )
+      };
+      changed = true;
+    }
+    if (next !== node) {
+      byId[id] = next;
+    }
+  }
+  if (!changed) {
+    return tree;
+  }
+  return {
+    ...tree,
+    byId,
+    activeRoundId: null,
+    // 展示树终态；不升格 / 不写 REST authority（P-AUTH）
+    status: tree.status === "done" || tree.status === "done_with_warning" ? tree.status : "error"
+  };
+}
+
 // src/core/store.ts
 function safeReduce(tree, event, eventIndex, options) {
   try {
@@ -1059,7 +1124,12 @@ function createSessionStore(initialTree = createEmptyTree(), storeOptions) {
       set({ tree, eventCount });
     },
     applyAuthoritySnapshot(snapshot) {
-      set({ authority: snapshot });
+      const current = get();
+      let tree = current.tree;
+      if (snapshot.status === "canceled") {
+        tree = finalizeSessionTreeOnAuthorityCanceled(tree);
+      }
+      set({ authority: snapshot, tree });
     },
     setAuthoritySubscribeFailed(failed) {
       set({ authoritySubscribeFailed: failed });
@@ -3383,6 +3453,38 @@ function safeReduceWorkflow(state, event, eventIndex, options) {
     };
   }
 }
+function finalizeWorkflowContainersOnAuthorityCanceled(state) {
+  const containersById = {
+    ...state.containerTree.containersById
+  };
+  let changed = false;
+  for (const [id, node] of Object.entries(containersById)) {
+    if (node.status !== "running" && node.status !== "pending") {
+      continue;
+    }
+    const display = node.display ? { ...node.display, metrics_running: 0 } : node.display;
+    containersById[id] = {
+      ...node,
+      status: "failed",
+      is_open: false,
+      display
+    };
+    changed = true;
+  }
+  if (!changed) {
+    return state;
+  }
+  return {
+    ...state,
+    containerTree: {
+      ...state.containerTree,
+      containersById,
+      activeContainerIds: state.containerTree.activeContainerIds.filter(
+        (id) => containersById[id]?.is_open
+      )
+    }
+  };
+}
 function finalizeWorkflowSessionOnStreamError(state, message, eventIndex, stack) {
   const containersById = {
     ...state.containerTree.containersById
@@ -3522,17 +3624,33 @@ function createWorkflowSessionStore(initialState = createEmptyWorkflowSessionSta
       }));
     },
     applyAuthoritySnapshot(snapshot) {
-      set((current) => ({
-        state: {
-          ...current.state,
-          authority: snapshot,
-          display: {
-            ...current.state.display,
-            awaitingAuthority: false,
-            authoritySyncFailed: false
+      set((current) => {
+        let nextState = current.state;
+        if (snapshot.status === "canceled") {
+          nextState = finalizeWorkflowContainersOnAuthorityCanceled(nextState);
+          const nextLoops = {
+            ...nextState.loopTreesBySessionId
+          };
+          for (const [loopSessionId, loopTree] of Object.entries(nextLoops)) {
+            nextLoops[loopSessionId] = finalizeSessionTreeOnAuthorityCanceled(loopTree);
           }
+          nextState = {
+            ...nextState,
+            loopTreesBySessionId: nextLoops
+          };
         }
-      }));
+        return {
+          state: {
+            ...nextState,
+            authority: snapshot,
+            display: {
+              ...nextState.display,
+              awaitingAuthority: false,
+              authoritySyncFailed: false
+            }
+          }
+        };
+      });
     },
     setAwaitingAuthority(value) {
       set((current) => ({
@@ -5392,8 +5510,17 @@ 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";
+  const showTransportError = (display.projectionStatus === "error" || state.internalErrors.length > 0) && banner.kind !== "projection_degraded" && banner.kind !== "authority_error" && banner.kind !== "canceled";
   return /* @__PURE__ */ jsxs14(Fragment4, { children: [
+    banner.kind === "canceled" ? /* @__PURE__ */ jsx17(
+      "div",
+      {
+        className: "lax-workflow-banner lax-workflow-banner--canceled",
+        "data-testid": "lax-workflow-canceled-banner",
+        role: "status",
+        children: banner.message
+      }
+    ) : null,
     banner.kind === "authority_error" ? /* @__PURE__ */ jsx17(
       "div",
       {
@@ -6138,6 +6265,7 @@ export {
   displayPath,
   entryKey,
   filterAgentLoopReplayEvents,
+  finalizeSessionTreeOnAuthorityCanceled,
   finalizeWorkflowDisplayOnStreamEnded,
   fixDiagramSvg,
   formatAgentTitle,
@@ -6156,6 +6284,7 @@ export {
   getNoopInteractionBus,
   hydrateAskFromSnapshot,
   hydrateWorkflowFromSnapshot,
+  isAgentInputLocked,
   isAgentLoopReplayEventType,
   isAtOrBeforeProjectionCursor,
   isAuthorityTerminal,