Make a change to test neo4j

This commit is contained in:
Vasilije 2023-12-21 23:20:52 +01:00
parent 523f893c5b
commit 98383b74ad

14
api.py
View file

@ -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.