Zdao032
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from chains_lab import Problem_rewrite
|
|
|
|
problem_rewrite = Problem_rewrite()
|
|
|
|
from vector_lab import intersection_of_three_lists
|
|
|
|
# input_str1 = "杆塔总基数是多少？"
|
|
# input_str2 = "单回路长度是多少？"
|
|
# input_str3 = "计算一下角钢塔的塔材装材费"
|
|
# input_str4 = "计算一下土石方总量"
|
|
# input_str5 = "板式塔基的各类基础数量占总塔基数比例是多少？"
|
|
# input_str6 = "基础混凝土总量是多少"
|
|
# input_str7 = "计算一下本体工程机械费"
|
|
# ipout_str8 = "项目建设技术服务费合计"
|
|
|
|
input_str = "项目建设技术服务费合计是多少？"
|
|
results = intersection_of_three_lists(input_str)
|
|
|
|
retriever = intersection_of_three_lists(input_str)[0]
|
|
|
|
print(f"输入:{input_str}")
|
|
|
|
keywords = problem_rewrite.invoke({"query":input_str, "retriever":retriever})
|
|
|
|
print(f"输出:{keywords}")
|
|
|
|
|
|
import json
|
|
|
|
with open('./data/data.json', 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
|
|
from utils import find_target_item, find_target_items, pre_mapping, pre_mapping2
|
|
|
|
input_neo4j = pre_mapping2(keywords, data)
|
|
|
|
print(f"检索目标：{input_neo4j}") |