Skip to content

Commit c60d836

Browse files
committed
revert milvus change
Signed-off-by: Lacewell, Chaunte W <chaunte.w.lacewell@intel.com>
1 parent 105468e commit c60d836

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

tests/retrievers/test_retrievers_milvus.sh

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ retriever_service_name="retriever-milvus"
1818

1919
function build_docker_images() {
2020
cd $WORKPATH
21-
mkdir -p ${DOCKER_VOLUME_DIRECTORY}/volumes/
2221
docker build --no-cache -t ${REGISTRY:-opea}/retriever:${TAG:-latest} --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy -f comps/retrievers/src/Dockerfile .
2322
if [ $? -ne 0 ]; then
2423
echo "opea/retriever built fail"
@@ -49,17 +48,19 @@ function start_service() {
4948
function validate_microservice() {
5049
local test_embedding="$1"
5150

51+
export PATH="${HOME}/miniforge3/bin:$PATH"
52+
source activate
5253
URL="http://${host_ip}:$RETRIEVER_PORT/v1/retrieval"
5354

54-
HTTP_STATUS=$(http_proxy="" curl -s -o /dev/null -w "%{http_code}" -X POST -d "{\"text\":\"test\",\"embedding\":${test_embedding}}" -H 'Content-Type: application/json' "$URL")
55+
HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" -X POST -d "{\"text\":\"test\",\"embedding\":${test_embedding}}" -H 'Content-Type: application/json' "$URL")
5556
if [ "$HTTP_STATUS" -eq 200 ]; then
5657
echo "[ retriever ] HTTP status is 200. Checking content..."
57-
local CONTENT=$(http_proxy="" curl -s -X POST -d "{\"text\":\"test\",\"embedding\":${test_embedding}}" -H 'Content-Type: application/json' "$URL" | tee ${LOG_PATH}/retriever.log)
58+
local CONTENT=$(curl -s -X POST -d "{\"text\":\"test\",\"embedding\":${test_embedding}}" -H 'Content-Type: application/json' "$URL" | tee ${LOG_PATH}/retriever.log)
5859

5960
if echo "$CONTENT" | grep -q "retrieved_docs"; then
6061
echo "[ retriever ] Content is as expected."
6162
else
62-
echo "[ retriever ] Content does not match the expected result: $CONTENT" >> ${LOG_PATH}/retriever.log
63+
echo "[ retriever ] Content does not match the expected result: $CONTENT"
6364
docker logs ${retriever_service_name} >> ${LOG_PATH}/retriever.log
6465
exit 1
6566
fi
@@ -76,11 +77,8 @@ function stop_docker() {
7677

7778
cd $WORKPATH/comps/retrievers/deployment/docker_compose
7879
docker compose -f compose.yaml down ${service_name} --remove-orphans
79-
80-
cid=$(docker ps -aq --filter "name=tei-embedding-serving" --filter "name=milvus-*")
80+
cid=$(docker ps -aq --filter "name=tei-embedding-serving")
8181
if [[ ! -z "$cid" ]]; then docker stop $cid && docker rm $cid && sleep 1s; fi
82-
83-
rm -rf ${DOCKER_VOLUME_DIRECTORY}/volumes/* || true
8482
}
8583

8684
function main() {
@@ -89,7 +87,7 @@ function main() {
8987
build_docker_images
9088

9189
start_service
92-
test_embedding=$(python3 -c "import random; embedding = [random.uniform(-1, 1) for _ in range(768)]; print(embedding)")
90+
test_embedding=$(python -c "import random; embedding = [random.uniform(-1, 1) for _ in range(768)]; print(embedding)")
9391
validate_microservice "$test_embedding"
9492

9593
stop_docker

0 commit comments

Comments
 (0)