rename function

src/huggingface_hub/inference/_providers/fal_ai.py

@@ -59,7 +59,7 @@ def prepare_request(
             raise ValueError(
                 "You must provide an api_key to work with fal.ai API or log in with `huggingface-cli login`."
             )
-        mapped_model = self.map_model(model)
+        mapped_model = self._map_model(model)
         headers = {
             **build_hf_headers(token=api_key),
             **headers,
@@ -85,10 +85,7 @@ def prepare_request(
             headers=headers,
         )
 
-    def map_model(
-        self,
-        model: Optional[str],
-    ) -> str:
+    def _map_model(self, model: Optional[str]) -> str:
         """Default implementation for mapping model HF model IDs to provider model IDs."""
         if model is None:
             raise ValueError("Please provide a HF model ID supported by fal.ai.")

src/huggingface_hub/inference/_providers/hf_inference.py

@@ -79,24 +79,23 @@ def prepare_request(
     ) -> RequestParameters:
         if extra_payload is None:
             extra_payload = {}
-        model = self.map_model(model)
-        url = self.build_url(model)
-        data, json = self._prepare_payload(inputs, parameters=parameters, model=model, extra_payload=extra_payload)
+        mapped_model = self._map_model(model)
+        url = self.build_url(mapped_model)
+        data, json = self._prepare_payload(
+            inputs, parameters=parameters, model=mapped_model, extra_payload=extra_payload
+        )
         headers = self.prepare_headers(headers=headers, api_key=api_key)
 
         return RequestParameters(
             url=url,
             task=self.task,
-            model=model,
+            model=mapped_model,
             json=json,
             data=data,
             headers=headers,
         )
 
-    def map_model(
-        self,
-        model: Optional[str],
-    ) -> str:
+    def _map_model(self, model: Optional[str]) -> str:
         return model if model is not None else get_recommended_model(self.task)
 
     def build_url(self, model: str) -> str:

src/huggingface_hub/inference/_providers/replicate.py

@@ -71,7 +71,7 @@ def prepare_request(
             base_url = BASE_URL
             logger.info("Calling Replicate provider directly.")
 
-        mapped_model = self.map_model(model)
+        mapped_model = self._map_model(model)
         url = _build_url(base_url, mapped_model)
 
         headers = {
@@ -91,7 +91,7 @@ def prepare_request(
             headers=headers,
         )
 
-    def map_model(self, model: Optional[str]) -> str:
+    def _map_model(self, model: Optional[str]) -> str:
         """Default implementation for mapping model HF model IDs to provider model IDs."""
         if model is None:
             raise ValueError("Please provide a HF model ID supported by Replicate.")

src/huggingface_hub/inference/_providers/sambanova.py

@@ -63,7 +63,7 @@ def prepare_request(
             logger.info("Calling Sambanova provider directly.")
         headers = {**build_hf_headers(token=api_key), **headers}
 
-        mapped_model = self.map_model(model)
+        mapped_model = self._map_model(model)
         payload = {
             "messages": inputs,
             **{k: v for k, v in parameters.items() if v is not None},
@@ -79,7 +79,7 @@ def prepare_request(
             headers=headers,
         )
 
-    def map_model(self, model: Optional[str]) -> str:
+    def _map_model(self, model: Optional[str]) -> str:
         """Default implementation for mapping model HF model IDs to provider model IDs."""
         if model is None:
             raise ValueError("Please provide a HF model ID supported by Sambanova.")

src/huggingface_hub/inference/_providers/together.py

@@ -105,7 +105,7 @@ def prepare_request(
         else:
             base_url = BASE_URL
             logger.info("Calling Together provider directly.")
-        mapped_model = self.map_model(model)
+        mapped_model = self._map_model(model)
         if "model" in parameters:
             parameters["model"] = mapped_model
         payload = self._prepare_payload(inputs, parameters=parameters)
@@ -119,7 +119,7 @@ def prepare_request(
             headers=headers,
         )
 
-    def map_model(self, model: Optional[str]) -> str:
+    def _map_model(self, model: Optional[str]) -> str:
         """Default implementation for mapping model HF model IDs to provider model IDs."""
         if model is None:
             raise ValueError("Please provide a HF model ID supported by Together.")