支持关闭知识库索引

Author: BetterAndBetterII

config.py

@@ -82,6 +82,7 @@ class Settings(BaseSettings):
     # 全局模式
     GLOBAL_MODE: Literal["public", "private"] = "public"
     GLOBAL_LLM: Literal["public", "private"] = "public"
+    DISABLE_KB_INDEXING: bool = False
 
     # Loki设置
     LOKI_URL: Optional[str] = None

pipeline/document_pipeline.py

@@ -415,6 +415,12 @@ async def stage_3(self, document: Document, db: Session) -> Document:
     async def stage_4(self, document: Document, db: Session) -> Document:
         """保存阶段：将处理后的内容保存到知识库."""
         logger.info(f"开始保存文档到知识库: {document.filename}")
+        # 临时关闭索引/保存到知识库阶段：
+        # 1) 可设置环境变量 DISABLE_KB_INDEXING=true
+        # 2) 或者直接注释掉下方调用 rag.upload_document(document)
+        if settings.DISABLE_KB_INDEXING:
+            logger.info("[stage_4] 索引/保存已被禁用 (DISABLE_KB_INDEXING)，跳过保存到知识库阶段")
+            return document
         if settings.GLOBAL_MODE == "public":
             namespace = "public"
         else:

rag/knowledgebase.py

@@ -150,7 +150,12 @@ async def upload_text(self, text: str, doc_id: str, metadata: dict = None):
         # 将用户提供的文本封装为一个Document对象
         doc = Document(text=text, id_=doc_id, metadata=metadata)
 
-        # 执行文档注入管道
+        # 如果希望临时跳过索引/保存到知识库，可注释掉下一行 self.pipeline.arun(...) 或开启环境变量开关。
+        if self.disable_indexing:
+            logger.info("[upload_text] 索引已被禁用 (DISABLE_KB_INDEXING)，跳过保存知识库阶段")
+            return doc_id
+
+        # 执行文档注入管道（索引与落库）
         await self.pipeline.arun(documents=[doc])
         return doc_id
 
@@ -178,7 +183,12 @@ async def upload_document(self, document: DBDocument):
             excluded_llm_metadata_keys=["owner"],
         )
 
-        # 执行文档注入管道
+        # 如果希望临时跳过索引/保存到知识库，可注释掉下一行 self.pipeline.arun(...) 或开启环境变量开关。
+        if self.disable_indexing:
+            logger.info("[upload_document] 索引已被禁用 (DISABLE_KB_INDEXING)，跳过保存知识库阶段")
+            return doc_id
+
+        # 执行文档注入管道（索引与落库）
         await self.pipeline.arun(documents=[doc])
         return doc_id