From df43cb7a90dd3cd3a055d3fd283ed7ebf4baf6a0 Mon Sep 17 00:00:00 2001
From: one <wangan.cs@gmail.com>
Date: Sun, 13 Jul 2025 20:22:02 +0800
Subject: [PATCH] fix(Knowledge): pass searchResultCount to embed-js (#8118)

---
 src/main/services/KnowledgeService.ts         | 4 +++-
 src/renderer/src/services/KnowledgeService.ts | 5 ++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/main/services/KnowledgeService.ts b/src/main/services/KnowledgeService.ts
index 2e5f3a44d0..8b2f22fc20 100644
--- a/src/main/services/KnowledgeService.ts
+++ b/src/main/services/KnowledgeService.ts
@@ -125,7 +125,8 @@ class KnowledgeService {
     apiKey,
     apiVersion,
     baseURL,
-    dimensions
+    dimensions,
+    documentCount
   }: KnowledgeBaseParams): Promise<RAGApplication> => {
     let ragApplication: RAGApplication
     const embeddings = new Embeddings({
@@ -141,6 +142,7 @@ class KnowledgeService {
         .setModel('NO_MODEL')
         .setEmbeddingModel(embeddings)
         .setVectorDatabase(new LibSqlDb({ path: path.join(this.storageDir, id) }))
+        .setSearchResultCount(documentCount || 30)
         .build()
     } catch (e) {
       Logger.error(e)
diff --git a/src/renderer/src/services/KnowledgeService.ts b/src/renderer/src/services/KnowledgeService.ts
index 4876de0a02..71fb67eae8 100644
--- a/src/renderer/src/services/KnowledgeService.ts
+++ b/src/renderer/src/services/KnowledgeService.ts
@@ -51,9 +51,8 @@ export const getKnowledgeBaseParams = (base: KnowledgeBase): KnowledgeBaseParams
     rerankModelProvider: rerankProvider.name.toLowerCase(),
     // topN: base.topN,
     // preprocessing: base.preprocessing,
-    preprocessOrOcrProvider: base.preprocessOrOcrProvider
-
-    // topN: base.topN
+    preprocessOrOcrProvider: base.preprocessOrOcrProvider,
+    documentCount: base.documentCount
   }
 }