"""Update Retriever Arguments

Revision ID: 03385f2000da
Revises: e476c2013352
Create Date: 2024-11-04 08:33:46.158354

"""
from alembic import op
import sqlalchemy as sa
import pgvector
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = '03385f2000da'
down_revision = 'e476c2013352'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('retriever', sa.Column('arguments', postgresql.JSONB(astext_type=sa.Text()), nullable=True))
    # ### end Alembic commands ###


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