ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add LangChain upstream ut test (#10349)

Browse source 
* Add LangChain upstream ut test

* Add LangChain upstream ut test

* Specify version numbers in yml script

* Correct langchain-community version
This commit is contained in:
Keyan (Kyrie) Zhang 2024-03-13 09:52:45 +08:00 committed by GitHub
parent aec83a8be6
commit 7cf01e6ec8
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.github/workflows/llm_unit_tests.yml vendored
View file

@ -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

6
python/llm/test/run-llm-langchain-tests-gpu.sh
View file

@ -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))