diff --git a/rag2_0/dify/DifyCompareTest.py b/rag2_0/dify/DifyCompareTest.py index d323464..7186023 100755 --- a/rag2_0/dify/DifyCompareTest.py +++ b/rag2_0/dify/DifyCompareTest.py @@ -158,8 +158,8 @@ class DifyCompareTest: # 构建结果 result_row = row.copy() result_row["message_id"] = message_id - result_row["回答"] = answer - # result_row["词条与工单同时回答对比"] = judge_result + result_row["本次回答"] = answer + result_row["回答对比"] = judge_result result_row["检索到的词条"] = self.get_wiki_list_by_msgid(message_id) logging.info(f"成功处理第 {index + 1} 行数据") return index, result_row @@ -167,9 +167,10 @@ class DifyCompareTest: except Exception as e: logging.error(f"处理第 {index + 1} 行数据时出错: {e}") result_row = row.copy() - result_row["回答"] = '' - result_row["检索到的词条"] = '' result_row["message_id"] = '' + result_row["本次回答"] = '' + result_row["回答对比"] = '' + result_row["检索到的词条"] = '' return index, result_row @@ -247,7 +248,7 @@ if __name__ == "__main__": # 处理第一个文件 excel_files = [ # ("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: