chore: linting

surfsense_backend/app/tasks/chat/streaming/context/deepagents_todos.py

@@ -19,9 +19,7 @@ def extract_todos_from_deepagents(command_output: Any) -> dict:
     elif isinstance(command_output, dict):
         if "todos" in command_output:
             todos_data = command_output.get("todos", [])
-        elif "update" in command_output and isinstance(
-            command_output["update"], dict
-        ):
+        elif "update" in command_output and isinstance(command_output["update"], dict):
             todos_data = command_output["update"].get("todos", [])
 
     return {"todos": todos_data}

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/auto_pin.py

@@ -69,17 +69,13 @@ async def resolve_initial_auto_pin(
                 "pin.requires_image_input": requires_image_input,
             },
         )
-        return AutoPinResult(
-            llm_config_id=pinned.resolved_llm_config_id, error=None
-        )
+        return AutoPinResult(llm_config_id=pinned.resolved_llm_config_id, error=None)
     except ValueError as pin_error:
         # The "no vision-capable cfg" path raises a ValueError whose message
         # we map to the friendly image-input SSE error so the user sees the
         # same message regardless of whether the gate fired in the resolver or
         # in ``llm_capability.assert_vision_capability_for_image_turn``.
-        is_vision_failure = (
-            requires_image_input and "vision-capable" in str(pin_error)
-        )
+        is_vision_failure = requires_image_input and "vision-capable" in str(pin_error)
         error_code = (
             "MODEL_DOES_NOT_SUPPORT_IMAGE_INPUT"
             if is_vision_failure

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/input_state.py

@@ -207,9 +207,7 @@ async def _resolve_mentions_for_query(
             try:
                 chip_objs.append(MentionedDocumentInfo.model_validate(raw))
             except Exception:
-                logger.debug(
-                    "stream_new_chat: dropping malformed mention chip %r", raw
-                )
+                logger.debug("stream_new_chat: dropping malformed mention chip %r", raw)
 
     resolved = await resolve_mentions(
         session,

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/llm_capability.py

@@ -48,9 +48,7 @@ def check_image_input_capability(
         return None
 
     model_label = agent_config.config_name or agent_config.model_name or "model"
-    ot.add_event(
-        "quota.denied", {"quota.code": "MODEL_DOES_NOT_SUPPORT_IMAGE_INPUT"}
-    )
+    ot.add_event("quota.denied", {"quota.code": "MODEL_DOES_NOT_SUPPORT_IMAGE_INPUT"})
     return (
         (
             f"The selected model ({model_label}) does not support "

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/orchestrator.py

@@ -259,7 +259,8 @@ async def stream_new_chat(
 
         if needs_premium_quota(agent_config, user_id):
             premium_reservation = await reserve_premium(
-                agent_config=agent_config, user_id=user_id  # type: ignore[arg-type]
+                agent_config=agent_config,
+                user_id=user_id,  # type: ignore[arg-type]
             )
             if not premium_reservation.allowed:
                 ot.add_event("quota.denied", {"quota.code": "PREMIUM_QUOTA_EXHAUSTED"})
@@ -492,7 +493,9 @@ async def stream_new_chat(
 
         # --- Block 4: First SSE frames ---
 
-        for sse in iter_initial_frames(streaming_service, turn_id=stream_result.turn_id):
+        for sse in iter_initial_frames(
+            streaming_service, turn_id=stream_result.turn_id
+        ):
             yield sse
 
         # --- Block 5: Persistence join + message-id frames ---
@@ -693,7 +696,9 @@ async def stream_new_chat(
             fallback_commit_search_space_id=search_space_id,
             fallback_commit_created_by_id=user_id,
             fallback_commit_filesystem_mode=(
-                filesystem_selection.mode if filesystem_selection else FilesystemMode.CLOUD
+                filesystem_selection.mode
+                if filesystem_selection
+                else FilesystemMode.CLOUD
             ),
             fallback_commit_thread_id=chat_id,
             runtime_context=runtime_context,
@@ -715,11 +720,7 @@ async def stream_new_chat(
                 title_emitted = True
             # Account for the case where the task completed but produced no
             # title — flip the flag anyway so we don't keep checking it.
-            if (
-                title_task is not None
-                and title_task.done()
-                and not title_emitted
-            ):
+            if title_task is not None and title_task.done() and not title_emitted:
                 title_emitted = True
 
         _perf_log.info(
@@ -811,9 +812,7 @@ async def stream_new_chat(
             end_turn(str(chat_id))
 
             if premium_reservation is not None and user_id:
-                await release_premium(
-                    reservation=premium_reservation, user_id=user_id
-                )
+                await release_premium(reservation=premium_reservation, user_id=user_id)
 
             await close_session_and_clear_ai_responding(session, chat_id)
 
@@ -852,9 +851,9 @@ async def stream_new_chat(
 
         # Break circular refs held by the agent graph, tools, and LLM
         # wrappers so the GC can reclaim them in a single pass.
-        agent = llm = connector_service = None  # noqa: F841
-        input_state = stream_result = None  # noqa: F841
-        session = None  # noqa: F841
+        agent = llm = connector_service = None
+        input_state = stream_result = None
+        session = None
 
         run_gc_pass(log_prefix="stream_new_chat", chat_id=chat_id)
         close_chat_request_span(

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/runtime_context.py

@@ -30,9 +30,7 @@ def build_new_chat_runtime_context(
     return SurfSenseContextSchema(
         search_space_id=search_space_id,
         mentioned_document_ids=list(mentioned_document_ids or []),
-        mentioned_folder_ids=list(
-            accepted_folder_ids or mentioned_folder_ids or []
-        ),
+        mentioned_folder_ids=list(accepted_folder_ids or mentioned_folder_ids or []),
         request_id=request_id,
         turn_id=turn_id,
     )

surfsense_backend/app/tasks/chat/streaming/flows/new_chat/title_gen.py

@@ -133,12 +133,8 @@ async def _generate_title(
             # inherited Azure endpoint — see ``provider_api_base`` for the
             # same bug repro on the image-gen / vision paths.
             raw_model = getattr(llm, "model", "") or ""
-            provider_prefix = (
-                raw_model.split("/", 1)[0] if "/" in raw_model else None
-            )
-            provider_value = (
-                agent_config.provider if agent_config is not None else None
-            )
+            provider_prefix = raw_model.split("/", 1)[0] if "/" in raw_model else None
+            provider_value = agent_config.provider if agent_config is not None else None
             title_api_base = resolve_api_base(
                 provider=provider_value,
                 provider_prefix=provider_prefix,

surfsense_backend/app/tasks/chat/streaming/flows/resume_chat/orchestrator.py

@@ -15,14 +15,10 @@ building blocks under ``flows/shared/``. Mirrors ``stream_new_chat`` but:
 from __future__ import annotations
 
 import contextlib
-import gc
 import logging
-import sys
 import time
-import uuid as _uuid
 from collections.abc import AsyncGenerator
 from functools import partial
-from typing import Any
 from uuid import UUID
 
 import anyio
@@ -32,7 +28,7 @@ from app.agents.new_chat.chat_deepagent import create_surfsense_deep_agent
 from app.agents.new_chat.filesystem_selection import FilesystemMode, FilesystemSelection
 from app.agents.new_chat.middleware.busy_mutex import end_turn
 from app.config import config as _app_config
-from app.db import ChatVisibility, async_session_maker, shielded_async_session
+from app.db import ChatVisibility, async_session_maker
 from app.observability import otel as ot
 from app.services.chat_session_state_service import set_ai_responding
 from app.services.new_streaming_service import VercelStreamingService
@@ -89,7 +85,7 @@ from app.tasks.chat.streaming.flows.shared.terminal_error import (
 )
 from app.tasks.chat.streaming.shared.stream_result import StreamResult
 from app.tasks.chat.streaming.shared.utils import resume_step_prefix
-from app.utils.perf import get_perf_logger, log_system_snapshot
+from app.utils.perf import get_perf_logger
 
 logger = logging.getLogger(__name__)
 _perf_log = get_perf_logger()
@@ -217,12 +213,11 @@ async def stream_resume_chat(
 
         if needs_premium_quota(agent_config, user_id):
             premium_reservation = await reserve_premium(
-                agent_config=agent_config, user_id=user_id  # type: ignore[arg-type]
+                agent_config=agent_config,
+                user_id=user_id,  # type: ignore[arg-type]
             )
             if not premium_reservation.allowed:
-                ot.add_event(
-                    "quota.denied", {"quota.code": "PREMIUM_QUOTA_EXHAUSTED"}
-                )
+                ot.add_event("quota.denied", {"quota.code": "PREMIUM_QUOTA_EXHAUSTED"})
                 if requested_llm_config_id == 0:
                     try:
                         pinned_fb = await resolve_or_get_pinned_llm_config_id(
@@ -396,7 +391,9 @@ async def stream_resume_chat(
 
         # --- First SSE frames ---
 
-        for sse in iter_initial_frames(streaming_service, turn_id=stream_result.turn_id):
+        for sse in iter_initial_frames(
+            streaming_service, turn_id=stream_result.turn_id
+        ):
             yield sse
 
         # --- Assistant-shell persistence + id frame ---
@@ -517,7 +514,9 @@ async def stream_resume_chat(
             fallback_commit_search_space_id=search_space_id,
             fallback_commit_created_by_id=user_id,
             fallback_commit_filesystem_mode=(
-                filesystem_selection.mode if filesystem_selection else FilesystemMode.CLOUD
+                filesystem_selection.mode
+                if filesystem_selection
+                else FilesystemMode.CLOUD
             ),
             fallback_commit_thread_id=chat_id,
             runtime_context=runtime_context,
@@ -589,9 +588,7 @@ async def stream_resume_chat(
             end_turn(str(chat_id))
 
             if premium_reservation is not None and user_id:
-                await release_premium(
-                    reservation=premium_reservation, user_id=user_id
-                )
+                await release_premium(reservation=premium_reservation, user_id=user_id)
 
             await close_session_and_clear_ai_responding(session, chat_id)
 
@@ -609,13 +606,11 @@ async def stream_resume_chat(
         if not busy_error_raised:
             with contextlib.suppress(Exception):
                 end_turn(str(chat_id))
-                _perf_log.info(
-                    "[stream_resume] end_turn cleanup (chat_id=%s)", chat_id
-                )
+                _perf_log.info("[stream_resume] end_turn cleanup (chat_id=%s)", chat_id)
 
-        agent = llm = connector_service = None  # noqa: F841
-        stream_result = None  # noqa: F841
-        session = None  # noqa: F841
+        agent = llm = connector_service = None
+        stream_result = None
+        session = None
 
         run_gc_pass(log_prefix="stream_resume", chat_id=chat_id)
         close_chat_request_span(

surfsense_backend/app/tasks/chat/streaming/flows/resume_chat/resume_routing.py

@@ -47,9 +47,7 @@ async def build_resume_routing(
         slice_decisions_by_tool_call,
     )
 
-    parent_state = await agent.aget_state(
-        {"configurable": {"thread_id": str(chat_id)}}
-    )
+    parent_state = await agent.aget_state({"configurable": {"thread_id": str(chat_id)}})
     pending = collect_pending_tool_calls(parent_state)
     _perf_log.info(
         "[hitl_route] resume_entry chat_id=%s decisions=%d pending_subagents=%d",

surfsense_backend/app/tasks/chat/streaming/flows/shared/assistant_finalize.py

@@ -49,9 +49,7 @@ async def finalize_assistant_message(
     was never assigned.
     """
     if not (
-        stream_result
-        and stream_result.turn_id
-        and stream_result.assistant_message_id
+        stream_result and stream_result.turn_id and stream_result.assistant_message_id
     ):
         return
 

surfsense_backend/app/tasks/chat/streaming/flows/shared/finally_cleanup.py

@@ -39,9 +39,7 @@ async def close_session_and_clear_ai_responding(
             async with shielded_async_session() as fresh_session:
                 await clear_ai_responding(fresh_session, chat_id)
         except Exception:
-            logger.warning(
-                "Failed to clear AI responding state for thread %s", chat_id
-            )
+            logger.warning("Failed to clear AI responding state for thread %s", chat_id)
 
     with contextlib.suppress(Exception):
         session.expunge_all()

surfsense_backend/app/tasks/chat/streaming/flows/shared/premium_quota.py

@@ -41,9 +41,7 @@ class PremiumReservation:
     allowed: bool
 
 
-def needs_premium_quota(
-    agent_config: AgentConfig | None, user_id: str | None
-) -> bool:
+def needs_premium_quota(agent_config: AgentConfig | None, user_id: str | None) -> bool:
     return bool(agent_config is not None and user_id and agent_config.is_premium)
 
 
@@ -61,8 +59,10 @@ async def reserve_premium(
     request_id = _uuid.uuid4().hex[:16]
     litellm_params = agent_config.litellm_params or {}
     base_model = (
-        litellm_params.get("base_model") if isinstance(litellm_params, dict) else None
-    ) or agent_config.model_name or ""
+        (litellm_params.get("base_model") if isinstance(litellm_params, dict) else None)
+        or agent_config.model_name
+        or ""
+    )
     reserve_amount_micros = estimate_call_reserve_micros(
         base_model=base_model,
         quota_reserve_tokens=agent_config.quota_reserve_tokens,

surfsense_backend/app/tasks/chat/streaming/flows/shared/span.py

@@ -6,8 +6,7 @@ import contextlib
 import sys
 from typing import Any, Literal
 
-from app.observability import metrics as ot_metrics
-from app.observability import otel as ot
+from app.observability import metrics as ot_metrics, otel as ot
 
 
 def open_chat_request_span(

surfsense_backend/app/tasks/chat/streaming/flows/shared/terminal_error.py

@@ -15,8 +15,7 @@ from collections.abc import Iterator
 from typing import Any, Literal
 
 from app.agents.new_chat.errors import BusyError
-from app.observability import metrics as ot_metrics
-from app.observability import otel as ot
+from app.observability import metrics as ot_metrics, otel as ot
 from app.services.new_streaming_service import VercelStreamingService
 from app.tasks.chat.streaming.errors.classifier import classify_stream_exception
 from app.tasks.chat.streaming.errors.emitter import emit_stream_terminal_error