apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-12 01:02:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Forward span metadata from report_progress thinking updates.

Browse source
This commit is contained in:
CREDO23 2026-05-08 22:47:50 +02:00
parent 1dcb08e925
commit f1d80ffe5d
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
surfsense_backend/app/tasks/chat/streaming/handlers/custom_event_dispatch.py
View file

@ -33,6 +33,7 @@ def iter_custom_event_frames(
last_active_step_items=state.last_active_step_items, last_active_step_items=state.last_active_step_items,
streaming_service=streaming_service, streaming_service=streaming_service,
content_builder=content_builder, content_builder=content_builder,
thinking_metadata=state.span_metadata_if_active(),
) )
if frame: if frame:
yield frame yield frame

2
surfsense_backend/app/tasks/chat/streaming/handlers/custom_events.py
View file

@ -15,6 +15,7 @@ def handle_report_progress(
last_active_step_items: list[str], last_active_step_items: list[str],
streaming_service: Any, streaming_service: Any,
content_builder: Any | None, content_builder: Any | None,
thinking_metadata: dict[str, Any] | None = None,
) -> tuple[str | None, list[str]]: ) -> tuple[str | None, list[str]]:
"""Update report step items; may emit one thinking SSE frame. """Update report step items; may emit one thinking SSE frame.
@ -50,6 +51,7 @@ def handle_report_progress(
title=last_active_step_title, title=last_active_step_title,
status="in_progress", status="in_progress",
items=new_items, items=new_items,
metadata=thinking_metadata,
) )
return frame, new_items return frame, new_items