phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat - show key alias on prometheus metrics

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-04 09:57:00 -07:00
parent 65866daafa
commit 09e22d8e33
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/prometheus.py
View file

@ -34,6 +34,7 @@ class PrometheusLogger:
labelnames=[ labelnames=[
"end_user", "end_user",
"hashed_api_key", "hashed_api_key",
"api_key_alias",
"model", "model",
"team", "team",
"team_alias", "team_alias",
@ -47,6 +48,7 @@ class PrometheusLogger:
labelnames=[ labelnames=[
"end_user", "end_user",
"hashed_api_key", "hashed_api_key",
"api_key_alias",
"model", "model",
"team", "team",
"team_alias", "team_alias",
@ -61,6 +63,7 @@ class PrometheusLogger:
labelnames=[ labelnames=[
"end_user", "end_user",
"hashed_api_key", "hashed_api_key",
"api_key_alias",
"model", "model",
"team", "team",
"team_alias", "team_alias",
@ -75,6 +78,7 @@ class PrometheusLogger:
labelnames=[ labelnames=[
"end_user", "end_user",
"hashed_api_key", "hashed_api_key",
"api_key_alias",
"model", "model",
"team", "team",
"team_alias", "team_alias",
@ -204,6 +208,7 @@ class PrometheusLogger:
self.litellm_requests_metric.labels( self.litellm_requests_metric.labels(
end_user_id, end_user_id,
user_api_key, user_api_key,
user_api_key_alias,
model, model,
user_api_team, user_api_team,
user_api_team_alias, user_api_team_alias,
@ -212,6 +217,7 @@ class PrometheusLogger:
self.litellm_spend_metric.labels( self.litellm_spend_metric.labels(
end_user_id, end_user_id,
user_api_key, user_api_key,
user_api_key_alias,
model, model,
user_api_team, user_api_team,
user_api_team_alias, user_api_team_alias,
@ -220,6 +226,7 @@ class PrometheusLogger:
self.litellm_tokens_metric.labels( self.litellm_tokens_metric.labels(
end_user_id, end_user_id,
user_api_key, user_api_key,
user_api_key_alias,
model, model,
user_api_team, user_api_team,
user_api_team_alias, user_api_team_alias,
@ -243,6 +250,7 @@ class PrometheusLogger:
self.litellm_llm_api_failed_requests_metric.labels( self.litellm_llm_api_failed_requests_metric.labels(
end_user_id, end_user_id,
user_api_key, user_api_key,
user_api_key_alias,
model, model,
user_api_team, user_api_team,
user_api_team_alias, user_api_team_alias,