Refactor visualization node label formatting in get_all_nodes function

MartinEBravo · MartinEBravo · commit 57ecebfe3520 · 2025-03-25T18:16:29.000+01:00
diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py
@@ -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];"
             )

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ def get_all_nodes(agent: Agent, parent: Optional[Agent] = None) -> str:`
`57`	`57`	`for handoff in agent.handoffs:`
`58`	`58`	`if isinstance(handoff, Handoff):`
`59`	`59`	`parts.append(`
`60`		`- f'"{handoff.agent_name}" [label="{handoff.agent_name}", shape=box, '`
	`60`	`+ f'"{handoff.agent_name}" [label="{handoff.agent_name}", '`
`61`	`61`	`f"shape=box, style=filled, style=rounded, "`
`62`	`62`	`f"fillcolor=lightyellow, width=1.5, height=0.8];"`
`63`	`63`	`)`