From 6a3e217fdfd670d97bb09d61f02042c1d2396d86 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 23 Jan 2025 08:54:19 -0800 Subject: [PATCH] add litellm_team_budget_reset_at_metric --- litellm/types/integrations/prometheus.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/litellm/types/integrations/prometheus.py b/litellm/types/integrations/prometheus.py index d8415945bf..ea32ee9e8b 100644 --- a/litellm/types/integrations/prometheus.py +++ b/litellm/types/integrations/prometheus.py @@ -81,6 +81,9 @@ DEFINED_PROMETHEUS_METRICS = Literal[ "litellm_output_tokens_metric", "litellm_deployment_successful_fallbacks", "litellm_deployment_failed_fallbacks", + "litellm_remaining_team_budget_metric", + "litellm_team_max_budget_metric", + "litellm_team_budget_reset_at_metric", ] @@ -186,6 +189,21 @@ class PrometheusMetricLabels: litellm_deployment_failed_fallbacks = litellm_deployment_successful_fallbacks + litellm_remaining_team_budget_metric = [ + UserAPIKeyLabelNames.TEAM.value, + UserAPIKeyLabelNames.TEAM_ALIAS.value, + ] + + litellm_team_max_budget_metric = [ + UserAPIKeyLabelNames.TEAM.value, + UserAPIKeyLabelNames.TEAM_ALIAS.value, + ] + + litellm_team_budget_reset_at_metric = [ + UserAPIKeyLabelNames.TEAM.value, + UserAPIKeyLabelNames.TEAM_ALIAS.value, + ] + @staticmethod def get_labels(label_name: DEFINED_PROMETHEUS_METRICS) -> List[str]: default_labels = getattr(PrometheusMetricLabels, label_name)