huggingface-hub 0.30.0rc3__py3-none-any.whl → 0.30.2__py3-none-any.whl

huggingface_hub/__init__.py

@@ -46,7 +46,7 @@ import sys
 from typing import TYPE_CHECKING
 
 
-__version__ = "0.30.0.rc3"
+__version__ = "0.30.2"
 
 # Alphabetical order of definitions is ensured in tests
 # WARNING: any comment added in this dictionary definition will be lost when

huggingface_hub/inference/_client.py

@@ -2400,8 +2400,8 @@ class InferenceClient:
         # Data can be a single element (dict) or an iterable of dicts where we select the first element of.
         if isinstance(data, list):
             data = data[0]
-
-        return TextGenerationOutput.parse_obj_as_instance(data) if details else data["generated_text"]
+        response = provider_helper.get_response(data, request_parameters)
+        return TextGenerationOutput.parse_obj_as_instance(response) if details else response["generated_text"]
 
     def text_to_image(
         self,

huggingface_hub/inference/_generated/_async_client.py

@@ -2456,8 +2456,8 @@ class AsyncInferenceClient:
         # Data can be a single element (dict) or an iterable of dicts where we select the first element of.
         if isinstance(data, list):
             data = data[0]
-
-        return TextGenerationOutput.parse_obj_as_instance(data) if details else data["generated_text"]
+        response = provider_helper.get_response(data, request_parameters)
+        return TextGenerationOutput.parse_obj_as_instance(response) if details else response["generated_text"]
 
     async def text_to_image(
         self,

huggingface_hub/inference/_providers/hf_inference.py

@@ -82,7 +82,7 @@ class HFInferenceBinaryInputTask(HFInferenceTask):
 
 class HFInferenceConversational(HFInferenceTask):
     def __init__(self):
-        super().__init__("text-generation")
+        super().__init__("conversational")
 
     def _prepare_payload_as_dict(self, inputs: Any, parameters: Dict, mapped_model: str) -> Optional[Dict]:
         payload_model = parameters.get("model") or mapped_model
@@ -151,6 +151,14 @@ def _check_supported_task(model: str, task: str) -> None:
             return  # Only conversational allowed if tagged as conversational
         raise ValueError("Non-conversational image-text-to-text task is not supported.")
 
+    if (
+        task in ("feature-extraction", "sentence-similarity")
+        and pipeline_tag in ("feature-extraction", "sentence-similarity")
+        and task in tags
+    ):
+        # feature-extraction and sentence-similarity are interchangeable for HF Inference
+        return
+
     # For all other tasks, just check pipeline tag
     if pipeline_tag != task:
         raise ValueError(

huggingface_hub/inference/_providers/nebius.py

@@ -14,6 +14,16 @@ class NebiusTextGenerationTask(BaseTextGenerationTask):
     def __init__(self):
         super().__init__(provider="nebius", base_url="https://api.studio.nebius.ai")
 
+    def get_response(self, response: Union[bytes, Dict], request_params: Optional[RequestParameters] = None) -> Any:
+        output = _as_dict(response)["choices"][0]
+        return {
+            "generated_text": output["text"],
+            "details": {
+                "finish_reason": output.get("finish_reason"),
+                "seed": output.get("seed"),
+            },
+        }
+
 
 class NebiusConversationalTask(BaseConversationalTask):
     def __init__(self):

huggingface_hub/inference/_providers/novita.py

@@ -22,6 +22,16 @@ class NovitaTextGenerationTask(BaseTextGenerationTask):
         # there is no v1/ route for novita
         return "/v3/openai/completions"
 
+    def get_response(self, response: Union[bytes, Dict], request_params: Optional[RequestParameters] = None) -> Any:
+        output = _as_dict(response)["choices"][0]
+        return {
+            "generated_text": output["text"],
+            "details": {
+                "finish_reason": output.get("finish_reason"),
+                "seed": output.get("seed"),
+            },
+        }
+
 
 class NovitaConversationalTask(BaseConversationalTask):
     def __init__(self):

huggingface_hub/inference/_providers/together.py

@@ -35,6 +35,16 @@ class TogetherTextGenerationTask(BaseTextGenerationTask):
     def __init__(self):
         super().__init__(provider=_PROVIDER, base_url=_BASE_URL)
 
+    def get_response(self, response: Union[bytes, Dict], request_params: Optional[RequestParameters] = None) -> Any:
+        output = _as_dict(response)["choices"][0]
+        return {
+            "generated_text": output["text"],
+            "details": {
+                "finish_reason": output.get("finish_reason"),
+                "seed": output.get("seed"),
+            },
+        }
+
 
 class TogetherConversationalTask(BaseConversationalTask):
     def __init__(self):

{huggingface_hub-0.30.0rc3.dist-info → huggingface_hub-0.30.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.30.0rc3
+Version: 0.30.2
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.

{huggingface_hub-0.30.0rc3.dist-info → huggingface_hub-0.30.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-huggingface_hub/__init__.py,sha256=-tkGfHRGo-dMkwcyCshwx-CxxmmzNNkVA7t-NF1JzHo,49372
+huggingface_hub/__init__.py,sha256=3XEq6NrvmanS_Vl2Ozamcs5TfIynXdQI0ima4zPAxus,49368
 huggingface_hub/_commit_api.py,sha256=cihZ2Zn4_AEjSVSuxwN7Sr42aKz2a38OkBFSqD-JB7I,38777
 huggingface_hub/_commit_scheduler.py,sha256=tfIoO1xWHjTJ6qy6VS6HIoymDycFPg0d6pBSZprrU2U,14679
 huggingface_hub/_inference_endpoints.py,sha256=SLoZOQtv_hNl0Xuafo34L--zuCZ3zSJja2tSkYkG5V4,17268
@@ -40,10 +40,10 @@ huggingface_hub/commands/upload_large_folder.py,sha256=P-EO44JWVl39Ax4b0E0Z873d0
 huggingface_hub/commands/user.py,sha256=M6Ef045YcyV4mFCbLaTRPciQDC6xtV9MMheeen69D0E,11168
 huggingface_hub/commands/version.py,sha256=vfCJn7GO1m-DtDmbdsty8_RTVtnZ7lX6MJsx0Bf4e-s,1266
 huggingface_hub/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-huggingface_hub/inference/_client.py,sha256=9QydbGgybtFeAsqxFVCG-Usb8GIUcK2lAU0xfQ_Tl_8,162678
+huggingface_hub/inference/_client.py,sha256=7m_WAbLasJKcB-TXgEGNU-QSXxhmz8T8Uv-3CWPgdds,162759
 huggingface_hub/inference/_common.py,sha256=iwCkq2fWE1MVoPTeeXN7UN5FZi7g5fZ3K8PHSOCi5dU,14591
 huggingface_hub/inference/_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-huggingface_hub/inference/_generated/_async_client.py,sha256=k20fioyWjC7B8UohguFZSIp6GR-CbJgp3fCvNhgzMGE,168883
+huggingface_hub/inference/_generated/_async_client.py,sha256=zE3dZSBh_RdTiH5wI8Fu8wZbA_KFNzQbFgQJxK-4pEw,168964
 huggingface_hub/inference/_generated/types/__init__.py,sha256=--r3nBmBRtgyQR9TkdQl53_crcKYJhWfTkFK2VE2gUk,6307
 huggingface_hub/inference/_generated/types/audio_classification.py,sha256=Jg3mzfGhCSH6CfvVvgJSiFpkz6v4nNA0G4LJXacEgNc,1573
 huggingface_hub/inference/_generated/types/audio_to_audio.py,sha256=2Ep4WkePL7oJwcp5nRJqApwviumGHbft9HhXE9XLHj4,891
@@ -84,14 +84,14 @@ huggingface_hub/inference/_providers/cerebras.py,sha256=YT1yFhXvDJiKZcqcJcA_7VZJ
 huggingface_hub/inference/_providers/cohere.py,sha256=GkFsuKSaqsyfeerPx0ewv-EX44MtJ8a3XXEfmAiTpb0,419
 huggingface_hub/inference/_providers/fal_ai.py,sha256=Pko6CHSB30jkuuxBNC6rpDWWWN07lvPCNxxIlteBtik,6099
 huggingface_hub/inference/_providers/fireworks_ai.py,sha256=6uDsaxJRaN2xWNQX8u1bvF8zO-8J31TAnHdsrf_TO5g,337
-huggingface_hub/inference/_providers/hf_inference.py,sha256=1vhEpgBbPF1JEcurm_5PqEQaTs2XoHrrnvudT6LJgdU,6865
+huggingface_hub/inference/_providers/hf_inference.py,sha256=5mP0NaeFdMwI4jYoQaehNd5gc8LlPW-RERO6nI1X5jI,7147
 huggingface_hub/inference/_providers/hyperbolic.py,sha256=ZIwD50fUv3QnL091XlKA7TOs_E33Df5stsewpnbDNZ4,1824
-huggingface_hub/inference/_providers/nebius.py,sha256=VcgjvQCNnFifU4GMFXJENgdqgQXSu4uHcu2hUljNDaY,1593
-huggingface_hub/inference/_providers/novita.py,sha256=_ofZzGxtKn9v9anmS1kElnOiJa9Hmoeyi4Exv_jVOVg,2023
+huggingface_hub/inference/_providers/nebius.py,sha256=wei1W80lMXpJ-3MGw9JRrO_LbRsqYVYgx67lSNYiV1Q,1980
+huggingface_hub/inference/_providers/novita.py,sha256=xlP1xZcFsouAJiJ_mXgaBngUOZ1x9OzEaHPH5DD2Dx0,2410
 huggingface_hub/inference/_providers/openai.py,sha256=J5YO5h6vZrZmf6WC5_sgXcu6qMv6yAd72U16vKXi628,873
 huggingface_hub/inference/_providers/replicate.py,sha256=WIO4DEz1Imgdn02XHxDYwkDxOxF1ir2eba3HgUn1cQg,2348
 huggingface_hub/inference/_providers/sambanova.py,sha256=pR2MajO3ffga9FxzruzrTfTm3eBQ3AC0TPeSIdiQeco,249
-huggingface_hub/inference/_providers/together.py,sha256=NXBZVtfOF5MWsafSJFwmjOfnr8BNVR3JT4uPfPdHvnc,2113
+huggingface_hub/inference/_providers/together.py,sha256=qa3Ns-b5HpJBdSkT8PchLobfE6Z48i8sIcfy6N1a5-A,2500
 huggingface_hub/serialization/__init__.py,sha256=kn-Fa-m4FzMnN8lNsF-SwFcfzug4CucexybGKyvZ8S0,1041
 huggingface_hub/serialization/_base.py,sha256=Df3GwGR9NzeK_SD75prXLucJAzPiNPgHbgXSw-_LTk8,8126
 huggingface_hub/serialization/_dduf.py,sha256=s42239rLiHwaJE36QDEmS5GH7DSmQ__BffiHJO5RjIg,15424
@@ -127,9 +127,9 @@ huggingface_hub/utils/insecure_hashlib.py,sha256=OjxlvtSQHpbLp9PWSrXBDJ0wHjxCBU-
 huggingface_hub/utils/logging.py,sha256=0A8fF1yh3L9Ka_bCDX2ml4U5Ht0tY8Dr3JcbRvWFuwo,4909
 huggingface_hub/utils/sha.py,sha256=OFnNGCba0sNcT2gUwaVCJnldxlltrHHe0DS_PCpV3C4,2134
 huggingface_hub/utils/tqdm.py,sha256=xAKcyfnNHsZ7L09WuEM5Ew5-MDhiahLACbbN2zMmcLs,10671
-huggingface_hub-0.30.0rc3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-huggingface_hub-0.30.0rc3.dist-info/METADATA,sha256=6e5RnSC9W2asip6cNWXB1H9KSvK5mKOZ3xzinFf2LwU,13554
-huggingface_hub-0.30.0rc3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-huggingface_hub-0.30.0rc3.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
-huggingface_hub-0.30.0rc3.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
-huggingface_hub-0.30.0rc3.dist-info/RECORD,,
+huggingface_hub-0.30.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+huggingface_hub-0.30.2.dist-info/METADATA,sha256=EIYpeFwEendNmFnsE0LVZoFFv0t1RzBhBaUjxc4OKuo,13551
+huggingface_hub-0.30.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+huggingface_hub-0.30.2.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
+huggingface_hub-0.30.2.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
+huggingface_hub-0.30.2.dist-info/RECORD,,