110 lines
5.7 KiB
Python
110 lines
5.7 KiB
Python
"""p6_llm
|
|
|
|
Revision ID: 3099be983359
|
|
Revises: a74b3a3db72f
|
|
Create Date: 2026-04-24 15:16:37.670788
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Sequence
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '3099be983359'
|
|
down_revision: str | Sequence[str] | None = 'a74b3a3db72f'
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('llm_provider',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('code', sa.String(length=64), nullable=False),
|
|
sa.Column('name', sa.String(length=64), nullable=False),
|
|
sa.Column('vendor', sa.String(length=32), nullable=False),
|
|
sa.Column('api_base', sa.String(length=255), nullable=False),
|
|
sa.Column('api_key_cipher', sa.String(length=512), nullable=True),
|
|
sa.Column('description', sa.String(length=255), nullable=True),
|
|
sa.Column('status', sa.String(length=16), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('code')
|
|
)
|
|
op.create_table('llm_model',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('provider_id', sa.Integer(), nullable=False),
|
|
sa.Column('model', sa.String(length=128), nullable=False),
|
|
sa.Column('name', sa.String(length=128), nullable=False),
|
|
sa.Column('description', sa.String(length=255), nullable=True),
|
|
sa.Column('max_context', sa.Integer(), nullable=False),
|
|
sa.Column('price_input_per_1k', sa.Numeric(precision=10, scale=6), nullable=False),
|
|
sa.Column('price_output_per_1k', sa.Numeric(precision=10, scale=6), nullable=False),
|
|
sa.Column('status', sa.String(length=16), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['provider_id'], ['llm_provider.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_llm_model_provider_id'), 'llm_model', ['provider_id'], unique=False)
|
|
op.create_table('llm_routing_rule',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('name', sa.String(length=128), nullable=False),
|
|
sa.Column('priority', sa.Integer(), nullable=False),
|
|
sa.Column('conditions', sa.JSON(), nullable=True),
|
|
sa.Column('model_id', sa.Integer(), nullable=False),
|
|
sa.Column('enabled', sa.Boolean(), nullable=False),
|
|
sa.Column('description', sa.String(length=255), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['model_id'], ['llm_model.id'], ondelete='RESTRICT'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('llm_usage',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('provider_id', sa.Integer(), nullable=True),
|
|
sa.Column('model_id', sa.Integer(), nullable=True),
|
|
sa.Column('provider_code', sa.String(length=64), nullable=True),
|
|
sa.Column('model', sa.String(length=128), nullable=True),
|
|
sa.Column('operator_id', sa.Integer(), nullable=True),
|
|
sa.Column('operator_name', sa.String(length=64), nullable=True),
|
|
sa.Column('source', sa.String(length=16), nullable=False),
|
|
sa.Column('task_id', sa.Integer(), nullable=True),
|
|
sa.Column('skill_id', sa.Integer(), nullable=True),
|
|
sa.Column('prompt_tokens', sa.Integer(), nullable=False),
|
|
sa.Column('completion_tokens', sa.Integer(), nullable=False),
|
|
sa.Column('total_tokens', sa.Integer(), nullable=False),
|
|
sa.Column('cost_usd', sa.Numeric(precision=12, scale=6), nullable=False),
|
|
sa.Column('result', sa.String(length=16), nullable=False),
|
|
sa.Column('message', sa.String(length=255), nullable=True),
|
|
sa.Column('cost_ms', sa.Integer(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['model_id'], ['llm_model.id'], ondelete='SET NULL'),
|
|
sa.ForeignKeyConstraint(['operator_id'], ['user.id'], ondelete='SET NULL'),
|
|
sa.ForeignKeyConstraint(['provider_id'], ['llm_provider.id'], ondelete='SET NULL'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_llm_usage_created_at'), 'llm_usage', ['created_at'], unique=False)
|
|
op.create_index(op.f('ix_llm_usage_model_id'), 'llm_usage', ['model_id'], unique=False)
|
|
op.create_index(op.f('ix_llm_usage_operator_id'), 'llm_usage', ['operator_id'], unique=False)
|
|
op.create_index(op.f('ix_llm_usage_provider_id'), 'llm_usage', ['provider_id'], unique=False)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_index(op.f('ix_llm_usage_provider_id'), table_name='llm_usage')
|
|
op.drop_index(op.f('ix_llm_usage_operator_id'), table_name='llm_usage')
|
|
op.drop_index(op.f('ix_llm_usage_model_id'), table_name='llm_usage')
|
|
op.drop_index(op.f('ix_llm_usage_created_at'), table_name='llm_usage')
|
|
op.drop_table('llm_usage')
|
|
op.drop_table('llm_routing_rule')
|
|
op.drop_index(op.f('ix_llm_model_provider_id'), table_name='llm_model')
|
|
op.drop_table('llm_model')
|
|
op.drop_table('llm_provider')
|
|
# ### end Alembic commands ###
|