"""simplify persona_template to name + system_prompt

Revision ID: 0006
Revises: 0005
Create Date: 2026-06-04
"""

from alembic import op
import sqlalchemy as sa

revision = "0006"
down_revision = "0005"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.drop_column("persona_template", "description")
    op.drop_column("persona_template", "agent_type")
    op.drop_column("persona_template", "provider_title")
    op.drop_column("persona_template", "model_id")
    op.drop_column("persona_template", "tools")
    op.drop_column("persona_template", "tags")
    op.drop_column("persona_template", "is_builtin")


def downgrade() -> None:
    op.add_column("persona_template", sa.Column("description", sa.Text(), nullable=False, server_default=""))
    op.add_column("persona_template", sa.Column("agent_type", sa.String(32), nullable=False, server_default="ordinary"))
    op.add_column("persona_template", sa.Column("provider_title", sa.String(50), nullable=True))
    op.add_column("persona_template", sa.Column("model_id", sa.String(100), nullable=True))
    op.add_column("persona_template", sa.Column("tools", sa.JSON(), server_default="[]"))
    op.add_column("persona_template", sa.Column("tags", sa.JSON(), server_default="[]"))
    op.add_column("persona_template", sa.Column("is_builtin", sa.Boolean(), nullable=False, server_default="false"))