AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 00:39:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: generate summary error when tokens=4097 (#488)

Browse Source
This commit is contained in:
John Wang 2023-06-29 12:54:50 +08:00 committed by GitHub
parent c6ab7eebd9
commit b9b0866a46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/generator/llm_generator.py
View File

@ -45,7 +45,7 @@ class LLMGenerator:
prompt = CONVERSATION_SUMMARY_PROMPT prompt = CONVERSATION_SUMMARY_PROMPT
prompt_with_empty_context = prompt.format(context='') prompt_with_empty_context = prompt.format(context='')
prompt_tokens = TokenCalculator.get_num_tokens(model, prompt_with_empty_context) prompt_tokens = TokenCalculator.get_num_tokens(model, prompt_with_empty_context)
rest_tokens = llm_constant.max_context_token_length[model] - prompt_tokens - max_tokens rest_tokens = llm_constant.max_context_token_length[model] - prompt_tokens - max_tokens - 1
context = '' context = ''
for message in messages: for message in messages:
@ -56,6 +56,9 @@ class LLMGenerator:
if rest_tokens - TokenCalculator.get_num_tokens(model, context + message_qa_text) > 0: if rest_tokens - TokenCalculator.get_num_tokens(model, context + message_qa_text) > 0:
context += message_qa_text context += message_qa_text
if not context:
return '[message too long, no summary]'
prompt = prompt.format(context=context) prompt = prompt.format(context=context)
llm: StreamableOpenAI = LLMBuilder.to_llm( llm: StreamableOpenAI = LLMBuilder.to_llm(

2
api/tasks/generate_conversation_summary_task.py
View File

@ -28,7 +28,7 @@ def generate_conversation_summary_task(conversation_id: str):
try: try:
# get conversation messages count # get conversation messages count
history_message_count = conversation.message_count history_message_count = conversation.message_count
if history_message_count >= 5: if history_message_count >= 5 and not conversation.summary:
app_model = conversation.app app_model = conversation.app
if not app_model: if not app_model:
return return