Fix web-retrievers hub client and tei endpoint issue (opea-project#1270)

* fix web-retrievers hub client and tei endpoint issue Signed-off-by: Spycsh <[email protected]>
arangoml · Feb 7, 2025 · ecb7f7b · ecb7f7b
1 parent 5baada8
commit ecb7f7b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/comps/web_retrievers/deployment/docker_compose/compose.yaml b/comps/web_retrievers/deployment/docker_compose/compose.yaml
@@ -15,6 +15,7 @@ services:
       GOOGLE_API_KEY: ${GOOGLE_API_KEY}
       GOOGLE_CSE_ID: ${GOOGLE_CSE_ID}
       HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
+      TEI_EMBEDDING_ENDPOINT: ${TEI_EMBEDDING_ENDPOINT}
     restart: unless-stopped
     depends_on:
       tei-embedding-service:

diff --git a/comps/web_retrievers/src/README.md b/comps/web_retrievers/src/README.md
@@ -46,6 +46,7 @@ export GOOGLE_CSE_ID=${GOOGLE_CSE_ID}
 export TEI_PORT=6060
 export no_proxy=$host_ip,$no_proxy
 export EMBEDDING_MODEL_ID=BAAI/bge-base-en-v1.5
+export TEI_EMBEDDING_ENDPOINT=http://${host_ip}:6060
 
 docker compose -f ../deployment/docker_compose/compose.yaml up web-retriever-service tei-embedding-service -d
 ```

diff --git a/comps/web_retrievers/src/requirements.txt b/comps/web_retrievers/src/requirements.txt
@@ -4,6 +4,7 @@ docarray[full]
 fastapi
 google-api-python-client>=2.100.0
 html2text
+huggingface-hub==0.27.1
 langchain-huggingface
 langchain_community
 opentelemetry-api