from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = '45f4e4262fef'
down_revision: str | None = '1b7585b54935'
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None

def upgrade() -> None:
    op.add_column('chat_messages', sa.Column('sources', sa.JSON(), nullable=True))

def downgrade() -> None:
    op.drop_column('chat_messages', 'sources')