Przeglądaj źródła

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

takatost 1 rok temu
rodzic
commit
02452421d5
1 zmienionych plików z 2 dodań i 1 usunięć
  1. 2 1
      api/core/conversation_message_task.py

+ 2 - 1
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