Refactor visualization node label formatting in get_all_nodes function

This commit is contained in:
Martín Bravo 2025-03-25 18:16:29 +01:00
parent 48fad9e2d3
commit 57ecebfe35

View file

@ -57,7 +57,7 @@ def get_all_nodes(agent: Agent, parent: Optional[Agent] = None) -> str:
for handoff in agent.handoffs:
if isinstance(handoff, Handoff):
parts.append(
f'"{handoff.agent_name}" [label="{handoff.agent_name}", shape=box, '
f'"{handoff.agent_name}" [label="{handoff.agent_name}", '
f"shape=box, style=filled, style=rounded, "
f"fillcolor=lightyellow, width=1.5, height=0.8];"
)