Skip to content

Commit

Permalink
debug ut issue
Browse files Browse the repository at this point in the history
Signed-off-by: minmin-intel <[email protected]>
  • Loading branch information
minmin-intel committed Dec 9, 2024
1 parent 7622cfd commit b80ee58
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 7 deletions.
13 changes: 8 additions & 5 deletions tests/agent/sql_agent_test/test_sql_agent.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,10 @@ export agent_container_name="test-comps-agent-endpoint"
export ip_address=$(hostname -I | awk '{print $1}')

vllm_port=8084
vllm_volume=$WORKPATH/data #${HF_CACHE_DIR}
vllm_volume=$WORKPATH #${HF_CACHE_DIR}
echo "vllm volume: $vllm_volume"
ls $vllm_volume

export model=meta-llama/Meta-Llama-3.1-70B-Instruct
export HUGGINGFACEHUB_API_TOKEN=${HF_TOKEN}
export LLM_MODEL_ID="meta-llama/Meta-Llama-3.1-70B-Instruct"
Expand Down Expand Up @@ -161,14 +164,14 @@ function run_benchmark() {
python3 test_tag_bench.py --query_file $query_file --output_dir $outdir --output_file $outfile
}

echo "Building docker image...."
build_docker_images
# echo "Building docker image...."
# build_docker_images

echo "Preparing data...."
prepare_data

echo "Building vllm docker image...."
build_vllm_docker_images
# echo "Building vllm docker image...."
# build_vllm_docker_images

echo "Launching vllm service...."
start_vllm_service
Expand Down
6 changes: 4 additions & 2 deletions tests/agent/test_agent_langchain_on_intel_hpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -309,6 +309,8 @@ function validate_sql_agent(){
if [ "$EXIT_CODE" == "1" ]; then
echo "==================SQL Agent logs ======================"
docker logs test-comps-agent-endpoint
echo "================== vllm gaudi service logs ======================"
docker logs test-comps-vllm-gaudi-service
exit 1
fi
}
Expand Down Expand Up @@ -406,10 +408,10 @@ function main() {
# stop_agent_docker

stop_docker

mv $WORKPATH/data $WORKPATH/hub
# test sql agent
validate_sql_agent

mv $WORKPATH/hub $WORKPATH/data
stop_docker

echo y | docker system prune 2>&1 > /dev/null
Expand Down

0 comments on commit b80ee58

Please sign in to comment.