scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

add options for /health/readiness and liveliness

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-19 12:13:35 -07:00
parent 4007c6c6e8
commit 19f0f66dbc
1 changed files with 19 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/health_endpoints/_health_endpoints.py
View file

@ -6,7 +6,7 @@ from datetime import datetime, timedelta
from typing import Literal, Optional
import fastapi
from fastapi import APIRouter, Depends, Header, HTTPException, Request, status
from fastapi import APIRouter, Depends, Header, HTTPException, Request, Response, status
import litellm
from litellm._logging import verbose_proxy_logger
@ -488,4 +488,21 @@ async def health_readiness_options():
"Access-Control-Allow-Methods": "GET, OPTIONS",
"Access-Control-Allow-Headers": "*",
}
return {"headers": response_headers}
return Response(headers=response_headers, status_code=200)
@router.options(
"/health/liveliness",
tags=["health"],
dependencies=[Depends(user_api_key_auth)],
)
async def health_liveliness_options():
"""
Options endpoint for health/liveliness check.
"""
response_headers = {
"Allow": "GET, OPTIONS",
"Access-Control-Allow-Methods": "GET, OPTIONS",
"Access-Control-Allow-Headers": "*",
}
return Response(headers=response_headers, status_code=200)