From 07cb72a5725d5335600f1c30d9442c4aa2fe5d8f Mon Sep 17 00:00:00 2001 From: cdietrich Date: Thu, 17 Feb 2022 16:16:43 +0100 Subject: [PATCH] add live and ready probes --- src/run_service.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/run_service.py b/src/run_service.py index 17f7492..81ac972 100644 --- a/src/run_service.py +++ b/src/run_service.py @@ -57,6 +57,18 @@ def run_server(resume): app = Flask(__name__) + @app.route("/ready", methods=["GET"]) + def ready(): + resp = jsonify("OK") + resp.status_code = 200 + return resp + + @app.route("/health", methods=["GET"]) + def healthy(): + resp = jsonify("OK") + resp.status_code = 200 + return resp + @app.route("/", methods=["POST"]) def predict_request():