Skip to content

Commit a672569

Browse files
authored
Fix security issue (#673)
Signed-off-by: XuhuiRen <[email protected]>
1 parent cff0a4d commit a672569

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

comps/retrievers/qdrant/haystack/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
docarray[full]
22
easyocr
33
fastapi
4-
haystack-ai==2.2.4
4+
haystack-ai==2.3.1
55
langchain_community
66
opentelemetry-api
77
opentelemetry-exporter-otlp

comps/retrievers/qdrant/haystack/retriever_qdrant.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import os
55

6-
from haystack.components.embedders import HuggingFaceTEITextEmbedder, SentenceTransformersTextEmbedder
6+
from haystack.components.embedders import HuggingFaceAPITextEmbedder, SentenceTransformersTextEmbedder
77
from haystack_integrations.components.retrievers.qdrant import QdrantEmbeddingRetriever
88
from haystack_integrations.document_stores.qdrant import QdrantDocumentStore
99
from qdrant_config import EMBED_DIMENSION, EMBED_ENDPOINT, EMBED_MODEL, INDEX_NAME, QDRANT_HOST, QDRANT_PORT
@@ -46,7 +46,7 @@ def retrieve(input: EmbedDoc) -> SearchedDoc:
4646
if __name__ == "__main__":
4747
if EMBED_ENDPOINT:
4848
# create embeddings using TEI endpoint service
49-
embedder = HuggingFaceTEITextEmbedder(url=EMBED_ENDPOINT)
49+
embedder = HuggingFaceAPITextEmbedder(api_type="text_embeddings_inference", api_params={"url": EMBED_ENDPOINT})
5050
else:
5151
# create embeddings using local embedding model
5252
embedder = SentenceTransformersTextEmbedder(model=EMBED_MODEL)

0 commit comments

Comments
 (0)