diff --git a/backend/open_webui/migrations/versions/922e7a387820_add_group_table.py b/backend/open_webui/migrations/versions/922e7a387820_add_group_table.py index 1f035f5d4f..4150f5ffdf 100644 --- a/backend/open_webui/migrations/versions/922e7a387820_add_group_table.py +++ b/backend/open_webui/migrations/versions/922e7a387820_add_group_table.py @@ -16,70 +16,55 @@ 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), - ) + conn = op.get_bind() + inspector = sa.inspect(conn) + existing_tables = set(inspector.get_table_names()) + + if 'group' not in existing_tables: + 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(), - ), - ) + model_cols = {c['name'] for c in inspector.get_columns('model')} + if 'access_control' not in model_cols: + op.add_column('model', sa.Column('access_control', sa.JSON(), nullable=True)) + if 'is_active' not in model_cols: + 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), - ) + knowledge_cols = {c['name'] for c in inspector.get_columns('knowledge')} + if 'access_control' not in knowledge_cols: + 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), - ) + prompt_cols = {c['name'] for c in inspector.get_columns('prompt')} + if 'access_control' not in prompt_cols: + 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), - ) + tool_cols = {c['name'] for c in inspector.get_columns('tool')} + if 'access_control' not in tool_cols: + 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') diff --git a/backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py b/backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py index 555224566b..ae096e0808 100644 --- a/backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py +++ b/backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py @@ -16,17 +16,22 @@ depends_on = None def upgrade(): - op.create_table( - 'note', - sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True), - sa.Column('user_id', sa.Text(), 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), - ) + conn = op.get_bind() + inspector = sa.inspect(conn) + existing_tables = set(inspector.get_table_names()) + + if 'note' not in existing_tables: + op.create_table( + 'note', + sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True), + sa.Column('user_id', sa.Text(), 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():