From c94032f97e82f20d88dbbf0e5a276c23ae280d17 Mon Sep 17 00:00:00 2001
From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com>
Date: Wed, 11 Sep 2024 12:11:04 +0800
Subject: [PATCH] Try to fix llamaindex ut again (#12061)

---
 .github/workflows/llm_unit_tests.yml | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 7ec27a3a..cdd2064a 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -261,9 +261,11 @@ jobs:
       - name: Run LLM llamaindex test
         shell: bash
         run: |
-          pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
+          pip install "llama-index-readers-file<0.2.0"
+          pip install "llama-index-vector-stores-postgres<0.2.0"
+          pip install "llama-index-embeddings-huggingface<0.3.0"
           pip install transformers==4.36.2
-          pip install "pydantic<2.9.1,>=2.0.0"
+          pip install "pydantic>=2.0.0"
           bash python/llm/test/run-llm-llamaindex-tests.sh
       - name: Run sentence-transformers uninstallation
         if: ${{ always() }}
@@ -471,7 +473,9 @@ jobs:
       - name: Run LLM llamaindex GPU test
         shell: bash
         run: |
-          pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
+          pip install "llama-index-readers-file<0.2.0"
+          pip install "llama-index-vector-stores-postgres<0.2.0"
+          pip install "llama-index-embeddings-huggingface<0.3.0"
           # Specific oneapi position on arc ut test machines
           if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
             pip install --pre --upgrade ipex-llm[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
@@ -485,7 +489,7 @@ jobs:
             fi
           fi
           pip install transformers==4.36.2
-          pip install "pydantic<2.9.1,>=2.0.0"
+          pip install "pydantic>=2.0.0"
           bash python/llm/test/run-llm-llamaindex-tests-gpu.sh
       - name: Run sentence-transformers uninstallation
         if: ${{ always() }}