From dd5229d5bae56e29a0c8a9fdf1c9c798335e75c2 Mon Sep 17 00:00:00 2001
From: Chen Tao <70054568+eeee0717@users.noreply.github.com>
Date: Mon, 12 May 2025 20:32:28 +0800
Subject: [PATCH] feat(knowledge): adjust default top-n to 10 (#5919)

---
 src/main/reranker/BaseReranker.ts                               | 2 +-
 .../src/pages/knowledge/components/KnowledgeSettingsPopup.tsx   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/reranker/BaseReranker.ts b/src/main/reranker/BaseReranker.ts
index 5a8bd6ee2a..a88d0883ae 100644
--- a/src/main/reranker/BaseReranker.ts
+++ b/src/main/reranker/BaseReranker.ts
@@ -38,7 +38,7 @@ export default abstract class BaseReranker {
   protected getRerankRequestBody(query: string, searchResults: ExtractChunkData[]) {
     const provider = this.base.rerankModelProvider
     const documents = searchResults.map((doc) => doc.pageContent)
-    const topN = this.base.topN || 5
+    const topN = this.base.topN || 10
 
     if (provider === 'voyageai') {
       return {
diff --git a/src/renderer/src/pages/knowledge/components/KnowledgeSettingsPopup.tsx b/src/renderer/src/pages/knowledge/components/KnowledgeSettingsPopup.tsx
index bf318fb265..f409990094 100644
--- a/src/renderer/src/pages/knowledge/components/KnowledgeSettingsPopup.tsx
+++ b/src/renderer/src/pages/knowledge/components/KnowledgeSettingsPopup.tsx
@@ -291,7 +291,7 @@ const PopupContainer: React.FC<Props> = ({ base: _base, resolve }) => {
             rules={[
               {
                 validator(_, value) {
-                  if (value && (value < 0 || value > 10)) {
+                  if (value && (value < 0 || value > 30)) {
                     return Promise.reject(new Error(t('knowledge.topN_too_large_or_small')))
                   }
                   return Promise.resolve()