feat: The prompt template for the Intent Recognition is configurable

Author: wangliang181230

apps/application/flow/step_node/intent_node/i_intent_node.py

@@ -17,6 +17,7 @@ class IntentBranchSerializer(serializers.Serializer):
 
 class IntentNodeSerializer(serializers.Serializer):
     model_id = serializers.CharField(required=True, label=_("Model id"))
+    prompt_template = serializers.CharField(required=False, allow_blank=True, allow_null=True, label=_("Prompt template"))
     content_list = serializers.ListField(required=True, label=_("Text content"))
     dialogue_number = serializers.IntegerField(required=True, label=
     _("Number of multi-round conversations"))

apps/application/flow/step_node/intent_node/impl/base_intent_node.py

@@ -12,7 +12,7 @@
 from application.flow.step_node.intent_node.i_intent_node import IIntentNode
 from models_provider.models import Model
 from models_provider.tools import get_model_instance_by_model_workspace_id, get_model_credential
-from .prompt_template import PROMPT_TEMPLATE
+from .prompt_template import DEFAULT_PROMPT_TEMPLATE
 
 
 def get_default_model_params_setting(model_id):
@@ -52,7 +52,7 @@ def save_context(self, details, workflow_manage):
         self.context['branch_id'] = details.get('branch_id')
         self.context['category'] = details.get('category')
 
-    def execute(self, model_id, dialogue_number, history_chat_record, user_input, branch, output_reason,
+    def execute(self, model_id, prompt_template, dialogue_number, history_chat_record, user_input, branch, output_reason,
                 model_params_setting=None, **kwargs) -> NodeResult:
 
         # 设置默认模型参数
@@ -73,7 +73,8 @@ def execute(self, model_id, dialogue_number, history_chat_record, user_input, br
         self.context['user_input'] = user_input
 
         # 构建分类提示词
-        prompt = self.build_classification_prompt(user_input, branch, output_reason)
+        prompt_template = self.workflow_manage.generate_prompt(prompt_template)
+        prompt = self.build_classification_prompt(prompt_template, user_input, branch, output_reason)
         self.context['system'] = prompt
 
         # 生成消息列表
@@ -130,7 +131,7 @@ def build_system_prompt(self) -> str:
         """构建系统提示词"""
         return "你是一个专业的意图识别助手，请根据用户输入和意图选项，准确识别用户的真实意图。"
 
-    def build_classification_prompt(self, user_input: str, branch: List[Dict], output_reason: bool) -> str:
+    def build_classification_prompt(self, prompt_template: str, user_input: str, branch: List[Dict], output_reason: bool) -> str:
         """构建分类提示词"""
 
         classification_list = []
@@ -152,10 +153,14 @@ def build_classification_prompt(self, user_input: str, branch: List[Dict], outpu
                 })
                 classification_id += 1
 
-        return PROMPT_TEMPLATE.format(
+        # 构建输出JSON结构
+        output_reason = ', "reason": ""' if output_reason is True else ''
+        output_json = f'{{"classificationId": 0{output_reason}}}'
+
+        return (prompt_template or DEFAULT_PROMPT_TEMPLATE).format(
             classification_list=classification_list,
             user_input=user_input,
-            output_reason=',\n"reason": ""' if output_reason is True else ''
+            output_json=output_json
         )
 
     def generate_message_list(self, system: str, prompt: str, history_message):

apps/application/flow/step_node/intent_node/impl/prompt_template.py

@@ -1,6 +1,6 @@
 
 
-PROMPT_TEMPLATE = """# Role
+DEFAULT_PROMPT_TEMPLATE = """# Role
 You are an intention classification expert, good at being able to judge which classification the user's input belongs to.
 
 ## Skills
@@ -20,9 +20,7 @@
 - Strictly ensure that the output is in a valid JSON format.
 - Do not add prefix ```json or suffix ```
 - The answer needs to include the following fields such as:
-{{
-"classificationId": 0{output_reason}
-}}
+{output_json}
 
 ## Limit
 - Please do not reply in text."""

ui/src/locales/lang/en-US/views/application.ts

@@ -87,6 +87,11 @@ export default {
          - Please use concise and professional language to answer the user's question.
          `,
     },
+    prompt_template: {
+      label: 'Prompt Template',
+      requiredMessage: 'Please enter Prompt template',
+      tooltip: 'Please pay attention to the placeholders `{xxx}` in the template',
+    },
     historyRecord: {
       label: 'Chat History',
     },

ui/src/locales/lang/en-US/workflow.ts

@@ -542,6 +542,7 @@ You are a master of problem optimization, adept at accurately inferring user int
   },
   SystemPromptPlaceholder: 'System Prompt, can reference variables in the system, such as',
   UserPromptPlaceholder: 'User Prompt, can reference variables in the system, such as',
+  PromptTemplatePlaceholder: 'Prompt Template, can reference variables in the system, such as',
   initiator: 'Iniiator',
   abnormalInformation: 'Abnormal Information',
 }

ui/src/locales/lang/zh-CN/views/application.ts

@@ -80,6 +80,11 @@ export default {
 回答要求：
 - 请使用中文回答用户问题`,
     },
+    prompt_template: {
+      label: '提示词模板',
+      requiredMessage: '请输入提示词模板',
+      tooltip: '请注意模板中的占位符 `{xxx}`',
+    },
     historyRecord: {
       label: '历史聊天记录',
     },

ui/src/locales/lang/zh-CN/workflow.ts

@@ -533,6 +533,7 @@ export default {
   },
   SystemPromptPlaceholder: '系统提示词，可以引用系统中的变量：如',
   UserPromptPlaceholder: '用户提示词，可以引用系统中的变量：如',
+  PromptTemplatePlaceholder: '提示词模板，可以引用系统中的变量：如',
   initiator: '发起人',
   abnormalInformation: '异常信息'
 

ui/src/locales/lang/zh-Hant/views/application.ts

@@ -80,6 +80,11 @@ export default {
 回答要求：
 - 請使用中文回答用戶問題`,
     },
+    prompt_template: {
+      label: '提示詞模板',
+      requiredMessage: '請輸入提示詞模板',
+      tooltip: '請注意模板中的佔位符 `{xxx}`',
+    },
     historyRecord: {
       label: '歷史對話紀錄',
     },

ui/src/locales/lang/zh-Hant/workflow.ts

@@ -527,6 +527,7 @@ export default {
   },
   SystemPromptPlaceholder: '系統提示詞，可以引用系統中的變量：如',
   UserPromptPlaceholder: '用戶提示詞，可以引用系統中的變量：如',
+  PromptTemplatePlaceholder: '提示詞模板，可以引用系統中的變量：如',
   initiator: '發起人',
   abnormalInformation: '異常信息',
 }

ui/src/workflow/nodes/intent-classify-node/index.vue

@@ -51,6 +51,38 @@
             :model-type="'LLM'"
           ></ModelSelect>
         </el-form-item>
+        <el-form-item
+          :label="$t('views.application.form.prompt_template.label')"
+          prop="prompt_template"
+          :rules="{
+            required: true,
+            message: $t('views.application.form.prompt_template.requiredMessage'),
+            trigger: 'blur',
+          }"
+        >
+          <template #label>
+            <div class="flex align-center">
+              <div class="mr-4">
+                <span
+                >{{ $t('views.application.form.prompt_template.label')
+                  }}<span class="color-danger">*</span></span
+                >
+              </div>
+              <el-tooltip effect="dark" placement="right" popper-class="max-w-200">
+                <template #content>{{ $t('views.application.form.prompt_template.tooltip') }} </template>
+                <AppIcon iconName="app-warning" class="app-warning-icon"></AppIcon>
+              </el-tooltip>
+            </div>
+          </template>
+          <MdEditorMagnify
+            @wheel="wheel"
+            :title="$t('views.application.form.prompt_template.label')"
+            v-model="form_data.prompt_template"
+            style="height: 100px"
+            @submitDialog="submitTemplateDialog"
+            :placeholder="`${t('workflow.PromptTemplatePlaceholder')}{{${t('workflow.nodes.startNode.label')}.question}}`"
+          />
+        </el-form-item>
         <el-form-item
           prop="content_list"
           :label="$t('workflow.nodes.intentNode.input.label')"
@@ -292,8 +324,34 @@ const model_change = (model_id?: string) => {
   }
 }
 
+const defaultPromptTemplate = `# Role
+You are an intention classification expert, good at being able to judge which classification the user's input belongs to.
+
+## Skills
+Skill 1: Clearly determine which of the following intention classifications the user's input belongs to.
+Intention classification list:
+{classification_list}
+
+Note:
+- Please determine the match between the user's input content and the Intention classification list content, without judging or categorizing the match with the classification ID.
+- **When classifying, you must give higher weight to the context and intent continuity shown in the historical conversation. Do not rely solely on the literal meaning of the current input; instead, prioritize the most consistent classification with the previous dialogue flow.**
+
+## User Input
+{user_input}
+
+## Reply requirements
+- The answer must be returned in JSON format.
+- Strictly ensure that the output is in a valid JSON format.
+- Do not add prefix \`\`\`json or suffix \`\`\`
+- The answer needs to include the following fields such as:
+{output_json}
+
+## Limit
+- Please do not reply in text.`
+
 const form = {
   model_id: '',
+  prompt_template: defaultPromptTemplate,
   branch: [
     {
       id: randomId(),
@@ -315,6 +373,10 @@ function refreshParam(data: any) {
   set(props.nodeModel.properties.node_data, 'model_params_setting', data)
 }
 
+function submitTemplateDialog(val: string) {
+  set(props.nodeModel.properties.node_data, 'prompt_template', val)
+}
+
 const openAIParamSettingDialog = (modelId: string) => {
   if (modelId) {
     AIModeParamSettingDialogRef.value?.open(modelId, id, form_data.value.model_params_setting)
@@ -323,6 +385,9 @@ const openAIParamSettingDialog = (modelId: string) => {
 const form_data = computed({
   get: () => {
     if (props.nodeModel.properties.node_data) {
+      if (!props.nodeModel.properties.node_data.prompt_template) {
+        set(props.nodeModel.properties.node_data, 'prompt_template', defaultPromptTemplate)
+      }
       return props.nodeModel.properties.node_data
     } else {
       set(props.nodeModel.properties, 'node_data', form)