scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix(prometheus): pick the series tracker with isinstance so every path of _admits returns

Browse source
This commit is contained in:
mateo-berri 2026-10-03 13:33:08 -07:00
parent 27b2ed7e25
commit ee8600b05b
1 changed files with 7 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/integrations/prometheus.py
View file

@ -239,17 +239,13 @@ class _LabeledMetric:
assert_never(self._tracker)
def _admits(self, kept_values: tuple[str, ...]) -> bool:
match self._tracker:
case SharedPrometheusSeriesAdmissions():
return self._limits.max_series is None or self._tracker.admit_series(
metric_name=self._metric_name, label_values=kept_values, max_series=self._limits.max_series
)
case BoundedPrometheusSeriesTracker():
return self._tracker.admit_series(
metric=self._metric, metric_name=self._metric_name, label_values=kept_values, limits=self._limits
)
case _:
return assert_never(self._tracker)
if isinstance(self._tracker, SharedPrometheusSeriesAdmissions):
return self._limits.max_series is None or self._tracker.admit_series(
metric_name=self._metric_name, label_values=kept_values, max_series=self._limits.max_series
)
return self._tracker.admit_series(
metric=self._metric, metric_name=self._metric_name, label_values=kept_values, limits=self._limits
)
def _kept_values(self, values: tuple[object, ...]) -> tuple[str, ...]:
return tuple(