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

Fix stream_new_chat interrupt resolution using LangGraph snapshot.interrupts.

Browse source
This commit is contained in:
CREDO23 2026-04-30 21:42:58 +02:00
parent ba131f5736
commit a6540b21c7
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
surfsense_backend/app/tasks/chat/stream_new_chat.py
View file

@ -1374,11 +1374,19 @@ async def _stream_agent_events(
result.agent_called_update_memory = called_update_memory result.agent_called_update_memory = called_update_memory
_log_file_contract("turn_outcome", result) _log_file_contract("turn_outcome", result)
is_interrupted = state.tasks and any(task.interrupts for task in state.tasks) snapshot_interrupts = getattr(state, "interrupts", ()) or ()
if is_interrupted: interrupt_value = None
if snapshot_interrupts:
interrupt_value = snapshot_interrupts[0].value
else:
for task in state.tasks or []:
if task.interrupts:
interrupt_value = task.interrupts[0].value
break
if interrupt_value is not None:
result.is_interrupted = True result.is_interrupted = True
result.interrupt_value = state.tasks[0].interrupts[0].value result.interrupt_value = interrupt_value
yield streaming_service.format_interrupt_request(result.interrupt_value) yield streaming_service.format_interrupt_request(interrupt_value)
async def stream_new_chat( async def stream_new_chat(