refactor: Add OpenAIModelProfile

Author: eliasto

pydantic_ai_slim/pydantic_ai/providers/ovhcloud.py

@@ -8,6 +8,12 @@
 from pydantic_ai import ModelProfile
 from pydantic_ai.exceptions import UserError
 from pydantic_ai.models import cached_async_http_client
+from pydantic_ai.profiles.deepseek import deepseek_model_profile
+from pydantic_ai.profiles.harmony import harmony_model_profile
+from pydantic_ai.profiles.meta import meta_model_profile
+from pydantic_ai.profiles.mistral import mistral_model_profile
+from pydantic_ai.profiles.openai import OpenAIJsonSchemaTransformer, OpenAIModelProfile
+from pydantic_ai.profiles.qwen import qwen_model_profile
 from pydantic_ai.providers import Provider
 
 try:
@@ -35,7 +41,21 @@ def client(self) -> AsyncOpenAI:
         return self._client
 
     def model_profile(self, model_name: str) -> ModelProfile | None:
-        return None
+        if 'deepseek' in model_name.lower():
+            profile = deepseek_model_profile(model_name)
+        elif 'llama' in model_name.lower():
+            profile = meta_model_profile(model_name)
+        elif 'mistral' in model_name.lower():
+            profile = mistral_model_profile(model_name)
+        elif 'qwen' in model_name.lower():
+            profile = qwen_model_profile(model_name)
+        elif 'openai' in model_name.lower() or 'gpt' in model_name:
+            profile = harmony_model_profile(model_name)
+        else:
+            profile = None
+
+        # As the OVHcloud AI Endpoints API is OpenAI-compatible, let's assume we also need OpenAIJsonSchemaTransformer.
+        return OpenAIModelProfile(json_schema_transformer=OpenAIJsonSchemaTransformer).update(profile)
 
     @overload
     def __init__(self) -> None: ...

tests/providers/test_ovhcloud.py

@@ -2,8 +2,16 @@
 
 import httpx
 import pytest
+from pytest_mock import MockerFixture
 
+from pydantic_ai._json_schema import InlineDefsJsonSchemaTransformer
 from pydantic_ai.exceptions import UserError
+from pydantic_ai.profiles.deepseek import deepseek_model_profile
+from pydantic_ai.profiles.harmony import harmony_model_profile
+from pydantic_ai.profiles.meta import meta_model_profile
+from pydantic_ai.profiles.mistral import mistral_model_profile
+from pydantic_ai.profiles.openai import OpenAIJsonSchemaTransformer
+from pydantic_ai.profiles.qwen import qwen_model_profile
 
 from ..conftest import TestEnv, try_import
 
@@ -51,3 +59,51 @@ def test_ovhcloud_pass_http_client():
     provider = OVHcloudProvider(api_key='your-api-key', http_client=http_client)
     assert isinstance(provider.client, openai.AsyncOpenAI)
     assert provider.client.api_key == 'your-api-key'
+
+
+def test_ovhcloud_model_profile(mocker: MockerFixture):
+    provider = OVHcloudProvider(api_key='your-api-key')
+
+    ns = 'pydantic_ai.providers.ovhcloud'
+
+    # Mock all profile functions
+    deepseek_mock = mocker.patch(f'{ns}.deepseek_model_profile', wraps=deepseek_model_profile)
+    harmony_mock = mocker.patch(f'{ns}.harmony_model_profile', wraps=harmony_model_profile)
+    meta_mock = mocker.patch(f'{ns}.meta_model_profile', wraps=meta_model_profile)
+    mistral_mock = mocker.patch(f'{ns}.mistral_model_profile', wraps=mistral_model_profile)
+    qwen_mock = mocker.patch(f'{ns}.qwen_model_profile', wraps=qwen_model_profile)
+
+    # Test deepseek provider
+    profile = provider.model_profile('DeepSeek-R1-Distill-Llama-70B')
+    deepseek_mock.assert_called_with('DeepSeek-R1-Distill-Llama-70B')
+    assert profile is not None
+    assert profile.json_schema_transformer == OpenAIJsonSchemaTransformer
+
+    # Test harmony (for openai gpt-oss) provider
+    profile = provider.model_profile('gpt-oss-120b')
+    harmony_mock.assert_called_with('gpt-oss-120b')
+    assert profile is not None
+    assert profile.json_schema_transformer == OpenAIJsonSchemaTransformer
+
+    # Test meta provider
+    meta_profile = provider.model_profile('Llama-3.3-70B-Instruct')
+    meta_mock.assert_called_with('Llama-3.3-70B-Instruct')
+    assert meta_profile is not None
+    assert meta_profile.json_schema_transformer == InlineDefsJsonSchemaTransformer
+
+    # Test mistral provider
+    profile = provider.model_profile('Mistral-Small-3.2-24B-Instruct-2506')
+    mistral_mock.assert_called_with('Mistral-Small-3.2-24B-Instruct-2506')
+    assert profile is not None
+    assert profile.json_schema_transformer == OpenAIJsonSchemaTransformer
+
+    # Test qwen provider
+    qwen_profile = provider.model_profile('Qwen3-32B')
+    qwen_mock.assert_called_with('Qwen3-32B')
+    assert qwen_profile is not None
+    assert qwen_profile.json_schema_transformer == InlineDefsJsonSchemaTransformer
+
+    # Test unknown provider
+    unknown_profile = provider.model_profile('unknown-model')
+    assert unknown_profile is not None
+    assert unknown_profile.json_schema_transformer == OpenAIJsonSchemaTransformer