Signed-off-by: Li Gang <gang.g.li@intel.com> Co-authored-by: ligang <ligang@ligang-nuc9v.bj.intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
103 lines
3.0 KiB
YAML
103 lines
3.0 KiB
YAML
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
version: "3.8"
|
|
|
|
services:
|
|
redis-vector-db:
|
|
image: redis/redis-stack:7.2.0-v9
|
|
container_name: redis-vector-db
|
|
ports:
|
|
- "6379:6379"
|
|
- "8001:8001"
|
|
dataprep-redis-service:
|
|
image: ${REGISTRY:-opea}/dataprep-redis:${TAG:-latest}
|
|
container_name: dataprep-redis-server
|
|
depends_on:
|
|
- redis-vector-db
|
|
ports:
|
|
- "6007:6007"
|
|
- "6008:6008"
|
|
- "6009:6009"
|
|
environment:
|
|
no_proxy: ${no_proxy}
|
|
http_proxy: ${http_proxy}
|
|
https_proxy: ${https_proxy}
|
|
REDIS_URL: redis://redis-vector-db:6379
|
|
REDIS_HOST: redis-vector-db
|
|
INDEX_NAME: ${INDEX_NAME:-rag-redis}
|
|
TEI_EMBEDDING_ENDPOINT: http://tei-embedding-service:80
|
|
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
|
|
tei-embedding-service:
|
|
image: ghcr.io/huggingface/text-embeddings-inference:cpu-1.5
|
|
container_name: tei-embedding-server
|
|
ports:
|
|
- "6006:80"
|
|
volumes:
|
|
- "/home/ligang/models:/data"
|
|
shm_size: 1g
|
|
environment:
|
|
no_proxy: ${no_proxy}
|
|
http_proxy: ${http_proxy}
|
|
https_proxy: ${https_proxy}
|
|
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
|
|
command: --model-id ${EMBEDDING_MODEL_ID} --auto-truncate
|
|
embedding:
|
|
image: ${REGISTRY:-opea}/embedding-tei:${TAG:-latest}
|
|
container_name: embedding-tei-server
|
|
ports:
|
|
- "6000:6000"
|
|
ipc: host
|
|
depends_on:
|
|
- tei-embedding-service
|
|
environment:
|
|
no_proxy: ${no_proxy}
|
|
http_proxy: ${http_proxy}
|
|
https_proxy: ${https_proxy}
|
|
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
|
|
TEI_EMBEDDING_ENDPOINT: http://tei-embedding-service:80
|
|
restart: unless-stopped
|
|
retriever:
|
|
image: ${REGISTRY:-opea}/retriever-redis:${TAG:-latest}
|
|
container_name: retriever-redis-server
|
|
depends_on:
|
|
- redis-vector-db
|
|
ports:
|
|
- "7000:7000"
|
|
ipc: host
|
|
environment:
|
|
no_proxy: ${no_proxy}
|
|
http_proxy: ${http_proxy}
|
|
https_proxy: ${https_proxy}
|
|
REDIS_URL: redis://redis-vector-db:6379
|
|
INDEX_NAME: ${INDEX_NAME:-rag-redis}
|
|
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
|
|
TEI_EMBEDDING_ENDPOINT: http://tei-embedding-service:80
|
|
restart: unless-stopped
|
|
doc-index-retriever-server:
|
|
image: ${REGISTRY:-opea}/doc-index-retriever:${TAG:-latest}
|
|
container_name: doc-index-retriever-server
|
|
depends_on:
|
|
- redis-vector-db
|
|
- tei-embedding-service
|
|
- embedding
|
|
- retriever
|
|
ports:
|
|
- "8889:8889"
|
|
environment:
|
|
no_proxy: ${no_proxy}
|
|
https_proxy: ${https_proxy}
|
|
http_proxy: ${http_proxy}
|
|
MEGA_SERVICE_HOST_IP: ${MEGA_SERVICE_HOST_IP:-0.0.0.0}
|
|
EMBEDDING_SERVICE_HOST_IP: embedding
|
|
EMBEDDING_SERVICE_PORT: ${EMBEDDING_SERVER_PORT:-6000}
|
|
RETRIEVER_SERVICE_HOST_IP: retriever
|
|
LOGFLAG: ${LOGFLAG}
|
|
ipc: host
|
|
restart: always
|
|
command: --without-rerank
|
|
|
|
networks:
|
|
default:
|
|
driver: bridge
|