From c785bb2d1073766798f4a143c27d69bfb36ba651 Mon Sep 17 00:00:00 2001 From: rohanray-jpg Date: Fri, 28 Aug 2026 23:22:47 +0530 Subject: [PATCH] Update transfer_to_agent_tool.py exam --- src/google/adk/tools/transfer_to_agent_tool.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/google/adk/tools/transfer_to_agent_tool.py b/src/google/adk/tools/transfer_to_agent_tool.py index fb419dd575a..bd71e3d8739 100644 --- a/src/google/adk/tools/transfer_to_agent_tool.py +++ b/src/google/adk/tools/transfer_to_agent_tool.py @@ -74,14 +74,12 @@ def _get_declaration(self) -> Optional[types.FunctionDeclaration]: return function_decl # Handle parameters (types.Schema object) - if function_decl.parameters: + if function_decl.parameters and function_decl.parameters.properties: agent_name_schema = function_decl.parameters.properties.get('agent_name') - if agent_name_schema: - agent_name_schema.enum = self._agent_names # Handle parameters_json_schema (dict) if function_decl.parameters_json_schema: - properties = function_decl.parameters_json_schema.get('properties', {}) + properties = function_decl.parameters_json_schema.get('properties') or {} if 'agent_name' in properties: properties['agent_name']['enum'] = self._agent_names