Skip to content

Commit 7dc9ff0

Browse files
committed
Update google.py
1 parent d7b063e commit 7dc9ff0

File tree

1 file changed

+2
-1
lines changed
  • pydantic_ai_slim/pydantic_ai/profiles

1 file changed

+2
-1
lines changed

pydantic_ai_slim/pydantic_ai/profiles/google.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from . import ModelProfile
88
from ._json_schema import JsonSchema, JsonSchemaTransformer
9-
9+
import time
1010

1111
def google_model_profile(model_name: str) -> ModelProfile | None:
1212
"""Get the model profile for a Google model."""
@@ -32,6 +32,7 @@ def __init__(self, schema: JsonSchema, *, strict: bool | None = None):
3232
super().__init__(schema, strict=strict, prefer_inlined_defs=True, simplify_nullable_unions=True)
3333

3434
def transform(self, schema: JsonSchema) -> JsonSchema:
35+
time.sleep(0.001)
3536
# Note: we need to remove `additionalProperties: False` since it is currently mishandled by Gemini
3637
additional_properties = schema.pop(
3738
'additionalProperties', None

0 commit comments

Comments
 (0)