diff --git a/src/renderer/src/services/KnowledgeService.ts b/src/renderer/src/services/KnowledgeService.ts
index 48a01adb..2858c0cf 100644
--- a/src/renderer/src/services/KnowledgeService.ts
+++ b/src/renderer/src/services/KnowledgeService.ts
@@ -90,8 +90,10 @@ export const getKnowledgeReferences = async (base: KnowledgeBase, message: Messa
     })
   )
 
+  const documentCount = base.documentCount || 6
+
   const references = await Promise.all(
-    take(_searchResults, 6).map(async (item, index) => {
+    take(_searchResults, documentCount).map(async (item, index) => {
       const baseItem = base.items.find((i) => i.uniqueId === item.metadata.uniqueLoaderId)
       return {
         id: index,