phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

correct use of healthy / unhealthy

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-06 13:48:30 +05:30
parent 578f9c91af
commit c133ba25d9
3 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/my-website/docs/proxy/team_logging.md
View file

@ -338,7 +338,7 @@ Response when logging callbacks are setup correctly:
Response when logging callbacks are not setup correctly: Response when logging callbacks are not setup correctly:
```json ```json
{ {
"key": "healthy", "key": "unhealthy",
"logging_callbacks": { "logging_callbacks": {
"callbacks": [ "callbacks": [
"gcs_bucket" "gcs_bucket"

4
litellm/proxy/_types.py
View file

@ -1927,10 +1927,10 @@ class CurrentItemRateLimit(TypedDict):
class LoggingCallbackStatus(TypedDict, total=False): class LoggingCallbackStatus(TypedDict, total=False):
callbacks: List[str] callbacks: List[str]
status: str status: Literal["healthy", "unhealthy"]
details: Optional[str] details: Optional[str]
class KeyHealthResponse(TypedDict, total=False): class KeyHealthResponse(TypedDict, total=False):
key: str key: Literal["healthy", "unhealthy"]
logging_callbacks: Optional[LoggingCallbackStatus] logging_callbacks: Optional[LoggingCallbackStatus]

8
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -1496,7 +1496,7 @@ async def key_health(
Response when logging callbacks are not setup correctly: Response when logging callbacks are not setup correctly:
```json ```json
{ {
"key": "healthy", "key": "unhealthy",
"logging_callbacks": { "logging_callbacks": {
"callbacks": [ "callbacks": [
"gcs_bucket" "gcs_bucket"
@ -1525,6 +1525,10 @@ async def key_health(
) )
health_status["logging_callbacks"] = logging_statuses health_status["logging_callbacks"] = logging_statuses
# Check if any logging callback is unhealthy
if logging_statuses.get("status") == "unhealthy":
health_status["key"] = "unhealthy"
return KeyHealthResponse(**health_status) return KeyHealthResponse(**health_status)
except Exception as e: except Exception as e:
@ -1591,7 +1595,7 @@ async def test_key_logging(
except Exception as e: except Exception as e:
return LoggingCallbackStatus( return LoggingCallbackStatus(
callbacks=logging_callbacks, callbacks=logging_callbacks,
status="error", status="unhealthy",
details=f"Logging test failed: {str(e)}", details=f"Logging test failed: {str(e)}",
) )