feat: Add health endpoint to py server (#3202)

feat: add health endpoint to py server Signed-off-by: Rob Howley <[email protected]> Signed-off-by: Rob Howley <[email protected]> Co-authored-by: Rob Howley <[email protected]>
feast-dev · Sep 12, 2022 · 43222f2 · 43222f2
1 parent cdf0faf
commit 43222f2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py
@@ -4,7 +4,7 @@
 
 import pandas as pd
 import uvicorn
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Request, Response, status
 from fastapi.logger import logger
 from fastapi.params import Depends
 from google.protobuf.json_format import MessageToDict, Parse
@@ -124,6 +124,10 @@ def write_to_online_store(body=Depends(get_body)):
             # Raise HTTPException to return the error message to the client
             raise HTTPException(status_code=500, detail=str(e))
 
+    @app.get("/health")
+    def health():
+        return Response(status_code=status.HTTP_200_OK)
+
     return app