"""tree AI model policy (ai_member_provider, ai_recommender_provider)

Revision ID: b2c3d4e5f6a7
Revises: a1b2c3d4e5f6
Create Date: 2026-06-09

"""
from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "b2c3d4e5f6a7"
down_revision: str | None = "a1b2c3d4e5f6"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.add_column("trees", sa.Column("ai_member_provider", sa.String(length=32), nullable=True))
    op.add_column("trees", sa.Column("ai_recommender_provider", sa.String(length=32), nullable=True))


def downgrade() -> None:
    op.drop_column("trees", "ai_recommender_provider")
    op.drop_column("trees", "ai_member_provider")