Edit name of class and text-generation

Author: eliasto

src/huggingface_hub/inference/_providers/__init__.py

@@ -38,7 +38,7 @@
 from .novita import NovitaConversationalTask, NovitaTextGenerationTask, NovitaTextToVideoTask
 from .nscale import NscaleConversationalTask, NscaleTextToImageTask
 from .openai import OpenAIConversationalTask
-from .ovhcloud import OVHcloudAIEndpointsConversationalTask, OVHcloudAIEndpointsTextGenerationTask
+from .ovhcloud import OVHcloudConversationalTask, OVHcloudTextGenerationTask
 from .publicai import PublicAIConversationalTask
 from .replicate import ReplicateImageToImageTask, ReplicateTask, ReplicateTextToImageTask, ReplicateTextToSpeechTask
 from .sambanova import SambanovaConversationalTask, SambanovaFeatureExtractionTask
@@ -169,8 +169,8 @@
         "conversational": OpenAIConversationalTask(),
     },
     "ovhcloud": {
-        "conversational": OVHcloudAIEndpointsConversationalTask(),
-        "text-generation": OVHcloudAIEndpointsTextGenerationTask(),
+        "conversational": OVHcloudConversationalTask(),
+        "text-generation": OVHcloudTextGenerationTask(),
     },
     "publicai": {
         "conversational": PublicAIConversationalTask(),

src/huggingface_hub/inference/_providers/ovhcloud.py

@@ -8,21 +8,15 @@
 _BASE_URL = "https://oai.endpoints.kepler.ai.cloud.ovh.net"
 
 
-class OVHcloudAIEndpointsConversationalTask(BaseConversationalTask):
+class OVHcloudConversationalTask(BaseConversationalTask):
     def __init__(self):
         super().__init__(provider=_PROVIDER, base_url=_BASE_URL)
 
-    def _prepare_route(self, mapped_model: str, api_key: str) -> str:
-        return "/v1/chat/completions"
 
-
-class OVHcloudAIEndpointsTextGenerationTask(BaseTextGenerationTask):
+class OVHcloudTextGenerationTask(BaseTextGenerationTask):
     def __init__(self):
         super().__init__(provider=_PROVIDER, base_url=_BASE_URL)
 
-    def _prepare_route(self, mapped_model: str, api_key: str) -> str:
-        return "/v1/chat/completions"
-
     def get_response(self, response: Union[bytes, dict], request_params: Optional[RequestParameters] = None) -> Any:
         output = _as_dict(response)["choices"][0]
         return {

tests/test_inference_providers.py

@@ -47,8 +47,8 @@
 from huggingface_hub.inference._providers.nscale import NscaleConversationalTask, NscaleTextToImageTask
 from huggingface_hub.inference._providers.openai import OpenAIConversationalTask
 from huggingface_hub.inference._providers.ovhcloud import (
-    OVHcloudAIEndpointsConversationalTask,
-    OVHcloudAIEndpointsTextGenerationTask,
+    OVHcloudConversationalTask,
+    OVHcloudTextGenerationTask,
 )
 from huggingface_hub.inference._providers.publicai import PublicAIConversationalTask
 from huggingface_hub.inference._providers.replicate import (
@@ -1428,17 +1428,17 @@ def test_prepare_url(self):
 
 class TestOVHcloudAIEndpointsProvider:
     def test_prepare_hf_url_conversational(self):
-        helper = OVHcloudAIEndpointsConversationalTask()
+        helper = OVHcloudConversationalTask()
         url = helper._prepare_url("hf_token", "username/repo_name")
         assert url == "https://router.huggingface.co/ovhcloud/v1/chat/completions"
 
     def test_prepare_url_conversational(self):
-        helper = OVHcloudAIEndpointsConversationalTask()
+        helper = OVHcloudConversationalTask()
         url = helper._prepare_url("ovhcloud_token", "username/repo_name")
         assert url == "https://oai.endpoints.kepler.ai.cloud.ovh.net/v1/chat/completions"
 
     def test_prepare_payload_as_dict(self):
-        helper = OVHcloudAIEndpointsConversationalTask()
+        helper = OVHcloudConversationalTask()
         payload = helper._prepare_payload_as_dict(
             [
                 {"role": "system", "content": "You are a helpful assistant"},
@@ -1473,23 +1473,23 @@ def test_prepare_payload_as_dict(self):
         }
 
     def test_prepare_route_conversational(self):
-        helper = OVHcloudAIEndpointsConversationalTask()
+        helper = OVHcloudConversationalTask()
         assert helper._prepare_route("username/repo_name", "hf_token") == "/v1/chat/completions"
 
     def test_prepare_url_text_generation(self):
-        helper = OVHcloudAIEndpointsTextGenerationTask()
+        helper = OVHcloudTextGenerationTask()
         url = helper._prepare_url("hf_token", "username/repo_name")
-        assert url == "https://router.huggingface.co/ovhcloud/v1/chat/completions"
+        assert url == "https://router.huggingface.co/ovhcloud/v1/completions"
 
         url = helper._prepare_url("ovhcloud_token", "username/repo_name")
-        assert url == "https://oai.endpoints.kepler.ai.cloud.ovh.net/v1/chat/completions"
+        assert url == "https://oai.endpoints.kepler.ai.cloud.ovh.net/v1/completions"
 
     def test_prepare_route_text_generation(self):
-        helper = OVHcloudAIEndpointsTextGenerationTask()
-        assert helper._prepare_route("username/repo_name", "hf_token") == "/v1/chat/completions"
+        helper = OVHcloudTextGenerationTask()
+        assert helper._prepare_route("username/repo_name", "hf_token") == "/v1/completions"
 
     def test_prepare_payload_as_dict_text_generation(self):
-        helper = OVHcloudAIEndpointsTextGenerationTask()
+        helper = OVHcloudTextGenerationTask()
         payload = helper._prepare_payload_as_dict(
             "Once upon a time",
             {"temperature": 0.7, "max_tokens": 100},
@@ -1509,7 +1509,7 @@ def test_prepare_payload_as_dict_text_generation(self):
         }
 
     def test_text_generation_get_response(self):
-        helper = OVHcloudAIEndpointsTextGenerationTask()
+        helper = OVHcloudTextGenerationTask()
         response = helper.get_response(
             {
                 "choices": [