diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py index 6e842e3b4..832b1d6d3 100644 --- a/pydantic_ai_slim/pydantic_ai/models/__init__.py +++ b/pydantic_ai_slim/pydantic_ai/models/__init__.py @@ -107,8 +107,8 @@ 'google-gla:gemini-2.0-flash-lite-preview-02-05', 'google-gla:gemini-2.0-pro-exp-02-05', 'google-gla:gemini-2.5-flash-preview-04-17', - 'google-gla:gemini-2.5-pro-exp-03-25', - 'google-gla:gemini-2.5-pro-preview-03-25', + 'google-gla:gemini-2.5-pro-exp-05-06', + 'google-gla:gemini-2.5-pro-preview-05-06', 'google-vertex:gemini-1.0-pro', 'google-vertex:gemini-1.5-flash', 'google-vertex:gemini-1.5-flash-8b', @@ -120,8 +120,8 @@ 'google-vertex:gemini-2.0-flash-lite-preview-02-05', 'google-vertex:gemini-2.0-pro-exp-02-05', 'google-vertex:gemini-2.5-flash-preview-04-17', - 'google-vertex:gemini-2.5-pro-exp-03-25', - 'google-vertex:gemini-2.5-pro-preview-03-25', + 'google-vertex:gemini-2.5-pro-exp-05-06', + 'google-vertex:gemini-2.5-pro-preview-05-06', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0125', 'gpt-3.5-turbo-0301', diff --git a/pydantic_ai_slim/pydantic_ai/models/gemini.py b/pydantic_ai_slim/pydantic_ai/models/gemini.py index ea325ada7..d1fe84b18 100644 --- a/pydantic_ai_slim/pydantic_ai/models/gemini.py +++ b/pydantic_ai_slim/pydantic_ai/models/gemini.py @@ -59,8 +59,8 @@ 'gemini-2.0-flash-lite-preview-02-05', 'gemini-2.0-pro-exp-02-05', 'gemini-2.5-flash-preview-04-17', - 'gemini-2.5-pro-exp-03-25', - 'gemini-2.5-pro-preview-03-25', + 'gemini-2.5-pro-exp-05-06', + 'gemini-2.5-pro-preview-05-06', ] """Latest Gemini models.""" diff --git a/tests/models/cassettes/test_gemini/test_image_as_binary_content_tool_response.yaml b/tests/models/cassettes/test_gemini/test_image_as_binary_content_tool_response.yaml index c408c4c3f..223837a87 100644 --- a/tests/models/cassettes/test_gemini/test_image_as_binary_content_tool_response.yaml +++ b/tests/models/cassettes/test_gemini/test_image_as_binary_content_tool_response.yaml @@ -23,7 +23,7 @@ interactions: functionDeclarations: - description: '' name: get_image - uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro-preview-03-25:generateContent + uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro-preview-05-06:generateContent response: headers: alt-svc: @@ -53,7 +53,7 @@ interactions: role: model finishReason: STOP index: 0 - modelVersion: gemini-2.5-pro-preview-03-25 + modelVersion: gemini-2.5-pro-preview-05-06 usageMetadata: candidatesTokenCount: 28 promptTokenCount: 38 @@ -107,7 +107,7 @@ interactions: functionDeclarations: - description: '' name: get_image - uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro-preview-03-25:generateContent + uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro-preview-05-06:generateContent response: headers: alt-svc: @@ -132,7 +132,7 @@ interactions: role: model finishReason: STOP index: 0 - modelVersion: gemini-2.5-pro-preview-03-25 + modelVersion: gemini-2.5-pro-preview-05-06 usageMetadata: candidatesTokenCount: 11 promptTokenCount: 360 diff --git a/tests/models/test_gemini.py b/tests/models/test_gemini.py index 2045598a6..07e13319d 100644 --- a/tests/models/test_gemini.py +++ b/tests/models/test_gemini.py @@ -964,7 +964,7 @@ def handler(request: httpx.Request) -> httpx.Response: async def test_image_as_binary_content_tool_response( allow_model_requests: None, gemini_api_key: str, image_content: BinaryContent ) -> None: - m = GeminiModel('gemini-2.5-pro-preview-03-25', provider=GoogleGLAProvider(api_key=gemini_api_key)) + m = GeminiModel('gemini-2.5-pro-preview-05-06', provider=GoogleGLAProvider(api_key=gemini_api_key)) agent = Agent(m) @agent.tool_plain @@ -992,7 +992,7 @@ async def get_image() -> BinaryContent: ), ToolCallPart(tool_name='get_image', args={}, tool_call_id=IsStr()), ], - model_name='gemini-2.5-pro-preview-03-25', + model_name='gemini-2.5-pro-preview-05-06', timestamp=IsDatetime(), ), ModelRequest( @@ -1014,7 +1014,7 @@ async def get_image() -> BinaryContent: ), ModelResponse( parts=[TextPart(content='The image shows a kiwi fruit, sliced in half.')], - model_name='gemini-2.5-pro-preview-03-25', + model_name='gemini-2.5-pro-preview-05-06', timestamp=IsDatetime(), ), ]