ouyangyouzhang/QueryRewrite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新DifyCompareTest类中的字段名称

Browse Source
This commit is contained in:
ouyangyouzhang
2025-07-31 10:25:24 +08:00
parent 11acc809c8
commit 9185d75e10
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rag2_0/dify/DifyCompareTest.py
+6 -5
View File
@@ -158,8 +158,8 @@ class DifyCompareTest:
# 构建结果 # 构建结果
result_row = row.copy() result_row = row.copy()
result_row["message_id"] = message_id result_row["message_id"] = message_id
result_row["回答"] = answer result_row["本次回答"] = answer
# result_row["词条与工单同时回答对比"] = judge_result result_row["回答对比"] = judge_result
result_row["检索到的词条"] = self.get_wiki_list_by_msgid(message_id) result_row["检索到的词条"] = self.get_wiki_list_by_msgid(message_id)
logging.info(f"成功处理第 {index + 1} 行数据") logging.info(f"成功处理第 {index + 1} 行数据")
return index, result_row return index, result_row
@@ -167,9 +167,10 @@ class DifyCompareTest:
except Exception as e: except Exception as e:
logging.error(f"处理第 {index + 1} 行数据时出错: {e}") logging.error(f"处理第 {index + 1} 行数据时出错: {e}")
result_row = row.copy() result_row = row.copy()
result_row["回答"] = ''
result_row["检索到的词条"] = ''
result_row["message_id"] = '' result_row["message_id"] = ''
result_row["本次回答"] = ''
result_row["回答对比"] = ''
result_row["检索到的词条"] = ''
return index, result_row return index, result_row
@@ -247,7 +248,7 @@ if __name__ == "__main__":
# 处理第一个文件 # 处理第一个文件
excel_files = [ excel_files = [
# ("data/excel/5月.xlsx", "data/excel/5月问答对比.xlsx"), # ("data/excel/5月.xlsx", "data/excel/5月问答对比.xlsx"),
("data/excel/7.30数据导出.xlsx", "data/excel/7.30数据导出_问答测试.xlsx") ("data/excel/第四轮点踩问题.xlsx", "data/excel/第四轮点踩问题_问答测试.xlsx")
] ]
for excel_path, save_path in excel_files: for excel_path, save_path in excel_files: