diff --git a/packages/dbgpt-app/src/dbgpt_app/scene/chat_dashboard/chat.py b/packages/dbgpt-app/src/dbgpt_app/scene/chat_dashboard/chat.py index 23e6d8039..ed3b11383 100644 --- a/packages/dbgpt-app/src/dbgpt_app/scene/chat_dashboard/chat.py +++ b/packages/dbgpt-app/src/dbgpt_app/scene/chat_dashboard/chat.py @@ -95,6 +95,19 @@ class ChatDashboard(BaseChat): "supported_chat_type": self.dashboard_template["supported_chart_type"], } + # Mapping variable names: compatible with custom prompt template variable names + # Get the input_variables of the current prompt + input_variables = [] + if hasattr(self.prompt_template, "prompt") and hasattr( + self.prompt_template.prompt, "input_variables" + ): + input_variables = self.prompt_template.prompt.input_variables + # Compatible with question and user_input + if "question" in input_variables: + input_values["question"] = self.current_user_input + if "user_input" in input_variables: + input_values["user_input"] = self.current_user_input + return input_values def do_action(self, prompt_response):