优化意图识别模块中的字符串格式化，更新分类信息提示内容，增强可读性和准确性。

2025-05-27 11:31:08 +08:00
parent 99017f0cb0
commit cd8e8bcad2
2 changed files with 13 additions and 7 deletions
@@ -101,9 +101,9 @@ class IntentRecognizer:
        Returns:
            分类结果
        """
-        formatted_prompt = classification_prompt.replace("{user_input}", query)
-        formatted_prompt = formatted_prompt.replace("{classification_info}", classification_info)
-        formatted_prompt = formatted_prompt.replace("{output_format}", self.classification_parser.get_format_instructions())
+        formatted_prompt = classification_prompt.format(user_input=query, 
+                                                        classification_info=classification_info, 
+                                                        output_format=self.classification_parser.get_format_instructions())
        # 将关键词列表转换为JSON字符串
        terms_dict = [term.model_dump() for term in keywords.terms] 
        keywords_str = json.dumps(terms_dict, ensure_ascii=False)
@@ -207,7 +207,9 @@ class IntentRecognizer:
        # 准备问题改写提示
        terms_dict = [term.model_dump(exclude={"description"}) for term in keywords.terms] 
        keywords_str = json.dumps(terms_dict, ensure_ascii=False)
-        formatted_prompt = query_rewrite_prompt.format(query=query, output_format=self.query_rewrite_parser.get_format_instructions(),keywords=keywords_str)
+        formatted_prompt = query_rewrite_prompt.format(query=query, 
+                                                       output_format=self.query_rewrite_parser.get_format_instructions(),
+                                                       keywords=keywords_str)

        
        # 调用LLM