diff --git a/service-ai/service-ai-knowledge/src/main/java/com/lanyuanxiaoyao/service/ai/knowledge/service/KnowledgeBaseService.java b/service-ai/service-ai-knowledge/src/main/java/com/lanyuanxiaoyao/service/ai/knowledge/service/KnowledgeBaseService.java index cf85fb8..411bb59 100644 --- a/service-ai/service-ai-knowledge/src/main/java/com/lanyuanxiaoyao/service/ai/knowledge/service/KnowledgeBaseService.java +++ b/service-ai/service-ai-knowledge/src/main/java/com/lanyuanxiaoyao/service/ai/knowledge/service/KnowledgeBaseService.java @@ -188,13 +188,14 @@ public class KnowledgeBaseService { .similarityThreshold(threshold) .build() ); - List rerankDocuments = rerankingModel.rerank( + // 如果只是一个知识库的话,似乎没有什么rerank的必要... + /* List 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); } } \ No newline at end of file