mirror of
https://github.com/open-webui/open-webui.git
synced 2026-07-10 12:01:29 -05:00
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""Add folder table
|
|
|
|
Revision ID: c69f45358db4
|
|
Revises: 3ab32c4b8f59
|
|
Create Date: 2024-10-16 02:02:35.241684
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = 'c69f45358db4'
|
|
down_revision = '3ab32c4b8f59'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn)
|
|
existing_tables = set(inspector.get_table_names())
|
|
|
|
if 'folder' not in existing_tables:
|
|
op.create_table(
|
|
'folder',
|
|
sa.Column('id', sa.Text(), nullable=False),
|
|
sa.Column('parent_id', sa.Text(), nullable=True),
|
|
sa.Column('user_id', sa.Text(), nullable=False),
|
|
sa.Column('name', sa.Text(), nullable=False),
|
|
sa.Column('items', sa.JSON(), nullable=True),
|
|
sa.Column('meta', sa.JSON(), nullable=True),
|
|
sa.Column('is_expanded', sa.Boolean(), default=False, nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now(), nullable=False),
|
|
sa.Column(
|
|
'updated_at',
|
|
sa.DateTime(),
|
|
nullable=False,
|
|
server_default=sa.func.now(),
|
|
onupdate=sa.func.now(),
|
|
),
|
|
sa.PrimaryKeyConstraint('id', 'user_id'),
|
|
)
|
|
|
|
chat_cols = {c['name'] for c in inspector.get_columns('chat')}
|
|
if 'folder_id' not in chat_cols:
|
|
op.add_column(
|
|
'chat',
|
|
sa.Column('folder_id', sa.Text(), nullable=True),
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_column('chat', 'folder_id')
|
|
|
|
op.drop_table('folder')
|