diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
index 916b5cd3c..6b4108585 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
@@ -295,25 +295,23 @@ public class KnowledgeBaseServiceImpl implements KnowledgeBaseService {
         if (baseDO == null) {
             throw exception(KNOWLEDGE_BASE_NOT_EXISTS);
         }
+        // 设置默认值
         Integer topK = 4;
+        Double score = 0.2;
+
+        // 处理 topK 的逻辑
         if (baseDO.getTopK() == null || baseDO.getTopK() <= 0) {
             testReqVO.setTopK(topK);
-            if (testReqVO.getTopK() == null){
-                testReqVO.setTopK(baseDO.getTopK());
-            }
         } else {
-            topK = testReqVO.getTopK();
+            topK = testReqVO.getTopK() != null ? testReqVO.getTopK() : baseDO.getTopK();
             testReqVO.setTopK(topK);
-            if (testReqVO.getScore() == null){
-                testReqVO.setScore(baseDO.getScore());
-            }
         }
 
-        Double score = 0.2;
-        if (baseDO.getScore() == null || baseDO.getTopK() <= 0.0 || baseDO.getScore() > 1) {
+        // 处理 score 的逻辑
+        if (baseDO.getScore() == null || baseDO.getScore() <= 0.0 || baseDO.getScore() > 1) {
             testReqVO.setScore(score);
         } else {
-            score = testReqVO.getScore();
+            score = testReqVO.getScore() != null ? testReqVO.getScore() : baseDO.getScore();
             testReqVO.setScore(score);
         }