From 98383b74ad5058b29fce2f763495de580b2ffa24 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Thu, 21 Dec 2023 23:20:52 +0100
Subject: [PATCH] Make a change to test neo4j

---
 api.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/api.py b/api.py
index 051507c98..1d4997601 100644
--- a/api.py
+++ b/api.py
@@ -63,8 +63,6 @@ def health_check():
     return {"status": "OK"}
 
 
-
-
 class Payload(BaseModel):
     payload: Dict[str, Any]
 
@@ -159,6 +157,18 @@ async def user_query_classfier(payload: Payload):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
+
+@app.post("/drop-db")
+async def drop_db(payload: Payload):
+    try:
+        decoded_payload = payload.payload
+        return JSONResponse(content={"response": "dropped"}, status_code=200)
+
+    except Exception as e:
+        return HTTPException(status_code=500, detail=str(e))
+
+
 def start_api_server(host: str = "0.0.0.0", port: int = 8000):
     """
     Start the API server using uvicorn.