From 77f04603b3633c809e03d6e4b7b4d79d18d6ce59 Mon Sep 17 00:00:00 2001 From: takatost Date: Sun, 25 Feb 2024 22:11:20 +0800 Subject: [PATCH] fix bugs --- api/core/prompt/simple_prompt_transform.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/api/core/prompt/simple_prompt_transform.py b/api/core/prompt/simple_prompt_transform.py index a929416be4..fcae0dc786 100644 --- a/api/core/prompt/simple_prompt_transform.py +++ b/api/core/prompt/simple_prompt_transform.py @@ -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