Просмотр исходного кода

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

tags/0.6.13
takatost 1 год назад
Родитель
Сommit
23dc6edb99
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 4 добавлений и 2 удалений
  1. 4
    2
      api/core/memory/token_buffer_memory.py

+ 4
- 2
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(

Загрузка…
Отмена
Сохранить