Add missing content type headers to /ping and /health (#17036) (#17786)

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:
Ximo Guanter
2025-05-10 08:13:32 +02:00
committed by GitHub
parent 246e3e0a36
commit fc4441a4ee
2 changed files with 5 additions and 5 deletions

View File

@@ -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)