import sqlalchemy as sa
from alembic import op

revision = '20260426_0001'
down_revision = 'consolidated_001'
branch_labels = None
depends_on = None

def upgrade():
    op.add_column('document_vectors', sa.Column('chunk_type', sa.String(20), nullable=True))
    op.add_column('document_vectors', sa.Column('media_path', sa.String(500), nullable=True))
    op.create_index('ix_document_vectors_chunk_type', 'document_vectors', ['chunk_type'])

def downgrade():
    op.drop_index('ix_document_vectors_chunk_type', 'document_vectors')
    op.drop_column('document_vectors', 'media_path')
    op.drop_column('document_vectors', 'chunk_type')