From a6be0cc135d805925380f68e59753b82b1ca25a4 Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Fri, 24 Oct 2025 14:28:27 +0800
Subject: [PATCH] feat: refresh knowledge base and document after uploading a
 document

---
 astrbot/core/knowledge_base/kb_helper.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/astrbot/core/knowledge_base/kb_helper.py b/astrbot/core/knowledge_base/kb_helper.py
index e6cef61be..32730f4c8 100644
--- a/astrbot/core/knowledge_base/kb_helper.py
+++ b/astrbot/core/knowledge_base/kb_helper.py
@@ -180,7 +180,8 @@ class KBHelper:
 
             vec_db: FaissVecDB = self.vec_db  # type: ignore
             await self.kb_db.update_kb_stats(kb_id=self.kb.kb_id, vec_db=vec_db)
-
+            await self.refresh_kb()
+            await self.refresh_document(doc_id)
             return doc
         except Exception as e:
             logger.error(f"上传文档失败: {e}")