diff --git a/src/main/java/com/gxwebsoft/ai/util/KnowledgeBaseUtil.java b/src/main/java/com/gxwebsoft/ai/util/KnowledgeBaseUtil.java index 4724288..ea0f495 100644 --- a/src/main/java/com/gxwebsoft/ai/util/KnowledgeBaseUtil.java +++ b/src/main/java/com/gxwebsoft/ai/util/KnowledgeBaseUtil.java @@ -34,7 +34,9 @@ public class KnowledgeBaseUtil { RetrieveRequest retrieveRequest = new RetrieveRequest(); retrieveRequest.setIndexId(indexId); retrieveRequest.setQuery(query); - retrieveRequest.setDenseSimilarityTopK(null); + retrieveRequest.setDenseSimilarityTopK(100); + retrieveRequest.setSparseSimilarityTopK(100); + retrieveRequest.setEnableReranking(false);//开启耗费巨量token RuntimeOptions runtime = new RuntimeOptions(); return client.retrieveWithOptions(workspaceId, retrieveRequest, null, runtime); }