From d83362725d08150832565668d8335db672526d97 Mon Sep 17 00:00:00 2001
From: Liang Wu <wuliang@google.com>
Date: Thu, 10 Jul 2025 14:44:27 -0700
Subject: [PATCH] feat(config): add disallow_transfer_to_parent and
 disallow_transfer_to_peers to LlmAgentConfig

PiperOrigin-RevId: 781690247
---
 .../agents/config_schemas/AgentConfig.json    | 24 +++++++++++++++++++
 src/google/adk/agents/llm_agent.py            | 10 ++++++++
 2 files changed, 34 insertions(+)

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."""