apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-26 23:51:14 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: declare citation channel for subagents

Browse source
This commit is contained in:
CREDO23 2026-07-22 19:34:56 +02:00
parent c1ba4215fd
commit 95b4afddeb
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
surfsense_backend/app/agents/chat/multi_agent_chat/subagents/middleware_stack.py
View file

@ -15,6 +15,9 @@ from __future__ import annotations
from typing import Any from typing import Any
from app.agents.chat.multi_agent_chat.shared.feature_flags import AgentFeatureFlags from app.agents.chat.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
from app.agents.chat.multi_agent_chat.shared.middleware.citation_state import (
build_citation_state_mw,
)
from app.agents.chat.multi_agent_chat.shared.middleware.resilience import ( from app.agents.chat.multi_agent_chat.shared.middleware.resilience import (
ResilienceMiddlewares, ResilienceMiddlewares,
) )
@ -45,6 +48,8 @@ def build_subagent_middleware_stack(
return { return {
"todos": build_todos_mw(), "todos": build_todos_mw(),
"permission": permission, "permission": permission,
# Declares the citation_registry channel so run [n]s merge up from tools.
"citation": build_citation_state_mw(),
"retry": resilience.retry, "retry": resilience.retry,
"fallback": resilience.fallback, "fallback": resilience.fallback,
"model_call_limit": resilience.model_call_limit, "model_call_limit": resilience.model_call_limit,