"""add_more_ai_log

Revision ID: 84deefdcf6c4
Revises: 8abfcf44523f
Create Date: 2024-07-17 11:30:27.035810

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '84deefdcf6c4'
down_revision = '8abfcf44523f'
branch_labels = None
depends_on = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('ai_log', sa.Column('app_id', sa.Integer(), nullable=True))
    op.add_column('ai_log', sa.Column('org_id', sa.String(length=64), nullable=True))
    op.add_column('ai_log', sa.Column('session_id', sa.String(length=64), nullable=True))
    op.add_column('ai_log', sa.Column('query_id', sa.String(length=64), nullable=True))
    op.add_column('ai_log', sa.Column('task_id', sa.String(length=64), nullable=True))
    op.add_column('ai_log', sa.Column('module', sa.String(length=64), nullable=True))
    op.add_column('ai_log', sa.Column('latency', sa.Integer(), nullable=True))
    op.add_column('ai_log', sa.Column('input_token', sa.Integer(), nullable=True))
    op.add_column('ai_log', sa.Column('output_token', sa.Integer(), nullable=True))
    op.add_column('ai_log', sa.Column('created_by', sa.String(length=64), nullable=True))
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('ai_log', 'created_by')
    op.drop_column('ai_log', 'output_token')
    op.drop_column('ai_log', 'input_token')
    op.drop_column('ai_log', 'latency')
    op.drop_column('ai_log', 'module')
    op.drop_column('ai_log', 'task_id')
    op.drop_column('ai_log', 'query_id')
    op.drop_column('ai_log', 'session_id')
    op.drop_column('ai_log', 'org_id')
    op.drop_column('ai_log', 'app_id')
    # ### end Alembic commands ###