优化DifyQueryRetrieval类,新增top_k参数以支持检索结果的数量控制,同时重构相关方法以提升性能和可维护性。更新DifyTool类,新增获取应用会话和消息信息的方法。修复DifyExporter类中的代码格式问题,调整日期参数的默认值。
This commit is contained in:
@@ -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时区)')
|
||||
|
||||
Reference in New Issue
Block a user