From ef432171172dbbe46e7ba781bd99f06a06d630c0 Mon Sep 17 00:00:00 2001
From: Waterwzy <2916963017@qq.com>
Date: Sun, 22 Mar 2026 13:36:18 +0800
Subject: [PATCH] feat: skip search when the entire knowledge base is empty
 (#6750)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* feat:增加知识库全为空时的跳过检索

* apply bot suggestions

* style:reformat code
---
 astrbot/core/astr_main_agent_resources.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/astrbot/core/astr_main_agent_resources.py b/astrbot/core/astr_main_agent_resources.py
index d0ef33b81..09e77b4cb 100644
--- a/astrbot/core/astr_main_agent_resources.py
+++ b/astrbot/core/astr_main_agent_resources.py
@@ -33,6 +33,7 @@ from astrbot.core.computer.tools import (
     RunBrowserSkillTool,
     SyncSkillReleaseTool,
 )
+from astrbot.core.knowledge_base.kb_helper import KBHelper
 from astrbot.core.message.message_event_result import MessageChain
 from astrbot.core.platform.message_session import MessageSession
 from astrbot.core.star.context import Context
@@ -396,6 +397,18 @@ class SendMessageToUserTool(FunctionTool[AstrAgentContext]):
         return f"Message sent to session {target_session}"
 
 
+def check_all_kb(kb_list: list[KBHelper | None]) -> bool:
+    """检查是否所有的知识库都为空
+    Args:
+        kb_list: 所选的知识库
+    Returns:
+        bool: 是否全为空
+    """
+    return not any(
+        kb and (kb.kb.doc_count != 0 or kb.kb.chunk_count != 0) for kb in kb_list
+    )
+
+
 async def retrieve_knowledge_base(
     query: str,
     umo: str,
@@ -454,6 +467,12 @@ async def retrieve_knowledge_base(
     if not kb_names:
         return
 
+    all_kbs = [await kb_mgr.get_kb_by_name(kb) for kb in kb_names]
+
+    if check_all_kb(all_kbs):
+        logger.debug("所配置的所有知识库全为空，跳过检索过程")
+        return
+
     logger.debug(f"[知识库] 开始检索知识库，数量: {len(kb_names)}, top_k={top_k}")
     kb_context = await kb_mgr.retrieve(
         query=query,