chore: format

Author: tjbck

backend/open_webui/main.py

@@ -650,6 +650,7 @@ async def lifespan(app: FastAPI):
     asyncio.create_task(periodic_session_pool_cleanup())
 
     from open_webui.utils.automations import automation_worker_loop
+
     asyncio.create_task(automation_worker_loop(app))
 
     if app.state.config.ENABLE_BASE_MODELS_CACHE:

backend/open_webui/migrations/versions/b7c8d9e0f1a2_add_last_read_at_to_chat.py

@@ -5,6 +5,7 @@
 Create Date: 2026-04-01 04:00:00.000000
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 

backend/open_webui/migrations/versions/d4e5f6a7b8c9_add_automation_tables.py

@@ -10,51 +10,51 @@
 from alembic import op
 import sqlalchemy as sa
 
-revision: str = "d4e5f6a7b8c9"
-down_revision: Union[str, None] = "a3dd5bedd151"
+revision: str = 'd4e5f6a7b8c9'
+down_revision: Union[str, None] = 'a3dd5bedd151'
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     op.create_table(
-        "automation",
-        sa.Column("id", sa.Text(), primary_key=True),
-        sa.Column("user_id", sa.Text(), nullable=False),
-        sa.Column("name", sa.Text(), nullable=False),
-        sa.Column("data", sa.JSON(), nullable=False),
-        sa.Column("meta", sa.JSON(), nullable=True),
+        'automation',
+        sa.Column('id', sa.Text(), primary_key=True),
+        sa.Column('user_id', sa.Text(), nullable=False),
+        sa.Column('name', sa.Text(), nullable=False),
+        sa.Column('data', sa.JSON(), nullable=False),
+        sa.Column('meta', sa.JSON(), nullable=True),
         sa.Column(
-            "is_active",
+            'is_active',
             sa.Boolean(),
             nullable=False,
-            server_default=sa.text("1"),
+            server_default=sa.text('1'),
         ),
-        sa.Column("last_run_at", sa.BigInteger(), nullable=True),
-        sa.Column("next_run_at", sa.BigInteger(), nullable=True),
-        sa.Column("created_at", sa.BigInteger(), nullable=False),
-        sa.Column("updated_at", sa.BigInteger(), nullable=False),
+        sa.Column('last_run_at', sa.BigInteger(), nullable=True),
+        sa.Column('next_run_at', sa.BigInteger(), nullable=True),
+        sa.Column('created_at', sa.BigInteger(), nullable=False),
+        sa.Column('updated_at', sa.BigInteger(), nullable=False),
     )
-    op.create_index("ix_automation_next_run", "automation", ["next_run_at"])
+    op.create_index('ix_automation_next_run', 'automation', ['next_run_at'])
 
     op.create_table(
-        "automation_run",
-        sa.Column("id", sa.Text(), primary_key=True),
-        sa.Column("automation_id", sa.Text(), nullable=False),
-        sa.Column("chat_id", sa.Text(), nullable=True),
-        sa.Column("status", sa.Text(), nullable=False),
-        sa.Column("error", sa.Text(), nullable=True),
-        sa.Column("created_at", sa.BigInteger(), nullable=False),
+        'automation_run',
+        sa.Column('id', sa.Text(), primary_key=True),
+        sa.Column('automation_id', sa.Text(), nullable=False),
+        sa.Column('chat_id', sa.Text(), nullable=True),
+        sa.Column('status', sa.Text(), nullable=False),
+        sa.Column('error', sa.Text(), nullable=True),
+        sa.Column('created_at', sa.BigInteger(), nullable=False),
     )
     op.create_index(
-        "ix_automation_run_automation_id",
-        "automation_run",
-        ["automation_id"],
+        'ix_automation_run_automation_id',
+        'automation_run',
+        ['automation_id'],
     )
 
 
 def downgrade():
-    op.drop_index("ix_automation_run_automation_id")
-    op.drop_table("automation_run")
-    op.drop_index("ix_automation_next_run")
-    op.drop_table("automation")
+    op.drop_index('ix_automation_run_automation_id')
+    op.drop_table('automation_run')
+    op.drop_index('ix_automation_next_run')
+    op.drop_table('automation')

backend/open_webui/models/automations.py

@@ -18,7 +18,7 @@
 
 
 class Automation(Base):
-    __tablename__ = "automation"
+    __tablename__ = 'automation'
 
     id = Column(Text, primary_key=True)
     user_id = Column(Text, nullable=False)
@@ -32,13 +32,11 @@ class Automation(Base):
     created_at = Column(BigInteger, nullable=False)
     updated_at = Column(BigInteger, nullable=False)
 
-    __table_args__ = (
-        Index("ix_automation_next_run", "next_run_at"),
-    )
+    __table_args__ = (Index('ix_automation_next_run', 'next_run_at'),)
 
 
 class AutomationRun(Base):
-    __tablename__ = "automation_run"
+    __tablename__ = 'automation_run'
 
     id = Column(Text, primary_key=True)
     automation_id = Column(Text, nullable=False)
@@ -47,9 +45,7 @@ class AutomationRun(Base):
     error = Column(Text, nullable=True)
     created_at = Column(BigInteger, nullable=False)
 
-    __table_args__ = (
-        Index("ix_automation_run_automation_id", "automation_id"),
-    )
+    __table_args__ = (Index('ix_automation_run_automation_id', 'automation_id'),)
 
 
 ####################
@@ -119,7 +115,6 @@ class AutomationListResponse(BaseModel):
 
 
 class AutomationTable:
-
     def insert(
         self,
         user_id: str,
@@ -145,9 +140,7 @@ def insert(
             db.refresh(row)
             return AutomationModel.model_validate(row)
 
-    def get_by_id(
-        self, id: str, db: Optional[Session] = None
-    ) -> Optional[AutomationModel]:
+    def get_by_id(self, id: str, db: Optional[Session] = None) -> Optional[AutomationModel]:
         with get_db_context(db) as db:
             row = db.get(Automation, id)
             return AutomationModel.model_validate(row) if row else None
@@ -242,9 +235,7 @@ def delete(self, id: str, db: Optional[Session] = None) -> bool:
             db.commit()
             return True
 
-    def claim_due(
-        self, now_ns: int, limit: int = 10, db: Optional[Session] = None
-    ) -> list[AutomationModel]:
+    def claim_due(self, now_ns: int, limit: int = 10, db: Optional[Session] = None) -> list[AutomationModel]:
         """
         Atomically claim due automations for execution.
 
@@ -263,7 +254,7 @@ def claim_due(
                 .limit(limit)
             )
 
-            if db.bind.dialect.name == "postgresql":
+            if db.bind.dialect.name == 'postgresql':
                 stmt = stmt.with_for_update(skip_locked=True)
 
             rows = db.execute(stmt).scalars().all()
@@ -272,7 +263,7 @@ def claim_due(
 
             for row in rows:
                 row.last_run_at = now_ns
-                row.next_run_at = next_run_ns(row.data.get("rrule", ""))
+                row.next_run_at = next_run_ns(row.data.get('rrule', ''))
 
             db.commit()
 
@@ -285,7 +276,6 @@ def claim_due(
 
 
 class AutomationRunTable:
-
     def insert(
         self,
         automation_id: str,
@@ -308,9 +298,7 @@ def insert(
             db.refresh(row)
             return AutomationRunModel.model_validate(row)
 
-    def get_latest(
-        self, automation_id: str, db: Optional[Session] = None
-    ) -> Optional[AutomationRunModel]:
+    def get_latest(self, automation_id: str, db: Optional[Session] = None) -> Optional[AutomationRunModel]:
         with get_db_context(db) as db:
             row = (
                 db.query(AutomationRun)
@@ -338,15 +326,9 @@ def get_by_automation(
             )
             return [AutomationRunModel.model_validate(r) for r in rows]
 
-    def delete_by_automation(
-        self, automation_id: str, db: Optional[Session] = None
-    ) -> int:
+    def delete_by_automation(self, automation_id: str, db: Optional[Session] = None) -> int:
         with get_db_context(db) as db:
-            count = (
-                db.query(AutomationRun)
-                .filter_by(automation_id=automation_id)
-                .delete()
-            )
+            count = db.query(AutomationRun).filter_by(automation_id=automation_id).delete()
             db.commit()
             return count
 

backend/open_webui/models/chats.py

@@ -402,9 +402,7 @@ def update_chat_by_id(self, id: str, chat: dict, db: Optional[Session] = None) -
         except Exception:
             return None
 
-    def update_chat_last_read_at_by_id(
-        self, id: str, user_id: str, db: Optional[Session] = None
-    ) -> bool:
+    def update_chat_last_read_at_by_id(self, id: str, user_id: str, db: Optional[Session] = None) -> bool:
         try:
             with get_db_context(db) as db:
                 chat = db.get(Chat, id)

backend/open_webui/routers/automations.py

@@ -60,9 +60,7 @@ def check_automation_access(automation, user):
         )
 
 
-def enrich_automation(
-    automation: AutomationModel, db: Session, tz: str = None
-) -> AutomationResponse:
+def enrich_automation(automation: AutomationModel, db: Session, tz: str = None) -> AutomationResponse:
     last_run = AutomationRuns.get_latest(automation.id, db=db)
     return AutomationResponse(
         **automation.model_dump(),
@@ -100,10 +98,7 @@ async def get_automation_items(
     )
 
     return {
-        'items': [
-            enrich_automation(item, db, tz=user.timezone)
-            for item in result.items
-        ],
+        'items': [enrich_automation(item, db, tz=user.timezone) for item in result.items],
         'total': result.total,
     }
 
@@ -139,9 +134,7 @@ async def create_new_automation(
             )
 
     tz = user.timezone
-    automation = Automations.insert(
-        user.id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db
-    )
+    automation = Automations.insert(user.id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
     return enrich_automation(automation, db, tz=tz)
 
 
@@ -198,9 +191,7 @@ async def update_automation_by_id(
             )
 
     tz = user.timezone
-    updated = Automations.update_by_id(
-        id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db
-    )
+    updated = Automations.update_by_id(id, form_data, next_run_ns(form_data.data.rrule, tz=tz), db=db)
     return enrich_automation(updated, db, tz=tz)
 
 
@@ -219,9 +210,7 @@ async def toggle_automation_by_id(
     check_automations_permission(request, user)
     automation = Automations.get_by_id(id, db=db)
     check_automation_access(automation, user)
-    toggled = Automations.toggle(
-        id, next_run_ns(automation.data['rrule'], tz=user.timezone), db=db
-    )
+    toggled = Automations.toggle(id, next_run_ns(automation.data['rrule'], tz=user.timezone), db=db)
     return enrich_automation(toggled, db, tz=user.timezone)
 
 
@@ -281,6 +270,3 @@ async def get_automation_runs(
     automation = Automations.get_by_id(id, db=db)
     check_automation_access(automation, user)
     return AutomationRuns.get_by_automation(id, skip=skip, limit=limit, db=db)
-
-
-

backend/open_webui/socket/main.py

@@ -501,10 +501,7 @@ async def chat_events(sid, data):
     event_type = event_data.get('type')
 
     if event_type == 'last_read_at':
-        await asyncio.to_thread(
-            Chats.update_chat_last_read_at_by_id,
-            data['chat_id'], user['id']
-        )
+        await asyncio.to_thread(Chats.update_chat_last_read_at_by_id, data['chat_id'], user['id'])
 
 
 def normalize_document_id(document_id: str) -> str:

backend/open_webui/tools/builtin.py

@@ -2339,9 +2339,9 @@ async def view_skill(
 
 
 class TaskItem(BaseModel):
-    id: Optional[str] = Field(None, description="Unique identifier for the task. Auto-generated if omitted.")
-    content: Optional[str] = Field(None, description="Task description. Aliases: title, name, description.")
-    status: Literal['pending', 'in_progress', 'completed', 'cancelled'] = Field('pending', description="Task status.")
+    id: Optional[str] = Field(None, description='Unique identifier for the task. Auto-generated if omitted.')
+    content: Optional[str] = Field(None, description='Task description. Aliases: title, name, description.')
+    status: Literal['pending', 'in_progress', 'completed', 'cancelled'] = Field('pending', description='Task status.')
 
 
 async def tasks(