Updated pull request based on feedbacks

Author: pulphix

pydantic_ai_slim/pydantic_ai/profiles/google.py

@@ -4,7 +4,7 @@
 
 from pydantic_ai.exceptions import UserError
 
-from .._json_schema import JsonSchema, JsonSchemaTransformer, flatten_allof
+from .._json_schema import JsonSchema, JsonSchemaTransformer
 from . import ModelProfile
 
 
@@ -35,8 +35,6 @@ def __init__(self, schema: JsonSchema, *, strict: bool | None = None):
         super().__init__(schema, strict=strict, prefer_inlined_defs=True, simplify_nullable_unions=True)
 
     def transform(self, schema: JsonSchema) -> JsonSchema:
-        # Flatten object-only allOf to improve compatibility
-        schema = flatten_allof(schema)
         # Note: we need to remove `additionalProperties: False` since it is currently mishandled by Gemini
         additional_properties = schema.pop(
             'additionalProperties', None