diff --git a/tests/entrypoints/openai/test_openai_schema.py b/tests/entrypoints/openai/test_openai_schema.py index 1ccb803a328d..5c585d54c429 100644 --- a/tests/entrypoints/openai/test_openai_schema.py +++ b/tests/entrypoints/openai/test_openai_schema.py @@ -44,6 +44,6 @@ schema = schemathesis.from_pytest_fixture("get_schema") @schema.parametrize() @schema.override(headers={"Content-Type": "application/json"}) -async def test_openapi_stateless(case): +def test_openapi_stateless(case: schemathesis.Case): #No need to verify SSL certificate for localhost - await case.call_and_validate(verify=False) + case.call_and_validate(verify=False) diff --git a/vllm/entrypoints/openai/api_server.py b/vllm/entrypoints/openai/api_server.py index e034eacb24ef..3699b2d6018b 100644 --- a/vllm/entrypoints/openai/api_server.py +++ b/vllm/entrypoints/openai/api_server.py @@ -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)