From 94dade9aca7db2de03afca40772254b9150a203c Mon Sep 17 00:00:00 2001
From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com>
Date: Wed, 11 Sep 2024 09:58:43 +0800
Subject: [PATCH] Fix UT of ipex_llm.llamaindex (#12055)

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

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 2268f031..7ec27a3a 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -263,7 +263,7 @@ jobs:
         run: |
           pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
           pip install transformers==4.36.2
-          pip install "pydantic>=2.0.0"
+          pip install "pydantic<2.9.1,>=2.0.0"
           bash python/llm/test/run-llm-llamaindex-tests.sh
       - name: Run sentence-transformers uninstallation
         if: ${{ always() }}
@@ -485,7 +485,7 @@ jobs:
             fi
           fi
           pip install transformers==4.36.2
-          pip install "pydantic>=2.0.0"
+          pip install "pydantic<2.9.1,>=2.0.0"
           bash python/llm/test/run-llm-llamaindex-tests-gpu.sh
       - name: Run sentence-transformers uninstallation
         if: ${{ always() }}