在heli_db_to_excel.py中添加SEQUENCE_ID字段到数据处理结果,并新增根据会话ID查询SEQUENCE_ID的方法;在DifyCompareTest.py中优化回答判断逻辑,增加对空回答的处理,调整文件处理的最大并发数。
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user