opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix bugs

This commit is contained in:
takatost 2024-02-25 22:11:20 +08:00
parent 34ed5e428c
commit 77f04603b3
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/prompt/simple_prompt_transform.py
View File

@ -178,11 +178,8 @@ class SimplePromptTransform(PromptTransform):
context=context
)
if prompt:
if query:
prompt_messages.append(SystemPromptMessage(content=prompt))
else:
prompt_messages.append(UserPromptMessage(content=prompt))
if prompt and query:
prompt_messages.append(SystemPromptMessage(content=prompt))
if memory:
prompt_messages = self._append_chat_histories(
@ -193,6 +190,8 @@ class SimplePromptTransform(PromptTransform):
if query:
prompt_messages.append(self.get_last_user_message(query, files))
else:
prompt_messages.append(self.get_last_user_message(prompt, files))
return prompt_messages, None