"""add benchmark_ids and source_fission_id to generation_tasks

Revision ID: 011
Revises: 010
Create Date: 2026-06-13

合并第2环S12+第11环两个需求：
- benchmark_ids: 关联标杆笔记ID列表（TEXT存JSON list，第2环标杆分析引用）
- source_fission_id: 裂变来源任务ID（NULL=普通任务，非NULL=裂变产出，第11环）
"""
from alembic import op
import sqlalchemy as sa

revision = "011"
down_revision = "010"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        "generation_tasks",
        sa.Column(
            "benchmark_ids",
            sa.Text,
            nullable=True,
            comment="关联标杆笔记ID列表（JSON list），第2环标杆分析引用",
        ),
    )
    op.add_column(
        "generation_tasks",
        sa.Column(
            "source_fission_id",
            sa.Integer,
            nullable=True,
            comment="裂变来源fission_task ID；NULL=普通任务，非NULL=裂变产出（第11环）",
        ),
    )


def downgrade():
    op.drop_column("generation_tasks", "source_fission_id")
    op.drop_column("generation_tasks", "benchmark_ids")