Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

chore: optimize memory messages fetch count limit (#6021)

Browse Source
This commit is contained in:
takatost 2024-07-06 03:25:38 +08:00 committed by GitHub
parent 79df8825c8
commit 23dc6edb99
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/memory/token_buffer_memory.py
View File

@ -36,9 +36,11 @@ class TokenBufferMemory:
).order_by(Message.created_at.desc()) ).order_by(Message.created_at.desc())
if message_limit and message_limit > 0: if message_limit and message_limit > 0:
messages = query.limit(message_limit).all() message_limit = message_limit if message_limit <= 500 else 500
else: else:
messages = query.all() message_limit = 500
messages = query.limit(message_limit).all()
messages = list(reversed(messages)) messages = list(reversed(messages))
message_file_parser = MessageFileParser( message_file_parser = MessageFileParser(