apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-19 18:45:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

chat/stream_resume: route a flat decisions list per paused subagent

Browse source
This commit is contained in:
CREDO23 2026-05-13 19:58:13 +02:00
parent 1001f56206
commit 03cf1466d3
1 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2863,14 +2863,35 @@ async def stream_resume_chat(
from langgraph.types import Command from langgraph.types import Command
from app.agents.multi_agent_chat.middleware.main_agent.checkpointed_subagent_middleware.resume_routing import (
collect_pending_tool_calls,
slice_decisions_by_tool_call,
)
# Each pending interrupt is stamped with its originating ``tool_call_id``
# (see ``checkpointed_subagent_middleware.propagation``) so we can route
# a flat ``decisions`` list back to the right paused subagent.
parent_state = await agent.aget_state(
{"configurable": {"thread_id": str(chat_id)}}
)
pending = collect_pending_tool_calls(parent_state)
_perf_log.info(
"[hitl_route] resume_entry chat_id=%s decisions=%d pending_subagents=%d",
chat_id,
len(decisions),
len(pending),
)
routed_resume_value = slice_decisions_by_tool_call(decisions, pending)
config = { config = {
"configurable": { "configurable": {
"thread_id": str(chat_id), "thread_id": str(chat_id),
"request_id": request_id or "unknown", "request_id": request_id or "unknown",
"turn_id": stream_result.turn_id, "turn_id": stream_result.turn_id,
# Side-channel consumed by ``SurfSenseCheckpointedSubAgentMiddleware`` # Per-``tool_call_id`` resume slices read by
# to forward the resume into a subagent's pending ``interrupt()``. # ``SurfSenseCheckpointedSubAgentMiddleware``. Parallel
"surfsense_resume_value": {"decisions": decisions}, # siblings each pop their own entry, so they never race.
"surfsense_resume_value": routed_resume_value,
}, },
# See ``stream_new_chat`` above for rationale: effectively # See ``stream_new_chat`` above for rationale: effectively
# uncapped to mirror the agent default and OpenCode's # uncapped to mirror the agent default and OpenCode's