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

Revert "set litellm_team_budget_reset_at_metric"

Browse source 
This reverts commit 39d959d72a.
This commit is contained in:
Ishaan Jaff 2025-01-23 09:05:14 -08:00
parent ddaed35f54
commit 8479d05d49
1 changed files with 8 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

35
litellm/integrations/prometheus.py
View file

@ -139,27 +139,7 @@ class PrometheusLogger(CustomLogger):
self.litellm_remaining_team_budget_metric = Gauge(
"litellm_remaining_team_budget_metric",
"Remaining budget for team",
labelnames=PrometheusMetricLabels.get_labels(
label_name="litellm_remaining_team_budget_metric"
),
)
# Max Budget for Team
self.litellm_team_max_budget_metric = Gauge(
"litellm_team_max_budget_metric",
"Maximum budget set for team",
labelnames=PrometheusMetricLabels.get_labels(
label_name="litellm_team_max_budget_metric"
),
)
# Team Budget Reset At
self.litellm_team_budget_reset_at_metric = Gauge(
"litellm_team_budget_reset_at_metric",
"Time when the team budget will be reset",
labelnames=PrometheusMetricLabels.get_labels(
label_name="litellm_team_budget_reset_at_metric"
),
labelnames=["team_id", "team_alias"],
)
# Remaining Budget for API Key
@ -169,6 +149,13 @@ class PrometheusLogger(CustomLogger):
labelnames=["hashed_api_key", "api_key_alias"],
)
# Max Budget for Team
self.litellm_team_max_budget_metric = Gauge(
"litellm_team_max_budget_metric",
"Maximum budget set for team",
labelnames=["team_id", "team_alias"],
)
# Max Budget for API Key
self.litellm_api_key_max_budget_metric = Gauge(
"litellm_api_key_max_budget_metric",
@ -1384,12 +1371,6 @@ class PrometheusLogger(CustomLogger):
)
)
if team.budget_reset_at is not None:
self.litellm_team_budget_reset_at_metric.labels(
team.team_id,
team.team_alias or "",
).set(team.budget_reset_at.timestamp())
def prometheus_label_factory(
supported_enum_labels: List[str],