"""add fileupload

Revision ID: aafae11896a3
Revises: 65626474a0df
Create Date: 2024-05-14 18:50:55.452162

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql

import tidb_vector

# revision identifiers, used by Alembic.
revision = 'aafae11896a3'
down_revision = '65626474a0df'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('image_corpus',
    sa.Column('id', sa.Integer(), nullable=False),
    sa.Column('image_embedding', tidb_vector.sqlalchemy.VectorType(dim=512), nullable=True),
    sa.Column('description', sa.String(length=1024), nullable=True),
    sa.Column('description_vector', tidb_vector.sqlalchemy.VectorType(dim=1536), nullable=True),
    sa.Column('filename', sa.String(length=255), nullable=False),
    sa.Column('slug', sa.String(length=255), nullable=False),
    sa.Column('mimetype', sa.String(length=255), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
    sa.Column('updated_at', sa.DateTime(), server_default=sa.text('now()'), nullable=True),
    sa.PrimaryKeyConstraint('id')
    )
    op.drop_table('file_uploads')


def downgrade():
    op.drop_table('image_corpus')