models(openai): add GPT-5 variants to OpenAIModelName; list both prefixed and bare GPT-5 IDs in KnownModelName

Author: zachmayer

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -307,6 +307,12 @@
         'openai:gpt-5-2025-08-07',
         'openai:gpt-5-mini-2025-08-07',
         'openai:gpt-5-nano-2025-08-07',
+        'gpt-5',
+        'gpt-5-mini',
+        'gpt-5-nano',
+        'gpt-5-2025-08-07',
+        'gpt-5-mini-2025-08-07',
+        'gpt-5-nano-2025-08-07',
         'openai:o1',
         'openai:o1-2024-12-17',
         'openai:o1-mini',

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -83,7 +83,19 @@
     'OpenAIModelName',
 )
 
-OpenAIModelName = Union[str, AllModels]
+OpenAIModelName = Union[
+    str,
+    AllModels,
+    # Include latest GPT-5 variants until the SDK enumerates them
+    Literal[
+        'gpt-5',
+        'gpt-5-mini',
+        'gpt-5-nano',
+        'gpt-5-2025-08-07',
+        'gpt-5-mini-2025-08-07',
+        'gpt-5-nano-2025-08-07',
+    ],
+]
 """
 Possible OpenAI model names.
 

tests/models/test_model_names.py

@@ -57,15 +57,6 @@ def get_model_names(model_name_type: Any) -> Iterator[str]:
     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') or n.startswith('o3')
     ]
-    # OpenAI SDK may lag new IDs; include latest prefixed GPT-5 variants explicitly
-    openai_names += [
-        'openai:gpt-5',
-        'openai:gpt-5-mini',
-        'openai:gpt-5-nano',
-        'openai:gpt-5-2025-08-07',
-        'openai:gpt-5-mini-2025-08-07',
-        'openai:gpt-5-nano-2025-08-07',
-    ]
     bedrock_names = [f'bedrock:{n}' for n in get_model_names(BedrockModelName)]
     deepseek_names = ['deepseek:deepseek-chat', 'deepseek:deepseek-reasoner']
     huggingface_names = [f'huggingface:{n}' for n in get_model_names(HuggingFaceModelName)]