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

add interrupt detection, SSE event, and resume schemas

Browse source
This commit is contained in:
CREDO23 2026-02-10 15:35:42 +02:00
parent c9542c8603
commit 9751918e41
3 changed files with 34 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1175,6 +1175,17 @@ async def stream_new_chat(
if completion_event:
yield completion_event
# Check if the graph was interrupted (human-in-the-loop)
state = await agent.aget_state(config)
is_interrupted = state.tasks and any(task.interrupts for task in state.tasks)
if is_interrupted:
interrupt_value = state.tasks[0].interrupts[0].value
yield streaming_service.format_interrupt_request(interrupt_value)
yield streaming_service.format_finish_step()
yield streaming_service.format_finish()
yield streaming_service.format_done()
return
# Generate LLM title for new chats after first response
# Check if this is the first assistant response by counting existing assistant messages
from sqlalchemy import func