the agent evetn graph is not able to be shown due to missing await keyword. this change is to fix missing await for build_graph

PiperOrigin-RevId: 756876056
This commit is contained in:
Yifan Wang 2025-05-09 13:00:46 -07:00 committed by Copybara-Service
parent 33acb26565
commit 415104743d

View File

@ -141,7 +141,7 @@ async def build_graph(graph, agent: BaseAgent, highlight_pairs):
async def get_agent_graph(root_agent, highlights_pairs, image=False):
print('build graph')
graph = graphviz.Digraph(graph_attr={'rankdir': 'LR', 'bgcolor': '#333537'})
build_graph(graph, root_agent, highlights_pairs)
await build_graph(graph, root_agent, highlights_pairs)
if image:
return graph.pipe(format='png')
else: