Skip to content

D
dify
Commit 15ce6e12 authored by takatost's avatar takatost
Browse files
Options

fix bugs

parent 3983ae15
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
api/core/prompt/simple_prompt_transform.py
View file @ 15ce6e12
...@@ -178,11 +178,8 @@ class SimplePromptTransform(PromptTransform): ...@@ -178,11 +178,8 @@ class SimplePromptTransform(PromptTransform):
context=context context=context
) )
if prompt: if prompt and query:
if query: prompt_messages.append(SystemPromptMessage(content=prompt))
prompt_messages.append(SystemPromptMessage(content=prompt))
else:
prompt_messages.append(UserPromptMessage(content=prompt))
if memory: if memory:
prompt_messages = self._append_chat_histories( prompt_messages = self._append_chat_histories(
...@@ -193,6 +190,8 @@ class SimplePromptTransform(PromptTransform): ...@@ -193,6 +190,8 @@ class SimplePromptTransform(PromptTransform):
if query: if query:
prompt_messages.append(self.get_last_user_message(query, files)) 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 return prompt_messages, None
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment