From c61d7fb3d79775ac4fb997620cbb5bd37a5cb48d Mon Sep 17 00:00:00 2001 From: zachary62 Date: Wed, 23 Apr 2025 13:51:37 -0400 Subject: [PATCH] fix start node in viz --- docs/utility_function/viz.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/utility_function/viz.md b/docs/utility_function/viz.md index 324a516..ca4cfb0 100644 --- a/docs/utility_function/viz.md +++ b/docs/utility_function/viz.md @@ -28,11 +28,11 @@ def build_mermaid(start): return parent and link(parent, get_id(node)) visited.add(node) if isinstance(node, Flow): - node.start and parent and link(parent, get_id(node.start)) + node.start_node and parent and link(parent, get_id(node.start_node)) lines.append(f"\n subgraph sub_flow_{get_id(node)}[{type(node).__name__}]") - node.start and walk(node.start) + node.start_node and walk(node.start_node) for nxt in node.successors.values(): - node.start and walk(nxt, get_id(node.start)) or (parent and link(parent, get_id(nxt))) or walk(nxt) + node.start_node and walk(nxt, get_id(node.start_node)) or (parent and link(parent, get_id(nxt))) or walk(nxt) lines.append(" end\n") else: lines.append(f" {(nid := get_id(node))}['{type(node).__name__}']")