diff --git a/comps/web_retrievers/deployment/docker_compose/compose.yaml b/comps/web_retrievers/deployment/docker_compose/compose.yaml index d626313eb1..276ca6ed4f 100644 --- 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 index b276a28b3e..cc96a7e4a1 100644 --- 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 index c3b0134961..d893a630de 100644 --- 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