Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: Remove load agent in recall cursor (#2286)

Browse Source
This commit is contained in:
Matthew Zhou
2024-12-19 11:33:00 -08:00
committed by GitHub
parent bb1ffcac94
commit 3602aec591
2 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
letta/server/server.py
View File

@@ -991,11 +991,6 @@ class SyncServer(Server):
)
actor = self.user_manager.get_user_or_default(user_id=user_id)
logger.info(
f"[{temp_rand_uuid}] {datetime.datetime.utcnow()} - Loading agent object for agent_id={agent_id} (Elapsed: {(datetime.datetime.utcnow() - start_time).total_seconds()}s)"
)
letta_agent = self.load_agent(agent_id=agent_id, actor=actor)
logger.info(
f"[{temp_rand_uuid}] {datetime.datetime.utcnow()} - Resolving start_date and end_date for filtering messages (Elapsed: {(datetime.datetime.utcnow() - start_time).total_seconds()}s)"
)
@@ -1005,7 +1000,7 @@ class SyncServer(Server):
logger.info(
f"[{temp_rand_uuid}] {datetime.datetime.utcnow()} - Fetching messages for agent_id={agent_id}, start_date={start_date}, end_date={end_date}, limit={limit}, reverse={reverse} (Elapsed: {(datetime.datetime.utcnow() - start_time).total_seconds()}s)"
)
records = letta_agent.message_manager.list_messages_for_agent(
records = self.message_manager.list_messages_for_agent(
agent_id=agent_id,
actor=actor,
start_date=start_date,
@@ -1014,11 +1009,6 @@ class SyncServer(Server):
ascending=not reverse,
)
logger.info(
f"[{temp_rand_uuid}] {datetime.datetime.utcnow()} - Validating message types (Elapsed: {(datetime.datetime.utcnow() - start_time).total_seconds()}s)"
)
assert all(isinstance(m, Message) for m in records)
if not return_message_object:
logger.info(
f"[{temp_rand_uuid}] {datetime.datetime.utcnow()} - Converting messages to LettaMessage objects (Elapsed: {(datetime.datetime.utcnow() - start_time).total_seconds()}s)"