opengrade-ai-server 0.1.0__tar.gz

opengrade_ai_server-0.1.0/.dockerignore

@@ -0,0 +1,8 @@
+.venv
+__pycache__
+*.pyc
+.pytest_cache
+.mypy_cache
+.ruff_cache
+.git
+.env

opengrade_ai_server-0.1.0/.env.example

@@ -0,0 +1,36 @@
+# Copy to backend/.env and adjust as needed:
+#   cp .env.example .env
+
+# Database (Compose publishes Postgres on host port 5434)
+DATABASE_URL=postgresql+asyncpg://opengrade:opengrade@localhost:5434/opengrade
+
+APP_HOST=0.0.0.0
+APP_PORT=8000
+
+# Capture prompt/completion content on traces (off by default).
+# Required for portfolio demo from-trace (OTLP input.value / output.value).
+CONTENT_CAPTURE_ENABLED=false
+
+# OTLP ingestion body size limit (bytes)
+OTLP_MAX_BODY_BYTES=4194304
+
+# App-wide JSON request body buffer cap (bytes). Middleware rejects larger JSON
+# before parsing. Raise this if you raise OTLP_MAX_BODY_BYTES for JSON OTLP.
+JSON_MAX_BODY_BYTES=4194304
+
+# --- LLM judges (LiteLLM) ---
+# Cloud example (Google AI Studio; default when LLM_MODEL is unset):
+LLM_MODEL=gemini/gemini-3.8-flash
+LLM_API_KEY=
+
+# Local Ollama example:
+# LLM_MODEL=ollama/gemma4:e2b
+# From Docker Compose API use host.docker.internal (extra_hosts in compose).
+# From a host-local API process use http://localhost:11434 instead.
+# LLM_API_BASE=http://host.docker.internal:11434
+# LLM_API_KEY=
+
+# Local Ollama cold-start can exceed 60s on small hosts
+LLM_TIMEOUT_SECONDS=300
+LLM_MAX_RETRIES=2
+LLM_MAX_CONCURRENCY=8

opengrade_ai_server-0.1.0/.gitignore

@@ -0,0 +1,42 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+.venv/
+venv/
+.env
+.env.local
+*.egg-info/
+dist/
+build/
+.mypy_cache/
+.ruff_cache/
+.pytest_cache/
+.coverage
+htmlcov/
+.uv/
+
+# IDE / OS
+.idea/
+.vscode/
+*.swp
+.DS_Store
+Thumbs.db
+
+# Docker volumes
+postgres_data/
+
+.worktrees/
+.cursor/
+docs/superpowers/
+
+# Packaged Next.js UI (built by scripts/package-ui.sh)
+backend/src/opengrade_server/_ui/**
+!backend/src/opengrade_server/_ui/.gitkeep
+
+opengrade.db
+**/opengrade.db
+
+.opengrade/

opengrade_ai_server-0.1.0/Dockerfile

@@ -0,0 +1,36 @@
+# One image for API and UI: the static UI export is built here and served by
+# FastAPI, exactly like `opengrade ui` serves the packaged wheel.
+
+FROM node:22-alpine AS ui
+WORKDIR /frontend
+COPY frontend/package.json frontend/package-lock.json ./
+RUN npm ci
+COPY frontend/ ./
+ENV NEXT_TELEMETRY_DISABLED=1
+RUN npm run build
+
+FROM python:3.12-slim
+
+WORKDIR /app
+
+RUN pip install --no-cache-dir uv
+
+COPY sdk/pyproject.toml sdk/README.md sdk/hatch_build.py /sdk/
+COPY sdk/src /sdk/src
+
+COPY backend/pyproject.toml backend/README.md backend/hatch_build.py ./
+COPY backend/src ./src
+COPY --from=ui /frontend/out ./src/opengrade_server/_ui
+COPY backend/alembic ./alembic
+COPY backend/alembic.ini ./
+COPY backend/docker-entrypoint.sh ./
+
+RUN uv pip install --system --no-cache . \
+    && python -c "from pathlib import Path; p = Path('docker-entrypoint.sh'); data = p.read_bytes().replace(b'\r\n', b'\n').replace(b'\r', b'\n').lstrip(b'\xef\xbb\xbf'); p.write_bytes(data); p.chmod(0o755)"
+
+ENV APP_HOST=0.0.0.0
+ENV APP_PORT=8000
+
+EXPOSE 8000
+
+ENTRYPOINT ["/app/docker-entrypoint.sh"]

opengrade_ai_server-0.1.0/PKG-INFO

@@ -0,0 +1,55 @@
+Metadata-Version: 2.5
+Name: opengrade-ai-server
+Version: 0.1.0
+Summary: AI Evaluation & Observability Platform server (FastAPI backend)
+Requires-Python: >=3.12
+Requires-Dist: aiosqlite>=0.20.0
+Requires-Dist: alembic>=1.14.0
+Requires-Dist: asyncpg>=0.30.0
+Requires-Dist: fastapi>=0.115.0
+Requires-Dist: httpx2>=2.13.0
+Requires-Dist: jsonschema>=4.23.0
+Requires-Dist: litellm>=1.55.0
+Requires-Dist: opengrade-ai==0.1.0
+Requires-Dist: opentelemetry-proto>=1.28.0
+Requires-Dist: protobuf>=5.0.0
+Requires-Dist: pydantic-settings>=2.6.0
+Requires-Dist: pydantic>=2.10.0
+Requires-Dist: python-multipart>=0.0.9
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: rich>=13.9.0
+Requires-Dist: sqlalchemy[asyncio]>=2.0.36
+Requires-Dist: typer>=0.15.0
+Requires-Dist: uvicorn[standard]>=0.32.0
+Provides-Extra: dev
+Requires-Dist: mypy>=1.13.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.24.0; extra == 'dev'
+Requires-Dist: pytest>=8.3.0; extra == 'dev'
+Requires-Dist: ruff>=0.8.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# opengrade-ai-server
+
+Server package for the AI Evaluation & Observability Platform.
+
+- Distribution: `opengrade-ai-server` (PyPI)
+- Import path: `opengrade_server` (e.g. `uvicorn opengrade_server.main:app`)
+- Console scripts: `opengrade` (`ui`, `check`; provided by the `opengrade-ai` dependency) and `opengrade-server` (same commands)
+
+## Local UI
+
+```bash
+# from repo root — build the Next static export into the package
+./scripts/package-ui.sh
+cd backend && uv sync
+uv run opengrade ui
+# → http://127.0.0.1:8000  (SQLite ./opengrade.db by default)
+```
+
+Postgres:
+
+```bash
+uv run opengrade ui --backend-store-uri 'postgresql+asyncpg://opengrade:opengrade@localhost:5434/opengrade'
+```
+
+See the repository root README for Docker Compose and SDK quickstart.

opengrade_ai_server-0.1.0/README.md

@@ -0,0 +1,25 @@
+# opengrade-ai-server
+
+Server package for the AI Evaluation & Observability Platform.
+
+- Distribution: `opengrade-ai-server` (PyPI)
+- Import path: `opengrade_server` (e.g. `uvicorn opengrade_server.main:app`)
+- Console scripts: `opengrade` (`ui`, `check`; provided by the `opengrade-ai` dependency) and `opengrade-server` (same commands)
+
+## Local UI
+
+```bash
+# from repo root — build the Next static export into the package
+./scripts/package-ui.sh
+cd backend && uv sync
+uv run opengrade ui
+# → http://127.0.0.1:8000  (SQLite ./opengrade.db by default)
+```
+
+Postgres:
+
+```bash
+uv run opengrade ui --backend-store-uri 'postgresql+asyncpg://opengrade:opengrade@localhost:5434/opengrade'
+```
+
+See the repository root README for Docker Compose and SDK quickstart.

opengrade_ai_server-0.1.0/alembic/env.py

@@ -0,0 +1,65 @@
+from logging.config import fileConfig
+
+from alembic import context
+from sqlalchemy import pool
+from sqlalchemy.engine import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+
+from opengrade_server.config import get_settings
+from opengrade_server.infrastructure.models import Base
+
+config = context.config
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+target_metadata = Base.metadata
+
+
+def get_url() -> str:
+    return get_settings().database_url
+
+
+def run_migrations_offline() -> None:
+    url = get_url()
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def do_run_migrations(connection: Connection) -> None:
+    context.configure(connection=connection, target_metadata=target_metadata)
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations() -> None:
+    configuration = config.get_section(config.config_ini_section) or {}
+    configuration["sqlalchemy.url"] = get_url()
+    connectable = async_engine_from_config(
+        configuration,
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    async with connectable.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await connectable.dispose()
+
+
+def run_migrations_online() -> None:
+    import asyncio
+
+    asyncio.run(run_async_migrations())
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

opengrade_ai_server-0.1.0/alembic/script.py.mako

@@ -0,0 +1,18 @@
+"""Alembic script template."""
+
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = None
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    pass
+
+
+def downgrade() -> None:
+    pass

opengrade_ai_server-0.1.0/alembic/versions/0001_create_projects.py

@@ -0,0 +1,39 @@
+"""create projects table
+
+Revision ID: 0001_create_projects
+Revises:
+Create Date: 2026-09-30
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+
+revision: str = "0001_create_projects"
+down_revision: str | None = None
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "projects",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("name", sa.String(length=200), nullable=False),
+        sa.Column("slug", sa.String(length=200), nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("slug"),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("projects")

opengrade_ai_server-0.1.0/alembic/versions/0002_create_traces_spans.py

@@ -0,0 +1,102 @@
+"""create traces and spans tables
+
+Revision ID: 0002_create_traces_spans
+Revises: 0001_create_projects
+Create Date: 2026-09-30
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+revision: str = "0002_create_traces_spans"
+down_revision: str | None = "0001_create_projects"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "traces",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Uuid(), nullable=False),
+        sa.Column("trace_id", sa.String(length=32), nullable=False),
+        sa.Column("name", sa.String(length=512), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column("start_time", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("end_time", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("input", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column("output", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("environment", sa.String(length=200), nullable=True),
+        sa.Column("user_id", sa.String(length=200), nullable=True),
+        sa.Column("session_id", sa.String(length=200), nullable=True),
+        sa.ForeignKeyConstraint(["project_id"], ["projects.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("project_id", "trace_id", name="uq_traces_project_trace_id"),
+    )
+    op.create_index(
+        "ix_traces_project_start_time",
+        "traces",
+        ["project_id", "start_time"],
+        unique=False,
+    )
+    op.create_index(
+        "ix_traces_project_status",
+        "traces",
+        ["project_id", "status"],
+        unique=False,
+    )
+
+    op.create_table(
+        "spans",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("trace_pk", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Uuid(), nullable=False),
+        sa.Column("span_id", sa.String(length=16), nullable=False),
+        sa.Column("parent_span_id", sa.String(length=16), nullable=True),
+        sa.Column("name", sa.String(length=512), nullable=False),
+        sa.Column("kind", sa.String(length=64), nullable=False),
+        sa.Column("start_time", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("end_time", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column(
+            "attributes",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "events",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'[]'::jsonb"),
+            nullable=False,
+        ),
+        sa.ForeignKeyConstraint(["trace_pk"], ["traces.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_spans_trace_pk", "spans", ["trace_pk"], unique=False)
+    op.create_index(
+        "ix_spans_project_start_time",
+        "spans",
+        ["project_id", "start_time"],
+        unique=False,
+    )
+
+
+def downgrade() -> None:
+    op.drop_index("ix_spans_project_start_time", table_name="spans")
+    op.drop_index("ix_spans_trace_pk", table_name="spans")
+    op.drop_table("spans")
+    op.drop_index("ix_traces_project_status", table_name="traces")
+    op.drop_index("ix_traces_project_start_time", table_name="traces")
+    op.drop_table("traces")

opengrade_ai_server-0.1.0/alembic/versions/0003_create_evaluation_tables.py

@@ -0,0 +1,190 @@
+"""create evaluation engine tables
+
+Revision ID: 0003_create_evaluation_tables
+Revises: 0002_create_traces_spans
+Create Date: 2026-10-01
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+revision: str = "0003_create_evaluation_tables"
+down_revision: str | None = "0002_create_traces_spans"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "datasets",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Uuid(), nullable=False),
+        sa.Column("name", sa.String(length=200), nullable=False),
+        sa.Column("version", sa.Integer(), server_default="1", nullable=False),
+        sa.Column("description", sa.String(length=2000), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.ForeignKeyConstraint(["project_id"], ["projects.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint(
+            "project_id", "name", "version", name="uq_datasets_project_name_version"
+        ),
+    )
+    op.create_index(
+        "ix_datasets_project_created_at",
+        "datasets",
+        ["project_id", "created_at"],
+        unique=False,
+    )
+
+    op.create_table(
+        "dataset_items",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("dataset_id", sa.Uuid(), nullable=False),
+        sa.Column("input", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
+        sa.Column("expected_output", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column("actual_output", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column("context", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("source_trace_id", sa.String(length=32), nullable=True),
+        sa.Column("source_span_id", sa.String(length=16), nullable=True),
+        sa.ForeignKeyConstraint(["dataset_id"], ["datasets.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_dataset_items_dataset_id", "dataset_items", ["dataset_id"], unique=False)
+
+    op.create_table(
+        "evaluators",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Uuid(), nullable=False),
+        sa.Column("name", sa.String(length=200), nullable=False),
+        sa.Column("type", sa.String(length=32), nullable=False),
+        sa.Column(
+            "config",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("version", sa.Integer(), server_default="1", nullable=False),
+        sa.ForeignKeyConstraint(["project_id"], ["projects.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint(
+            "project_id", "name", "version", name="uq_evaluators_project_name_version"
+        ),
+    )
+    op.create_index("ix_evaluators_project_id", "evaluators", ["project_id"], unique=False)
+
+    op.create_table(
+        "experiments",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Uuid(), nullable=False),
+        sa.Column("name", sa.String(length=200), nullable=False),
+        sa.Column("dataset_id", sa.Uuid(), nullable=False),
+        sa.Column(
+            "model_config",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("version", sa.String(length=200), nullable=True),
+        sa.Column("baseline_experiment_id", sa.Uuid(), nullable=True),
+        sa.Column("status", sa.String(length=32), server_default="created", nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.ForeignKeyConstraint(["project_id"], ["projects.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["dataset_id"], ["datasets.id"], ondelete="RESTRICT"),
+        sa.ForeignKeyConstraint(
+            ["baseline_experiment_id"], ["experiments.id"], ondelete="SET NULL"
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(
+        "ix_experiments_project_created_at",
+        "experiments",
+        ["project_id", "created_at"],
+        unique=False,
+    )
+
+    op.create_table(
+        "evaluation_runs",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("experiment_id", sa.Uuid(), nullable=False),
+        sa.Column("evaluator_id", sa.Uuid(), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("finished_at", sa.DateTime(timezone=True), nullable=True),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.ForeignKeyConstraint(["experiment_id"], ["experiments.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["evaluator_id"], ["evaluators.id"], ondelete="RESTRICT"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(
+        "ix_evaluation_runs_experiment_id",
+        "evaluation_runs",
+        ["experiment_id"],
+        unique=False,
+    )
+
+    op.create_table(
+        "evaluation_results",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("run_id", sa.Uuid(), nullable=False),
+        sa.Column("dataset_item_id", sa.Uuid(), nullable=False),
+        sa.Column("score", sa.Float(), nullable=True),
+        sa.Column("label", sa.String(length=64), nullable=True),
+        sa.Column("explanation", sa.String(length=4000), nullable=True),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("duration_ms", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(["run_id"], ["evaluation_runs.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["dataset_item_id"], ["dataset_items.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(
+        "ix_evaluation_results_run_item",
+        "evaluation_results",
+        ["run_id", "dataset_item_id"],
+        unique=False,
+    )
+
+
+def downgrade() -> None:
+    op.drop_index("ix_evaluation_results_run_item", table_name="evaluation_results")
+    op.drop_table("evaluation_results")
+    op.drop_index("ix_evaluation_runs_experiment_id", table_name="evaluation_runs")
+    op.drop_table("evaluation_runs")
+    op.drop_index("ix_experiments_project_created_at", table_name="experiments")
+    op.drop_table("experiments")
+    op.drop_index("ix_evaluators_project_id", table_name="evaluators")
+    op.drop_table("evaluators")
+    op.drop_index("ix_dataset_items_dataset_id", table_name="dataset_items")
+    op.drop_table("dataset_items")
+    op.drop_index("ix_datasets_project_created_at", table_name="datasets")
+    op.drop_table("datasets")

opengrade_ai_server-0.1.0/alembic/versions/0004_rename_experiment_application_version.py

@@ -0,0 +1,60 @@
+"""rename experiments.application_version to version
+
+Revision ID: 0004_rename_experiment_version
+Revises: 0003_create_evaluation_tables
+Create Date: 2026-10-02
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from alembic import op
+
+revision: str = "0004_rename_experiment_version"
+down_revision: str | None = "0003_create_evaluation_tables"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute(
+        """
+        DO $$
+        BEGIN
+            IF EXISTS (
+                SELECT 1
+                FROM information_schema.columns
+                WHERE table_name = 'experiments'
+                  AND column_name = 'application_version'
+            ) THEN
+                ALTER TABLE experiments
+                RENAME COLUMN application_version TO version;
+            END IF;
+        END $$;
+        """
+    )
+
+
+def downgrade() -> None:
+    op.execute(
+        """
+        DO $$
+        BEGIN
+            IF EXISTS (
+                SELECT 1
+                FROM information_schema.columns
+                WHERE table_name = 'experiments'
+                  AND column_name = 'version'
+            ) AND NOT EXISTS (
+                SELECT 1
+                FROM information_schema.columns
+                WHERE table_name = 'experiments'
+                  AND column_name = 'application_version'
+            ) THEN
+                ALTER TABLE experiments
+                RENAME COLUMN version TO application_version;
+            END IF;
+        END $$;
+        """
+    )