From 57ecebfe35209cf6052d87d3569dfc3983a5845a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mart=C3=ADn=20Bravo?= <123977407+MartinEBravo@users.noreply.github.com> Date: Tue, 25 Mar 2025 18:16:29 +0100 Subject: [PATCH] Refactor visualization node label formatting in get_all_nodes function --- src/agents/extensions/visualization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py index dd75acf..f607985 100644 --- 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];" )