fix: completion stop invalid (#355)

ab115b5f · John Wang · GitHub · 3bbc4ad3 · ab115b5f
Unverified Commit ab115b5f authored Jun 13, 2023 by John Wang Committed by GitHub Jun 13, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

llm_callback_handler.py api/core/callback_handler/llm_callback_handler.py +1 -0

No files found.
--- 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