diff --git a/src/google/adk/agents/config_schemas/AgentConfig.json b/src/google/adk/agents/config_schemas/AgentConfig.json index 98be5dfd..9cbaca81 100644 --- a/src/google/adk/agents/config_schemas/AgentConfig.json +++ b/src/google/adk/agents/config_schemas/AgentConfig.json @@ -76,6 +76,27 @@ ], "default": null, "title": "Disallow Transfer To Peers" + }, + "output_key": { + "anyOf": [ + { + "type": "string" + }, + { + "type": "null" + } + ], + "default": null, + "title": "Output Key" + }, + "include_contents": { + "default": "default", + "enum": [ + "default", + "none" + ], + "title": "Include Contents", + "type": "string" } }, "required": [ diff --git a/src/google/adk/agents/llm_agent.py b/src/google/adk/agents/llm_agent.py index 0926518c..b9b03285 100644 --- a/src/google/adk/agents/llm_agent.py +++ b/src/google/adk/agents/llm_agent.py @@ -537,6 +537,10 @@ class LlmAgent(BaseAgent): agent.disallow_transfer_to_parent = config.disallow_transfer_to_parent if config.disallow_transfer_to_peers: agent.disallow_transfer_to_peers = config.disallow_transfer_to_peers + if config.include_contents != 'default': + agent.include_contents = config.include_contents + if config.output_key: + agent.output_key = config.output_key return agent @@ -562,3 +566,9 @@ class LlmAgentConfig(BaseAgentConfig): disallow_transfer_to_peers: Optional[bool] = None """Optional. LlmAgent.disallow_transfer_to_peers.""" + + output_key: Optional[str] = None + """Optional. LlmAgent.output_key.""" + + include_contents: Literal['default', 'none'] = 'default' + """Optional. LlmAgent.include_contents."""