#/bin/bash
# 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/getting_started/weather_forecast/
docker compose up weather_forecast_service --build -d
cd -

log building and installing plano cli
log ==================================
cd ../../cli
uv sync
uv tool install .
cd -

log building docker image for plano gateway
log ======================================
cd ../../
planoai build
cd -

# Once we build plano we have to install the dependencies again to a new virtual environment.
uv sync

log startup plano gateway with function calling demo
cd ../../
planoai down --docker
planoai up --docker demos/getting_started/weather_forecast/config.yaml
cd -

log running e2e tests for prompt gateway
log ====================================
uv run pytest test_prompt_gateway.py

log shutting down the plano gateway service for prompt_gateway demo
log ===============================================================
planoai down --docker

log startup plano gateway with model alias routing demo
cd ../../
planoai up --docker demos/llm_routing/model_alias_routing/config_with_aliases.yaml
cd -

log running e2e tests for model alias routing
log ========================================
uv run pytest test_model_alias_routing.py

log running e2e tests for openai responses api client
log ========================================
uv run pytest test_openai_responses_api_client.py

log startup plano gateway with state storage for openai responses api client demo
planoai down --docker
planoai up --docker config_memory_state_v1_responses.yaml

log running e2e tests for openai responses api client
log ========================================
uv run pytest test_openai_responses_api_client_with_state.py

log shutting down the weather_forecast demo
log =======================================
cd ../../demos/getting_started/weather_forecast
docker compose down
cd -