Explorar el Código

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

takatost hace 1 año
padre
commit
02452421d5
Se han modificado 1 ficheros con 2 adiciones y 1 borrados
  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