| @@ -75,7 +75,11 @@ class LLMCallbackHandler(BaseCallbackHandler): | |||
| self.conversation_message_task.save_message(self.llm_message) | |||
| def on_llm_new_token(self, token: str, **kwargs: Any) -> None: | |||
| self.conversation_message_task.append_message_text(token) | |||
| try: | |||
| self.conversation_message_task.append_message_text(token) | |||
| except ConversationTaskStoppedException as ex: | |||
| self.on_llm_error(error=ex) | |||
| self.llm_message.completion += token | |||
| def on_llm_error( | |||