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

2025-07-10 08:36:12 +08:00
parent ec3db656a5
commit 23f522dde5
4 changed files with 28 additions and 15 deletions
@@ -93,7 +93,8 @@ async def retrieve(request: RetrieveRequest):
        results = await dify_query_retrieval.retrieve_api_async(
            request.original_query, 
            query_list, 
-            data_set_list
+            data_set_list,
+            top_k=3
        )
        end_time = time.time()