fix: completion stop invalid (#355)

3 years ago · ab115b5f87
parent 3bbc4ad3db
commit ab115b5f87
1 changed files with 1 additions and 0 deletions
--- 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