Skip to content

Opik Optimizer E2E Tests 7460/merge by @awkoy #1376

Opik Optimizer E2E Tests 7460/merge by @awkoy

Opik Optimizer E2E Tests 7460/merge by @awkoy #1376

name: "Opik Optimizer - E2E Tests"
run-name: "Opik Optimizer E2E Tests ${{ github.ref_name }} by @${{ github.actor }}"
concurrency:
group: opik-optimizer-e2e-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true
on:
workflow_dispatch:
pull_request:
paths:
- 'sdks/opik_optimizer/**'
- '.github/workflows/opik-optimizer-e2e-tests.yaml'
push:
branches:
- 'main'
paths:
- 'sdks/opik_optimizer/**'
- '.github/workflows/opik-optimizer-e2e-tests.yaml'
permissions:
contents: read
packages: write
env:
OPIK_ENABLE_LITELLM_MODELS_MONITORING: False
OPIK_SENTRY_ENABLE: False
HF_HUB_ENABLE_PROGRESS_BARS: "0"
HF_HUB_DISABLE_TELEMETRY: "1"
OPIK_URL_OVERRIDE: http://localhost:8080
OPIK_CONSOLE_LOGGING_LEVEL: DEBUG
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
OPENAI_ORG_ID: ${{ secrets.OPENAI_ORG_ID }}
jobs:
build-opik:
uses: ./.github/workflows/build_e2e_docker.yaml
with:
include_guardrails: true
run-e2e-tests:
name: Opik Optimizer E2E Tests Python ${{matrix.python_version}}
needs: build-opik
runs-on: ubuntu-latest
timeout-minutes: 40
defaults:
run:
working-directory: sdks/opik_optimizer/
strategy:
fail-fast: false
matrix:
python_version: ${{ fromJSON(vars.PYTHON_VERSIONS) }}
steps:
- name: Check out code
uses: actions/checkout@v6
- name: Login to GHCR
uses: docker/login-action@v3
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Pull Docker images from GHCR
env:
TAG: ${{ needs.build-opik.outputs.image_tag }}
run: |
docker pull "ghcr.io/comet-ml/opik/opik-backend:$TAG"
docker pull "ghcr.io/comet-ml/opik/opik-python-backend:$TAG"
docker pull "ghcr.io/comet-ml/opik/opik-guardrails-backend:$TAG"
docker tag "ghcr.io/comet-ml/opik/opik-backend:$TAG" ghcr.io/comet-ml/opik/opik-backend:latest
docker tag "ghcr.io/comet-ml/opik/opik-python-backend:$TAG" ghcr.io/comet-ml/opik/opik-python-backend:latest
docker tag "ghcr.io/comet-ml/opik/opik-guardrails-backend:$TAG" ghcr.io/comet-ml/opik/opik-guardrails-backend:latest
- name: Setup Python ${{matrix.python_version}}
uses: actions/setup-python@v5
with:
python-version: ${{matrix.python_version}}
cache: 'pip'
cache-dependency-path: |
sdks/opik_optimizer/pyproject.toml
sdks/opik_optimizer/setup.py
sdks/opik_optimizer/tests/test_requirements.txt
- name: Run latest Opik server
env:
OPIK_USAGE_REPORT_ENABLED: false
# Avoid Buildx Bake concurrent image export collisions in CI.
COMPOSE_BAKE: false
TOGGLE_RUNNERS_ENABLED: "true"
OPIK_BACKEND_PULL_POLICY: never
PYTHON_BACKEND_PULL_POLICY: never
OPIK_GUARDRAILS_BACKEND_PULL_POLICY: never
run: |
cd ${{ github.workspace }}
./opik.sh --backend --port-mapping --guardrails
- name: Check Opik server availability
shell: bash
run: |
chmod +x ${{ github.workspace }}/tests_end_to_end/installer_utils/*.sh
cd ${{ github.workspace }}/deployment/docker-compose
echo "Check Docker pods are up"
${{ github.workspace }}/tests_end_to_end/installer_utils/check_docker_compose_pods.sh
echo "Check backend health"
${{ github.workspace }}/tests_end_to_end/installer_utils/check_backend.sh
- name: Cache HuggingFace datasets
uses: actions/cache@v4
with:
path: ~/.cache/huggingface
key: hf-datasets-${{ hashFiles('**/datasets/*.py') }}
restore-keys: |
hf-datasets-
- name: Cache LiteLLM cache
uses: actions/cache@v4
with:
path: ~/.litellm_cache
key: litellm-cache-${{ hashFiles('**/requirements.txt', '**/pyproject.toml', '**/setup.py') }}
restore-keys: |
litellm-cache-
- name: Install opik_optimizer
run: pip install .[dev]
- name: Install test requirements
run: |
cd ./tests
pip install --no-cache-dir --disable-pip-version-check -r test_requirements.txt
- name: Run E2E tests
run: pytest tests/e2e -vv --junitxml=${{ github.workspace }}/test_results_${{matrix.python_version}}.xml
- name: Upload test artifacts on failure
if: failure()
uses: actions/upload-artifact@v7
with:
name: opik-optimizer-e2e-logs-p${{matrix.python_version}}
path: |
*.log
tests/**/*.log
retention-days: 7
- name: Stop opik server
if: always()
run: |
cd ${{ github.workspace }}
./opik.sh --stop --guardrails
dataset-source-check:
name: Opik Optimizer Integration Smoke Tests
runs-on: ubuntu-latest
timeout-minutes: 30
needs: run-e2e-tests
env:
OPIK_USE_HF_STREAMING: "true"
defaults:
run:
working-directory: sdks/opik_optimizer/
steps:
- name: Check out code
uses: actions/checkout@v6
- name: Setup Python 3.12
uses: actions/setup-python@v5
with:
python-version: "3.12"
cache: 'pip'
cache-dependency-path: |
sdks/opik_optimizer/pyproject.toml
sdks/opik_optimizer/setup.py
sdks/opik_optimizer/tests/test_requirements.txt
- name: Cache HuggingFace datasets
uses: actions/cache@v4
with:
path: ~/.cache/huggingface
key: hf-datasets-${{ hashFiles('**/datasets/*.py') }}
restore-keys: |
hf-datasets-
- name: Install opik_optimizer
run: pip install .
- name: Install test requirements
run: |
cd ./tests
pip install --no-cache-dir --disable-pip-version-check -r test_requirements.txt
- name: Run dataset integration check
run: pytest -n auto --dist loadfile tests/integration/datasets/test_dataset_sources.py -vv