优化DifyQueryRetrieval类,新增top_k参数以支持检索结果的数量控制,同时重构相关方法以提升性能和可维护性。更新DifyTool类,新增获取应用会话和消息信息的方法。修复DifyExporter类中的代码格式问题,调整日期参数的默认值。

This commit is contained in:
2025-07-10 08:36:12 +08:00
parent ec3db656a5
commit 23f522dde5
4 changed files with 28 additions and 15 deletions
+2 -4
View File
@@ -174,8 +174,6 @@ class DifyExporter:
"""
conversations = self.dify_pgsql.get_app_conversations(appid=self.app_id)
for conversation in conversations:
if conversation['conversation_id'] == '10d04219-0359-42f7-b9da-2ba039bf87a2':
breakpoint()
messages = self.dify_pgsql.get_conversation_messages(conversation_id=conversation['conversation_id'])
message_chain_new = self.process_message_chain(messages)
if len(message_chain_new) != len(messages):
@@ -198,7 +196,7 @@ class DifyExporter:
message_info = self.extract_message_info(message)
if message_info:
self.message_info_list.append(message_info)
return self.message_info_list
def save_to_excel(self, message_info_list, output_file):
@@ -323,7 +321,7 @@ if __name__ == "__main__":
help='Dify应用ID')
parser.add_argument('--query_log_file', '-q', type=str, default="data/query_logs/answer_type_logs.json",
help='查询日志文件路径')
parser.add_argument('--start_date', '-s', type=str, default=None,
parser.add_argument('--start_date', '-s', type=str, default="2025-07-09 13",
help='开始日期时间,格式为YYYY-MM-DD HH,例如2025-07-08 14表示2025年7月8日14时(UTC+8时区)')
parser.add_argument('--end_date', '-e', type=str, default=None,
help='结束日期时间,格式为YYYY-MM-DD HH,例如2025-07-08 18表示2025年7月8日18时(UTC+8时区)')