Update pydantic_ai_slim/pydantic_ai/models/google.py

DouweM · web-flow · commit dd63c0b37261 · 2025-11-12T17:49:08.000-06:00
diff --git a/pydantic_ai_slim/pydantic_ai/models/google.py b/pydantic_ai_slim/pydantic_ai/models/google.py
@@ -267,7 +267,7 @@ async def count_tokens(
             messages, model_settings, model_request_parameters
         )
 
-        # Annoyingly, the type of `GenerateContentConfigDict.get` is "partially `Unknown`" because `response_json_schema` includes `typing._UnionGenericAlias`,
+        # Annoyingly, the type of `GenerateContentConfigDict.get` is "partially `Unknown`" because `response_schema` includes `typing._UnionGenericAlias`,
         # so without this we'd need `pyright: ignore[reportUnknownMemberType]` on every line and wouldn't get type checking anyway.
         generation_config = cast(dict[str, Any], generation_config)
 

Original file line number	Diff line number	Diff line change
`@@ -267,7 +267,7 @@ async def count_tokens(`
`267`	`267`	`messages, model_settings, model_request_parameters`
`268`	`268`	`)`
`269`	`269`
`270`		- # Annoyingly, the type of `GenerateContentConfigDict.get` is "partially `Unknown`" because `response_json_schema` includes `typing._UnionGenericAlias`,
	`270`	+ # Annoyingly, the type of `GenerateContentConfigDict.get` is "partially `Unknown`" because `response_schema` includes `typing._UnionGenericAlias`,
`271`	`271`	# so without this we'd need `pyright: ignore[reportUnknownMemberType]` on every line and wouldn't get type checking anyway.
`272`	`272`	`generation_config = cast(dict[str, Any], generation_config)`
`273`	`273`