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

Skip registry subagents missing connectors when assembling the stack.

Browse source
This commit is contained in:
CREDO23 2026-05-02 00:43:56 +02:00
parent 06cef04e57
commit ffed829cf8
1 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
surfsense_backend/app/agents/multi_agent_with_deepagents/main_agent/graph/middleware/deepagent_stack.py
View file

@ -23,8 +23,10 @@ from langchain_core.language_models import BaseChatModel
from langchain_core.tools import BaseTool
from ...context_prune.prune_tool_names import safe_exclude_tools
from ...permissions.connector_deny_rules import synthesize_connector_deny_rules
from app.agents.multi_agent_with_deepagents.subagents.registry import build_subagents
from app.agents.multi_agent_with_deepagents.subagents import (
build_subagents,
get_subagents_to_exclude,
)
from app.agents.multi_agent_with_deepagents.subagents.shared.permissions import (
ToolsPermissions,
)
@ -130,6 +132,7 @@ def build_main_agent_deepagent_middleware(
model=llm,
extra_middleware=subagent_extra_middleware,
mcp_tools_by_agent=mcp_tools_by_agent or {},
exclude=get_subagents_to_exclude(available_connectors),
)
logging.info(
"Registry subagents: %s",
@ -263,12 +266,6 @@ def build_main_agent_deepagent_middleware(
origin="desktop_safety",
)
)
if permission_enabled:
synthesized = synthesize_connector_deny_rules(
available_connectors=available_connectors,
enabled_tool_names={t.name for t in tools},
)
rulesets.append(Ruleset(rules=synthesized, origin="connector_synthesized"))
permission_mw = PermissionMiddleware(rulesets=rulesets)
action_log_mw: ActionLogMiddleware | None = None