perf(knowledge): 暂时关闭rerank提升性能
This commit is contained in:
@@ -188,13 +188,14 @@ public class KnowledgeBaseService {
|
||||
.similarityThreshold(threshold)
|
||||
.build()
|
||||
);
|
||||
List<org.noear.solon.ai.rag.Document> rerankDocuments = rerankingModel.rerank(
|
||||
// 如果只是一个知识库的话,似乎没有什么rerank的必要...
|
||||
/* List<org.noear.solon.ai.rag.Document> rerankDocuments = rerankingModel.rerank(
|
||||
text,
|
||||
documents.stream()
|
||||
.map(doc -> new org.noear.solon.ai.rag.Document(doc.getId(), doc.getText(), doc.getMetadata(), doc.getScore()))
|
||||
.toList()
|
||||
);
|
||||
return Lists.immutable.ofAll(rerankDocuments)
|
||||
.collect(org.noear.solon.ai.rag.Document::getContent);
|
||||
); */
|
||||
return Lists.immutable.ofAll(documents)
|
||||
.collect(Document::getText);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user