"""add status_update to ai_session_steps step_type constraint

Revision ID: 069
Revises: 068
Create Date: 2026-03-29
"""
from alembic import op


revision = "069"
down_revision = "068"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.drop_constraint("ck_ai_session_steps_step_type", "ai_session_steps", type_="check")
    op.create_check_constraint(
        "ck_ai_session_steps_step_type", "ai_session_steps",
        "step_type IN ('question', 'action', 'script_generation', 'verification', "
        "'info_request', 'note', 'intake_analysis', 'fork', 'status_update')",
    )


def downgrade() -> None:
    op.drop_constraint("ck_ai_session_steps_step_type", "ai_session_steps", type_="check")
    op.create_check_constraint(
        "ck_ai_session_steps_step_type", "ai_session_steps",
        "step_type IN ('question', 'action', 'script_generation', 'verification', "
        "'info_request', 'note', 'intake_analysis', 'fork')",
    )
