github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-23 20:26:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rag): Fix bug fail to delete files when deleting the knowledge space (#2185)

Browse Source
This commit is contained in:
toralee 2024-12-11 14:45:47 +08:00 committed by GitHub
parent a801a973bf
commit 23aedea092
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
dbgpt/app/knowledge/api.py
View File

@ -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}")