Skip to content

Commit f4edc1f

Browse files
committed
style: improve string formatting in visualization functions
1 parent b7627cb commit f4edc1f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/agents/extensions/visualization.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,21 +44,21 @@ def get_all_nodes(agent: Agent, parent: Agent = None) -> str:
4444
if not parent:
4545
parts.append(
4646
f'"{agent.name}" [label="{agent.name}", shape=box, style=filled, '
47-
'fillcolor=lightyellow, width=1.5, height=0.8];'
47+
"fillcolor=lightyellow, width=1.5, height=0.8];"
4848
)
4949

5050
# Smaller tools (ellipse, green)
5151
for tool in agent.tools:
5252
parts.append(
5353
f'"{tool.name}" [label="{tool.name}", shape=ellipse, style=filled, '
54-
f'fillcolor=lightgreen, width=0.5, height=0.3];'
54+
f"fillcolor=lightgreen, width=0.5, height=0.3];"
5555
)
5656

5757
# Bigger handoffs (rounded box, yellow)
5858
for handoff in agent.handoffs:
5959
parts.append(
6060
f'"{handoff.name}" [label="{handoff.name}", shape=box, style=filled, style=rounded, '
61-
f'fillcolor=lightyellow, width=1.5, height=0.8];'
61+
f"fillcolor=lightyellow, width=1.5, height=0.8];"
6262
)
6363
parts.append(get_all_nodes(handoff))
6464

0 commit comments

Comments
 (0)