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)