From 9e426546c3b6998242dafa4a50ca96a333ecefc2 Mon Sep 17 00:00:00 2001 From: "wangjiaju.716" Date: Tue, 12 Aug 2025 13:11:34 +0800 Subject: [PATCH] change subagents type to BaseAgent. --- .gitignore | 2 +- veadk/agent.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 5d228f04..a311a991 100644 --- a/.gitignore +++ b/.gitignore @@ -98,7 +98,7 @@ ipython_config.py # Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control. # This is especially recommended for binary packages to ensure reproducibility, and is more # commonly ignored for libraries. -#uv.lock +uv.lock # poetry # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control. diff --git a/veadk/agent.py b/veadk/agent.py index b7c09a5e..16472133 100644 --- a/veadk/agent.py +++ b/veadk/agent.py @@ -39,6 +39,7 @@ from veadk.tracing.base_tracer import BaseTracer from veadk.utils.logger import get_logger from veadk.utils.patches import patch_asyncio +from google.adk.agents.base_agent import BaseAgent patch_asyncio() logger = get_logger(__name__) @@ -77,7 +78,7 @@ class Agent(LlmAgent): tools: list[ToolUnion] = [] """The tools provided to agent.""" - sub_agents: list[Agent] = Field(default_factory=list, exclude=True) + sub_agents: list[BaseAgent] = Field(default_factory=list, exclude=True) """The sub agents provided to agent.""" knowledgebase: Optional[KnowledgeBase] = None