Function get_agent_graph

src/google/adk/cli/agent_graph.py:300–311

(
    root_agent, highlights_pairs, image=False, dark_mode=True
)

async def get_agent_graph(
    root_agent, highlights_pairs, image=False, dark_mode=True
):
  bg_color = '#333537' if dark_mode else '#ffffff'
  graph = graphviz.Digraph(
      graph_attr={'rankdir': 'LR', 'bgcolor': bg_color}, strict=True
  )
  await build_graph(graph, root_agent, highlights_pairs)
  if image:
    return graph.pipe(format='png')
  else:
    return graph