fix: pub generate message text return null (#1037)

3 years ago · 02452421d5
parent 3a5c7c75ad
commit 02452421d5
1 changed files with 2 additions and 1 deletions
--- a/api/core/conversation_message_task.py
+++ b/api/core/conversation_message_task.py
@ -137,7 +137,8 @@ class ConversationMessageTask:
        db.session.flush()

    def append_message_text(self, text: str):
-        self._pub_handler.pub_text(text)
+        if text is not None:
+            self._pub_handler.pub_text(text)

    def save_message(self, llm_message: LLMMessage, by_stopped: bool = False):
        message_tokens = llm_message.prompt_tokens