|
|
|
@ -65,10 +65,10 @@ async def get_chat_log_by_session(mysql_pool, session_id, page=1, page_size=10):
|
|
|
|
|
)
|
|
|
|
|
total = (await cur.fetchone())[0]
|
|
|
|
|
|
|
|
|
|
# 查询分页数据
|
|
|
|
|
# 查询分页数据,按 id 降序排列
|
|
|
|
|
await cur.execute(
|
|
|
|
|
"SELECT id, session_id, user_input, model_response, audio_url, duration, create_time "
|
|
|
|
|
"FROM t_chat_log WHERE session_id = %s ORDER BY id asc LIMIT %s OFFSET %s",
|
|
|
|
|
"FROM t_chat_log WHERE session_id = %s ORDER BY id DESC LIMIT %s OFFSET %s",
|
|
|
|
|
(session_id, page_size, offset)
|
|
|
|
|
)
|
|
|
|
|
records = await cur.fetchall()
|
|
|
|
@ -114,11 +114,12 @@ async def update_risk(mysql_pool, session_id, risk_memo):
|
|
|
|
|
(risk_memo.replace('\n', '').replace("NO", ""), last_id)
|
|
|
|
|
)
|
|
|
|
|
await conn.commit()
|
|
|
|
|
print(f"已更新 session_id={session_id} 的最后一条记录 (id={last_id}) 的 risk_flag 和 risk_memo。")
|
|
|
|
|
logger.info(f"已更新 session_id={session_id} 的最后一条记录 (id={last_id}) 的 risk_flag 和 risk_memo。")
|
|
|
|
|
else:
|
|
|
|
|
print(f"未找到 session_id={session_id} 的记录。")
|
|
|
|
|
logger.warning(f"未找到 session_id={session_id} 的记录。")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 查询有风险的聊天记录
|
|
|
|
|
async def get_risk_chat_log_page(mysql_pool, risk_flag, page=1, page_size=10):
|
|
|
|
|
"""
|
|
|
|
|
查询有风险的聊天记录,并按 id 降序分页
|
|
|
|
@ -136,7 +137,7 @@ async def get_risk_chat_log_page(mysql_pool, risk_flag, page=1, page_size=10):
|
|
|
|
|
"SELECT COUNT(*) FROM t_chat_log WHERE risk_flag = %s", (risk_flag,)
|
|
|
|
|
)
|
|
|
|
|
total = (await cur.fetchone())[0]
|
|
|
|
|
print(f"总记录数: {total}")
|
|
|
|
|
logger.info(f"总记录数: {total}")
|
|
|
|
|
|
|
|
|
|
# 查询分页数据
|
|
|
|
|
query = (
|
|
|
|
@ -144,11 +145,11 @@ async def get_risk_chat_log_page(mysql_pool, risk_flag, page=1, page_size=10):
|
|
|
|
|
"FROM t_chat_log WHERE risk_flag = %s ORDER BY id DESC LIMIT %s OFFSET %s"
|
|
|
|
|
)
|
|
|
|
|
params = (risk_flag, page_size, offset)
|
|
|
|
|
print(f"执行查询: {query % params}") # 打印 SQL 查询
|
|
|
|
|
logger.debug(f"执行查询: {query % params}") # 打印 SQL 查询
|
|
|
|
|
|
|
|
|
|
await cur.execute(query, params)
|
|
|
|
|
records = await cur.fetchall()
|
|
|
|
|
print(f"查询结果: {records}") # 打印查询结果
|
|
|
|
|
logger.debug(f"查询结果: {records}") # 打印查询结果
|
|
|
|
|
|
|
|
|
|
# 将查询结果转换为字典列表
|
|
|
|
|
result = [
|
|
|
|
@ -170,4 +171,4 @@ async def get_risk_chat_log_page(mysql_pool, risk_flag, page=1, page_size=10):
|
|
|
|
|
"total": total,
|
|
|
|
|
"page": page,
|
|
|
|
|
"page_size": page_size
|
|
|
|
|
}
|
|
|
|
|
}
|