yc/java-10561
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

关闭知识库召回时Reranking排序

Browse Source
This commit is contained in:
yuance
2026-01-12 15:13:34 +08:00
parent a284472782
commit 45a88a94ea
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/gxwebsoft/ai/util/KnowledgeBaseUtil.java
View File

@@ -34,7 +34,9 @@ public class KnowledgeBaseUtil {
RetrieveRequest retrieveRequest = new RetrieveRequest(); RetrieveRequest retrieveRequest = new RetrieveRequest();
retrieveRequest.setIndexId(indexId); retrieveRequest.setIndexId(indexId);
retrieveRequest.setQuery(query); retrieveRequest.setQuery(query);
retrieveRequest.setDenseSimilarityTopK(null); retrieveRequest.setDenseSimilarityTopK(100);
retrieveRequest.setSparseSimilarityTopK(100);
retrieveRequest.setEnableReranking(false);//开启耗费巨量token
RuntimeOptions runtime = new RuntimeOptions(); RuntimeOptions runtime = new RuntimeOptions();
return client.retrieveWithOptions(workspaceId, retrieveRequest, null, runtime); return client.retrieveWithOptions(workspaceId, retrieveRequest, null, runtime);
} }