Skip to content

Commit

Permalink
fix(rag): Fix bug fail to delete files when deleting the knowledge sp…
Browse files Browse the repository at this point in the history
…ace (#2185)
  • Loading branch information
toralee authored Dec 11, 2024
1 parent a801a97 commit 23aedea
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions dbgpt/app/knowledge/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}")
Expand Down

0 comments on commit 23aedea

Please sign in to comment.