Increase security and reformat

core/messaging/history.py

@@ -1,12 +1,13 @@
-from asgiref.sync import sync_to_async
-from django.conf import settings
 import time
 import uuid
 
+from asgiref.sync import sync_to_async
+from django.conf import settings
+
 from core.events.ledger import append_event
 from core.messaging.utils import messages_to_string
-from core.observability.tracing import ensure_trace_id
 from core.models import ChatSession, Message, QueuedMessage
+from core.observability.tracing import ensure_trace_id
 from core.util import logs
 
 log = logs.get_logger("history")
@@ -272,7 +273,9 @@ async def store_own_message(
                 trace_id=ensure_trace_id(trace_id, message_meta or {}),
             )
         except Exception as exc:
-            log.warning("Event ledger append failed for own message=%s: %s", msg.id, exc)
+            log.warning(
+                "Event ledger append failed for own message=%s: %s", msg.id, exc
+            )
 
     return msg
 

core/messaging/reply_sync.py

@@ -335,8 +335,12 @@ def extract_reply_ref(service: str, raw_payload: dict[str, Any]) -> dict[str, st
     svc = _clean(service).lower()
     payload = _as_dict(raw_payload)
     if svc == "xmpp":
-        reply_id = _clean(payload.get("reply_source_message_id") or payload.get("reply_id"))
-        reply_chat = _clean(payload.get("reply_source_chat_id") or payload.get("reply_chat_id"))
+        reply_id = _clean(
+            payload.get("reply_source_message_id") or payload.get("reply_id")
+        )
+        reply_chat = _clean(
+            payload.get("reply_source_chat_id") or payload.get("reply_chat_id")
+        )
         if reply_id:
             return {
                 "reply_source_message_id": reply_id,
@@ -363,7 +367,9 @@ def extract_origin_tag(raw_payload: dict[str, Any] | None) -> str:
     return _find_origin_tag(_as_dict(raw_payload))
 
 
-async def resolve_reply_target(user, session, reply_ref: dict[str, str]) -> Message | None:
+async def resolve_reply_target(
+    user, session, reply_ref: dict[str, str]
+) -> Message | None:
     if not reply_ref or session is None:
         return None
     reply_source_message_id = _clean(reply_ref.get("reply_source_message_id"))