plano/tests/e2e/run_prompt_gateway_tests.sh
2025-12-25 17:41:32 -08:00

64 lines
1.5 KiB
Bash
Executable file

#!/bin/bash
# Run prompt gateway e2e tests
# if any of the commands fail, the script will exit
set -e
. ./common_scripts.sh
print_disk_usage
mkdir -p ~/plano_logs
touch ~/plano_logs/modelserver.log
print_debug() {
log "Received signal to stop"
log "Printing debug logs for docker"
log "===================================="
tail -n 100 ../build.log
planoai logs --debug | tail -n 100
}
trap 'print_debug' INT TERM ERR
log starting > ../build.log
log building and running function_calling demo
log ===========================================
cd ../../demos/samples_python/weather_forecast/
docker compose up weather_forecast_service --build -d
cd -
log building and installing plano cli
log ==================================
cd ../../cli
poetry install
cd -
log building docker image for arch gateway
log ======================================
cd ../../
planoai build
cd -
# Once we build plano we have to install the dependencies again to a new virtual environment.
poetry install
log startup arch gateway with function calling demo
cd ../../
planoai down
planoai up demos/samples_python/weather_forecast/config.yaml
cd -
log running e2e tests for prompt gateway
log ====================================
poetry run pytest test_prompt_gateway.py
log shutting down the arch gateway service for prompt_gateway demo
log ===============================================================
planoai down
log shutting down the weather_forecast demo
log =======================================
cd ../../demos/samples_python/weather_forecast
docker compose down
cd -