From 2889d2f2d5c3f7d7bc00b4c74b3246fee9f97bfc Mon Sep 17 00:00:00 2001
From: cthomas <caren@letta.com>
Date: Wed, 30 Apr 2025 21:08:26 -0700
Subject: [PATCH] feat: add mcp support to sleeptime (#1964)

---
 letta/groups/helpers.py               | 1 +
 letta/groups/sleeptime_multi_agent.py | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/letta/groups/helpers.py b/letta/groups/helpers.py
index 039230df..f66269c7 100644
--- a/letta/groups/helpers.py
+++ b/letta/groups/helpers.py
@@ -76,6 +76,7 @@ def load_multi_agent(
                 agent_state=agent_state,
                 interface=interface,
                 user=actor,
+                mcp_clients=mcp_clients,
                 group_id=group.id,
                 agent_ids=group.agent_ids,
                 description=group.description,
diff --git a/letta/groups/sleeptime_multi_agent.py b/letta/groups/sleeptime_multi_agent.py
index 6349b57b..87f49b10 100644
--- a/letta/groups/sleeptime_multi_agent.py
+++ b/letta/groups/sleeptime_multi_agent.py
@@ -1,9 +1,10 @@
 import asyncio
 import threading
 from datetime import datetime, timezone
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from letta.agent import Agent, AgentState
+from letta.functions.mcp_client.base_client import BaseMCPClient
 from letta.groups.helpers import stringify_message
 from letta.interface import AgentInterface
 from letta.orm import User
@@ -26,6 +27,7 @@ class SleeptimeMultiAgent(Agent):
         interface: AgentInterface,
         agent_state: AgentState,
         user: User,
+        mcp_clients: Optional[Dict[str, BaseMCPClient]] = None,
         # custom
         group_id: str = "",
         agent_ids: List[str] = [],
@@ -115,6 +117,7 @@ class SleeptimeMultiAgent(Agent):
                 agent_state=participant_agent_state,
                 interface=StreamingServerInterface(),
                 user=self.user,
+                mcp_clients=self.mcp_clients,
             )
 
             prior_messages = []
@@ -212,6 +215,7 @@ class SleeptimeMultiAgent(Agent):
                 agent_state=self.agent_state,
                 interface=self.interface,
                 user=self.user,
+                mcp_clients=self.mcp_clients,
             )
             # Perform main agent step
             usage_stats = main_agent.step(