在heli_db_to_excel.py中添加SEQUENCE_ID字段到数据处理结果,并新增根据会话ID查询SEQUENCE_ID的方法;在DifyCompareTest.py中优化回答判断逻辑,增加对空回答的处理,调整文件处理的最大并发数。
This commit is contained in:
@@ -69,7 +69,8 @@ class DifyCompareTest:
|
||||
|
||||
max_retries = 3
|
||||
retry_count = 0
|
||||
|
||||
if len(old_answer) == 0 or len(now_answer) == 0:
|
||||
return "回答基本不相同"
|
||||
while retry_count < max_retries:
|
||||
try:
|
||||
response = self.llm.invoke(user_prompt=user_prompt, need_retry=False, response_format={"type": "json_object"})
|
||||
@@ -100,15 +101,14 @@ class DifyCompareTest:
|
||||
)
|
||||
result = response.json()
|
||||
answer = result.get('answer', "")
|
||||
answer = answer.split("----------------------------------------")[0].strip()
|
||||
if len(answer) == 0:
|
||||
raise Exception(f"回答为空: {result}")
|
||||
if old_answer:
|
||||
judge_result = self.llm_judge_answer(old_answer=old_answer, now_answer=answer)
|
||||
else:
|
||||
judge_result=""
|
||||
judge_result=""
|
||||
# 只取回答的前半部分
|
||||
answer = answer.split("----------------------------------------")[0].strip()
|
||||
message_id = result.get('message_id', "")
|
||||
return answer, judge_result, message_id
|
||||
except Exception as e:
|
||||
@@ -247,14 +247,14 @@ if __name__ == "__main__":
|
||||
|
||||
# 处理第一个文件
|
||||
excel_files = [
|
||||
# ("data/excel/5月.xlsx", "data/excel/5月问答对比.xlsx"),
|
||||
("data/excel/第四轮点踩问题.xlsx", "data/excel/第四轮点踩问题_问答测试.xlsx")
|
||||
("data/excel/300专业提问.xlsx", "data/excel/300专业提问_问答测试.xlsx"),
|
||||
# ("data/excel/有知识的.xlsx", "data/excel/有知识的_问答测试.xlsx")
|
||||
]
|
||||
|
||||
for excel_path, save_path in excel_files:
|
||||
logging.info(f"开始处理文件: {excel_path}")
|
||||
try:
|
||||
dify_compare_test.run(excel_path=excel_path, save_path=save_path, max_workers=3)
|
||||
dify_compare_test.run(excel_path=excel_path, save_path=save_path, max_workers=10)
|
||||
logging.info(f"文件处理完成: {excel_path}")
|
||||
except Exception as e:
|
||||
logging.error(f"处理文件 {excel_path} 时出错: {e}")
|
||||
|
||||
Reference in New Issue
Block a user