上传文件至 kg_lab_6.13

6.18 更新数据配置路径统一，和前端demo

kg_lab_6.13/main2.py

@@ -1,8 +1,13 @@
 from chains_lab import Problem_rewrite
+from chains_lab import booway_cypher_chain
+from chains_lab import question_answer, question_answer_calculation
 from vector_lab import intersection_of_three_lists
 from utils import find_target_item, find_target_items, pre_mapping, pre_mapping2
+from utils import extract_concrete_info, extract_query_prefix_list, split_chinese_bracketed_phrases
 import json
 
+from extraction_info import info_data_json
+
 # 样例
 # input_str1 = "杆塔总基数是多少？"
 # input_str2 = "单回路长度是多少？"
@@ -13,21 +18,15 @@ import json
 # input_str7 = "计算一下本体工程机械费"
 # input_str8 = "项目建设技术服务费合计"
 
-# 初始化
+# 初始化chians
 problem_rewrite = Problem_rewrite()
-
-from utils import extract_concrete_info, extract_query_prefix_list, split_chinese_bracketed_phrases
-
-from chains_lab import question_answer, question_answer_calculation
-
 qa_chains = question_answer()
-
 calculation_chains = question_answer_calculation()
 
-from chains_lab import  booway_cypher_chain
+
 
 # 加载数据
-with open('./data/data.json', 'r', encoding='utf-8') as file:
+with open(info_data_json, 'r', encoding='utf-8') as file:
     data = json.load(file)
 
 print("📥 请输入查询内容，输入 'exit' 可退出程序。\n")
@@ -91,11 +90,16 @@ while True:
             print(ques)
 
             retriever_info = []
-            for idx, i in enumerate(ques):
-                response = booway_cypher_chain.invoke(i)    
-                temp = response.get("result")
-                retriever_info.append(temp)
-        
+            for i in ques[:12]:
+                try:
+                    response = booway_cypher_chain.invoke(i)
+                    temp = response.get("result")
+                    # todo: 重复筛选策略
+                    retriever_info.append(temp)
+                except Exception as e:
+                    print(f"处理问题时出错: {e}")
+                    retriever_info.append(None) 
+                
 
             retriever_keywords = ques_info[0]
             calculation = ques_info[-1]