mirror of
https://github.com/open-webui/open-webui.git
synced 2026-04-25 17:15:16 +02:00
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
"""Add group table
|
|
|
|
Revision ID: 922e7a387820
|
|
Revises: 4ace53fd72c8
|
|
Create Date: 2024-11-14 03:00:00.000000
|
|
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = '922e7a387820'
|
|
down_revision = '4ace53fd72c8'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.create_table(
|
|
'group',
|
|
sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
|
|
sa.Column('user_id', sa.Text(), nullable=True),
|
|
sa.Column('name', sa.Text(), nullable=True),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('data', sa.JSON(), nullable=True),
|
|
sa.Column('meta', sa.JSON(), nullable=True),
|
|
sa.Column('permissions', sa.JSON(), nullable=True),
|
|
sa.Column('user_ids', sa.JSON(), nullable=True),
|
|
sa.Column('created_at', sa.BigInteger(), nullable=True),
|
|
sa.Column('updated_at', sa.BigInteger(), nullable=True),
|
|
)
|
|
|
|
# Add 'access_control' column to 'model' table
|
|
op.add_column(
|
|
'model',
|
|
sa.Column('access_control', sa.JSON(), nullable=True),
|
|
)
|
|
|
|
# Add 'is_active' column to 'model' table
|
|
op.add_column(
|
|
'model',
|
|
sa.Column(
|
|
'is_active',
|
|
sa.Boolean(),
|
|
nullable=False,
|
|
server_default=sa.sql.expression.true(),
|
|
),
|
|
)
|
|
|
|
# Add 'access_control' column to 'knowledge' table
|
|
op.add_column(
|
|
'knowledge',
|
|
sa.Column('access_control', sa.JSON(), nullable=True),
|
|
)
|
|
|
|
# Add 'access_control' column to 'prompt' table
|
|
op.add_column(
|
|
'prompt',
|
|
sa.Column('access_control', sa.JSON(), nullable=True),
|
|
)
|
|
|
|
# Add 'access_control' column to 'tools' table
|
|
op.add_column(
|
|
'tool',
|
|
sa.Column('access_control', sa.JSON(), nullable=True),
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_table('group')
|
|
|
|
# Drop 'access_control' column from 'model' table
|
|
op.drop_column('model', 'access_control')
|
|
|
|
# Drop 'is_active' column from 'model' table
|
|
op.drop_column('model', 'is_active')
|
|
|
|
# Drop 'access_control' column from 'knowledge' table
|
|
op.drop_column('knowledge', 'access_control')
|
|
|
|
# Drop 'access_control' column from 'prompt' table
|
|
op.drop_column('prompt', 'access_control')
|
|
|
|
# Drop 'access_control' column from 'tools' table
|
|
op.drop_column('tool', 'access_control')
|