"""empty message

Revision ID: e859bc7fc0d4
Revises: 0998f67d7bd4
Create Date: 2024-09-13 00:20:57.597375

"""
from alembic import op
import sqlalchemy as sa
import sqlmodel.sql.sqltypes
from tidb_vector.sqlalchemy import VectorType


# revision identifiers, used by Alembic.
revision = 'e859bc7fc0d4'
down_revision = '0998f67d7bd4'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('regression_dataset', sa.Column('gold_answer', sa.JSON(), nullable=True))
    op.add_column('regression_dataset', sa.Column('dataset_name', sqlmodel.sql.sqltypes.AutoString(), nullable=True))
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('regression_dataset', 'dataset_name')
    op.drop_column('regression_dataset', 'gold_answer')
    # ### end Alembic commands ###