From 6710c2ee5d4454beb9adae807c8a223f7ce2ca26 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 9 Feb 2024 18:11:33 -0800 Subject: [PATCH] fix(proxy_server.py): fix /spend/logs endpoint --- litellm/proxy/proxy_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index f719ec0cc9..8946faea43 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3185,7 +3185,7 @@ async def view_spend_logs( # SQL query response = await prisma_client.db.litellm_spendlogs.group_by( - by=["startTime"], + by=["api_key", "startTime"], where=filter_query, # type: ignore sum={ "spend": True, @@ -3205,9 +3205,13 @@ async def view_spend_logs( date = dt_object.date() if date not in result: result[date] = {} + api_key = record["api_key"] result[date]["spend"] = ( result[date].get("spend", 0) + record["_sum"]["spend"] ) + result[date][api_key] = ( + result[date].get(api_key, 0) + record["_sum"]["spend"] + ) return_list = [] final_date = None for k, v in sorted(result.items()):