"""Added rfp_profile_ids_to_search and rfp mode to rfp table

Revision ID: 16c2a9de1d3f
Revises: c086ad70498b
Create Date: 2026-02-20 15:56:34.422157

"""
from typing import Sequence, Union

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

# revision identifiers, used by Alembic.
revision: str = '16c2a9de1d3f'
down_revision: Union[str, Sequence[str], None] = 'c086ad70498b'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('rfp', sa.Column('rfp_profile_ids_to_search', postgresql.ARRAY(sa.Integer()), nullable=True))
    op.add_column('rfp', sa.Column('rfp_mode', sa.Enum('EXTRACTION_ONLY', 'EXTRACTION_AND_ANSWER', name='rfpmode', native_enum=False), nullable=True))
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('rfp', 'rfp_mode')
    op.drop_column('rfp', 'rfp_profile_ids_to_search')
    # ### end Alembic commands ###
