Add o3 support for OpenAIModel (#853)

Author: sydney-runkle

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -112,6 +112,8 @@
     'o1-mini-2024-09-12',
     'o1-preview',
     'o1-preview-2024-09-12',
+    'o3-mini',
+    'o3-mini-2025-01-31',
     'openai:chatgpt-4o-latest',
     'openai:gpt-3.5-turbo',
     'openai:gpt-3.5-turbo-0125',
@@ -149,6 +151,8 @@
     'openai:o1-mini-2024-09-12',
     'openai:o1-preview',
     'openai:o1-preview-2024-09-12',
+    'openai:o3-mini',
+    'openai:o3-mini-2025-01-31',
     'test',
 ]
 """Known model names that can be used with the `model` parameter of [`Agent`][pydantic_ai.Agent].
@@ -311,7 +315,7 @@ def infer_model(model: Model | KnownModelName) -> Model:
         from .openai import OpenAIModel
 
         return OpenAIModel(model[7:])
-    elif model.startswith(('gpt', 'o1')):
+    elif model.startswith(('gpt', 'o1', 'o3')):
         from .openai import OpenAIModel
 
         return OpenAIModel(model)

pydantic_ai_slim/pyproject.toml

@@ -45,7 +45,7 @@ dependencies = [
 [project.optional-dependencies]
 # WARNING if you add optional groups, please update docs/install.md
 logfire = ["logfire>=2.3"]
-openai = ["openai>=1.59.0"]
+openai = ["openai>=1.61.0"]
 cohere = ["cohere>=5.13.11"]
 vertexai = ["google-auth>=2.36.0", "requests>=2.32.3"]
 anthropic = ["anthropic>=0.40.0"]

tests/models/test_model_names.py

@@ -38,7 +38,7 @@ def get_model_names(model_name_type: Any) -> Iterator[str]:
     groq_names = [f'groq:{n}' for n in get_model_names(GroqModelName)]
     mistral_names = [f'mistral:{n}' for n in get_model_names(MistralModelName)]
     openai_names = [f'openai:{n}' for n in get_model_names(OpenAIModelName)] + [
-        n for n in get_model_names(OpenAIModelName) if n.startswith('o1') or n.startswith('gpt')
+        n for n in get_model_names(OpenAIModelName) if n.startswith('o1') or n.startswith('gpt') or n.startswith('o3')
     ]
     extra_names = ['test']