From 7cf01e6ec839f51388330791a086e6c4a5cf11e5 Mon Sep 17 00:00:00 2001
From: "Keyan (Kyrie) Zhang" <79576162+Zhangky11@users.noreply.github.com>
Date: Wed, 13 Mar 2024 09:52:45 +0800
Subject: [PATCH] Add LangChain upstream ut test (#10349)

* Add LangChain upstream ut test

* Add LangChain upstream ut test

* Specify version numbers in yml script

* Correct langchain-community version
---
 .github/workflows/llm_unit_tests.yml           | 1 +
 python/llm/test/run-llm-langchain-tests-gpu.sh | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 1c6dab55..c8db1b7d 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -363,6 +363,7 @@ jobs:
           pip install -U langchain==0.0.184
           pip install -U chromadb==0.3.25
           pip install -U pandas==2.0.3
+          pip install -U langchain-community==0.0.27
           # Specific oneapi position on arc ut test machines
           if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
             source /opt/intel/oneapi/setvars.sh
diff --git a/python/llm/test/run-llm-langchain-tests-gpu.sh b/python/llm/test/run-llm-langchain-tests-gpu.sh
index bff28198..130fd400 100644
--- a/python/llm/test/run-llm-langchain-tests-gpu.sh
+++ b/python/llm/test/run-llm-langchain-tests-gpu.sh
@@ -15,6 +15,12 @@ start=$(date "+%s")
 
 python -m pytest -s ${LLM_INFERENCE_TEST_DIR}
 
+mkdir ${LLM_INFERENCE_TEST_DIR}/tmp_wget_dir
+wget https://raw.githubusercontent.com/langchain-ai/langchain/master/libs/community/tests/integration_tests/llms/test_bigdl.py -P ${LLM_INFERENCE_TEST_DIR}/tmp_wget_dir
+sed -i "s,model_id=\"[^\"]*\",model_id=\"$LLAMA2_7B_ORIGIN_PATH\",g" ${LLM_INFERENCE_TEST_DIR}/tmp_wget_dir/test_bigdl.py
+python -m pytest -s ${LLM_INFERENCE_TEST_DIR}/tmp_wget_dir
+rm -rf ${LLM_INFERENCE_TEST_DIR}/tmp_wget_dir
+
 now=$(date "+%s")
 time=$((now-start))