From 52d5e02d8be015b97dd829d0fecc1e3b49e6c86d Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 24 Apr 2024 08:08:42 -0700 Subject: [PATCH] fix(prometheus.py): fix user-id get for prometheus --- litellm/integrations/prometheus.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/integrations/prometheus.py b/litellm/integrations/prometheus.py index b9d89fd3e13..e3c6e8e7742 100644 --- a/litellm/integrations/prometheus.py +++ b/litellm/integrations/prometheus.py @@ -71,7 +71,7 @@ class PrometheusLogger: litellm_params = kwargs.get("litellm_params", {}) or {} proxy_server_request = litellm_params.get("proxy_server_request") or {} end_user_id = proxy_server_request.get("body", {}).get("user", None) - user_id = proxy_server_request.get("metadata", {}).get( + user_id = litellm_params.get("metadata", {}).get( "user_api_key_user_id", None ) user_api_key = litellm_params.get("metadata", {}).get("user_api_key", None)