style: ruff autofix (auto)

Author: autofix-ci[bot]

src/agent.py

@@ -112,6 +112,7 @@ def _stream_error_chunk(message: str, response_id: str | None = None) -> bytes:
         payload["response_id"] = response_id
     return (json.dumps(payload) + "\n").encode("utf-8")
 
+
 async def get_user_conversations(user_id: str):
     """Get conversation metadata for a user from persistent storage."""
     return await conversation_persistence.get_user_conversations(user_id)
@@ -163,9 +164,7 @@ async def store_conversation_thread(user_id: str, response_id: str, conversation
         content = first_user_msg.get("content", "")
         title = content[:50] + "..." if len(content) > 50 else content
 
-    user_assistant_messages = [
-        msg for msg in messages if msg.get("role") in ["user", "assistant"]
-    ]
+    user_assistant_messages = [msg for msg in messages if msg.get("role") in ["user", "assistant"]]
     metadata_only = {
         "response_id": response_id,
         "title": title,

src/api/chat.py

@@ -1,17 +1,17 @@
-from typing import Optional, Any, Dict
+from typing import Any, Dict, Optional
 
 from fastapi import Depends, HTTPException
-from pydantic import BaseModel
 from fastapi.responses import JSONResponse, StreamingResponse
-from utils.logging_config import get_logger
+from pydantic import BaseModel
 
 from dependencies import (
     get_chat_service,
-    get_session_manager,
     get_current_user,
+    get_session_manager,
     require_permission,
 )
 from session_manager import User
+from utils.logging_config import get_logger
 
 logger = get_logger(__name__)
 
@@ -20,7 +20,7 @@ def _openrag_user_id(user: User) -> str:
     return getattr(user, "db_user_id", None) or user.user_id
 
 
-async def _assert_owns(session_id: Optional[str], user_id: str) -> None:
+async def _assert_owns(session_id: str | None, user_id: str) -> None:
     """Raise 403 if `session_id` is set but not owned by `user_id`.
 
     No-op when `session_id` is None (new conversation, nothing to check).
@@ -30,6 +30,7 @@ async def _assert_owns(session_id: Optional[str], user_id: str) -> None:
     if not session_id:
         return
     from services.session_ownership_service import session_ownership_service
+
     owner = await session_ownership_service.get_session_owner(session_id)
     if owner is None:
         raise HTTPException(status_code=404, detail={"error": "session_not_found"})
@@ -39,15 +40,15 @@ async def _assert_owns(session_id: Optional[str], user_id: str) -> None:
 
 class ChatBody(BaseModel):
     prompt: str
-    previous_response_id: Optional[str] = None
+    previous_response_id: str | None = None
     # OpenRAG sidebar/thread id. Distinct from previous_response_id so a retry
     # after an error can stay in the same chat while starting a fresh Langflow session.
-    conversation_id: Optional[str] = None
+    conversation_id: str | None = None
     stream: bool = False
-    filters: Optional[Dict[str, Any]] = None
+    filters: dict[str, Any] | None = None
     limit: int = 10
     scoreThreshold: float = 0
-    filter_id: Optional[str] = None
+    filter_id: str | None = None
 
 
 async def chat_endpoint(
@@ -67,9 +68,11 @@ async def chat_endpoint(
 
     if body.filters:
         from auth_context import set_search_filters
+
         set_search_filters(body.filters)
 
-    from auth_context import set_search_limit, set_score_threshold
+    from auth_context import set_score_threshold, set_search_limit
+
     set_search_limit(body.limit)
     set_score_threshold(body.scoreThreshold)
 
@@ -123,9 +126,11 @@ async def langflow_endpoint(
 
     if body.filters:
         from auth_context import set_search_filters
+
         set_search_filters(body.filters)
 
-    from auth_context import set_search_limit, set_score_threshold
+    from auth_context import set_score_threshold, set_search_limit
+
     set_search_limit(body.limit)
     set_score_threshold(body.scoreThreshold)
 
@@ -171,9 +176,7 @@ async def langflow_endpoint(
 
     except Exception as e:
         logger.exception("[CHAT] Langflow request failed")
-        return JSONResponse(
-            {"error": f"Langflow request failed: {str(e)}"}, status_code=500
-        )
+        return JSONResponse({"error": f"Langflow request failed: {str(e)}"}, status_code=500)
 
 
 async def chat_history_endpoint(
@@ -186,9 +189,7 @@ async def chat_history_endpoint(
         return JSONResponse(history)
     except Exception as e:
         logger.exception("[CHAT] Failed to get chat history")
-        return JSONResponse(
-            {"error": f"Failed to get chat history: {str(e)}"}, status_code=500
-        )
+        return JSONResponse({"error": f"Failed to get chat history: {str(e)}"}, status_code=500)
 
 
 async def langflow_history_endpoint(
@@ -201,9 +202,7 @@ async def langflow_history_endpoint(
         return JSONResponse(history)
     except Exception as e:
         logger.exception("[CHAT] Failed to get langflow history")
-        return JSONResponse(
-            {"error": f"Failed to get langflow history: {str(e)}"}, status_code=500
-        )
+        return JSONResponse({"error": f"Failed to get langflow history: {str(e)}"}, status_code=500)
 
 
 async def delete_session_endpoint(
@@ -221,11 +220,8 @@ async def delete_session_endpoint(
             return JSONResponse({"message": "Session deleted successfully"})
         else:
             return JSONResponse(
-                {"error": result.get("error", "Failed to delete session")},
-                status_code=500
+                {"error": result.get("error", "Failed to delete session")}, status_code=500
             )
     except Exception as e:
         logger.error(f"Error deleting session: {e}")
-        return JSONResponse(
-            {"error": f"Failed to delete session: {str(e)}"}, status_code=500
-        )
+        return JSONResponse({"error": f"Failed to delete session: {str(e)}"}, status_code=500)

src/services/chat_service.py

@@ -843,9 +843,7 @@ async def get_langflow_history(self, user_id: str):
                 ]
 
             if response_id in listed_ids:
-                existing = next(
-                    c for c in all_conversations if c["response_id"] == response_id
-                )
+                existing = next(c for c in all_conversations if c["response_id"] == response_id)
                 if len(messages) > len(existing.get("messages") or []):
                     existing["messages"] = messages
                     existing["total_messages"] = len(messages)
@@ -864,9 +862,7 @@ async def get_langflow_history(self, user_id: str):
                 else "New Chat"
             )
             created_at = metadata.get("created_at") or conversation_state.get("created_at")
-            last_activity = metadata.get("last_activity") or conversation_state.get(
-                "last_activity"
-            )
+            last_activity = metadata.get("last_activity") or conversation_state.get("last_activity")
             all_conversations.append(
                 {
                     "response_id": response_id,

tests/unit/test_langflow_stream_provider_errors.py

@@ -206,9 +206,7 @@ async def raise_provider_error(*_args, **_kwargs) -> AsyncIterator[bytes]:
 
 
 @pytest.mark.asyncio
-async def test_langflow_follow_up_after_error_reuses_store_id(
-    monkeypatch, store_in_memory
-):
+async def test_langflow_follow_up_after_error_reuses_store_id(monkeypatch, store_in_memory):
     """Retrying in the same chat must not create a second sidebar conversation."""
     provider_message = "Rate limit exceeded for watsonx.ai."
     stream_calls: list[dict] = []
@@ -263,9 +261,7 @@ async def raise_provider_error(*_args, **kwargs) -> AsyncIterator[bytes]:
 
 
 @pytest.mark.asyncio
-async def test_langflow_follow_up_ignores_broken_previous_response_id(
-    monkeypatch, store_in_memory
-):
+async def test_langflow_follow_up_ignores_broken_previous_response_id(monkeypatch, store_in_memory):
     """Even if the client still sends a failed Langflow session id, do not chain it."""
     provider_message = "Invalid API key for Anthropic. Check your credentials."
     stream_calls: list[dict] = []