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

test(prometheus): cover retiring a tracked series that is already gone

Browse source 
The tracked labelset can outlive the child series it names, for instance
when a cardinality cap evicts it. Pin that retiring it does not break the
emission that triggered the retirement, which was the last uncovered
branch in this change.
This commit is contained in:
DanBrima 2026-08-22 13:44:56 +00:00
parent 359aa24774
commit 4e1d1fac2e
No known key found for this signature in database
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/test_litellm/integrations/test_prometheus_team_rate_limit_metrics.py
View file

@ -472,3 +472,29 @@ def test_retires_the_old_alias_when_the_limit_is_removed_after_a_rename():
assert registry.get_sample_value("litellm_team_rpm_limit", TEAM_LABELS) is None
renamed = {**TEAM_LABELS, "team_alias": "ml-research"}
assert registry.get_sample_value("litellm_team_rpm_limit", renamed) is None
def test_rename_survives_a_tracked_series_that_is_already_gone():
"""
The tracked labelset can outlive the child series it names, for instance
when a cardinality cap evicts it. Retiring it must not break the emission
that triggered the retirement.
"""
registry = CollectorRegistry()
gauge = Gauge("litellm_team_rpm_limit", "doc", labelnames=list(ORIGINAL_LABELNAMES), registry=registry)
logger = _logger_with_real_gauge("litellm_team_rpm_limit", gauge)
headers = {"x-ratelimit-model_per_team-limit-requests": 60}
_set_team_metrics(logger, _payload_with_headers(headers))
gauge.remove(*(TEAM_LABELS[name] for name in ORIGINAL_LABELNAMES))
assert registry.get_sample_value("litellm_team_rpm_limit", TEAM_LABELS) is None
logger._set_team_rate_limit_metrics(
user_api_team="team-abc",
user_api_team_alias="ml-research",
model_group="gpt-4o-mini",
standard_logging_payload=_payload_with_headers(headers),
)
renamed = {**TEAM_LABELS, "team_alias": "ml-research"}
assert registry.get_sample_value("litellm_team_rpm_limit", renamed) == 60