langchain_agentx_stream_ui 0.3.4 → 0.3.7

package/dist/{chunk-6ORA5RMV.js → chunk-24FLS4TF.js}

@@ -15,7 +15,7 @@ import {
   extractLastToolInfoFromProgress,
   formatExploreSummaryText,
   formatThoughtDurationPrefix
-} from "./chunk-FL2G7SV3.js";
+} from "./chunk-QL3DIGFI.js";
 
 // src/types/tree.ts
 function createEmptyTree() {
@@ -77,6 +77,8 @@ var DEFAULT_EVENT_TIERS = {
   "subagent-model-step": "collapsed",
   "hook-progress": "debug",
   "hook-finished": "debug",
+  "api-retry": "debug",
+  "assistant-turn-superseded": "main",
   "compact-start": "debug",
   "compact-end": "debug",
   "task-started": "debug",
@@ -483,23 +485,31 @@ function parseSnapshotTasks(rawTasks) {
 }
 
 // src/types/modelRetry.ts
-function parseModelRetryHookData(data, timestampMs) {
-  if (data.name !== "model_retry") return null;
+function parseModelRetryFields(data, timestampMs) {
   const attempt = typeof data.attempt === "number" ? data.attempt : Number(data.attempt);
   const maxRetries = typeof data.max_retries === "number" ? data.max_retries : Number(data.max_retries);
   const retryDelayMs = typeof data.retry_delay_ms === "number" ? data.retry_delay_ms : Number(data.retry_delay_ms ?? 0);
   if (!Number.isFinite(attempt) || !Number.isFinite(maxRetries)) return null;
+  const errorCategory = typeof data.error_category === "string" ? data.error_category : typeof data.error === "string" ? data.error : "api_error";
   return {
     attempt: Math.max(1, Math.floor(attempt)),
     maxRetries: Math.max(1, Math.floor(maxRetries)),
     retryDelayMs: Number.isFinite(retryDelayMs) ? Math.max(0, retryDelayMs) : 0,
-    errorCategory: typeof data.error_category === "string" ? data.error_category : "api_error",
+    errorCategory,
     errorMessage: typeof data.error_message === "string" ? data.error_message : "",
     errorStatus: typeof data.error_status === "number" ? data.error_status : data.error_status != null ? Number(data.error_status) : null,
     phase: typeof data.phase === "string" ? data.phase : "scheduled",
     updatedAt: timestampMs
   };
 }
+function parseApiRetryEventData(data, timestampMs) {
+  if (data.subtype !== "api_retry") return null;
+  return parseModelRetryFields(data, timestampMs);
+}
+function parseModelRetryHookData(data, timestampMs) {
+  if (data.name !== "model_retry") return null;
+  return parseModelRetryFields(data, timestampMs);
+}
 function formatModelRetryCategoryLabel(category) {
   switch (category) {
     case "server_overload":
@@ -650,8 +660,14 @@ function streamChunk(data) {
 function appendStreamChunk(current, data) {
   return (current ?? "") + streamChunk(data);
 }
+function reactivateSessionIfIdle(tree) {
+  if (tree.status === "error") return;
+  if (tree.status !== "running" && tree.status !== "connecting") {
+    tree.status = "running";
+  }
+}
 function handleStart(tree, event) {
-  if (tree.status === "done" || tree.status === "error") {
+  if (tree.status === "error") {
     return;
   }
   tree.status = "running";
@@ -781,6 +797,43 @@ function appendTaskDebugEvent(tree, event, eventIndex) {
     data: event.data
   });
 }
+function findLastMainStageTextNodeId(tree) {
+  const childIds = getActiveChildIds(tree);
+  for (let i = childIds.length - 1; i >= 0; i -= 1) {
+    const node = tree.byId[childIds[i]];
+    if (node?.kind === "text") {
+      return node.id;
+    }
+  }
+  return void 0;
+}
+function removeMainStageTextNode(tree, nodeId) {
+  delete tree.byId[nodeId];
+  tree.rootChildren = tree.rootChildren.filter((id) => id !== nodeId);
+}
+function handleAssistantTurnSuperseded(tree) {
+  clearModelRetry(tree);
+  const hadDeferredPartial = Boolean(tree.deferredMainText);
+  tree.deferredMainText = "";
+  const streaming = findStreamingNode(tree, "text");
+  if (streaming) {
+    removeMainStageTextNode(tree, streaming.id);
+    return;
+  }
+  if (!hadDeferredPartial) {
+    const lastTextId = findLastMainStageTextNodeId(tree);
+    if (lastTextId) {
+      removeMainStageTextNode(tree, lastTextId);
+    }
+  }
+}
+function handleApiRetry(tree, event) {
+  const data = event.data;
+  const modelRetry = parseApiRetryEventData(data, eventTimeToMs(event.timestamp));
+  if (modelRetry) {
+    tree.modelRetry = modelRetry;
+  }
+}
 function handleHookProgress(tree, event) {
   const data = event.data;
   const modelRetry = parseModelRetryHookData(data, eventTimeToMs(event.timestamp));
@@ -848,6 +901,7 @@ function flushDeferredMainTextToTree(tree, event, seq) {
   appendMainStageChild(tree, id, event, seq);
 }
 function handleTextStart(tree, event, seq) {
+  reactivateSessionIfIdle(tree);
   clearModelRetry(tree);
   const id = `text:${seq}`;
   const node = {
@@ -887,6 +941,7 @@ function handleTextEnd(tree, event) {
   };
 }
 function handleReasoningStart(tree, event, seq) {
+  reactivateSessionIfIdle(tree);
   const id = `reasoning:${seq}`;
   const roundId = `round-${seq}`;
   const node = {
@@ -1031,6 +1086,7 @@ function maybeSynthesizeLeakedToolProgress(tree, options, params) {
   });
 }
 function handleToolCall(tree, event, seq, options) {
+  reactivateSessionIfIdle(tree);
   const id = toolNodeId(tree, event);
   const data = event.data;
   const existing = tree.byId[id];
@@ -1337,6 +1393,9 @@ function handleDebug(tree, event, eventIndex, parentNodeId, collectDebug) {
     appendTaskDebugEvent(tree, event, eventIndex);
   }
   switch (event.event_type) {
+    case "api-retry":
+      handleApiRetry(tree, event);
+      break;
     case "hook-progress":
       handleHookProgress(tree, event);
       break;
@@ -1533,6 +1592,9 @@ function applyClassified(tree, classified, eventIndex, options) {
       handleTextEnd(tree, event);
       break;
     }
+    case "assistant-turn-superseded":
+      handleAssistantTurnSuperseded(tree);
+      break;
     case "reasoning-start":
       if (isDroppedByEmitterRoute(event.data)) break;
       if (isSubagentScopedReasoningEvent(event.data)) break;
@@ -5498,4 +5560,4 @@ export {
   VirtualTimeline,
   SessionTimeline
 };
-//# sourceMappingURL=chunk-6ORA5RMV.js.map
+//# sourceMappingURL=chunk-24FLS4TF.js.map