From 5c92f23e6a0c5325fa33b3edbb54703b29fdd047 Mon Sep 17 00:00:00 2001
From: Saurab-Shrestha <shresthasaurab030@gmail.com>
Date: Sun, 11 Feb 2024 11:41:15 +0545
Subject: [PATCH] small update in delete

---
 private_gpt/server/ingest/ingest_router.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/private_gpt/server/ingest/ingest_router.py b/private_gpt/server/ingest/ingest_router.py
index c591acbd..875b680f 100644
--- a/private_gpt/server/ingest/ingest_router.py
+++ b/private_gpt/server/ingest/ingest_router.py
@@ -151,7 +151,8 @@ def delete_file(
         for doc_id in doc_ids:
             service.delete(doc_id)
         document = crud.documents.get_by_filename(db,file_name=filename)
-        crud.documents.remove(db=db, id=document.id)
+        if document:
+            crud.documents.remove(db=db, id=document.id)
         return {"status": "SUCCESS", "message": f"{filename}' successfully deleted."}
     except Exception as e:
         logger.error(