phowell
/
pocketflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix start node in viz

This commit is contained in:
zachary62 2025-04-23 13:51:37 -04:00
parent ba17a42a1c
commit c61d7fb3d7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/utility_function/viz.md
View File

@ -28,11 +28,11 @@ def build_mermaid(start):
return parent and link(parent, get_id(node)) return parent and link(parent, get_id(node))
visited.add(node) visited.add(node)
if isinstance(node, Flow): 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__}]") 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(): 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") lines.append(" end\n")
else: else:
lines.append(f" {(nid := get_id(node))}['{type(node).__name__}']") lines.append(f" {(nid := get_id(node))}['{type(node).__name__}']")