From eefdd1d7f00953b0a4f759884c7c936663fb3198 Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Sat, 2 May 2026 00:43:46 +0200 Subject: [PATCH] Add connector routing entries and per-subagent connector token map. --- .../multi_agent_with_deepagents/constants.py | 27 ++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/surfsense_backend/app/agents/multi_agent_with_deepagents/constants.py b/surfsense_backend/app/agents/multi_agent_with_deepagents/constants.py index 775027764..972677502 100644 --- a/surfsense_backend/app/agents/multi_agent_with_deepagents/constants.py +++ b/surfsense_backend/app/agents/multi_agent_with_deepagents/constants.py @@ -1,4 +1,4 @@ -"""Map connector type strings to the agent route key used for tools and MCP slices.""" +"""Connector-type to subagent name; subagent name to availability tokens for build_subagents.""" from __future__ import annotations @@ -15,4 +15,29 @@ CONNECTOR_TYPE_TO_CONNECTOR_AGENT_MAPS: dict[str, str] = { "CLICKUP_CONNECTOR": "clickup", "SLACK_CONNECTOR": "slack", "AIRTABLE_CONNECTOR": "airtable", + "NOTION_CONNECTOR": "notion", + "CONFLUENCE_CONNECTOR": "confluence", + "GOOGLE_DRIVE_CONNECTOR": "google_drive", + "COMPOSIO_GOOGLE_DRIVE_CONNECTOR": "google_drive", + "DROPBOX_CONNECTOR": "dropbox", + "ONEDRIVE_CONNECTOR": "onedrive", +} + +SUBAGENT_TO_REQUIRED_CONNECTOR_MAP: dict[str, frozenset[str]] = { + "deliverables": frozenset(), + "airtable": frozenset({"AIRTABLE_CONNECTOR"}), + "calendar": frozenset({"GOOGLE_CALENDAR_CONNECTOR"}), + "clickup": frozenset({"CLICKUP_CONNECTOR"}), + "confluence": frozenset({"CONFLUENCE_CONNECTOR"}), + "discord": frozenset({"DISCORD_CONNECTOR"}), + "dropbox": frozenset({"DROPBOX_FILE"}), + "gmail": frozenset({"GOOGLE_GMAIL_CONNECTOR"}), + "google_drive": frozenset({"GOOGLE_DRIVE_FILE"}), + "jira": frozenset({"JIRA_CONNECTOR"}), + "linear": frozenset({"LINEAR_CONNECTOR"}), + "luma": frozenset({"LUMA_CONNECTOR"}), + "notion": frozenset({"NOTION_CONNECTOR"}), + "onedrive": frozenset({"ONEDRIVE_FILE"}), + "slack": frozenset({"SLACK_CONNECTOR"}), + "teams": frozenset({"TEAMS_CONNECTOR"}), }