scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

refactor(proxy): build the team member alert threshold config without mutable collections

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
ryan 2026-09-24 23:21:18 +00:00
parent b602c94839
commit 280abea2a8
1 changed files with 25 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
litellm/proxy/auth/auth_checks.py
View file

@ -18,7 +18,7 @@ from types import MappingProxyType
from typing import TYPE_CHECKING, Any, Final, Generic, Literal, Optional, Protocol, TypeAlias
from fastapi import HTTPException, Request, status
from pydantic import BaseModel, TypeAdapter
from pydantic import BaseModel, TypeAdapter, ValidationError
from typing_extensions import NotRequired, ReadOnly, Required, TypedDict, Unpack
import litellm
@ -5598,6 +5598,27 @@ async def _virtual_key_max_budget_alert_check(
TEAM_MEMBER_MAX_BUDGET_ALERT_EMAILS_KEY: Final = "team_member_max_budget_alert_emails"
_TEAM_MEMBER_ALERT_CONFIG_ADAPTER: Final[TypeAdapter[Mapping[str, object]]] = TypeAdapter(Mapping[str, object])
def _is_valid_alert_threshold_pct(pct: str) -> bool:
return pct.isdigit() and len(pct) <= 3 and 1 <= int(pct) <= 100
def _alert_recipients(raw: object) -> Sequence[str] | None:
if isinstance(raw, str) or isinstance(raw, Sequence):
return _parse_email_list(raw)
return None
def _valid_alert_threshold_config(raw_config: object) -> Mapping[str, str | Sequence[object] | None] | None:
try:
config: Final = _TEAM_MEMBER_ALERT_CONFIG_ADAPTER.validate_python(raw_config)
except ValidationError:
return None
return MappingProxyType(
{pct: _alert_recipients(emails) for pct, emails in config.items() if _is_valid_alert_threshold_pct(pct)}
)
def _team_member_max_budget_alert_check(
@ -5611,16 +5632,10 @@ def _team_member_max_budget_alert_check(
spend: float,
max_budget: float,
) -> None:
raw_config: Final = (team_metadata or {}).get(TEAM_MEMBER_MAX_BUDGET_ALERT_EMAILS_KEY)
merged_config: Final = _merge_budget_alert_email_configs(
global_cfg=None,
per_key_cfg=raw_config if isinstance(raw_config, Mapping) else None,
raw_config: Final = team_metadata.get(TEAM_MEMBER_MAX_BUDGET_ALERT_EMAILS_KEY) if team_metadata else None
alert_email_config: Final = _merge_budget_alert_email_configs(
global_cfg=None, per_key_cfg=_valid_alert_threshold_config(raw_config)
)
alert_email_config: Final = {
pct: emails
for pct, emails in (merged_config or {}).items()
if pct.isdigit() and len(pct) <= 3 and 1 <= int(pct) <= 100
}
if not alert_email_config or spend <= 0:
return
min_pct: Final = min(int(pct) for pct in alert_email_config)