diff --git a/letta/server/server.py b/letta/server/server.py index 0adcb459..c885b965 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -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)" diff --git a/letta/services/message_manager.py b/letta/services/message_manager.py index b4151944..6c317f64 100644 --- a/letta/services/message_manager.py +++ b/letta/services/message_manager.py @@ -147,6 +147,7 @@ class MessageManager: limit=limit, filters=message_filters, query_text=query_text, + ascending=ascending, ) @enforce_types