"""add_ai_analysis_columns

Revision ID: 6e8f4c2d1b0a
Revises: 1f59799a5083
Create Date: 2026-05-21 10:00:00.000000

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '6e8f4c2d1b0a'
down_revision: Union[str, Sequence[str], None] = '1f59799a5083'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """添加 AI 分析相关字段"""
    op.add_column('announcements', sa.Column('ai_relevant', sa.Boolean(), nullable=True))
    op.add_column('announcements', sa.Column('ai_analysis', sa.Text(), nullable=True))


def downgrade() -> None:
    """回滚"""
    op.drop_column('announcements', 'ai_analysis')
    op.drop_column('announcements', 'ai_relevant')