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

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 langchain_core.tools import BaseTool
from ...context_prune.prune_tool_names import safe_exclude_tools 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 import (
from app.agents.multi_agent_with_deepagents.subagents.registry import build_subagents build_subagents,
get_subagents_to_exclude,
)
from app.agents.multi_agent_with_deepagents.subagents.shared.permissions import ( from app.agents.multi_agent_with_deepagents.subagents.shared.permissions import (
ToolsPermissions, ToolsPermissions,
) )
@ -130,6 +132,7 @@ def build_main_agent_deepagent_middleware(
model=llm, model=llm,
extra_middleware=subagent_extra_middleware, extra_middleware=subagent_extra_middleware,
mcp_tools_by_agent=mcp_tools_by_agent or {}, mcp_tools_by_agent=mcp_tools_by_agent or {},
exclude=get_subagents_to_exclude(available_connectors),
) )
logging.info( logging.info(
"Registry subagents: %s", "Registry subagents: %s",
@ -263,12 +266,6 @@ def build_main_agent_deepagent_middleware(
origin="desktop_safety", 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) permission_mw = PermissionMiddleware(rulesets=rulesets)
action_log_mw: ActionLogMiddleware | None = None action_log_mw: ActionLogMiddleware | None = None