optimize

Author: wangliang181230

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

@@ -189,8 +189,14 @@ def get_branch_by_id(category_id: int):
             return None
 
         try:
-            result_json = json.loads(result)
-            classification_id = result_json.get('classificationId')
+            # 先尝试解析为数字（自定义提示词模板时，可提示大模型只输出ID值）
+            classification_id = self.to_int(result)
+
+            # 再尝试解析为 JSON
+            if classification_id is None:
+                result_json = json.loads(result)
+                classification_id = result_json.get('classificationId')
+
             # 如果是 0 ，返回其他分支
             matched_branch = get_branch_by_id(classification_id)
             if matched_branch:
@@ -230,6 +236,12 @@ def parse_result_reason(self, result: str):
 
             return ''
 
+    def to_int(self, str):
+        try:
+            return int(str)
+        except ValueError:
+            return None
+
     def find_other_branch(self, branch: List[Dict]) -> Dict[str, Any] | None:
         """查找其他分支"""
         for b in branch: