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
   }
 }