"""Remove unused embeddings table

Revision ID: 003
Revises: 002
Create Date: 2024-12-24 12:00:00.000000

"""
from alembic import op
import sqlalchemy as sa


revision = '003'
down_revision = '002'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.drop_table('embeddings')


def downgrade() -> None:
    op.create_table(
        'embeddings',
        sa.Column('embedding_id', sa.dialects.postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('document_id', sa.dialects.postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('embedding', sa.dialects.postgresql.JSON(astext_type=sa.Text()), nullable=True),
        sa.Column('model_version', sa.String(), nullable=True),
        sa.Column('created_at', sa.DateTime(), nullable=False),
        sa.ForeignKeyConstraint(['document_id'], ['documents.document_id'], ),
        sa.PrimaryKeyConstraint('embedding_id')
    )