From fb4f2a1e6039857da73d9f18ff928d80bedf5e80 Mon Sep 17 00:00:00 2001
From: 18586361686 <1207669650@qq.com>
Date: 星期六, 12 四月 2025 18:45:44 +0800
Subject: [PATCH] fix: Dynamically set the number of knowledge base entries obtained

---
 aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java b/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java
index 7d8e67e..2e0ba49 100644
--- a/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java
+++ b/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java
@@ -95,7 +95,7 @@
                         }
                         documentStore.setEmbeddingModel(aiLlm.toLlm());
                         SearchWrapper wrapper = new SearchWrapper();
-                        wrapper.setMaxResults(Integer.valueOf(10));
+                        wrapper.setMaxResults(Integer.valueOf(limit));
                         wrapper.setText(keyword);
                         StoreOptions options = StoreOptions.ofCollectionName(aiKnowledge.getVectorStoreCollection());
 

--
Gitblit v1.8.0