test.yml

name: Test

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

jobs:
  test:

    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.9", "3.10", "3.11"]
        flavor: [all, api, base-api, openai, huggingface, anthropic, local]

    steps:
    - uses: actions/checkout@v3
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v3
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install dependencies
      run: |
        if [ "${{ matrix.flavor }}" = "base-api" ]; then
          pip install -e .[test]
        else
          pip install -e .[${{ matrix.flavor }},test]
        fi
    - name: Test with pytest
      run: |
        case "${{ matrix.flavor }}" in
          all)
            pytest tests
            ;;
          api)
            pytest tests/llm_api_client
            ;;
          base-api)
            pytest tests/llm_api_client/ai21_client tests/llm_api_client/google_client
            ;;
          local)
            pytest tests/llm_client/local_client
            ;;
          *)
            pytest tests/llm_api_client/${{ matrix.flavor }}_client
            ;;
        esac