diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py index 14d3a45284..6fd0a3c793 100644 --- a/pydantic_ai_slim/pydantic_ai/models/__init__.py +++ b/pydantic_ai_slim/pydantic_ai/models/__init__.py @@ -217,6 +217,9 @@ 'heroku:claude-3-7-sonnet', 'heroku:claude-4-sonnet', 'heroku:claude-3-haiku', + 'heroku:gpt-oss-120b', + 'heroku:nova-lite', + 'heroku:nova-pro', 'huggingface:Qwen/QwQ-32B', 'huggingface:Qwen/Qwen2.5-72B-Instruct', 'huggingface:Qwen/Qwen3-235B-A22B', diff --git a/tests/models/cassettes/test_model_names/test_known_model_names.yaml b/tests/models/cassettes/test_model_names/test_known_model_names.yaml index 4fc94b1460..e66c3747bd 100644 --- a/tests/models/cassettes/test_model_names/test_known_model_names.yaml +++ b/tests/models/cassettes/test_model_names/test_known_model_names.yaml @@ -15,7 +15,7 @@ interactions: response: headers: content-length: - - '550' + - '760' content-security-policy: - default-src 'none'; frame-ancestors 'none' content-type: @@ -56,6 +56,23 @@ interactions: - us type: - text-to-embedding + - model_id: gpt-oss-120b + regions: + - us + type: + - text-to-text + - model_id: nova-lite + regions: + - eu + - us + type: + - text-to-text + - model_id: nova-pro + regions: + - eu + - us + type: + - text-to-text - model_id: stable-image-ultra regions: - us