feat(config): support input/output schema by fully-qualified code reference

wuliang229 · copybara-github · commit dfee06ac067e · 2025-07-21T12:53:13.000-07:00
PiperOrigin-RevId: 785541326
diff --git a/src/google/adk/agents/config_schemas/AgentConfig.json b/src/google/adk/agents/config_schemas/AgentConfig.json
@@ -163,6 +163,28 @@
           "default": null,
           "title": "Disallow Transfer To Peers"
         },
+        "input_schema": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/CodeConfig"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
+        "output_schema": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/CodeConfig"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
         "output_key": {
           "anyOf": [
             {
diff --git a/src/google/adk/agents/llm_agent.py b/src/google/adk/agents/llm_agent.py
@@ -563,6 +563,7 @@ def from_config(
       config_abs_path: str,
   ) -> LlmAgent:
     from .config_agent_utils import resolve_callbacks
+    from .config_agent_utils import resolve_code_reference
 
     agent = super().from_config(config, config_abs_path)
     if config.model:
@@ -575,6 +576,10 @@ def from_config(
       agent.disallow_transfer_to_peers = config.disallow_transfer_to_peers
     if config.include_contents != 'default':
       agent.include_contents = config.include_contents
+    if config.input_schema:
+      agent.input_schema = resolve_code_reference(config.input_schema)
+    if config.output_schema:
+      agent.output_schema = resolve_code_reference(config.output_schema)
     if config.output_key:
       agent.output_key = config.output_key
     if config.tools:
@@ -619,6 +624,12 @@ class LlmAgentConfig(BaseAgentConfig):
   disallow_transfer_to_peers: Optional[bool] = None
   """Optional. LlmAgent.disallow_transfer_to_peers."""
 
+  input_schema: Optional[CodeConfig] = None
+  """Optional. LlmAgent.input_schema."""
+
+  output_schema: Optional[CodeConfig] = None
+  """Optional. LlmAgent.output_schema."""
+
   output_key: Optional[str] = None
   """Optional. LlmAgent.output_key."""
 
