From 23aedea0928db031c2ecd1e87b92db167c714211 Mon Sep 17 00:00:00 2001 From: toralee Date: Wed, 11 Dec 2024 14:45:47 +0800 Subject: [PATCH] fix(rag): Fix bug fail to delete files when deleting the knowledge space (#2185) --- dbgpt/app/knowledge/api.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/dbgpt/app/knowledge/api.py b/dbgpt/app/knowledge/api.py index 8dad87459..49e49fbd1 100644 --- a/dbgpt/app/knowledge/api.py +++ b/dbgpt/app/knowledge/api.py @@ -97,7 +97,20 @@ def space_list(request: KnowledgeSpaceRequest): @router.post("/knowledge/space/delete") def space_delete(request: KnowledgeSpaceRequest): print(f"/space/delete params:") + print(request.name) try: + # delete Files in 'pilot/data/ + safe_space_name = os.path.basename(request.name) + + # obtain absolute paths of uploaded space-docfiles + space_dir = os.path.abspath( + os.path.join(KNOWLEDGE_UPLOAD_ROOT_PATH, safe_space_name) + ) + try: + if os.path.exists(space_dir): + shutil.rmtree(space_dir) + except Exception as e: + print(e) return Result.succ(knowledge_space_service.delete_space(request.name)) except Exception as e: return Result.failed(code="E000X", msg=f"space delete error {e}")