feat: Configurable output of reason for Intent recognition

Author: wangliang181230

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

@@ -23,6 +23,7 @@ class IntentNodeSerializer(serializers.Serializer):
     model_params_setting = serializers.DictField(required=False,
                                                  label=_("Model parameter settings"))
     branch = IntentBranchSerializer(many=True)
+    output_reason = serializers.BooleanField(required=False, label=_("Output reason"), default=True)
 
 
 class IIntentNode(INode):

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

@@ -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,
+    def execute(self, model_id, dialogue_number, history_chat_record, user_input, branch, output_reason,
                 model_params_setting=None, **kwargs) -> NodeResult:
 
         # 设置默认模型参数
@@ -73,7 +73,7 @@ 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)
+        prompt = self.build_classification_prompt(user_input, branch, output_reason)
 
         # 生成消息列表
         system = self.build_system_prompt()
@@ -129,7 +129,7 @@ def build_system_prompt(self) -> str:
         """构建系统提示词"""
         return "你是一个专业的意图识别助手，请根据用户输入和意图选项，准确识别用户的真实意图。"
 
-    def build_classification_prompt(self, user_input: str, branch: List[Dict]) -> str:
+    def build_classification_prompt(self, user_input: str, branch: List[Dict], output_reason: bool) -> str:
         """构建分类提示词"""
 
         classification_list = []
@@ -153,7 +153,8 @@ def build_classification_prompt(self, user_input: str, branch: List[Dict]) -> st
 
         return PROMPT_TEMPLATE.format(
             classification_list=classification_list,
-            user_input=user_input
+            user_input=user_input,
+            output_reason=',\n"reason": ""' if output_reason is True else ""
         )
 
     def generate_message_list(self, system: str, prompt: str, history_message):

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

@@ -21,8 +21,7 @@
 - Do not add prefix ```json or suffix ```
 - The answer needs to include the following fields such as:
 {{
-"classificationId": 0,
-"reason": ""
+"classificationId": 0{output_reason}
 }}
 
 ## Limit

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

@@ -447,6 +447,7 @@ You are a master of problem optimization, adept at accurately inferring user int
       classify: {
         label: 'Intent classify',
       },
+      output_reason: 'Output Reason',
       input: {
         label: 'Input',
       },

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

@@ -439,6 +439,7 @@ export default {
       classify: {
         label: '意图分类',
       },
+      output_reason: '输出理由',
       input: {
         label: '输入',
       },

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

@@ -439,6 +439,7 @@ export default {
       classify: {
         label: '意圖分類',
       },
+      output_reason: '輸出理由',
       input: {
         label: '輸入',
       },

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

@@ -138,6 +138,19 @@
             </div>
           </div>
         </el-form-item>
+        <el-form-item>
+          <div class="flex-between w-full">
+            <div>
+              <span>{{ $t('workflow.nodes.intentNode.output_reason') }}</span>
+            </div>
+            <div>
+              <el-switch
+                size="small"
+                v-model="form_data.output_reason"
+              />
+            </div>
+          </div>
+        </el-form-item>
       </el-form>
     </el-card>
 
@@ -295,6 +308,7 @@ const form = {
   ],
   dialogue_number: 1,
   content_list: [],
+  output_reason: true,
 }
 
 function refreshParam(data: any) {