scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

add litellm_team_budget_reset_at_metric

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-23 08:54:19 -08:00
parent f201888f69
commit 6a3e217fdf
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/types/integrations/prometheus.py
View file

@ -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)