From cabc5b5c86b0b602bae2a52ef99a1cbfd93ce624 Mon Sep 17 00:00:00 2001
From: CREDO23 <bakerathierry@gmail.com>
Date: Sun, 19 Jul 2026 08:45:27 +0200
Subject: [PATCH] feat(walmart): register subagent in roster and connector map

---
 .../app/agents/chat/multi_agent_chat/constants.py             | 1 +
 .../app/agents/chat/multi_agent_chat/subagents/registry.py    | 4 ++++
 .../unit/agents/multi_agent_chat/test_subagent_composition.py | 1 +
 3 files changed, 6 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 333d4d274..2c44dbefe 100644
--- a/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py
+++ b/surfsense_backend/app/agents/chat/multi_agent_chat/constants.py
@@ -39,6 +39,7 @@ SUBAGENT_TO_REQUIRED_CONNECTOR_MAP: dict[str, frozenset[str]] = {
     "reddit": frozenset(),
     "instagram": frozenset(),
     "tiktok": frozenset(),
+    "walmart": frozenset(),
     "mcp_discovery": frozenset(
         {
             "SLACK_CONNECTOR",
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 7b8b8a6f1..48de3e245 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
@@ -42,6 +42,9 @@ from app.agents.chat.multi_agent_chat.subagents.builtins.reddit.agent import (
 from app.agents.chat.multi_agent_chat.subagents.builtins.tiktok.agent import (
     build_subagent as build_tiktok_subagent,
 )
+from app.agents.chat.multi_agent_chat.subagents.builtins.walmart.agent import (
+    build_subagent as build_walmart_subagent,
+)
 from app.agents.chat.multi_agent_chat.subagents.builtins.web_crawler.agent import (
     build_subagent as build_web_crawler_subagent,
 )
@@ -96,6 +99,7 @@ SUBAGENT_BUILDERS_BY_NAME: dict[str, SubagentBuilder] = {
     "onedrive": build_onedrive_subagent,
     "reddit": build_reddit_subagent,
     "tiktok": build_tiktok_subagent,
+    "walmart": build_walmart_subagent,
     "web_crawler": build_web_crawler_subagent,
     "youtube": build_youtube_subagent,
 }
diff --git a/surfsense_backend/tests/unit/agents/multi_agent_chat/test_subagent_composition.py b/surfsense_backend/tests/unit/agents/multi_agent_chat/test_subagent_composition.py
index f578f3457..bc12f1308 100644
--- a/surfsense_backend/tests/unit/agents/multi_agent_chat/test_subagent_composition.py
+++ b/surfsense_backend/tests/unit/agents/multi_agent_chat/test_subagent_composition.py
@@ -40,6 +40,7 @@ _EXPECTED_SUBAGENTS = frozenset(
         "onedrive",
         "reddit",
         "tiktok",
+        "walmart",
         "web_crawler",
         "youtube",
     }