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

fix(proxy_server.py): fix /spend/logs endpoint

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-09 18:11:33 -08:00
parent d1d4568dc5
commit 6710c2ee5d
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/proxy_server.py
View file

@ -3185,7 +3185,7 @@ async def view_spend_logs(
# SQL query # SQL query
response = await prisma_client.db.litellm_spendlogs.group_by( response = await prisma_client.db.litellm_spendlogs.group_by(
by=["startTime"], by=["api_key", "startTime"],
where=filter_query, # type: ignore where=filter_query, # type: ignore
sum={ sum={
"spend": True, "spend": True,
@ -3205,9 +3205,13 @@ async def view_spend_logs(
date = dt_object.date() date = dt_object.date()
if date not in result: if date not in result:
result[date] = {} result[date] = {}
api_key = record["api_key"]
result[date]["spend"] = ( result[date]["spend"] = (
result[date].get("spend", 0) + record["_sum"]["spend"] result[date].get("spend", 0) + record["_sum"]["spend"]
) )
result[date][api_key] = (
result[date].get(api_key, 0) + record["_sum"]["spend"]
)
return_list = [] return_list = []
final_date = None final_date = None
for k, v in sorted(result.items()): for k, v in sorted(result.items()):