delete_docs (chatchat-space#499)

删除文档的时候重新生成知识库（FAISS）内的数据
byyc0769 · May 30, 2023 · 8477368 · 8477368
1 parent 8d669cd
commit 8477368
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/api.py b/api.py
@@ -175,7 +175,7 @@ async def list_docs(
 
 async def delete_docs(
         knowledge_base_id: str = Query(...,
-                                       description="Knowledge Base Name(注意此方法仅删除上传的文件并不会删除知识库(FAISS)内数据)",
+                                       description="Knowledge Base Name",
                                        example="kb1"),
         doc_name: Optional[str] = Query(
             None, description="doc name", example="doc_name_1.pdf"
@@ -188,18 +188,20 @@ async def delete_docs(
         doc_path = get_file_path(knowledge_base_id, doc_name)
         if os.path.exists(doc_path):
             os.remove(doc_path)
+
+            # 删除上传的文件后重新生成知识库（FAISS）内的数据
+            remain_docs = await list_docs(knowledge_base_id)
+            if len(remain_docs.data) == 0:
+                shutil.rmtree(get_folder_path(knowledge_base_id), ignore_errors=True)
+            else:
+                local_doc_qa.init_knowledge_vector_store(
+                    get_folder_path(knowledge_base_id), get_vs_path(knowledge_base_id)
+                )
+
             return BaseResponse(code=200, msg=f"document {doc_name} delete success")
         else:
             return BaseResponse(code=1, msg=f"document {doc_name} not found")
 
-        remain_docs = await list_docs(knowledge_base_id)
-        remain_docs = remain_docs.json()
-        if len(remain_docs["data"]) == 0:
-            shutil.rmtree(get_folder_path(knowledge_base_id), ignore_errors=True)
-        else:
-            local_doc_qa.init_knowledge_vector_store(
-                get_folder_path(knowledge_base_id), get_vs_path(knowledge_base_id)
-            )
     else:
         shutil.rmtree(get_folder_path(knowledge_base_id))
         return BaseResponse(code=200, msg=f"Knowledge Base {knowledge_base_id} delete success")