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

feat(config): add GenerateContentConfig to LlmAgentConfig

Browse Source 
PiperOrigin-RevId: 789631181
This commit is contained in:
Liang Wu
2025-07-31 23:32:56 -07:00
committed by Copybara-Service
parent db975dfe2a
commit 9656ccc407
3 changed files with 3859 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/agents/config_schemas/AgentConfig.json
+3853 -2
View File
File diff suppressed because it is too large Load Diff
src/google/adk/agents/llm_agent.py
+2
View File
@@ -628,6 +628,8 @@ class LlmAgent(BaseAgent):
)
if config.after_tool_callbacks:
agent.after_tool_callback = resolve_callbacks(config.after_tool_callbacks)
if config.generate_content_config:
agent.generate_content_config = config.generate_content_config
return agent
src/google/adk/agents/llm_agent_config.py
+4
View File
@@ -19,6 +19,7 @@ from typing import List
from typing import Literal
from typing import Optional
from google.genai import types
from pydantic import ConfigDict
from ..tools.base_tool import ToolConfig
@@ -138,3 +139,6 @@ class LlmAgentConfig(BaseAgentConfig):
after_tool_callbacks: Optional[List[CodeConfig]] = None
"""Optional. LlmAgent.after_tool_callbacks."""
generate_content_config: Optional[types.GenerateContentConfig] = None
"""Optional. LlmAgent.generate_content_config."""