apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-06 20:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix(e2e): delegate connector work via task in fake LLM

Browse source
This commit is contained in:
CREDO23 2026-06-05 18:49:57 +02:00
parent bfadde93b7
commit 52ff304d64
1 changed files with 43 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

43
surfsense_backend/tests/e2e/fakes/chat_llm.py
View file

@ -553,6 +553,49 @@ class FakeChatLLM(BaseChatModel):
latest_tool_name = getattr(latest_tool, "name", None)
latest_tool_text = _content_to_text(latest_tool.content) if latest_tool else ""
# Marker unique to a connector subagent's prompt: the main agent must
# delegate via ``task``; only the subagent has connector tools registered.
in_connector_subagent = (
"specialist for the user's connected" in _messages_to_text(messages)
)
# Main agent: delegate live-tool connector work to its subagent (which
# then runs the real tools below). Indexed connectors are absent here.
if not in_connector_subagent and latest_tool is None:
connector_delegations = (
("gmail", ("gmail", "email", "message", GMAIL_CANARY_SUBJECT)),
("calendar", ("calendar", "event", "meeting", CALENDAR_CANARY_SUMMARY)),
(
"jira",
(
"jira",
"atlassian",
JIRA_CANARY_SUMMARY,
JIRA_CANARY_KEY,
"surfsense-e2e.atlassian.net",
"fake-jira-cloud-001",
),
),
("linear", ("linear", "issue", LINEAR_CANARY_TITLE)),
("slack", ("slack", SLACK_CANARY_TOKEN)),
("clickup", ("clickup", CLICKUP_CANARY_TITLE)),
)
for subagent_type, needles in connector_delegations:
if _contains_any(latest_human, needles):
return AIMessage(
content="",
tool_calls=[
{
"name": "task",
"args": {
"subagent_type": subagent_type,
"description": latest_human,
},
"id": f"call_e2e_task_{subagent_type}",
}
],
)
if (
latest_tool_name == "search_gmail"
and GMAIL_CANARY_MESSAGE_ID in latest_tool_text