estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
takatost
2023-08-28 16:43:54 +08:00
committed by GitHub
parent 3a5c7c75ad
commit 02452421d5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/conversation_message_task.py
View File

@@ -137,7 +137,8 @@ class ConversationMessageTask:
db.session.flush() db.session.flush()
def append_message_text(self, text: str): 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): def save_message(self, llm_message: LLMMessage, by_stopped: bool = False):
message_tokens = llm_message.prompt_tokens message_tokens = llm_message.prompt_tokens