"""Add asset_select_mode to generation_tasks

Revision ID: 032
Revises: 031
Create Date: 2026-07-07

素材库自动匹配功能：为 generation_tasks 表添加 asset_select_mode 字段。
"""

import sqlalchemy as sa

from alembic import op

revision = "032"
down_revision = "031"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "generation_tasks",
        sa.Column("asset_select_mode", sa.String(20), nullable=False, server_default=""),
    )


def downgrade() -> None:
    op.drop_column("generation_tasks", "asset_select_mode")
