github/MemGPT
7
0
Fork 0
mirror of https://github.com/cpacker/MemGPT.git synced 2025-06-03 04:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add mcp support to sleeptime (#1964)

Browse Source
This commit is contained in:
cthomas 2025-04-30 21:08:26 -07:00 committed by GitHub
parent cb4caf103a
commit e3819cf066
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
letta/groups/helpers.py
View File

@ -76,6 +76,7 @@ def load_multi_agent(
agent_state=agent_state, agent_state=agent_state,
interface=interface, interface=interface,
user=actor, user=actor,
mcp_clients=mcp_clients,
group_id=group.id, group_id=group.id,
agent_ids=group.agent_ids, agent_ids=group.agent_ids,
description=group.description, description=group.description,

6
letta/groups/sleeptime_multi_agent.py
View File

@ -1,9 +1,10 @@
import asyncio import asyncio
import threading import threading
from datetime import datetime, timezone from datetime import datetime, timezone
from typing import List, Optional from typing import Dict, List, Optional
from letta.agent import Agent, AgentState from letta.agent import Agent, AgentState
from letta.functions.mcp_client.base_client import BaseMCPClient
from letta.groups.helpers import stringify_message from letta.groups.helpers import stringify_message
from letta.interface import AgentInterface from letta.interface import AgentInterface
from letta.orm import User from letta.orm import User
@ -26,6 +27,7 @@ class SleeptimeMultiAgent(Agent):
interface: AgentInterface, interface: AgentInterface,
agent_state: AgentState, agent_state: AgentState,
user: User, user: User,
mcp_clients: Optional[Dict[str, BaseMCPClient]] = None,
# custom # custom
group_id: str = "", group_id: str = "",
agent_ids: List[str] = [], agent_ids: List[str] = [],
@ -115,6 +117,7 @@ class SleeptimeMultiAgent(Agent):
agent_state=participant_agent_state, agent_state=participant_agent_state,
interface=StreamingServerInterface(), interface=StreamingServerInterface(),
user=self.user, user=self.user,
mcp_clients=self.mcp_clients,
) )
prior_messages = [] prior_messages = []
@ -212,6 +215,7 @@ class SleeptimeMultiAgent(Agent):
agent_state=self.agent_state, agent_state=self.agent_state,
interface=self.interface, interface=self.interface,
user=self.user, user=self.user,
mcp_clients=self.mcp_clients,
) )
# Perform main agent step # Perform main agent step
usage_stats = main_agent.step( usage_stats = main_agent.step(