update tests

Signed-off-by: Mustafa <[email protected]>

Author: MSCetin37

CodeGen/tests/test_compose_on_gaudi.sh

@@ -48,7 +48,7 @@ function build_docker_images() {
     git checkout ${VLLM_VER} &> /dev/null && cd ../
 
     echo "Build all the images with --no-cache, check docker_image_build.log for details..."
-    service_list="codegen codegen-gradio-ui"
+    service_list="codegen codegen-gradio-ui llm-textgen vllm-gaudi dataprep retriever embedding"
     docker compose -f build.yaml build ${service_list} --no-cache > ${LOG_PATH}/docker_image_build.log
 
     docker images && sleep 1s

CodeGen/tests/test_compose_on_xeon.sh

@@ -48,7 +48,7 @@ function build_docker_images() {
     cd ../
 
     echo "Build all the images with --no-cache, check docker_image_build.log for details..."
-    service_list="codegen codegen-gradio-ui"
+    service_list="codegen codegen-gradio-ui llm-textgen vllm dataprep retriever embedding"
 
     docker compose -f build.yaml build ${service_list} --no-cache > ${LOG_PATH}/docker_image_build.log