small fixes

deepset-ai · Apr 5, 2024 · 14cedf4 · 14cedf4
1 parent fb21201
commit 14cedf4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/haystack/components/embedders/hugging_face_api_text_embedder.py b/haystack/components/embedders/hugging_face_api_text_embedder.py
@@ -40,7 +40,7 @@ class HuggingFaceAPITextEmbedder:
     ```python
     from haystack.components.embedders import HuggingFaceAPITextEmbedder
     from haystack.utils import Secret
-    text_embedder = HuggingFaceAPITextEmbedder(api_type="serverless_inference_api",
+    text_embedder = HuggingFaceAPITextEmbedder(api_type="inference_endpoints",
                                                api_params={"model": "BAAI/bge-small-en-v1.5"},
                                                token=Secret.from_token("<your-api-key>"))
 

diff --git a/test/components/embedders/test_hugging_face_api_text_embedder.py b/test/components/embedders/test_hugging_face_api_text_embedder.py
@@ -1,12 +1,10 @@
-from unittest.mock import MagicMock, Mock, patch
+from unittest.mock import MagicMock, patch
 
 import pytest
-from huggingface_hub import TextGenerationOutputToken, TextGenerationStreamDetails, TextGenerationStreamOutput
 from huggingface_hub.utils import RepositoryNotFoundError
 from numpy import array, random
 
 from haystack.components.embedders import HuggingFaceAPITextEmbedder
-from haystack.dataclasses import StreamingChunk
 from haystack.utils.auth import Secret
 from haystack.utils.hf import HFEmbeddingAPIType