chore: linting

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