"""asset_atom_clips 新增 caption/embedding 字段（#2035 语义标签增强）

Revision ID: 085_atom_clip_caption_embedding
Revises: 084_lipsync_jobs_style
Create Date: 2026-09-25
"""

import sqlalchemy as sa

from alembic import op

revision = "085_atom_clip_caption_embedding"
down_revision = "084_lipsync_jobs_style"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # caption: 中文画面描述（10-30字）
    op.add_column(
        "asset_atom_clips",
        sa.Column("caption", sa.Text(), nullable=True),
    )
    # embedding: caption 对应的向量（豆包 embedding 接口返回，JSON 存 float 数组）
    op.add_column(
        "asset_atom_clips",
        sa.Column("embedding", sa.JSON(), nullable=True),
    )


def downgrade() -> None:
    op.drop_column("asset_atom_clips", "embedding")
    op.drop_column("asset_atom_clips", "caption")