diff --git a/src/google/adk/tools/agent_tool.py b/src/google/adk/tools/agent_tool.py index 460a83f..3efafdb 100644 --- a/src/google/adk/tools/agent_tool.py +++ b/src/google/adk/tools/agent_tool.py @@ -45,10 +45,9 @@ class AgentTool(BaseTool): skip_summarization: Whether to skip summarization of the agent output. """ - def __init__(self, agent: BaseAgent): + def __init__(self, agent: BaseAgent, skip_summarization: bool = False): self.agent = agent - self.skip_summarization: bool = False - """Whether to skip summarization of the agent output.""" + self.skip_summarization: bool = skip_summarization super().__init__(name=agent.name, description=agent.description)