scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 175

fix(proxy): keep day-zero deprecation emails when the marker lifetime is under a day

Browse source
This commit is contained in:
Rameshwaran Mohan 2026-09-15 05:39:46 +05:30
parent 185c8ce3f9
commit 121e3ba7e2
2 changed files with 19 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/common_utils/model_deprecation_notifications.py
View file

@ -93,7 +93,7 @@ def select_milestone(days_until: int, thresholds: Sequence[int]) -> int | None:
def _max_age_days(alerting_args: SlackAlertingArgs) -> int:
return alerting_args.model_deprecation_email_ttl // (24 * 60 * 60)
return max(1, alerting_args.model_deprecation_email_ttl // (24 * 60 * 60))
def _reached(

18
tests/test_litellm/proxy/common_utils/test_model_deprecation_notifications.py
View file

@ -423,6 +423,11 @@ FRESH_DEPLOYMENT: Final = {
"litellm_params": {"model": "openai/fresh-model-with-no-cost-map-entry"},
"model_info": {"id": "x"},
}
TODAY_DEPLOYMENT: Final = {
"model_name": "today-alias",
"litellm_params": {"model": "openai/today-model"},
"model_info": {"id": "5", "deprecation_date": TODAY.isoformat(), "litellm_provider": "openai"},
}
FAR_DEPLOYMENT: Final = {
"model_name": "far-alias",
"litellm_params": {"model": "openai/far-model"},
@ -669,6 +674,19 @@ class TestSendModelDeprecationEmails:
assert lock.calls == []
assert await cache.async_get_cache(key=SlackAlertingCacheKeys.deprecation_email_pass_key.value) is not None
@pytest.mark.asyncio
async def test_should_still_email_the_day_zero_milestone_when_the_marker_lifetime_is_under_a_day(self):
deliverer: Final = _Deliverer()
ctx: Final = _context(
_router([TODAY_DEPLOYMENT, DEAD_DEPLOYMENT]),
_prisma(TWO_TEAMS[:1], TWO_ADMINS),
deliverer,
alerting_args=SlackAlertingArgs(model_deprecation_email_ttl=3600),
)
assert await send_model_deprecation_emails(ctx) == 1
assert len(deliverer.sent) == 1
@pytest.mark.asyncio
async def test_should_keep_going_and_leave_keys_unstamped_when_one_team_fails(self):
cache: Final = DualCache()