From 35c07306df319a25757f679005aec5d25030cd6b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 21 Jun 2024 14:36:38 -0700 Subject: [PATCH] show correct key aliases on ui --- .../analytics_endpoints.py | 24 ++++++++++++------- 1 file changed, 15 insertions(+), 9 deletions(-) diff --git a/litellm/proxy/analytics_endpoints/analytics_endpoints.py b/litellm/proxy/analytics_endpoints/analytics_endpoints.py index ebb7d43acfd..54ac8bc48fb 100644 --- a/litellm/proxy/analytics_endpoints/analytics_endpoints.py +++ b/litellm/proxy/analytics_endpoints/analytics_endpoints.py @@ -74,18 +74,24 @@ async def get_global_activity( sql_query = """ SELECT - "api_key", - "call_type", - "model", + CASE + WHEN vt."key_alias" IS NOT NULL THEN vt."key_alias" + ELSE 'Unnamed Key' + END AS api_key, + sl."call_type", + sl."model", COUNT(*) AS total_rows, - SUM(CASE WHEN "cache_hit" = 'True' THEN 1 ELSE 0 END) AS cache_hit_true_rows - FROM "LiteLLM_SpendLogs" + SUM(CASE WHEN sl."cache_hit" = 'True' THEN 1 ELSE 0 END) AS cache_hit_true_rows, + SUM(CASE WHEN sl."cache_hit" = 'True' THEN sl."completion_tokens" ELSE 0 END) AS cached_completion_tokens, + SUM(CASE WHEN sl."cache_hit" != 'True' THEN sl."completion_tokens" ELSE 0 END) AS generated_completion_tokens + FROM "LiteLLM_SpendLogs" sl + LEFT JOIN "LiteLLM_VerificationToken" vt ON sl."api_key" = vt."token" WHERE - "startTime" BETWEEN $1::date AND $2::date + interval '1 day' + sl."startTime" BETWEEN $1::date AND $2::date + interval '1 day' GROUP BY - "api_key", - "call_type", - "model" + vt."key_alias", + sl."call_type", + sl."model" """ db_response = await prisma_client.db.query_raw( sql_query, start_date_obj, end_date_obj