tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-09 14:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files
litellm/types/integrations/prometheus.py
+18
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)