fix: Error in obtaining variables for character settings

1Panel-dev · Dec 30, 2024 · 262040c · 262040c
1 parent ad9c837
commit 262040c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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
@@ -111,7 +111,7 @@ def execute(self, model_id, system, prompt, dialogue_number, history_chat_record
         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
@@ -93,7 +93,7 @@ def execute(self, model_id, system, prompt, dialogue_number, history_chat_record
         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: