"""Chunking information added to Document models

Revision ID: 324ed734ed9b
Revises: 3bc7bbed1ec6
Create Date: 2024-05-08 17:15:18.472857

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '324ed734ed9b'
down_revision = '3bc7bbed1ec6'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('document_language', sa.Column('user_context', sa.Text(), nullable=True))
    op.add_column('document_language', sa.Column('system_context', sa.Text(), nullable=True))
    op.add_column('embedding_mistral', sa.Column('chunk', sa.Text(), nullable=False))
    op.add_column('embedding_small_open_ai', sa.Column('chunk', sa.Text(), nullable=False))
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('embedding_small_open_ai', 'chunk')
    op.drop_column('embedding_mistral', 'chunk')
    op.drop_column('document_language', 'system_context')
    op.drop_column('document_language', 'user_context')
    # ### end Alembic commands ###
