Enable GraphRAG and ProductivitySuite model cache for docker compose test. (#1608)

Signed-off-by: ZePan110 <ze.pan@intel.com>
This commit is contained in:
ZePan110
2025-03-13 11:23:03 +08:00
committed by GitHub
parent 0701b8cfff
commit d670dbf0aa
4 changed files with 8 additions and 6 deletions

View File

@@ -39,7 +39,7 @@ services:
ports:
- "${TEI_EMBEDDER_PORT:-12000}:80"
volumes:
- "./data:/data"
- "${MODEL_CACHE:-./data}:/data"
shm_size: 1g
environment:
no_proxy: ${no_proxy}
@@ -58,7 +58,7 @@ services:
ports:
- ${LLM_ENDPOINT_PORT:-8008}:80
volumes:
- "${DATA_PATH:-./data}:/data"
- "${MODEL_CACHE:-./data}:/data"
environment:
no_proxy: ${no_proxy}
http_proxy: ${http_proxy}

View File

@@ -9,6 +9,7 @@ echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
echo "TAG=IMAGE_TAG=${IMAGE_TAG}"
export REGISTRY=${IMAGE_REPO}
export TAG=${IMAGE_TAG}
export MODEL_CACHE=${model_cache:-"./data"}
WORKPATH=$(dirname "$PWD")
LOG_PATH="$WORKPATH/tests"

View File

@@ -39,7 +39,7 @@ services:
ports:
- "6006:80"
volumes:
- "./data_embedding:/data"
- "${MODEL_CACHE:-./data_embedding}:/data"
shm_size: 1g
environment:
no_proxy: ${no_proxy}
@@ -95,7 +95,7 @@ services:
ports:
- "8808:80"
volumes:
- "./data_tei:/data"
- "${MODEL_CACHE:-./data_tei}:/data"
shm_size: 1g
environment:
no_proxy: ${no_proxy}
@@ -136,7 +136,7 @@ services:
ports:
- "9009:80"
volumes:
- "./data:/data"
- "${MODEL_CACHE:-./data}:/data"
shm_size: 1g
environment:
no_proxy: ${no_proxy}
@@ -209,7 +209,7 @@ services:
ports:
- "8028:80"
volumes:
- "./data_codegen:/data"
- "${MODEL_CACHE:-./data_codegen}:/data"
shm_size: 1g
environment:
no_proxy: ${no_proxy}

View File

@@ -9,6 +9,7 @@ echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
echo "TAG=IMAGE_TAG=${IMAGE_TAG}"
export REGISTRY=${IMAGE_REPO}
export TAG=${IMAGE_TAG}
export MODEL_CACHE=${model_cache:-"./data"}
WORKPATH=$(dirname "$PWD")
LOG_PATH="$WORKPATH/tests"