From a35980a5f488dc839fcd212573b4ce6f800ed76b Mon Sep 17 00:00:00 2001 From: v-zhangjc9 Date: Thu, 5 Jun 2025 15:21:03 +0800 Subject: [PATCH] =?UTF-8?q?perf(knowledge):=20=E6=9A=82=E6=97=B6=E5=85=B3?= =?UTF-8?q?=E9=97=ADrerank=E6=8F=90=E5=8D=87=E6=80=A7=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ai/knowledge/service/KnowledgeBaseService.java | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) 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