From 21aa010319912757ae5cfb92bac5fdb5171296f3 Mon Sep 17 00:00:00 2001 From: Matthew Zhou Date: Fri, 20 Dec 2024 17:33:53 -0800 Subject: [PATCH] fix: Fix limit for get_messages_by_ids (#2301) --- letta/services/message_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/letta/services/message_manager.py b/letta/services/message_manager.py index c369aa2c..48851f58 100644 --- a/letta/services/message_manager.py +++ b/letta/services/message_manager.py @@ -32,7 +32,7 @@ class MessageManager: def get_messages_by_ids(self, message_ids: List[str], actor: PydanticUser) -> List[PydanticMessage]: """Fetch messages by ID and return them in the requested order.""" with self.session_maker() as session: - results = MessageModel.list(db_session=session, id=message_ids, organization_id=actor.organization_id) + results = MessageModel.list(db_session=session, id=message_ids, organization_id=actor.organization_id, limit=len(message_ids)) if len(results) != len(message_ids): raise NoResultFound(