diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index ead083364..86061dba0 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -111,7 +111,7 @@ class BaseChatNode(IChatNode): question = self.generate_prompt_question(prompt) self.context['question'] = question.content system = self.workflow_manage.generate_prompt(system) - self.context['system'] = question.content + self.context['system'] = system message_list = self.generate_message_list(system, prompt, history_message) self.context['message_list'] = message_list if stream: diff --git a/apps/application/flow/step_node/question_node/impl/base_question_node.py b/apps/application/flow/step_node/question_node/impl/base_question_node.py index f67206bda..d20b5a5f8 100644 --- a/apps/application/flow/step_node/question_node/impl/base_question_node.py +++ b/apps/application/flow/step_node/question_node/impl/base_question_node.py @@ -93,7 +93,7 @@ class BaseQuestionNode(IQuestionNode): question = self.generate_prompt_question(prompt) self.context['question'] = question.content system = self.workflow_manage.generate_prompt(system) - self.context['system'] = question.content + self.context['system'] = system message_list = self.generate_message_list(system, prompt, history_message) self.context['message_list'] = message_list if stream: