chore: format

backend/open_webui/migrations/versions/3781e22d8b01_update_message_table.py

@@ -53,7 +53,9 @@ def upgrade():
     if 'channel_member' not in existing_tables:
         op.create_table(
             'channel_member',
-            sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),  # Record ID for the membership row
+            sa.Column(
+                'id', sa.Text(), nullable=False, primary_key=True, unique=True
+            ),  # Record ID for the membership row
             sa.Column('channel_id', sa.Text(), nullable=False),  # Associated channel
             sa.Column('user_id', sa.Text(), nullable=False),  # Associated user
             sa.Column('created_at', sa.BigInteger(), nullable=True),  # Timestamp of when the user joined the channel

backend/open_webui/migrations/versions/38d63c18f30f_add_oauth_session_table.py

@@ -42,9 +42,9 @@ def upgrade() -> None:
 
     # Create indexes (idempotent — no-ops when table was just created
     # with the columns above, and safe to call if indexes already exist).
-    existing_indexes = {
-        idx['name'] for idx in inspector.get_indexes('oauth_session')
-    } if 'oauth_session' in existing_tables else set()
+    existing_indexes = (
+        {idx['name'] for idx in inspector.get_indexes('oauth_session')} if 'oauth_session' in existing_tables else set()
+    )
 
     if 'idx_oauth_session_user_id' not in existing_indexes:
         op.create_index('idx_oauth_session_user_id', 'oauth_session', ['user_id'])

backend/open_webui/migrations/versions/3c9b0ca343fd_add_knowledge_directory_table.py

@@ -5,6 +5,7 @@ Revises: a0b1c2d3e4f5
 Create Date: 2026-05-13 21:58:40.832482
 
 """
+
 from typing import Sequence, Union
 
 from alembic import op
@@ -36,7 +37,9 @@ def upgrade() -> None:
             sa.ForeignKeyConstraint(['knowledge_id'], ['knowledge.id'], ondelete='CASCADE'),
             sa.ForeignKeyConstraint(['parent_id'], ['knowledge_directory.id'], ondelete='CASCADE'),
             sa.PrimaryKeyConstraint('id'),
-            sa.UniqueConstraint('knowledge_id', 'parent_id', 'name', name='uq_knowledge_directory_knowledge_parent_name'),
+            sa.UniqueConstraint(
+                'knowledge_id', 'parent_id', 'name', name='uq_knowledge_directory_knowledge_parent_name'
+            ),
         )
         op.create_index('ix_knowledge_directory_knowledge_id', 'knowledge_directory', ['knowledge_id'])
         op.create_index('ix_knowledge_directory_parent_id', 'knowledge_directory', ['parent_id'])

backend/open_webui/migrations/versions/461111b60977_add_missing_primary_keys_to_legacy_.py

@@ -5,6 +5,7 @@ Revises: 3c9b0ca343fd
 Create Date: 2026-05-14 04:38:14.000000
 
 """
+
 from typing import Sequence, Union
 
 import sqlalchemy as sa
@@ -21,9 +22,18 @@ depends_on: Union[str, Sequence[str], None] = None
 # already have correct PKs from 7e5b5dc7342b_init.py.
 # 'tag' uses a composite PK since the same tag name can exist for multiple users.
 LEGACY_TABLES = {
-    'auth': ['id'], 'chat': ['id'], 'chatidtag': ['id'], 'document': ['id'],
-    'file': ['id'], 'function': ['id'], 'memory': ['id'], 'model': ['id'],
-    'prompt': ['id'], 'tag': ['id', 'user_id'], 'tool': ['id'], 'user': ['id'],
+    'auth': ['id'],
+    'chat': ['id'],
+    'chatidtag': ['id'],
+    'document': ['id'],
+    'file': ['id'],
+    'function': ['id'],
+    'memory': ['id'],
+    'model': ['id'],
+    'prompt': ['id'],
+    'tag': ['id', 'user_id'],
+    'tool': ['id'],
+    'user': ['id'],
 }
 
 

backend/open_webui/migrations/versions/4de81c2a3af1_add_pinned_note_table.py

@@ -52,13 +52,18 @@ def upgrade() -> None:
             column('created_at', sa.BigInteger),
         )
 
-        notes = conn.execute(select(note_table.c.id, note_table.c.user_id).where(note_table.c.is_pinned == True)).fetchall()
+        notes = conn.execute(
+            select(note_table.c.id, note_table.c.user_id).where(note_table.c.is_pinned == True)
+        ).fetchall()
 
         if notes:
             now = int(time.time_ns())
             conn.execute(
                 insert(pinned_note_table),
-                [{'id': str(uuid.uuid4()), 'user_id': note[1], 'note_id': note[0], 'created_at': now} for note in notes],
+                [
+                    {'id': str(uuid.uuid4()), 'user_id': note[1], 'note_id': note[0], 'created_at': now}
+                    for note in notes
+                ],
             )
 
         with op.batch_alter_table('note', schema=None) as batch_op:

backend/open_webui/migrations/versions/56359461a091_add_calendar_tables.py

@@ -95,7 +95,9 @@ def upgrade() -> None:
     inspector.clear_cache()
     if 'calendar_event_attendee' in inspector.get_table_names():
         if not _index_exists(inspector, 'ix_calendar_event_attendee_user', 'calendar_event_attendee'):
-            op.create_index('ix_calendar_event_attendee_user', 'calendar_event_attendee', ['user_id', 'status'], unique=False)
+            op.create_index(
+                'ix_calendar_event_attendee_user', 'calendar_event_attendee', ['user_id', 'status'], unique=False
+            )
 
 
 def downgrade() -> None:

backend/open_webui/migrations/versions/6283dc0e4d8d_add_channel_file_table.py

@@ -48,7 +48,9 @@ def upgrade() -> None:
             sa.Index('ix_channel_file_file_id', 'file_id'),
             sa.Index('ix_channel_file_user_id', 'user_id'),
             # unique constraints
-            sa.UniqueConstraint('channel_id', 'file_id', name='uq_channel_file_channel_file'),  # prevent duplicate entries
+            sa.UniqueConstraint(
+                'channel_id', 'file_id', name='uq_channel_file_channel_file'
+            ),  # prevent duplicate entries
         )
 
 

backend/open_webui/migrations/versions/7e5b5dc7342b_init.py

@@ -9,7 +9,8 @@ import sqlalchemy as sa
 from alembic import op
 from open_webui.internal.db import JSONField
 from open_webui.migrations.util import get_existing_tables
-revision: str = "7e5b5dc7342b"
+
+revision: str = '7e5b5dc7342b'
 down_revision: str | None = None
 branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None
@@ -175,12 +176,16 @@ _INITIAL_TABLES: list[tuple[str, list[sa.Column], list]] = [
         ],
     ),
 ]
+
+
 # --- migration execution ---
 def upgrade() -> None:  # deploy initial schema tables
     existing_tables = set(get_existing_tables())
     for name, columns, constraints in _INITIAL_TABLES:
         if name not in existing_tables:
             op.create_table(name, *columns, *constraints)
+
+
 # --- rollback function ---
 def downgrade() -> None:  # rollback initial schema tables
     for table_name, _, _ in reversed(_INITIAL_TABLES):

backend/open_webui/migrations/versions/b10670c03dd5_update_user_table.py

@@ -182,27 +182,18 @@ def upgrade() -> None:
 
     # ── Migrate oauth_sub → oauth JSON (only if old column still exists)
     if 'oauth_sub' in user_columns:
-        rows = conn.execute(
-            sa.select(_user.c.id, _user.c.oauth_sub)
-            .where(_user.c.oauth_sub.is_not(None))
-        ).fetchall()
+        rows = conn.execute(sa.select(_user.c.id, _user.c.oauth_sub).where(_user.c.oauth_sub.is_not(None))).fetchall()
 
         for uid, oauth_sub in rows:
             if oauth_sub:
-                provider, sub = (oauth_sub.split('@', 1) if '@' in oauth_sub
-                                 else ('oidc', oauth_sub))
+                provider, sub = oauth_sub.split('@', 1) if '@' in oauth_sub else ('oidc', oauth_sub)
                 conn.execute(
-                    sa.update(_user)
-                    .where(_user.c.id == uid)
-                    .values(oauth=json.dumps({provider: {'sub': sub}}))
+                    sa.update(_user).where(_user.c.id == uid).values(oauth=json.dumps({provider: {'sub': sub}}))
                 )
 
     # ── Migrate api_key column → api_key table (only if old column still exists)
     if 'api_key' in user_columns:
-        rows = conn.execute(
-            sa.select(_user.c.id, _user.c.api_key)
-            .where(_user.c.api_key.is_not(None))
-        ).fetchall()
+        rows = conn.execute(sa.select(_user.c.id, _user.c.api_key).where(_user.c.api_key.is_not(None))).fetchall()
         now = int(time.time())
 
         for uid, key_val in rows:
@@ -233,10 +224,7 @@ def downgrade() -> None:
     op.add_column('user', sa.Column('oauth_sub', sa.Text(), nullable=True))
 
     conn = op.get_bind()
-    rows = conn.execute(
-        sa.select(_user.c.id, _user.c.oauth)
-        .where(_user.c.oauth.is_not(None))
-    ).fetchall()
+    rows = conn.execute(sa.select(_user.c.id, _user.c.oauth).where(_user.c.oauth.is_not(None))).fetchall()
 
     for uid, oauth in rows:
         try:
@@ -247,11 +235,7 @@ def downgrade() -> None:
         except Exception:
             oauth_sub = None
 
-        conn.execute(
-            sa.update(_user)
-            .where(_user.c.id == uid)
-            .values(oauth_sub=oauth_sub)
-        )
+        conn.execute(sa.update(_user).where(_user.c.id == uid).values(oauth_sub=oauth_sub))
 
     op.drop_column('user', 'oauth')
 
@@ -260,11 +244,7 @@ def downgrade() -> None:
 
     keys = conn.execute(sa.select(_api_key.c.user_id, _api_key.c.key)).fetchall()
     for uid, key in keys:
-        conn.execute(
-            sa.update(_user)
-            .where(_user.c.id == uid)
-            .values(api_key=key)
-        )
+        conn.execute(sa.update(_user).where(_user.c.id == uid).values(api_key=key))
 
     op.drop_table('api_key')
 

backend/open_webui/migrations/versions/c1d2e3f4a5b6_add_shared_chat_table.py

@@ -101,9 +101,7 @@ def upgrade():
 
         # Check if shared_chat record already exists (idempotent)
         existing_shared = conn.execute(
-            sa.select(shared_chat_t.c.id).where(
-                shared_chat_t.c.id == share_token
-            )
+            sa.select(shared_chat_t.c.id).where(shared_chat_t.c.id == share_token)
         ).fetchone()
 
         if not existing_shared: