bhetherman/open-webui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-05-14 13:45:39 +09:00
parent 6b1df94bf9
commit 98d3b23085
2 changed files with 51 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/migrations/versions/922e7a387820_add_group_table.py
+35 -50
View File
@@ -16,70 +16,55 @@ depends_on = None
def upgrade(): def upgrade():
op.create_table( conn = op.get_bind()
'group', inspector = sa.inspect(conn)
sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True), existing_tables = set(inspector.get_table_names())
sa.Column('user_id', sa.Text(), nullable=True),
sa.Column('name', sa.Text(), nullable=True), if 'group' not in existing_tables:
sa.Column('description', sa.Text(), nullable=True), op.create_table(
sa.Column('data', sa.JSON(), nullable=True), 'group',
sa.Column('meta', sa.JSON(), nullable=True), sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
sa.Column('permissions', sa.JSON(), nullable=True), sa.Column('user_id', sa.Text(), nullable=True),
sa.Column('user_ids', sa.JSON(), nullable=True), sa.Column('name', sa.Text(), nullable=True),
sa.Column('created_at', sa.BigInteger(), nullable=True), sa.Column('description', sa.Text(), nullable=True),
sa.Column('updated_at', sa.BigInteger(), 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 # Add 'access_control' column to 'model' table
op.add_column( model_cols = {c['name'] for c in inspector.get_columns('model')}
'model', if 'access_control' not in model_cols:
sa.Column('access_control', sa.JSON(), nullable=True), op.add_column('model', sa.Column('access_control', sa.JSON(), nullable=True))
) if 'is_active' not in model_cols:
op.add_column(
# Add 'is_active' column to 'model' table 'model',
op.add_column( sa.Column('is_active', sa.Boolean(), nullable=False, server_default=sa.sql.expression.true()),
'model', )
sa.Column(
'is_active',
sa.Boolean(),
nullable=False,
server_default=sa.sql.expression.true(),
),
)
# Add 'access_control' column to 'knowledge' table # Add 'access_control' column to 'knowledge' table
op.add_column( knowledge_cols = {c['name'] for c in inspector.get_columns('knowledge')}
'knowledge', if 'access_control' not in knowledge_cols:
sa.Column('access_control', sa.JSON(), nullable=True), op.add_column('knowledge', sa.Column('access_control', sa.JSON(), nullable=True))
)
# Add 'access_control' column to 'prompt' table # Add 'access_control' column to 'prompt' table
op.add_column( prompt_cols = {c['name'] for c in inspector.get_columns('prompt')}
'prompt', if 'access_control' not in prompt_cols:
sa.Column('access_control', sa.JSON(), nullable=True), op.add_column('prompt', sa.Column('access_control', sa.JSON(), nullable=True))
)
# Add 'access_control' column to 'tools' table # Add 'access_control' column to 'tools' table
op.add_column( tool_cols = {c['name'] for c in inspector.get_columns('tool')}
'tool', if 'access_control' not in tool_cols:
sa.Column('access_control', sa.JSON(), nullable=True), op.add_column('tool', sa.Column('access_control', sa.JSON(), nullable=True))
)
def downgrade(): def downgrade():
op.drop_table('group') op.drop_table('group')
# Drop 'access_control' column from 'model' table
op.drop_column('model', 'access_control') op.drop_column('model', 'access_control')
# Drop 'is_active' column from 'model' table
op.drop_column('model', 'is_active') op.drop_column('model', 'is_active')
# Drop 'access_control' column from 'knowledge' table
op.drop_column('knowledge', 'access_control') op.drop_column('knowledge', 'access_control')
# Drop 'access_control' column from 'prompt' table
op.drop_column('prompt', 'access_control') op.drop_column('prompt', 'access_control')
# Drop 'access_control' column from 'tools' table
op.drop_column('tool', 'access_control') op.drop_column('tool', 'access_control')
backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py
+16 -11
View File
@@ -16,17 +16,22 @@ depends_on = None
def upgrade(): def upgrade():
op.create_table( conn = op.get_bind()
'note', inspector = sa.inspect(conn)
sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True), existing_tables = set(inspector.get_table_names())
sa.Column('user_id', sa.Text(), nullable=True),
sa.Column('title', sa.Text(), nullable=True), if 'note' not in existing_tables:
sa.Column('data', sa.JSON(), nullable=True), op.create_table(
sa.Column('meta', sa.JSON(), nullable=True), 'note',
sa.Column('access_control', sa.JSON(), nullable=True), sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
sa.Column('created_at', sa.BigInteger(), nullable=True), sa.Column('user_id', sa.Text(), nullable=True),
sa.Column('updated_at', sa.BigInteger(), nullable=True), sa.Column('title', sa.Text(), nullable=True),
) sa.Column('data', sa.JSON(), nullable=True),
sa.Column('meta', sa.JSON(), nullable=True),
sa.Column('access_control', sa.JSON(), nullable=True),
sa.Column('created_at', sa.BigInteger(), nullable=True),
sa.Column('updated_at', sa.BigInteger(), nullable=True),
)
def downgrade(): def downgrade():