chore: format

backend/open_webui/migrations/versions/2f1211949ecc_update_message_and_channel_member_table.py

@@ -12,7 +12,6 @@ from alembic import op
 import sqlalchemy as sa
 import open_webui.internal.db
 
-
 # revision identifiers, used by Alembic.
 revision: str = "2f1211949ecc"
 down_revision: Union[str, None] = "37f288994c47"

backend/open_webui/migrations/versions/374d2f66af06_add_prompt_history_table.py

@@ -12,7 +12,6 @@ import uuid
 from alembic import op
 import sqlalchemy as sa
 
-
 revision: str = "374d2f66af06"
 down_revision: Union[str, None] = "c440947495f3"
 branch_labels: Union[str, Sequence[str], None] = None

backend/open_webui/migrations/versions/37f288994c47_add_group_member_table.py

@@ -14,7 +14,6 @@ from typing import Sequence, Union
 from alembic import op
 import sqlalchemy as sa
 
-
 # revision identifiers, used by Alembic.
 revision: str = "37f288994c47"
 down_revision: Union[str, None] = "a5c220713937"

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

@@ -11,7 +11,6 @@ from typing import Sequence, Union
 from alembic import op
 import sqlalchemy as sa
 
-
 # revision identifiers, used by Alembic.
 revision: str = "38d63c18f30f"
 down_revision: Union[str, None] = "3af16a1c9fb6"

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

@@ -12,7 +12,6 @@ from alembic import op
 import sqlalchemy as sa
 import open_webui.internal.db
 
-
 # revision identifiers, used by Alembic.
 revision: str = "6283dc0e4d8d"
 down_revision: Union[str, None] = "3e0e00844bb0"

backend/open_webui/migrations/versions/6a39f3d8e55c_add_knowledge_table.py

@@ -11,7 +11,6 @@ import sqlalchemy as sa
 from sqlalchemy.sql import table, column, select
 import json
 
-
 revision = "6a39f3d8e55c"
 down_revision = "c0fbf31ca0db"
 branch_labels = None

backend/open_webui/migrations/versions/81cc2ce44d79_update_channel_file_and_knowledge_table.py

@@ -12,7 +12,6 @@ from alembic import op
 import sqlalchemy as sa
 import open_webui.internal.db
 
-
 # revision identifiers, used by Alembic.
 revision: str = "81cc2ce44d79"
 down_revision: Union[str, None] = "6283dc0e4d8d"

backend/open_webui/migrations/versions/8452d01d26d7_add_chat_message_table.py

@@ -165,7 +165,9 @@ def upgrade() -> None:
                 log.warning(f"Failed to insert message {message_id}: {e}")
                 continue
 
-    log.info(f"Backfilled {messages_inserted} messages into chat_message table ({messages_failed} failed)")
+    log.info(
+        f"Backfilled {messages_inserted} messages into chat_message table ({messages_failed} failed)"
+    )
 
 
 def downgrade() -> None:

backend/open_webui/migrations/versions/90ef40d4714e_update_channel_and_channel_members_table.py

@@ -12,7 +12,6 @@ from alembic import op
 import sqlalchemy as sa
 import open_webui.internal.db
 
-
 # revision identifiers, used by Alembic.
 revision: str = "90ef40d4714e"
 down_revision: Union[str, None] = "b10670c03dd5"

backend/open_webui/migrations/versions/b10670c03dd5_update_user_table.py

@@ -173,12 +173,10 @@ def upgrade() -> None:
     for uid, api_key in users_with_keys:
         if api_key:
             conn.execute(
-                sa.text(
-                    """
+                sa.text("""
                     INSERT INTO api_key (id, user_id, key, created_at, updated_at)
                     VALUES (:id, :user_id, :key, :created_at, :updated_at)
-                """
-                ),
+                """),
                 {
                     "id": f"key_{uid}",
                     "user_id": uid,

backend/open_webui/migrations/versions/c29facfe716b_update_file_table_path.py

@@ -12,7 +12,6 @@ import json
 from sqlalchemy.sql import table, column
 from sqlalchemy import String, Text, JSON, and_
 
-
 revision = "c29facfe716b"
 down_revision = "c69f45358db4"
 branch_labels = None

backend/open_webui/migrations/versions/c440947495f3_add_chat_file_table.py

@@ -11,7 +11,6 @@ from typing import Sequence, Union
 from alembic import op
 import sqlalchemy as sa
 
-
 # revision identifiers, used by Alembic.
 revision: str = "c440947495f3"
 down_revision: Union[str, None] = "81cc2ce44d79"

backend/open_webui/migrations/versions/f1e2d3c4b5a6_add_access_grant_table.py

@@ -98,26 +98,27 @@ def upgrade() -> None:
             # Could be Python None (SQL NULL) or string "null" (JSON null)
             # EXCEPTION: files with NULL are PRIVATE (owner-only), not public
             is_null = (
-                access_control_json is None or 
-                access_control_json == "null" or
-                (isinstance(access_control_json, str) and access_control_json.strip().lower() == "null")
+                access_control_json is None
+                or access_control_json == "null"
+                or (
+                    isinstance(access_control_json, str)
+                    and access_control_json.strip().lower() == "null"
+                )
             )
             if is_null:
                 # Files: NULL = private (no entry needed, owner has implicit access)
                 # Other resources: NULL = public (insert user:* for read)
                 if resource_type == "file":
                     continue  # Private - no entry needed
-                
+
                 key = (resource_type, resource_id, "user", "*", "read")
                 if key not in inserted:
                     try:
                         conn.execute(
-                            sa.text(
-                                """
+                            sa.text("""
                                 INSERT INTO access_grant (id, resource_type, resource_id, principal_type, principal_id, permission, created_at)
                                 VALUES (:id, :resource_type, :resource_id, :principal_type, :principal_id, :permission, :created_at)
-                            """
-                            ),
+                            """),
                             {
                                 "id": str(uuid.uuid4()),
                                 "resource_type": resource_type,
@@ -174,12 +175,10 @@ def upgrade() -> None:
                         continue
                     try:
                         conn.execute(
-                            sa.text(
-                                """
+                            sa.text("""
                                 INSERT INTO access_grant (id, resource_type, resource_id, principal_type, principal_id, permission, created_at)
                                 VALUES (:id, :resource_type, :resource_id, :principal_type, :principal_id, :permission, :created_at)
-                            """
-                            ),
+                            """),
                             {
                                 "id": str(uuid.uuid4()),
                                 "resource_type": resource_type,
@@ -200,12 +199,10 @@ def upgrade() -> None:
                         continue
                     try:
                         conn.execute(
-                            sa.text(
-                                """
+                            sa.text("""
                                 INSERT INTO access_grant (id, resource_type, resource_id, principal_type, principal_id, permission, created_at)
                                 VALUES (:id, :resource_type, :resource_id, :principal_type, :principal_id, :permission, :created_at)
-                            """
-                            ),
+                            """),
                             {
                                 "id": str(uuid.uuid4()),
                                 "resource_type": resource_type,
@@ -233,9 +230,9 @@ def upgrade() -> None:
 
 def downgrade() -> None:
     import json
-    
+
     conn = op.get_bind()
-    
+
     # Resource tables mapping: (table_name, resource_type)
     resource_tables = [
         ("knowledge", "knowledge"),
@@ -265,7 +262,7 @@ def downgrade() -> None:
                     FROM access_grant
                     WHERE resource_type = :resource_type
                 """),
-                {"resource_type": resource_type}
+                {"resource_type": resource_type},
             )
             rows = result.fetchall()
         except Exception:
@@ -287,39 +284,61 @@ def downgrade() -> None:
                 }
 
             # Handle public access (user:* for read)
-            if principal_type == "user" and principal_id == "*" and permission == "read":
+            if (
+                principal_type == "user"
+                and principal_id == "*"
+                and permission == "read"
+            ):
                 resource_grants[resource_id]["is_public"] = True
                 continue
 
             # Add to appropriate list
             if permission in ["read", "write"]:
                 if principal_type == "group":
-                    if principal_id not in resource_grants[resource_id][permission]["group_ids"]:
-                        resource_grants[resource_id][permission]["group_ids"].append(principal_id)
+                    if (
+                        principal_id
+                        not in resource_grants[resource_id][permission]["group_ids"]
+                    ):
+                        resource_grants[resource_id][permission]["group_ids"].append(
+                            principal_id
+                        )
                 elif principal_type == "user":
-                    if principal_id not in resource_grants[resource_id][permission]["user_ids"]:
-                        resource_grants[resource_id][permission]["user_ids"].append(principal_id)
+                    if (
+                        principal_id
+                        not in resource_grants[resource_id][permission]["user_ids"]
+                    ):
+                        resource_grants[resource_id][permission]["user_ids"].append(
+                            principal_id
+                        )
 
         # Step 3: Update each resource with reconstructed JSON
         for resource_id, grants in resource_grants.items():
             if grants["is_public"]:
                 # Public = NULL
                 access_control_value = None
-            elif (not grants["read"]["group_ids"] and not grants["read"]["user_ids"] and
-                  not grants["write"]["group_ids"] and not grants["write"]["user_ids"]):
+            elif (
+                not grants["read"]["group_ids"]
+                and not grants["read"]["user_ids"]
+                and not grants["write"]["group_ids"]
+                and not grants["write"]["user_ids"]
+            ):
                 # No grants = should not happen (would mean no entries), default to {}
                 access_control_value = json.dumps({})
             else:
                 # Custom permissions
-                access_control_value = json.dumps({
-                    "read": grants["read"],
-                    "write": grants["write"],
-                })
+                access_control_value = json.dumps(
+                    {
+                        "read": grants["read"],
+                        "write": grants["write"],
+                    }
+                )
 
             try:
                 conn.execute(
-                    sa.text(f'UPDATE "{table_name}" SET access_control = :access_control WHERE id = :id'),
-                    {"access_control": access_control_value, "id": resource_id}
+                    sa.text(
+                        f'UPDATE "{table_name}" SET access_control = :access_control WHERE id = :id'
+                    ),
+                    {"access_control": access_control_value, "id": resource_id},
                 )
             except Exception:
                 pass
@@ -330,15 +349,15 @@ def downgrade() -> None:
         if resource_type != "file":
             try:
                 conn.execute(
-                    sa.text(f'''
+                    sa.text(f"""
                         UPDATE "{table_name}" 
                         SET access_control = :private_value
                         WHERE id NOT IN (
                             SELECT DISTINCT resource_id FROM access_grant WHERE resource_type = :resource_type
                         )
                         AND access_control IS NULL
-                    '''),
-                    {"private_value": json.dumps({}), "resource_type": resource_type}
+                    """),
+                    {"private_value": json.dumps({}), "resource_type": resource_type},
                 )
             except Exception:
                 pass