Skip to content

Commit 3b9e55c

Browse files
authored
Minor fix DocIndexRetriever test (#1266)
1 parent 7d9b34c commit 3b9e55c

File tree

3 files changed

+6
-86
lines changed

3 files changed

+6
-86
lines changed

DocIndexRetriever/docker_compose/intel/hpu/gaudi/compose.yaml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ services:
6161
http_proxy: ${http_proxy}
6262
https_proxy: ${https_proxy}
6363
TEI_EMBEDDING_ENDPOINT: ${TEI_EMBEDDING_ENDPOINT}
64+
LOGFLAG: ${LOGFLAG}
6465
restart: unless-stopped
6566
retriever:
6667
image: ${REGISTRY:-opea}/retriever-redis:${TAG:-latest}
@@ -76,6 +77,7 @@ services:
7677
https_proxy: ${https_proxy}
7778
REDIS_URL: ${REDIS_URL}
7879
INDEX_NAME: ${INDEX_NAME}
80+
LOGFLAG: ${LOGFLAG}
7981
restart: unless-stopped
8082
tei-reranking-service:
8183
image: ghcr.io/huggingface/text-embeddings-inference:cpu-1.5
@@ -109,6 +111,7 @@ services:
109111
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
110112
HF_HUB_DISABLE_PROGRESS_BARS: 1
111113
HF_HUB_ENABLE_HF_TRANSFER: 0
114+
LOGFLAG: ${LOGFLAG}
112115
restart: unless-stopped
113116
doc-index-retriever-server:
114117
image: ${REGISTRY:-opea}/doc-index-retriever:${TAG:-latest}
@@ -130,6 +133,7 @@ services:
130133
- RETRIEVER_SERVICE_HOST_IP=${RETRIEVER_SERVICE_HOST_IP}
131134
- RERANK_SERVICE_HOST_IP=${RERANK_SERVICE_HOST_IP}
132135
- LLM_SERVICE_HOST_IP=${LLM_SERVICE_HOST_IP}
136+
- LOGFLAG=${LOGFLAG}
133137
ipc: host
134138
restart: always
135139

DocIndexRetriever/tests/test.py

Lines changed: 0 additions & 70 deletions
This file was deleted.

DocIndexRetriever/tests/test_compose_on_gaudi.sh

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ function start_services() {
4949

5050
# Start Docker Containers
5151
docker compose up -d
52-
sleep 20
52+
sleep 30
5353
echo "Docker services started!"
5454
}
5555

@@ -93,25 +93,11 @@ function validate_megaservice() {
9393
if [ "$EXIT_CODE" == "1" ]; then
9494
docker logs tei-embedding-gaudi-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
9595
docker logs retriever-redis-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
96-
docker logs reranking-tei-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
96+
docker logs reranking-tei-gaudi-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
9797
docker logs doc-index-retriever-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
9898
exit 1
9999
fi
100100

101-
echo "==============Testing retriever service: ChatCompletion Request================"
102-
cd $WORKPATH/tests
103-
local CONTENT=$(python test.py --host_ip ${ip_address} --request_type chat_completion)
104-
local EXIT_CODE=$(validate "$CONTENT" "OPEA" "doc-index-retriever-service-gaudi")
105-
echo "$EXIT_CODE"
106-
local EXIT_CODE="${EXIT_CODE:0-1}"
107-
echo "return value is $EXIT_CODE"
108-
if [ "$EXIT_CODE" == "1" ]; then
109-
docker logs tei-embedding-gaudi-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
110-
docker logs retriever-redis-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
111-
docker logs reranking-tei-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
112-
docker logs doc-index-retriever-server | tee -a ${LOG_PATH}/doc-index-retriever-service-gaudi.log
113-
exit 1
114-
fi
115101
}
116102

117103
function stop_docker() {

0 commit comments

Comments
 (0)