diff --git a/rag2_0/intent_recognition/IntentRecognition.py b/rag2_0/intent_recognition/IntentRecognition.py index 3a77e3a..f07d86c 100644 --- a/rag2_0/intent_recognition/IntentRecognition.py +++ b/rag2_0/intent_recognition/IntentRecognition.py @@ -210,6 +210,8 @@ class IntentRecognizer: current_key_terms = set() # 添加向量检索结果 for result in vector_results: + if isinstance(result.get('synonymous', []), str): + result['synonymous'] = result['synonymous'].split(';') term = Term( name=result.get('name'), synonymous=result.get('synonymous', []), diff --git a/rag2_0/intent_recognition/PromptTemplates.py b/rag2_0/intent_recognition/PromptTemplates.py index e24156b..5abe18f 100644 --- a/rag2_0/intent_recognition/PromptTemplates.py +++ b/rag2_0/intent_recognition/PromptTemplates.py @@ -15,8 +15,9 @@ extract_nouns_prompt=""" - 软件相关:功能模块/操作步骤/报错提示/扩展名后缀名 - 造价专业:费用类型/计算标准/行业规范 - 电力工程:项目类型/设备型号/工程阶段 -3. 自动展开缩写(如将'导excel'转为'Excel导入') +3. 自动展开缩写(如将'导excel'转为'导入Excel') 4. 严格基于用户问题提取关键词,不要输出与用户问题无关的关键词 +5. 不要擅自揣摩、猜测、推断关键词,不要修改关键的信息(如:粘贴excel->导入excel) 三、输出格式: {output_format}