apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-24 23:41:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(walmart): register subagent in roster and connector map

Browse source
This commit is contained in:
CREDO23 2026-07-19 08:45:27 +02:00
parent f6a8a2a566
commit cabc5b5c86
3 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
surfsense_backend/app/agents/chat/multi_agent_chat/constants.py
View file

@ -39,6 +39,7 @@ SUBAGENT_TO_REQUIRED_CONNECTOR_MAP: dict[str, frozenset[str]] = {
"reddit": frozenset(), "reddit": frozenset(),
"instagram": frozenset(), "instagram": frozenset(),
"tiktok": frozenset(), "tiktok": frozenset(),
"walmart": frozenset(),
"mcp_discovery": frozenset( "mcp_discovery": frozenset(
{ {
"SLACK_CONNECTOR", "SLACK_CONNECTOR",

4
surfsense_backend/app/agents/chat/multi_agent_chat/subagents/registry.py
View file

@ -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 ( from app.agents.chat.multi_agent_chat.subagents.builtins.tiktok.agent import (
build_subagent as build_tiktok_subagent, 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 ( from app.agents.chat.multi_agent_chat.subagents.builtins.web_crawler.agent import (
build_subagent as build_web_crawler_subagent, build_subagent as build_web_crawler_subagent,
) )
@ -96,6 +99,7 @@ SUBAGENT_BUILDERS_BY_NAME: dict[str, SubagentBuilder] = {
"onedrive": build_onedrive_subagent, "onedrive": build_onedrive_subagent,
"reddit": build_reddit_subagent, "reddit": build_reddit_subagent,
"tiktok": build_tiktok_subagent, "tiktok": build_tiktok_subagent,
"walmart": build_walmart_subagent,
"web_crawler": build_web_crawler_subagent, "web_crawler": build_web_crawler_subagent,
"youtube": build_youtube_subagent, "youtube": build_youtube_subagent,
} }

1
surfsense_backend/tests/unit/agents/multi_agent_chat/test_subagent_composition.py
View file

@ -40,6 +40,7 @@ _EXPECTED_SUBAGENTS = frozenset(
"onedrive", "onedrive",
"reddit", "reddit",
"tiktok", "tiktok",
"walmart",
"web_crawler", "web_crawler",
"youtube", "youtube",
} }