From 5ae945c1e540b59130a6b2066793c3b89658950e Mon Sep 17 00:00:00 2001 From: yangdx Date: Tue, 24 Jun 2025 01:12:25 +0800 Subject: [PATCH] Improved error handling for document deletion Added HTTPException for not_found status Added HTTPException for fail status --- lightrag/api/routers/document_routes.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/lightrag/api/routers/document_routes.py b/lightrag/api/routers/document_routes.py index 813bdc48..01a0d00f 100644 --- a/lightrag/api/routers/document_routes.py +++ b/lightrag/api/routers/document_routes.py @@ -1438,14 +1438,19 @@ def create_document_routes( try: 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: 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: error_msg = f"Error deleting document {doc_id}: {str(e)}"