@@ -42,21 +42,24 @@ def get_all_nodes(agent: Agent, parent: Agent = None) -> str:
42
42
parts = []
43
43
# Ensure parent agent node is colored
44
44
if not parent :
45
- parts .append (f"""
46
- "{ agent .name } " [label="{ agent .name } ", shape=box, style=filled,
47
- fillcolor=lightyellow, width=1.5, height=0.8];""" )
45
+ parts .append (
46
+ f'"{ agent .name } " [label="{ agent .name } ", shape=box, style=filled, '
47
+ 'fillcolor=lightyellow, width=1.5, height=0.8];'
48
+ )
48
49
49
50
# Smaller tools (ellipse, green)
50
51
for tool in agent .tools :
51
- parts .append (f"""
52
- "{ tool .name } " [label="{ tool .name } ", shape=ellipse, style=filled,
53
- fillcolor=lightgreen, width=0.5, height=0.3];""" )
52
+ parts .append (
53
+ f'"{ tool .name } " [label="{ tool .name } ", shape=ellipse, style=filled, '
54
+ f'fillcolor=lightgreen, width=0.5, height=0.3];'
55
+ )
54
56
55
57
# Bigger handoffs (rounded box, yellow)
56
58
for handoff in agent .handoffs :
57
- parts .append (f"""
58
- "{ handoff .name } " [label="{ handoff .name } ", shape=box, style=filled,
59
- style=rounded, fillcolor=lightyellow, width=1.5, height=0.8];""" )
59
+ parts .append (
60
+ f'"{ handoff .name } " [label="{ handoff .name } ", shape=box, style=filled, style=rounded, '
61
+ f'fillcolor=lightyellow, width=1.5, height=0.8];'
62
+ )
60
63
parts .append (get_all_nodes (handoff ))
61
64
62
65
return "" .join (parts )
0 commit comments