"""Add tuning to Retriever

Revision ID: 331f8100eb87
Revises: 6c5ca750e60c
Create Date: 2024-10-31 07:17:22.579376

"""
from alembic import op
import sqlalchemy as sa
import pgvector


# revision identifiers, used by Alembic.
revision = '331f8100eb87'
down_revision = '6c5ca750e60c'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('retriever', sa.Column('tuning', sa.Boolean(), nullable=True))
    # ### end Alembic commands ###


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