document enum

deepset-ai · Apr 5, 2024 · 430c7d6 · 430c7d6
1 parent 537008b
commit 430c7d6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/haystack/utils/hf.py b/haystack/utils/hf.py
@@ -26,8 +26,13 @@ class HFGenerationAPIType(Enum):
     API type to use for Hugging Face API Generators.
     """
 
+    # HF [Text Generation Inference (TGI)](https://github.com/huggingface/text-generation-inference)
     TEXT_GENERATION_INFERENCE = "text_generation_inference"
+
+    # HF [Inference Endpoints](https://huggingface.co/inference-endpoints)
     INFERENCE_ENDPOINTS = "inference_endpoints"
+
+    # HF [Serverless Inference API](https://huggingface.co/inference-api)
     SERVERLESS_INFERENCE_API = "serverless_inference_api"
 
     def __str__(self):