langchain_agentx_stream_ui 0.3.2 → 0.3.4

package/dist/index.js

@@ -1426,6 +1426,42 @@ 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 normalizeSnapshotContainerContentBlocks(containersById) {
+  const next = {};
+  for (const [containerId, node] of Object.entries(containersById)) {
+    next[containerId] = { ...node, content_blocks: node.content_blocks ?? [] };
+  }
+  return next;
+}
 function resolveContentScopeKey(data) {
   for (const keyName of ["stage_key", "item_key", "branch_key", "aggregate_key", "debug_key"]) {
     const value = readString(data, keyName);
@@ -1503,7 +1539,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) => {
@@ -1581,7 +1621,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,
@@ -1594,7 +1639,7 @@ var WorkflowContainerStateMachine = class {
       pattern,
       parent_container_id: parentId,
       display,
-      content_blocks: [],
+      content_blocks: existing?.content_blocks ?? [],
       is_open: true
     };
     this.active.open(node);
@@ -1672,9 +1717,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 {
@@ -2802,6 +2845,10 @@ function isAtOrBeforeProjectionCursor(projectionCursor, eventId) {
 }
 function hydrateWorkflowFromSnapshot(state, snapshot, options = {}) {
   const authority = options.preserveAuthority ? state.authority : null;
+  const containerTree = structuredClone(snapshot.containerTree);
+  containerTree.containersById = normalizeSnapshotContainerContentBlocks(
+    sanitizeContainerTreeParentLinks(containerTree.containersById)
+  );
   return {
     ...createEmptyWorkflowSessionState(),
     authority,
@@ -2809,7 +2856,7 @@ function hydrateWorkflowFromSnapshot(state, snapshot, options = {}) {
       ...snapshot.display,
       streamEnded: false
     },
-    containerTree: structuredClone(snapshot.containerTree),
+    containerTree,
     projectionCursor: snapshot.cursor,
     snapshotVersion: snapshot.snapshotVersion,
     meta: {
@@ -4502,22 +4549,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") {