From f4edc1f3726a2038458e5fe1ca192de28c46d26c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mart=C3=ADn=20Bravo?= <123977407+MartinEBravo@users.noreply.github.com> Date: Tue, 18 Mar 2025 10:15:41 +0100 Subject: [PATCH] style: improve string formatting in visualization functions --- src/agents/extensions/visualization.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/agents/extensions/visualization.py b/src/agents/extensions/visualization.py index 1b5b7e5..81b6338 100644 --- a/src/agents/extensions/visualization.py +++ b/src/agents/extensions/visualization.py @@ -44,21 +44,21 @@ def get_all_nodes(agent: Agent, parent: Agent = None) -> str: if not parent: parts.append( f'"{agent.name}" [label="{agent.name}", shape=box, style=filled, ' - 'fillcolor=lightyellow, width=1.5, height=0.8];' + "fillcolor=lightyellow, width=1.5, height=0.8];" ) # Smaller tools (ellipse, green) for tool in agent.tools: parts.append( f'"{tool.name}" [label="{tool.name}", shape=ellipse, style=filled, ' - f'fillcolor=lightgreen, width=0.5, height=0.3];' + f"fillcolor=lightgreen, width=0.5, height=0.3];" ) # Bigger handoffs (rounded box, yellow) for handoff in agent.handoffs: parts.append( f'"{handoff.name}" [label="{handoff.name}", shape=box, style=filled, style=rounded, ' - f'fillcolor=lightyellow, width=1.5, height=0.8];' + f"fillcolor=lightyellow, width=1.5, height=0.8];" ) parts.append(get_all_nodes(handoff))