name: e2e demo preference based routing tests

permissions:
  contents: read

on:
  push:
    branches:
      - main
  pull_request:

jobs:
  e2e_demo_tests:
    runs-on: ubuntu-latest-m

    steps:
      - name: Checkout code
        uses: actions/checkout@v3

      - name: Set up Python
        uses: actions/setup-python@v4
        with:
          python-version: "3.12"

      - name: build arch docker image
        run: |
          docker build  -f Dockerfile . -t katanemo/plano -t katanemo/plano:0.4.0

      - name: install poetry
        run: |
          export POETRY_VERSION=2.2.1
          curl -sSL https://install.python-poetry.org | python3 -

      - name: setup python venv
        run: |
          python -m venv venv

      - name: install hurl
        run: |
          curl --location --remote-name https://github.com/Orange-OpenSource/hurl/releases/download/4.0.0/hurl_4.0.0_amd64.deb
          sudo dpkg -i hurl_4.0.0_amd64.deb

      - name: install arch gateway and test dependencies
        run: |
          source venv/bin/activate
          cd cli && echo "installing plano cli" && poetry install
          cd ../demos/shared/test_runner && echo "installing test dependencies" && poetry install

      - name: run demo tests
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
          MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }}
          GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }}
          ARCH_API_KEY: ${{ secrets.ARCH_API_KEY }}
          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
        run: |
          source venv/bin/activate
          cd demos/shared/test_runner && sh run_demo_tests.sh use_cases/preference_based_routing