unit_tests.yml

name: Unit tests

on:
  push:
    branches: [main]
  pull_request:
    branches: [main]
  workflow_dispatch:

jobs:
  unit-tests:

    strategy:
      matrix:
        os: [ubuntu-latest, windows-latest, macos-latest]
        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
        model:
          - "gpt2cpu"
          - "phi2cpu"
          # - "transformers_mistral_7b" See Issue 713
          - "hfllama7b"
          - "hfllama_mistral_7b"

    runs-on: ${{ matrix.os }}
    steps:
      - uses: actions/checkout@v4
      - name: Set up Python ${{ matrix.python-version }}
        uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}
      - name: Install dependencies
        shell: bash
        run: |
          python -m pip install --upgrade pip
          pip install pytest
          pip install -e .[schemas,test]
          if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
      - name: Install model-specific dependencies
        run: |
          pip install "llama-cpp-python!=0.2.58"
      - name: Run tests (except server)
        shell: bash
        run: |
          pytest --cov=guidance --cov-report=xml --cov-report=term-missing \
          --selected_model ${{ matrix.model }} \
          -m "not (needs_credentials or use_gpu or server)" \
          ./tests/
      - name: Upload coverage reports to Codecov
        uses: codecov/codecov-action@v4
        with:
          token: ${{ secrets.CODECOV_TOKEN }}