checkout image build branch

Signed-off-by: chensuyue <[email protected]>
opea-project · Sep 26, 2024 · 2cf7782 · 2cf7782
1 parent 7f66fe8
commit 2cf7782
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/_comps-workflow.yml b/.github/workflows/_comps-workflow.yml
@@ -65,6 +65,7 @@ jobs:
           cd ${{ github.workspace }}
           if [[ $(grep -c "vllm-openvino:" ${docker_compose_yml}) != 0 ]]; then
               git clone https://github.com/vllm-project/vllm.git vllm-openvino
+              cd ./vllm-openvino && git checkout v0.6.1 && cd ../
           fi
 
       - name: Build Image

diff --git a/comps/llms/text-generation/vllm/langchain/dependency/build_docker_vllm_openvino.sh b/comps/llms/text-generation/vllm/langchain/dependency/build_docker_vllm_openvino.sh
@@ -5,6 +5,6 @@
 
 BASEDIR="$( cd "$( dirname "$0" )" && pwd )"
 git clone https://github.com/vllm-project/vllm.git vllm
-cd ./vllm/
+cd ./vllm/ && git checkout v0.6.1
 docker build -t vllm:openvino -f Dockerfile.openvino . --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy
 cd $BASEDIR && rm -rf vllm
diff --git a/comps/llms/text-generation/vllm/llama_index/dependency/build_docker_vllm_openvino.sh b/comps/llms/text-generation/vllm/llama_index/dependency/build_docker_vllm_openvino.sh
@@ -5,6 +5,6 @@
 
 BASEDIR="$( cd "$( dirname "$0" )" && pwd )"
 git clone https://github.com/vllm-project/vllm.git vllm
-cd ./vllm/
+cd ./vllm/ && git checkout v0.6.1
 docker build -t vllm:openvino -f Dockerfile.openvino . --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy
 cd $BASEDIR && rm -rf vllm
diff --git a/tests/llms/test_llms_text-generation_vllm_langchain_openvino.sh b/tests/llms/test_llms_text-generation_vllm_langchain_openvino.sh
@@ -15,8 +15,8 @@ CONTAINER_NAME="test-comps-vllm-openvino-container"
 function build_container() {
     cd $WORKPATH
     git clone https://github.com/vllm-project/vllm.git vllm-openvino
-    cd ./vllm-openvino/ && git checkout v0.6.1
-    # for test, something wrong with main branch image build
+    cd ./vllm-openvino/ && git checkout v0.6.1 # something wrong with main branch image build
+
     docker build --no-cache -t $DOCKER_IMAGE \
       -f Dockerfile.openvino \
       . \