"""add ai_providers table

Revision ID: add_ai_providers
Revises: 0798c5c09c8c
Create Date: 2026-05-27

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import UUID, JSONB
import uuid

revision = 'add_ai_providers'
down_revision = '0798c5c09c8c'
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        'ai_providers',
        sa.Column('id', UUID(as_uuid=True), primary_key=True, default=uuid.uuid4),
        sa.Column('name', sa.String(100), nullable=False),
        sa.Column('provider_type', sa.String(50), nullable=False),
        sa.Column('api_key', sa.Text(), nullable=True),
        sa.Column('api_secret', sa.Text(), nullable=True),
        sa.Column('base_url', sa.String(500), nullable=True),
        sa.Column('model_name', sa.String(100), nullable=False),
        sa.Column('extra_config', JSONB(), default={}),
        sa.Column('priority', sa.Integer(), default=0),
        sa.Column('enabled', sa.Boolean(), default=True),
        sa.Column('created_at', sa.DateTime(), default=sa.func.now()),
        sa.Column('updated_at', sa.DateTime(), default=sa.func.now(), onupdate=sa.func.now()),
    )


def downgrade():
    op.drop_table('ai_providers')