opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: completion stop invalid (#355)

This commit is contained in:
John Wang 2023-06-13 17:47:42 +08:00 committed by GitHub
parent 3bbc4ad3db
commit ab115b5f87
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/callback_handler/llm_callback_handler.py
View File

@ -79,6 +79,7 @@ class LLMCallbackHandler(BaseCallbackHandler):
self.conversation_message_task.append_message_text(token) self.conversation_message_task.append_message_text(token)
except ConversationTaskStoppedException as ex: except ConversationTaskStoppedException as ex:
self.on_llm_error(error=ex) self.on_llm_error(error=ex)
raise ex
self.llm_message.completion += token self.llm_message.completion += token