"""Add execution_trace to scoring_runs.

Revision ID: 0009_execution_trace
Revises: 0008_retention_redaction_state
Create Date: 2026-09-04

"""

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql

revision: str = "0009_execution_trace"
down_revision: str | None = "0008_retention_redaction_state"
branch_labels: str | None = None
depends_on: str | None = None


def upgrade() -> None:
    op.add_column(
        "scoring_runs",
        sa.Column("execution_trace", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
    )


def downgrade() -> None:
    op.drop_column("scoring_runs", "execution_trace")