ci.yml

name: CI

on: [pull_request, workflow_dispatch]
permissions:
  contents: read

jobs:
  build-and-test-extension:
    services:
      ollama:
        image: ollama/ollama:latest
        ports:
          - 11434:11434

    runs-on: ubuntu-latest
    strategy:
      matrix:
        postgres-version: [ 16, 17 ]
      fail-fast: false

    steps:
      - uses: actions/checkout@v4

      - uses: taiki-e/install-action@just

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: "3.10"

      - name: Verify Docker installation
        run: |
          docker --version
          docker info

      - name: Build Docker image
        run: just ext docker-build
        env:
          PG_MAJOR: ${{ matrix.postgres-version }}

      - name: Run Docker container
        run: just ext docker-run

      - name: Build
        run: docker exec pgai-ext just build

      - name: Lint SQL and Python
        run: docker exec pgai-ext just lint

      - name: Check Python Formatting
        run: docker exec pgai-ext just format-py

      - name: Install extension
        run: docker exec pgai-ext just install

      - name: Run test server
        run: docker exec -d
          -e OPENAI_API_KEY
          -e ANTHROPIC_API_KEY
          -e COHERE_API_KEY
          -e VOYAGE_API_KEY
          pgai-ext just test-server
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
          COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
          VOYAGE_API_KEY: ${{ secrets.VOYAGE_API_KEY }}

      - name: Run tests
        run: docker exec
          -e OPENAI_API_KEY
          -e ANTHROPIC_API_KEY
          -e COHERE_API_KEY
          -e VOYAGE_API_KEY
          -e OLLAMA_HOST
          pgai-ext just test
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
          COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
          VOYAGE_API_KEY: ${{ secrets.VOYAGE_API_KEY }}
          OLLAMA_HOST: http://localhost:11434

      - name: Stop and remove Docker container
        run: |
          just ext docker-stop
          just ext docker-rm

  build-and-test-pgai:
    services:
      ollama:
        image: ollama/ollama:latest
        ports:
          - 11434:11434
    runs-on: ubuntu-latest

    steps:
      - uses: actions/checkout@v4

      - uses: taiki-e/install-action@just

      - name: Install uv
        uses: astral-sh/setup-uv@v3
        with:
          version: "0.4.29"
          enable-cache: true
          cache-dependency-glob: "./projects/pgai/uv.lock"

      - name: "Set up Python"
        uses: actions/setup-python@v5
        with:
          python-version-file: "./projects/pgai/.python-version"

      - name: Install dependencies
        working-directory: ./projects/pgai
        run: uv sync --all-extras

      - name: Lint
        run: just pgai lint

      - name: Check Typing
        run: just pgai type-check

      - name: Check Formatting
        run: just pgai format

      - name: Run Tests
        run: just pgai test
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
        

      - name: Build the pgai distributable and check artifacts
        run: just pgai build