在heli_db_to_excel.py中添加SEQUENCE_ID字段到数据处理结果,并新增根据会话ID查询SEQUENCE_ID的方法;在DifyCompareTest.py中优化回答判断逻辑,增加对空回答的处理,调整文件处理的最大并发数。

This commit is contained in:
2025-08-01 13:45:58 +08:00
parent 9b957665b5
commit 948827d553
2 changed files with 18 additions and 7 deletions
+12 -1
View File
@@ -277,6 +277,7 @@ class DataProcessor:
"消息发送者": record["message_sender"],
"发送者昵称": record["sender_nickname"],
"创建时间": record["CREATE_TIME"],
"SEQUENCE_ID": record["SEQUENCE_ID"],
})
return result
@@ -337,11 +338,21 @@ class MariaDBClient:
df, _ = self.execute_query(sql, (start_date, end_date))
return df
def get_sequence_id_by_session_id(self, session_id: str) -> Optional[pd.DataFrame]:
"""根据会话ID查询消息详情"""
sql = """
SELECT SEQUENCE_ID
FROM crm_hlyj.crm_hlyj_dmri
WHERE SESSION_ID = %s
"""
df, _ = self.execute_query(sql, (session_id,))
return df
def query_messages_by_session_id(self, session_id: str) -> Optional[pd.DataFrame]:
"""根据会话ID查询消息详情"""
sql = """
SELECT CREATE_TIME, CUS_NICK_NAME, MODE, MSG_TYPE, AGENT_NAME, CONTENT,
SESSION_ID, ACCOUNT, SYSTEM_MODE_MESSAGE_TYPE
SESSION_ID, ACCOUNT, SYSTEM_MODE_MESSAGE_TYPE, SEQUENCE_ID
FROM crm_hlyj.crm_hlyj_dmri
WHERE SESSION_ID = %s
ORDER BY CREATE_TIME