fix: build_graph is not awaited.

Jacksunwei · copybara-github · commit f4c7d788b01c · 2025-05-27T23:32:21.000-07:00
Fixes google#927 PiperOrigin-RevId: 764113488
diff --git a/src/google/adk/cli/agent_graph.py b/src/google/adk/cli/agent_graph.py
@@ -130,7 +130,7 @@ def draw_edge(from_name, to_name):
 
   draw_node(agent)
   for sub_agent in agent.sub_agents:
-    build_graph(graph, sub_agent, highlight_pairs)
+    await build_graph(graph, sub_agent, highlight_pairs)
     draw_edge(agent.name, sub_agent.name)
   if isinstance(agent, LlmAgent):
     for tool in await agent.canonical_tools():
