vent_ai_platform/alembic/versions/0006_processing_run_document_index.py

"""add processing run document created index

Revision ID: 0006_processing_run_document_index
Revises: 0005_extraction_judgement_status
Create Date: 2026-10-13
"""

from __future__ import annotations

import sqlalchemy as sa
from alembic import op


revision = "0006_processing_task_index"
down_revision = "0005_extraction_judgement_status"
branch_labels = None
depends_on = None


def upgrade() -> None:
    inspector = sa.inspect(op.get_bind())
    index_name = "ix_document_processing_runs_document_created"
    if index_name not in {
        item["name"]
        for item in inspector.get_indexes("document_processing_runs")
    }:
        op.create_index(
            index_name,
            "document_processing_runs",
            ["document_id", "created_at"],
        )


def downgrade() -> None:
    inspector = sa.inspect(op.get_bind())
    index_name = "ix_document_processing_runs_document_created"
    if index_name in {
        item["name"]
        for item in inspector.get_indexes("document_processing_runs")
    }:
        op.drop_index(
            index_name,
            table_name="document_processing_runs",
        )