From 69848b4c71004650af45b313484e5234f20039a4 Mon Sep 17 00:00:00 2001 From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com> Date: Wed, 15 Jul 2026 10:24:45 +0530 Subject: [PATCH] feat(amazon): register amazon subagent --- .../app/agents/chat/multi_agent_chat/constants.py | 1 + .../app/agents/chat/multi_agent_chat/subagents/registry.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py b/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py index 6bbd808fb..333d4d274 100644 --- a/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py +++ b/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py @@ -29,6 +29,7 @@ CONNECTOR_TYPE_TO_CONNECTOR_AGENT_MAPS: dict[str, str] = { # connected app. Tokens are searchable-type strings (Composio Gmail/Calendar # map to the GOOGLE_* tokens in connector_searchable_types). SUBAGENT_TO_REQUIRED_CONNECTOR_MAP: dict[str, frozenset[str]] = { + "amazon": frozenset(), "deliverables": frozenset(), "knowledge_base": frozenset(), "web_crawler": frozenset(), diff --git a/surfsense_backend/app/agents/chat/multi_agent_chat/subagents/registry.py b/surfsense_backend/app/agents/chat/multi_agent_chat/subagents/registry.py index de37edfb9..7b8b8a6f1 100644 --- a/surfsense_backend/app/agents/chat/multi_agent_chat/subagents/registry.py +++ b/surfsense_backend/app/agents/chat/multi_agent_chat/subagents/registry.py @@ -12,6 +12,9 @@ from langchain_core.tools import BaseTool from app.agents.chat.multi_agent_chat.constants import ( SUBAGENT_TO_REQUIRED_CONNECTOR_MAP, ) +from app.agents.chat.multi_agent_chat.subagents.builtins.amazon.agent import ( + build_subagent as build_amazon_subagent, +) from app.agents.chat.multi_agent_chat.subagents.builtins.deliverables.agent import ( build_subagent as build_deliverables_subagent, ) @@ -80,6 +83,7 @@ class SubagentBuilder(Protocol): SUBAGENT_BUILDERS_BY_NAME: dict[str, SubagentBuilder] = { + "amazon": build_amazon_subagent, "deliverables": build_deliverables_subagent, "dropbox": build_dropbox_subagent, "google_drive": build_google_drive_subagent,