encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Prevent agent_graph.py throwing when workflow agent is root agent

Browse Source 
Partial fix for https://github.com/google/adk-python/issues/1170

TODOs:
- UI rendering still has issue to match the event with the correct agent.
- graph building needs further fix when there is a workflow agent in the tree.
PiperOrigin-RevId: 767711701
This commit is contained in:
Wei Sun (Jack)
2025-06-05 12:15:14 -07:00
committed by Copybara-Service
parent 7dbd8c6a5b
commit 4b1c218cbe
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/cli/agent_graph.py
+3 -3
View File
@@ -145,7 +145,7 @@ async def build_graph(
return False return False
def build_cluster(child: graphviz.Digraph, agent: BaseAgent, name: str): def build_cluster(child: graphviz.Digraph, agent: BaseAgent, name: str):
if isinstance(agent, LoopAgent): if isinstance(agent, LoopAgent) and parent_agent:
# Draw the edge from the parent agent to the first sub-agent # Draw the edge from the parent agent to the first sub-agent
draw_edge(parent_agent.name, agent.sub_agents[0].name) draw_edge(parent_agent.name, agent.sub_agents[0].name)
length = len(agent.sub_agents) length = len(agent.sub_agents)
@@ -162,7 +162,7 @@ async def build_graph(
].name, ].name,
) )
currLength += 1 currLength += 1
elif isinstance(agent, SequentialAgent): elif isinstance(agent, SequentialAgent) and parent_agent:
# Draw the edge from the parent agent to the first sub-agent # Draw the edge from the parent agent to the first sub-agent
draw_edge(parent_agent.name, agent.sub_agents[0].name) draw_edge(parent_agent.name, agent.sub_agents[0].name)
length = len(agent.sub_agents) length = len(agent.sub_agents)
@@ -179,7 +179,7 @@ async def build_graph(
) if currLength != length - 1 else None ) if currLength != length - 1 else None
currLength += 1 currLength += 1
elif isinstance(agent, ParallelAgent): elif isinstance(agent, ParallelAgent) and parent_agent:
# Draw the edge from the parent agent to every sub-agent # Draw the edge from the parent agent to every sub-agent
for sub_agent in agent.sub_agents: for sub_agent in agent.sub_agents:
build_graph(child, sub_agent, highlight_pairs) build_graph(child, sub_agent, highlight_pairs)