diff --git a/src/google/adk/agents/config_schemas/AgentConfig.json b/src/google/adk/agents/config_schemas/AgentConfig.json index b89b0b9f..4a91f9f3 100644 --- a/src/google/adk/agents/config_schemas/AgentConfig.json +++ b/src/google/adk/agents/config_schemas/AgentConfig.json @@ -37,6 +37,30 @@ "instruction": { "title": "Instruction", "type": "string" + }, + "disallow_transfer_to_parent": { + "anyOf": [ + { + "type": "boolean" + }, + { + "type": "null" + } + ], + "default": null, + "title": "Disallow Transfer To Parent" + }, + "disallow_transfer_to_peers": { + "anyOf": [ + { + "type": "boolean" + }, + { + "type": "null" + } + ], + "default": null, + "title": "Disallow Transfer To Peers" } }, "required": [ diff --git a/src/google/adk/agents/llm_agent.py b/src/google/adk/agents/llm_agent.py index 12b5b7c4..6c5e109a 100644 --- a/src/google/adk/agents/llm_agent.py +++ b/src/google/adk/agents/llm_agent.py @@ -532,6 +532,10 @@ class LlmAgent(BaseAgent): agent.model = config.model if config.instruction: agent.instruction = config.instruction + if config.disallow_transfer_to_parent: + 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 return agent @@ -551,3 +555,9 @@ class LlmAgentConfig(BaseAgentConfig): instruction: str """Required. LlmAgent.instruction.""" + + disallow_transfer_to_parent: Optional[bool] = None + """Optional. LlmAgent.disallow_transfer_to_parent.""" + + disallow_transfer_to_peers: Optional[bool] = None + """Optional. LlmAgent.disallow_transfer_to_peers."""