Add OpenAI o1-pro, o3-pro, o3-deep-research, computer-use models (#2234)

Author: zachmayer

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -176,6 +176,7 @@
         'gpt-4o-audio-preview',
         'gpt-4o-audio-preview-2024-10-01',
         'gpt-4o-audio-preview-2024-12-17',
+        'gpt-4o-audio-preview-2025-06-03',
         'gpt-4o-mini',
         'gpt-4o-mini-2024-07-18',
         'gpt-4o-mini-audio-preview',
@@ -229,11 +230,18 @@
         'o1-mini-2024-09-12',
         'o1-preview',
         'o1-preview-2024-09-12',
+        'o1-pro',
+        'o1-pro-2025-03-19',
         'o3',
         'o3-2025-04-16',
+        'o3-deep-research',
+        'o3-deep-research-2025-06-26',
         'o3-mini',
         'o3-mini-2025-01-31',
+        'o3-pro',
+        'o3-pro-2025-06-10',
         'openai:chatgpt-4o-latest',
+        'openai:codex-mini-latest',
         'openai:gpt-3.5-turbo',
         'openai:gpt-3.5-turbo-0125',
         'openai:gpt-3.5-turbo-0301',
@@ -266,6 +274,7 @@
         'openai:gpt-4o-audio-preview',
         'openai:gpt-4o-audio-preview-2024-10-01',
         'openai:gpt-4o-audio-preview-2024-12-17',
+        'openai:gpt-4o-audio-preview-2025-06-03',
         'openai:gpt-4o-mini',
         'openai:gpt-4o-mini-2024-07-18',
         'openai:gpt-4o-mini-audio-preview',
@@ -280,12 +289,22 @@
         'openai:o1-mini-2024-09-12',
         'openai:o1-preview',
         'openai:o1-preview-2024-09-12',
+        'openai:o1-pro',
+        'openai:o1-pro-2025-03-19',
         'openai:o3',
         'openai:o3-2025-04-16',
+        'openai:o3-deep-research',
+        'openai:o3-deep-research-2025-06-26',
         'openai:o3-mini',
         'openai:o3-mini-2025-01-31',
         'openai:o4-mini',
         'openai:o4-mini-2025-04-16',
+        'openai:o4-mini-deep-research',
+        'openai:o4-mini-deep-research-2025-06-26',
+        'openai:o3-pro',
+        'openai:o3-pro-2025-06-10',
+        'openai:computer-use-preview',
+        'openai:computer-use-preview-2025-03-11',
         'test',
     ],
 )

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -50,7 +50,7 @@
 
 try:
     from openai import NOT_GIVEN, APIStatusError, AsyncOpenAI, AsyncStream, NotGiven
-    from openai.types import ChatModel, chat, responses
+    from openai.types import AllModels, chat, responses
     from openai.types.chat import (
         ChatCompletionChunk,
         ChatCompletionContentPartImageParam,
@@ -80,7 +80,7 @@
     'OpenAIModelName',
 )
 
-OpenAIModelName = Union[str, ChatModel]
+OpenAIModelName = Union[str, AllModels]
 """
 Possible OpenAI model names.
 

pydantic_ai_slim/pyproject.toml

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