fix: build_graph is not awaited.

Fixes https://github.com/google/adk-python/issues/927

PiperOrigin-RevId: 764113488
This commit is contained in:
Wei Sun (Jack) 2025-05-27 23:31:48 -07:00 committed by Copybara-Service
parent 15109c6879
commit f4c7d788b0

View File

@ -130,7 +130,7 @@ async def build_graph(graph, agent: BaseAgent, highlight_pairs):
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():