"""add ai_provider and ai_api_key_enc to users

Revision ID: 0003
Revises: 0002
Create Date: 2026-04-22
"""
from alembic import op
import sqlalchemy as sa

revision = "0003"
down_revision = "0002"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("users", sa.Column("ai_provider", sa.Text, nullable=True))
    op.add_column("users", sa.Column("ai_api_key_enc", sa.LargeBinary, nullable=True))


def downgrade() -> None:
    op.drop_column("users", "ai_api_key_enc")
    op.drop_column("users", "ai_provider")