Merge pull request #1602 from DavidSchmidt00:fix_graph

PiperOrigin-RevId: 782090252
This commit is contained in:
Copybara-Service
2025-07-11 13:27:57 -07:00
+3 -1
View File
@@ -284,7 +284,9 @@ async def build_graph(
async def get_agent_graph(root_agent, highlights_pairs, image=False):
print('build graph')
graph = graphviz.Digraph(graph_attr={'rankdir': 'LR', 'bgcolor': '#333537'})
graph = graphviz.Digraph(
graph_attr={'rankdir': 'LR', 'bgcolor': '#333537'}, strict=True
)
await build_graph(graph, root_agent, highlights_pairs)
if image:
return graph.pipe(format='png')