fix: handle boolean MCP schema leaves

Author: pragnyanramtha

google/genai/_mcp_utils.py

@@ -24,7 +24,6 @@
 import google.auth
 from google.auth.transport.requests import Request
 
-from . import _common
 from . import types
 from ._api_client import _MULTI_REGIONAL_LOCATIONS
 
@@ -124,9 +123,12 @@ def set_mcp_usage_header(headers: dict[str, str]) -> None:
 
 
 def _filter_to_supported_schema(
-    schema: _common.StringDict,
-) -> _common.StringDict:
+    schema: Any,
+) -> Any:
   """Filters the schema to only include fields that are supported by JSONSchema."""
+  if not isinstance(schema, dict):
+    return schema
+
   supported_fields: set[str] = set(types.JSONSchema.model_fields.keys())
 
   supported_fields.update([

google/genai/tests/mcp/test_mcp_to_gemini_tools.py

@@ -198,6 +198,45 @@ def test_properties_conversion():
   ]
 
 
+def test_additional_properties_boolean_conversion():
+  """Test conversion of MCP tools with boolean additionalProperties."""
+  mcp_tools = [
+      mcp_types.Tool(
+          name='tool',
+          description='tool-description',
+          inputSchema={
+              'type': 'object',
+              'properties': {
+                  'key1': {
+                      'type': 'object',
+                      'additionalProperties': False,
+                  },
+              },
+              'additionalProperties': False,
+          },
+      ),
+  ]
+
+  result = _mcp_utils.mcp_to_gemini_tools(mcp_tools)
+
+  assert result == [
+      types.Tool(
+          function_declarations=[
+              types.FunctionDeclaration(
+                  name='tool',
+                  description='tool-description',
+                  parameters=types.Schema(
+                      type='OBJECT',
+                      properties={
+                          'key1': types.Schema(type='OBJECT'),
+                      },
+                  ),
+              ),
+          ],
+      ),
+  ]
+
+
 def test_defs_conversion():
     """Test conversion of MCP tools with shared definitions ($defs)."""
     mcp_tools = [