NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Fix formatting

Browse Source 
PiperOrigin-RevId: 766407362
This commit is contained in:
Selcuk Gun 2025-06-02 17:27:46 -07:00 committed by Copybara-Service
parent 174afb3975
commit eb2b9ef88f
1 changed files with 57 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/google/adk/cli/agent_graph.py
View File

@ -19,7 +19,10 @@ from typing import Union
import graphviz
from ..agents import BaseAgent, SequentialAgent, LoopAgent, ParallelAgent
from ..agents import BaseAgent
from ..agents import LoopAgent
from ..agents import ParallelAgent
from ..agents import SequentialAgent
from ..agents.llm_agent import LlmAgent
from ..tools.agent_tool import AgentTool
from ..tools.base_tool import BaseTool
@ -35,7 +38,12 @@ else:
retrieval_tool_module_loaded = True
async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs, parent_agent=None):
async def build_graph(
graph: graphviz.Digraph,
agent: BaseAgent,
highlight_pairs,
parent_agent=None,
):
"""
Build a graph of the agent and its sub-agents.
Args:
@ -56,11 +64,11 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
if isinstance(tool_or_agent, BaseAgent):
# Added Workflow Agent checks for different agent types
if isinstance(tool_or_agent, SequentialAgent):
return tool_or_agent.name + f" (Sequential Agent)"
return tool_or_agent.name + f' (Sequential Agent)'
elif isinstance(tool_or_agent, LoopAgent):
return tool_or_agent.name + f" (Loop Agent)"
return tool_or_agent.name + f' (Loop Agent)'
elif isinstance(tool_or_agent, ParallelAgent):
return tool_or_agent.name + f" (Parallel Agent)"
return tool_or_agent.name + f' (Parallel Agent)'
else:
return tool_or_agent.name
elif isinstance(tool_or_agent, BaseTool):
@ -147,7 +155,12 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
build_graph(child, sub_agent_int_sequential, highlight_pairs)
# Draw the edge between the current sub-agent and the next one
# If it's the last sub-agent, draw an edge to the first one to indicating a loop
draw_edge(agent.sub_agents[currLength].name, agent.sub_agents[0 if currLength == length - 1 else currLength+1 ].name)
draw_edge(
agent.sub_agents[currLength].name,
agent.sub_agents[
0 if currLength == length - 1 else currLength + 1
].name,
)
currLength += 1
elif isinstance(agent, SequentialAgent):
# Draw the edge from the parent agent to the first sub-agent
@ -160,7 +173,10 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
build_graph(child, sub_agent_int_sequential, highlight_pairs)
# Draw the edge between the current sub-agent and the next one
# If it's the last sub-agent, don't draw an edge to avoid a loop
draw_edge(agent.sub_agents[currLength].name, agent.sub_agents[currLength+1].name) if currLength != length - 1 else None
draw_edge(
agent.sub_agents[currLength].name,
agent.sub_agents[currLength + 1].name,
) if currLength != length - 1 else None
currLength += 1
elif isinstance(agent, ParallelAgent):
@ -191,7 +207,9 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
if name in highlight_tuple:
# if in highlight, draw highlight node
if asCluster:
cluster = graphviz.Digraph(name='cluster_' + name) # adding "cluster_" to the name makes the graph render as a cluster subgraph
cluster = graphviz.Digraph(
name='cluster_' + name
) # adding "cluster_" to the name makes the graph render as a cluster subgraph
build_cluster(cluster, agent, name)
graph.subgraph(cluster)
else:
@ -208,7 +226,9 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
# if not in highlight, draw non-highlight node
if asCluster:
cluster = graphviz.Digraph(name='cluster_' + name) # adding "cluster_" to the name makes the graph render as a cluster subgraph
cluster = graphviz.Digraph(
name='cluster_' + name
) # adding "cluster_" to the name makes the graph render as a cluster subgraph
build_cluster(cluster, agent, name)
graph.subgraph(cluster)
@ -234,9 +254,13 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
graph.edge(from_name, to_name, color=light_green, dir='back')
return
# if no need to highlight, color gray
if (should_build_agent_cluster(agent)):
if should_build_agent_cluster(agent):
graph.edge(from_name, to_name, color=light_gray, )
graph.edge(
from_name,
to_name,
color=light_gray,
)
else:
graph.edge(from_name, to_name, arrowhead='none', color=light_gray)
@ -244,7 +268,11 @@ async def build_graph(graph: graphviz.Digraph, agent: BaseAgent, highlight_pairs
for sub_agent in agent.sub_agents:
build_graph(graph, sub_agent, highlight_pairs, agent)
if (not should_build_agent_cluster(sub_agent) and not should_build_agent_cluster(agent)): # This is to avoid making a node for a Workflow Agent
if not should_build_agent_cluster(
sub_agent
) and not should_build_agent_cluster(
agent
): # This is to avoid making a node for a Workflow Agent
draw_edge(agent.name, sub_agent.name)
if isinstance(agent, LlmAgent):
for tool in await agent.canonical_tools():