From ab115b5f87b0c595d2c5f48b4f98e029f15e47b6 Mon Sep 17 00:00:00 2001 From: John Wang Date: Tue, 13 Jun 2023 17:47:42 +0800 Subject: [PATCH] fix: completion stop invalid (#355) --- api/core/callback_handler/llm_callback_handler.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/core/callback_handler/llm_callback_handler.py b/api/core/callback_handler/llm_callback_handler.py index da38fcabf5..89a7630737 100644 --- a/api/core/callback_handler/llm_callback_handler.py +++ b/api/core/callback_handler/llm_callback_handler.py @@ -79,6 +79,7 @@ class LLMCallbackHandler(BaseCallbackHandler): self.conversation_message_task.append_message_text(token) except ConversationTaskStoppedException as ex: self.on_llm_error(error=ex) + raise ex self.llm_message.completion += token