lanyuanxiaoyao/hudi-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(knowledge): 暂时关闭rerank提升性能

Browse Source
This commit is contained in:
v-zhangjc9
2025-06-05 15:21:03 +08:00
parent e359bed97c
commit a35980a5f4
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
service-ai/service-ai-knowledge/src/main/java/com/lanyuanxiaoyao/service/ai/knowledge/service/KnowledgeBaseService.java
View File

@@ -188,13 +188,14 @@ public class KnowledgeBaseService {
.similarityThreshold(threshold) .similarityThreshold(threshold)
.build() .build()
); );
List<org.noear.solon.ai.rag.Document> rerankDocuments = rerankingModel.rerank( // 如果只是一个知识库的话，似乎没有什么rerank的必要...
/* List<org.noear.solon.ai.rag.Document> rerankDocuments = rerankingModel.rerank(
text, text,
documents.stream() documents.stream()
.map(doc -> new org.noear.solon.ai.rag.Document(doc.getId(), doc.getText(), doc.getMetadata(), doc.getScore())) .map(doc -> new org.noear.solon.ai.rag.Document(doc.getId(), doc.getText(), doc.getMetadata(), doc.getScore()))
.toList() .toList()
); ); */
return Lists.immutable.ofAll(rerankDocuments) return Lists.immutable.ofAll(documents)
.collect(org.noear.solon.ai.rag.Document::getContent); .collect(Document::getText);
} }
} }