From 4bdcb051194ae4f8d6d25f2d813491d5a1ceda9f Mon Sep 17 00:00:00 2001 From: Rob Howley Date: Fri, 9 Sep 2022 10:26:08 -0400 Subject: [PATCH] feat: add health endpoint to py server Signed-off-by: Rob Howley --- sdk/python/feast/feature_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py index bef88923e18..c2596d411c6 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