rag-gateway/alembic/versions/75d247bc0248_knowledgebases_tables_add.py

"""knowledgebases tables add 

Revision ID: 75d247bc0248
Revises: 16efd20cce44
Create Date: 2024-12-17 19:38:44.939447

"""
from typing import Sequence, Union

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

# revision identifiers, used by Alembic.
revision: str = '75d247bc0248'
down_revision: Union[str, None] = '16efd20cce44'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('knowledgebases',
    sa.Column('id', sa.String(length=36), nullable=False),
    sa.Column('name', sa.String(length=128), nullable=True),
    sa.Column('create_date', sa.DateTime(), nullable=True),
    sa.Column('update_date', sa.DateTime(), nullable=True),
    sa.Column('permission', sa.String(length=32), nullable=True),
    sa.Column('tenant_id', sa.String(length=32), nullable=True),
    sa.Column('description', sa.Text(), nullable=True),
    sa.Column('status', sa.String(length=1), nullable=True),
    sa.Column('documents', sa.Integer(), nullable=True),
    sa.Column('icon', sa.String(length=128), nullable=True),
    sa.Column('knowledge_type', sa.String(length=1), nullable=True),
    sa.PrimaryKeyConstraint('id')
    )
    op.drop_table('knowledgebase')
    op.drop_constraint('group_Knowledge_ibfk_2', 'group_Knowledge', type_='foreignkey')
    op.create_foreign_key(None, 'group_Knowledge', 'knowledgebases', ['knowledge_id'], ['id'], ondelete='CASCADE')
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint(None, 'group_Knowledge', type_='foreignkey')
    op.create_foreign_key('group_Knowledge_ibfk_2', 'group_Knowledge', 'knowledgebase', ['knowledge_id'], ['id'], ondelete='CASCADE')
    op.create_table('knowledgebase',
    sa.Column('id', mysql.VARCHAR(length=36), nullable=False),
    sa.Column('name', mysql.VARCHAR(length=128), nullable=True),
    sa.Column('create_date', mysql.DATETIME(), nullable=True),
    sa.Column('update_date', mysql.DATETIME(), nullable=True),
    sa.Column('tenant_id', mysql.VARCHAR(length=32), nullable=True),
    sa.Column('description', mysql.TEXT(), nullable=True),
    sa.Column('status', mysql.VARCHAR(length=1), nullable=True),
    sa.Column('documents', mysql.INTEGER(), autoincrement=False, nullable=True),
    sa.Column('icon', mysql.VARCHAR(length=128), nullable=True),
    sa.Column('knowledge_type', mysql.VARCHAR(length=1), nullable=True),
    sa.Column('permission', mysql.VARCHAR(length=32), nullable=True),
    sa.PrimaryKeyConstraint('id'),
    mysql_collate='utf8mb4_0900_ai_ci',
    mysql_default_charset='utf8mb4',
    mysql_engine='InnoDB'
    )
    op.drop_table('knowledgebases')
    # ### end Alembic commands ###