apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-28 18:36:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

chore: linting

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-04-22 01:05:31 -07:00
parent c85c3a438e
commit 80d3f624d4
9 changed files with 310 additions and 127 deletions
Download patch file Download diff file Expand all files Collapse all files

10
surfsense_backend/app/agents/new_chat/middleware/safe_summarization.py
View file

@ -46,7 +46,7 @@ if TYPE_CHECKING:
logger = logging.getLogger(__name__)
def _sanitize_message_content(msg: "AnyMessage") -> "AnyMessage":
def _sanitize_message_content(msg: AnyMessage) -> AnyMessage:
"""Return ``msg`` with ``content`` coerced to a non-``None`` value.
``get_buffer_string`` reads ``m.text`` which iterates ``self.content``;
@ -90,16 +90,14 @@ class SafeSummarizationMiddleware(SummarizationMiddleware):
implementations from upstream.
"""
def _filter_summary_messages(
self, messages: "list[AnyMessage]"
) -> "list[AnyMessage]":
def _filter_summary_messages(self, messages: list[AnyMessage]) -> list[AnyMessage]:
filtered = super()._filter_summary_messages(messages)
return [_sanitize_message_content(m) for m in filtered]
def create_safe_summarization_middleware(
model: "BaseChatModel",
backend: "BACKEND_TYPES",
model: BaseChatModel,
backend: BACKEND_TYPES,
) -> SafeSummarizationMiddleware:
"""Drop-in replacement for ``create_summarization_middleware``.