Skip to content

Commit 94c7f95

Browse files
committed
chore: add support for overriding skip_summarization in AgentTool
1 parent 21d2047 commit 94c7f95

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/google/adk/tools/agent_tool.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,10 +45,9 @@ class AgentTool(BaseTool):
4545
skip_summarization: Whether to skip summarization of the agent output.
4646
"""
4747

48-
def __init__(self, agent: BaseAgent):
48+
def __init__(self, agent: BaseAgent, skip_summarization: bool = False):
4949
self.agent = agent
50-
self.skip_summarization: bool = False
51-
"""Whether to skip summarization of the agent output."""
50+
self.skip_summarization: bool = skip_summarization
5251

5352
super().__init__(name=agent.name, description=agent.description)
5453

0 commit comments

Comments
 (0)