Bump openai to v2.9.0 (#3708)

Author: DouweM

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -314,6 +314,7 @@
         'gateway/openai:gpt-5.1-chat-latest',
         'gateway/openai:gpt-5.1-codex',
         'gateway/openai:gpt-5.1-mini',
+        'gateway/openai:gpt-5.1-codex-max',
         'gateway/openai:o1',
         'gateway/openai:o1-2024-12-17',
         'gateway/openai:o1-mini',
@@ -489,6 +490,7 @@
         'openai:gpt-5.1-chat-latest',
         'openai:gpt-5.1-codex',
         'openai:gpt-5.1-mini',
+        'openai:gpt-5.1-codex-max',
         'openai:o1',
         'openai:o1-2024-12-17',
         'openai:o1-mini',

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -8,7 +8,7 @@
 from contextlib import asynccontextmanager
 from dataclasses import dataclass, field, replace
 from datetime import datetime
-from typing import TYPE_CHECKING, Any, Literal, cast, overload
+from typing import Any, Literal, cast, overload
 
 from pydantic import ValidationError
 from pydantic_core import to_json
@@ -55,7 +55,7 @@
 from . import Model, ModelRequestParameters, StreamedResponse, check_allow_model_requests, download_item, get_user_agent
 
 try:
-    from openai import NOT_GIVEN, APIConnectionError, APIStatusError, AsyncOpenAI, AsyncStream
+    from openai import NOT_GIVEN, APIConnectionError, APIStatusError, AsyncOpenAI, AsyncStream, Omit, omit
     from openai.types import AllModels, chat, responses
     from openai.types.chat import (
         ChatCompletionChunk,
@@ -90,28 +90,14 @@
     from openai.types.responses.response_status import ResponseStatus
     from openai.types.shared import ReasoningEffort
     from openai.types.shared_params import Reasoning
+
+    OMIT = omit
 except ImportError as _import_error:
     raise ImportError(
         'Please install `openai` to use the OpenAI model, '
         'you can use the `openai` optional group — `pip install "pydantic-ai-slim[openai]"`'
     ) from _import_error
 
-if TYPE_CHECKING:
-    from openai import Omit, omit
-
-    OMIT = omit
-else:
-    # Backward compatibility with openai<2
-    try:
-        from openai import Omit, omit
-
-        OMIT = omit
-    except ImportError:  # pragma: lax no cover
-        from openai import NOT_GIVEN, NotGiven
-
-        OMIT = NOT_GIVEN
-        Omit = NotGiven
-
 
 __all__ = (
     'OpenAIModel',

pydantic_ai_slim/pyproject.toml

@@ -67,7 +67,7 @@ dependencies = [
 # WARNING if you add optional groups, please update docs/install.md
 logfire = ["logfire[httpx]>=3.14.1"]
 # Models
-openai = ["openai>=1.107.2"]
+openai = ["openai>=2.9.0"]
 cohere = ["cohere>=5.18.0; platform_system != 'Emscripten'"]
 vertexai = ["google-auth>=2.36.0", "requests>=2.32.2"]
 google = ["google-genai>=1.51.0"]

tests/models/test_model_names.py

@@ -95,7 +95,16 @@ def get_model_names(model_name_type: Any) -> Iterator[str]:
     generated_names = sorted(all_generated_names + gateway_names + heroku_names + cerebras_names + extra_names)
 
     known_model_names = sorted(get_args(KnownModelName.__value__))
-    assert generated_names == known_model_names
+
+    if generated_names != known_model_names:
+        errors: list[str] = []
+        missing_names = set(generated_names) - set(known_model_names)
+        if missing_names:
+            errors.append(f'Missing names: {missing_names}')
+        extra_names = set(known_model_names) - set(generated_names)
+        if extra_names:
+            errors.append(f'Extra names: {extra_names}')
+        raise AssertionError('\n'.join(errors))
 
 
 class HerokuModel(TypedDict):