components/dbo/alembic/versions/12bb1ebae3ff_logs_refactoring.py

"""Logs refactoring

Revision ID: 12bb1ebae3ff
Revises: 6635b061c086
Create Date: 2025-04-16 12:00:40.247356

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '12bb1ebae3ff'
down_revision: Union[str, None] = '6635b061c086'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('log', sa.Column('user_request', sa.String(), nullable=True))
    op.add_column('log', sa.Column('qe_result', sa.String(), nullable=True))
    op.add_column('log', sa.Column('search_result', sa.String(), nullable=True))
    op.add_column('log', sa.Column('llm_result', sa.String(), nullable=True))
    op.add_column('log', sa.Column('llm_settings', sa.String(), nullable=True))
    op.add_column('log', sa.Column('user_name', sa.String(), nullable=True))
    op.add_column('log', sa.Column('error', sa.String(), nullable=True))
    op.drop_column('log', 'query_type')
    op.drop_column('log', 'llm_classifier')
    op.drop_column('log', 'llmResponse')
    op.drop_column('log', 'userRequest')
    op.drop_column('log', 'userName')
    op.drop_column('log', 'llmPrompt')
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('log', sa.Column('llmPrompt', sa.VARCHAR(), nullable=True))
    op.add_column('log', sa.Column('userName', sa.VARCHAR(), nullable=True))
    op.add_column('log', sa.Column('userRequest', sa.VARCHAR(), nullable=True))
    op.add_column('log', sa.Column('llmResponse', sa.VARCHAR(), nullable=True))
    op.add_column('log', sa.Column('llm_classifier', sa.VARCHAR(), nullable=True))
    op.add_column('log', sa.Column('query_type', sa.VARCHAR(), nullable=True))
    op.drop_column('log', 'error')
    op.drop_column('log', 'user_name')
    op.drop_column('log', 'llm_settings')
    op.drop_column('log', 'llm_result')
    op.drop_column('log', 'search_result')
    op.drop_column('log', 'qe_result')
    op.drop_column('log', 'user_request')
    # ### end Alembic commands ###