Signed-off-by: Ximo Guanter <ximo.guanter@gmail.com> Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -390,10 +390,10 @@ def engine_client(request: Request) -> EngineClient:
|
||||
|
||||
|
||||
@router.get("/health")
|
||||
async def health(raw_request: Request) -> Response:
|
||||
async def health(raw_request: Request) -> JSONResponse:
|
||||
"""Health check."""
|
||||
await engine_client(raw_request).check_health()
|
||||
return Response(status_code=200)
|
||||
return JSONResponse(content={}, status_code=200)
|
||||
|
||||
|
||||
@router.get("/load")
|
||||
@@ -415,7 +415,7 @@ async def get_server_load_metrics(request: Request):
|
||||
|
||||
|
||||
@router.api_route("/ping", methods=["GET", "POST"])
|
||||
async def ping(raw_request: Request) -> Response:
|
||||
async def ping(raw_request: Request) -> JSONResponse:
|
||||
"""Ping check. Endpoint required for SageMaker"""
|
||||
return await health(raw_request)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user