From 23dc6edb99db3cc4ea3ffa1a2a3c72f77b0b80cb Mon Sep 17 00:00:00 2001 From: takatost Date: Sat, 6 Jul 2024 03:25:38 +0800 Subject: [PATCH] chore: optimize memory messages fetch count limit (#6021) --- api/core/memory/token_buffer_memory.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/core/memory/token_buffer_memory.py b/api/core/memory/token_buffer_memory.py index 4f4f5045f5..f38175020c 100644 --- a/api/core/memory/token_buffer_memory.py +++ b/api/core/memory/token_buffer_memory.py @@ -36,9 +36,11 @@ class TokenBufferMemory: ).order_by(Message.created_at.desc()) if message_limit and message_limit > 0: - messages = query.limit(message_limit).all() + message_limit = message_limit if message_limit <= 500 else 500 else: - messages = query.all() + message_limit = 500 + + messages = query.limit(message_limit).all() messages = list(reversed(messages)) message_file_parser = MessageFileParser(