"""Drop data path column

Revision ID: c27cc6cc9239
Revises: 915c200914f5
Create Date: 2024-01-08 14:04:18.780152

"""

import sqlalchemy as sa
from alembic import op

# revision identifiers, used by Alembic.
revision = "c27cc6cc9239"
down_revision = "915c200914f5"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column("inference_prompts", "data_path")
    op.drop_column("inference_responses", "data_path")
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        "inference_responses",
        sa.Column("data_path", sa.VARCHAR(), autoincrement=False, nullable=False),
    )
    op.add_column(
        "inference_prompts",
        sa.Column("data_path", sa.VARCHAR(), autoincrement=False, nullable=False),
    )
    # ### end Alembic commands ###
