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

02452421 · takatost · GitHub · 3a5c7c75 · 02452421
Unverified Commit 02452421 authored Aug 28, 2023 by takatost Committed by GitHub Aug 28, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

conversation_message_task.py api/core/conversation_message_task.py +2 -1

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