mirror of
https://github.com/open-webui/open-webui
synced 2024-11-25 05:18:15 +00: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")
|