From 09e22d8e3368c2a7c5beff8c7faae99e681bd0c8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 4 Jul 2024 09:57:00 -0700 Subject: [PATCH] feat - show key alias on prometheus metrics --- litellm/integrations/prometheus.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index 6cd7469079..4a271d6e00 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -34,6 +34,7 @@ class PrometheusLogger: labelnames=[ "end_user", "hashed_api_key", + "api_key_alias", "model", "team", "team_alias", @@ -47,6 +48,7 @@ class PrometheusLogger: labelnames=[ "end_user", "hashed_api_key", + "api_key_alias", "model", "team", "team_alias", @@ -61,6 +63,7 @@ class PrometheusLogger: labelnames=[ "end_user", "hashed_api_key", + "api_key_alias", "model", "team", "team_alias", @@ -75,6 +78,7 @@ class PrometheusLogger: labelnames=[ "end_user", "hashed_api_key", + "api_key_alias", "model", "team", "team_alias", @@ -204,6 +208,7 @@ class PrometheusLogger: self.litellm_requests_metric.labels( end_user_id, user_api_key, + user_api_key_alias, model, user_api_team, user_api_team_alias, @@ -212,6 +217,7 @@ class PrometheusLogger: self.litellm_spend_metric.labels( end_user_id, user_api_key, + user_api_key_alias, model, user_api_team, user_api_team_alias, @@ -220,6 +226,7 @@ class PrometheusLogger: self.litellm_tokens_metric.labels( end_user_id, user_api_key, + user_api_key_alias, model, user_api_team, user_api_team_alias, @@ -243,6 +250,7 @@ class PrometheusLogger: self.litellm_llm_api_failed_requests_metric.labels( end_user_id, user_api_key, + user_api_key_alias, model, user_api_team, user_api_team_alias,