opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 19:27:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct indentation in TokenBufferMemory get_history_prompt_messages method

Browse Source
This commit is contained in:
-LAN- 2025-09-07 12:48:50 +08:00
parent 7aef0b54e5
commit cc1d437dc1
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -167,11 +167,11 @@ class TokenBufferMemory:
else: else:
prompt_messages.append(AssistantPromptMessage(content=message.answer)) prompt_messages.append(AssistantPromptMessage(content=message.answer))
if not prompt_messages: if not prompt_messages:
return [] return []
# prune the chat message if it exceeds the max token limit # prune the chat message if it exceeds the max token limit
curr_message_tokens = self.model_instance.get_llm_num_tokens(prompt_messages) curr_message_tokens = self.model_instance.get_llm_num_tokens(prompt_messages)
if curr_message_tokens > max_token_limit: if curr_message_tokens > max_token_limit:
while curr_message_tokens > max_token_limit and len(prompt_messages) > 1: while curr_message_tokens > max_token_limit and len(prompt_messages) > 1: