Zdao032
52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
"""
|
||
===================================
|
||
@Auther：WenZ
|
||
@Company: BooWay
|
||
@project：booway_dm
|
||
===================================
|
||
"""
|
||
from vector_load import Mixed_retrieval, interface_search
|
||
|
||
xizang_input_path = '../data/temp/Z1_keywords_software_usage.txt'
|
||
retriever_txt_faiss1, retriever_txt_faiss2, retriever_txt_faiss3 = Mixed_retrieval(xizang_input_path)
|
||
|
||
cuceng_input_path = '../data/temp/C1_keywords_software_usage.txt'
|
||
retriever_txt_faiss4, retriever_txt_faiss5, retriever_txt_faiss6 = Mixed_retrieval(cuceng_input_path)
|
||
|
||
jigai_input_path = '../data/temp/T1_keywords_software_usage.txt'
|
||
retriever_txt_faiss7, retriever_txt_faiss8, retriever_txt_faiss9 = Mixed_retrieval(jigai_input_path)
|
||
|
||
from vector_load import Building_search_dictionary, Official_website_kg_search
|
||
|
||
input_index_csv_path = "/data/Z_project/rahulnyk/DM_lab/data/temp/data_index.csv"
|
||
xizang_input_csv_path = "/data/Z_project/rahulnyk/DM_lab/data/temp/西藏造价FAQ数据集.csv"
|
||
cuceng_input_csv_path = "/data/Z_project/rahulnyk/DM_lab/data/temp/新型储能电站建设计价通C1.csv"
|
||
jigai_input_csv_path = "/data/Z_project/rahulnyk/DM_lab/data/temp/技改检修计价通T1.csv"
|
||
|
||
|
||
def process_domain_category(nlu_info, retrievers, input_csv_path, index_csv_path):
|
||
"""
|
||
处理不同领域类别的检索逻辑。
|
||
|
||
:param nlu_info: 包含领域类别和检索关键词的对象
|
||
:param retrievers: 用于检索的 FAISS 索引列表
|
||
:param input_csv_path: 领域特定的 CSV 输入路径
|
||
:param index_csv_path: 索引 CSV 路径
|
||
:return: 处理后的 QA RAG 结果列表
|
||
"""
|
||
index_keywords = interface_search(nlu_info.retrieve_keywords, *retrievers)
|
||
qa_rag = []
|
||
|
||
for keyword in index_keywords:
|
||
# todo: bug修改: 避免output_id为None情况
|
||
output_path, output_id = Building_search_dictionary(input_csv_path, index_csv_path, keyword)
|
||
if output_path is not None or output_id is not None:
|
||
qa_rag.append(f"检索知识：{output_path}")
|
||
qa_rag.append(Official_website_kg_search(output_id))
|
||
else:
|
||
continue
|
||
|
||
return qa_rag
|
||
|
||
|