gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improved error handling for document deletion

Browse source 
Added HTTPException for not_found status
Added HTTPException for fail status
This commit is contained in:
yangdx 2025-06-24 01:12:25 +08:00
parent fd69c895cd
commit 5ae945c1e5
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
lightrag/api/routers/document_routes.py
View file

@ -1438,14 +1438,19 @@ def create_document_routes(
try: try:
result = await rag.adelete_by_doc_id(doc_id) result = await rag.adelete_by_doc_id(doc_id)
response_data = {
"doc_id": result.doc_id,
"message": result.message,
"status": result.status,
}
if "history_messages" in pipeline_status: if "history_messages" in pipeline_status:
pipeline_status["history_messages"].append(result.message) pipeline_status["history_messages"].append(result.message)
return DeleteDocByIdResponse(**response_data)
if result.status == "not_found":
raise HTTPException(status_code=404, detail=result.message)
if result.status == "fail":
raise HTTPException(status_code=500, detail=result.message)
return DeleteDocByIdResponse(
doc_id=result.doc_id,
message=result.message,
status=result.status,
)
except Exception as e: except Exception as e:
error_msg = f"Error deleting document {doc_id}: {str(e)}" error_msg = f"Error deleting document {doc_id}: {str(e)}"