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

This commit is contained in:
Krrish Dholakia 2024-02-09 18:11:33 -08:00
parent d1d4568dc5
commit 6710c2ee5d

View file

@ -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()):