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

This commit is contained in:
takatost 2023-08-28 16:43:54 +08:00 committed by GitHub
parent 3a5c7c75ad
commit 02452421d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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