e2e-llama-cpp-python.yaml

# End-to-end testing that deploys Supabase and the API, and deploy/tests llama-cpp-python, text-embeddings, and whisper

name: e2e-llama-cpp-python
on:
  pull_request:
    types:
      - ready_for_review
      - review_requested
      - synchronize
      - milestoned
    paths:
      # Catch-all
      - "**"

      # Ignore updates to the .github directory, unless it's this current file
      - "!.github/**"
      - ".github/workflows/e2e-llama-cpp-python.yaml"

      # Ignore docs and website things
      - "!**.md"
      - "!docs/**"
      - "!adr/**"
      - "!website/**"
      - "!netlify.toml"

      # Ignore updates to generic github metadata files
      - "!CODEOWNERS"
      - "!.gitignore"
      - "!LICENSE"

      # Ignore local development files
      - "!.pre-commit-config.yaml"

      # Ignore non e2e tests changes
      - "!tests/pytest/**"

      # Ignore LFAI-UI source code changes
      - "!src/leapfrogai_ui/**"

      # Ignore changes to unrelated packages
      - "!packages/k3d-gpu/**"
      - "!packages/repeater/**"
      - "!packages/text-embeddings/**"
      - "!packages/ui/**"
      - "!packages/vllm/**"
      - "!packages/whisper/**"

concurrency:
  group: e2e-llama-cpp-python-${{ github.ref }}
  cancel-in-progress: true

jobs:
  e2e_llama:
    runs-on: ai-ubuntu-big-boy-8-core
    if: ${{ !github.event.pull_request.draft }}

    steps:
        - name: Checkout Repo
          uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1

        - name: Setup Python
          uses: actions/setup-python@0a5c61591373683505ea898e09a3ea4f39ef2b9c #v5.0.0
          with:
            python-version-file: 'pyproject.toml'

        - name: Install Python Deps
          run: python -m pip install "."

        - name: Setup UDS Environment
          uses: defenseunicorns/uds-common/.github/actions/setup@05f42bb3117b66ebef8c72ae050b34bce19385f5
          with:
            username: ${{ secrets.IRON_BANK_ROBOT_USERNAME }}
            password: ${{ secrets.IRON_BANK_ROBOT_PASSWORD }}

        - name: Create UDS Cluster
          run: |
            uds deploy k3d-core-slim-dev:0.22.2 --confirm

        ##########
        # Supabase
        ##########
        - name: Deploy Supabase
          run: |
            make build-supabase LOCAL_VERSION=e2e-test
            docker image prune -af
            uds zarf tools kubectl create namespace leapfrogai
            uds zarf package deploy packages/supabase/zarf-package-supabase-amd64-e2e-test.tar.zst --confirm
            rm packages/supabase/zarf-package-supabase-amd64-e2e-test.tar.zst

        - name: Set environment variable
          id: set-env-var
          run: |
            echo "ANON_KEY=$(uds zarf tools kubectl get secret supabase-bootstrap-jwt -n leapfrogai -o jsonpath='{.data.anon-key}' | base64 -d)" >> "$GITHUB_ENV"

        ##########
        # API
        ##########
        - name: Deploy LFAI-API
          run: |
            make build-api LOCAL_VERSION=e2e-test
            docker image prune -af
            uds zarf package deploy packages/api/zarf-package-leapfrogai-api-amd64-e2e-test.tar.zst --confirm
            rm packages/api/zarf-package-leapfrogai-api-amd64-e2e-test.tar.zst

        ##########
        # llama
        ##########
        - name: Deploy llama-cpp-python
          run: |
            make build-llama-cpp-python LOCAL_VERSION=e2e-test
            docker image prune -af
            uds zarf package deploy packages/llama-cpp-python/zarf-package-llama-cpp-python-amd64-e2e-test.tar.zst -l=trace --confirm
            rm packages/llama-cpp-python/zarf-package-llama-cpp-python-amd64-e2e-test.tar.zst

        - name: Test llama-cpp-python
          run: |
            python -m pytest ./tests/e2e/test_llama.py -v