diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py index bef88923e1..c2596d411c 100644 --- 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