Add new OpenAI GPT-5 models (#2503)

Author: medaminezghal

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -194,6 +194,13 @@
         'gpt-4o-mini-search-preview-2025-03-11',
         'gpt-4o-search-preview',
         'gpt-4o-search-preview-2025-03-11',
+        'gpt-5',
+        'gpt-5-2025-08-07',
+        'gpt-5-chat-latest',
+        'gpt-5-mini',
+        'gpt-5-mini-2025-08-07',
+        'gpt-5-nano',
+        'gpt-5-nano-2025-08-07',
         'grok:grok-4',
         'grok:grok-4-0709',
         'grok:grok-3',
@@ -313,11 +320,18 @@
         'openai:gpt-4o-mini-search-preview-2025-03-11',
         'openai:gpt-4o-search-preview',
         'openai:gpt-4o-search-preview-2025-03-11',
+        'openai:gpt-5',
+        'openai:gpt-5-2025-08-07',
         'openai:o1',
+        'openai:gpt-5-chat-latest',
         'openai:o1-2024-12-17',
+        'openai:gpt-5-mini',
         'openai:o1-mini',
+        'openai:gpt-5-mini-2025-08-07',
         'openai:o1-mini-2024-09-12',
+        'openai:gpt-5-nano',
         'openai:o1-preview',
+        'openai:gpt-5-nano-2025-08-07',
         'openai:o1-preview-2024-09-12',
         'openai:o1-pro',
         'openai:o1-pro-2025-03-19',

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -59,6 +59,11 @@
     from openai.types.chat.chat_completion_content_part_image_param import ImageURL
     from openai.types.chat.chat_completion_content_part_input_audio_param import InputAudio
     from openai.types.chat.chat_completion_content_part_param import File, FileFile
+    from openai.types.chat.chat_completion_message_custom_tool_call import ChatCompletionMessageCustomToolCall
+    from openai.types.chat.chat_completion_message_function_tool_call import ChatCompletionMessageFunctionToolCall
+    from openai.types.chat.chat_completion_message_function_tool_call_param import (
+        ChatCompletionMessageFunctionToolCallParam,
+    )
     from openai.types.chat.chat_completion_prediction_content_param import ChatCompletionPredictionContentParam
     from openai.types.chat.completion_create_params import (
         WebSearchOptions,
@@ -416,7 +421,14 @@ def _process_response(self, response: chat.ChatCompletion | str) -> ModelRespons
             items.extend(split_content_into_text_and_thinking(choice.message.content, self.profile.thinking_tags))
         if choice.message.tool_calls is not None:
             for c in choice.message.tool_calls:
-                part = ToolCallPart(c.function.name, c.function.arguments, tool_call_id=c.id)
+                if isinstance(c, ChatCompletionMessageFunctionToolCall):
+                    part = ToolCallPart(c.function.name, c.function.arguments, tool_call_id=c.id)
+                elif isinstance(c, ChatCompletionMessageCustomToolCall):  # pragma: no cover
+                    # NOTE: Custom tool calls are not supported.
+                    # See <https://github.com/pydantic/pydantic-ai/issues/2513> for more details.
+                    raise RuntimeError('Custom tool calls are not supported')
+                else:
+                    assert_never(c)
                 part.tool_call_id = _guard_tool_call_id(part)
                 items.append(part)
         return ModelResponse(
@@ -476,7 +488,7 @@ async def _map_messages(self, messages: list[ModelMessage]) -> list[chat.ChatCom
                     openai_messages.append(item)
             elif isinstance(message, ModelResponse):
                 texts: list[str] = []
-                tool_calls: list[chat.ChatCompletionMessageToolCallParam] = []
+                tool_calls: list[ChatCompletionMessageFunctionToolCallParam] = []
                 for item in message.parts:
                     if isinstance(item, TextPart):
                         texts.append(item.content)
@@ -507,8 +519,8 @@ async def _map_messages(self, messages: list[ModelMessage]) -> list[chat.ChatCom
         return openai_messages
 
     @staticmethod
-    def _map_tool_call(t: ToolCallPart) -> chat.ChatCompletionMessageToolCallParam:
-        return chat.ChatCompletionMessageToolCallParam(
+    def _map_tool_call(t: ToolCallPart) -> ChatCompletionMessageFunctionToolCallParam:
+        return ChatCompletionMessageFunctionToolCallParam(
             id=_guard_tool_call_id(t=t),
             type='function',
             function={'name': t.tool_name, 'arguments': t.args_as_json_str()},

pydantic_ai_slim/pyproject.toml

@@ -62,7 +62,7 @@ dependencies = [
 # WARNING if you add optional groups, please update docs/install.md
 logfire = ["logfire>=3.14.1"]
 # Models
-openai = ["openai>=1.92.0"]
+openai = ["openai>=1.99.9"]
 cohere = ["cohere>=5.16.0; platform_system != 'Emscripten'"]
 vertexai = ["google-auth>=2.36.0", "requests>=2.32.2"]
 google = ["google-genai>=1.28.0"]

tests/models/cassettes/test_model_names/test_known_model_names.yaml

@@ -14,12 +14,20 @@ interactions:
     uri: https://us.inference.heroku.com/available-models
   response:
     headers:
+      cache-control:
+      - no-cache, no-store, must-revalidate
+      connection:
+      - keep-alive
       content-length:
       - '760'
       content-security-policy:
       - default-src 'none'; frame-ancestors 'none'
       content-type:
       - application/json
+      expires:
+      - '0'
+      pragma:
+      - no-cache
       strict-transport-security:
       - max-age=63072000
     parsed_body:

tests/models/test_openai.py

@@ -60,6 +60,7 @@
         ChoiceDeltaToolCallFunction,
     )
     from openai.types.chat.chat_completion_message import ChatCompletionMessage
+    from openai.types.chat.chat_completion_message_function_tool_call import ChatCompletionMessageFunctionToolCall
     from openai.types.chat.chat_completion_message_tool_call import Function
     from openai.types.chat.chat_completion_token_logprob import ChatCompletionTokenLogprob
     from openai.types.completion_usage import CompletionUsage, PromptTokensDetails
@@ -241,7 +242,7 @@ async def test_request_structured_response(allow_model_requests: None):
             content=None,
             role='assistant',
             tool_calls=[
-                chat.ChatCompletionMessageToolCall(
+                ChatCompletionMessageFunctionToolCall(
                     id='123',
                     function=Function(arguments='{"response": [1, 2, 123]}', name='final_result'),
                     type='function',
@@ -292,7 +293,7 @@ async def test_request_tool_call(allow_model_requests: None):
                 content=None,
                 role='assistant',
                 tool_calls=[
-                    chat.ChatCompletionMessageToolCall(
+                    ChatCompletionMessageFunctionToolCall(
                         id='1',
                         function=Function(arguments='{"loc_name": "San Fransisco"}', name='get_location'),
                         type='function',
@@ -311,7 +312,7 @@ async def test_request_tool_call(allow_model_requests: None):
                 content=None,
                 role='assistant',
                 tool_calls=[
-                    chat.ChatCompletionMessageToolCall(
+                    ChatCompletionMessageFunctionToolCall(
                         id='2',
                         function=Function(arguments='{"loc_name": "London"}', name='get_location'),
                         type='function',
@@ -722,7 +723,7 @@ async def test_parallel_tool_calls(allow_model_requests: None, parallel_tool_cal
             content=None,
             role='assistant',
             tool_calls=[
-                chat.ChatCompletionMessageToolCall(
+                ChatCompletionMessageFunctionToolCall(
                     id='123',
                     function=Function(arguments='{"response": [1, 2, 3]}', name='final_result'),
                     type='function',