phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fixes

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-08 21:54:48 -08:00
parent e98437104d
commit 3a4ac8be79
5 changed files with 37 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/proxy_server.py
View file

@ -3041,12 +3041,7 @@ async def spend_key_fn(
f"Database not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys"
)
if start_date is None and end_date is None:
key_info = await prisma_client.get_data(
table_name="key", query_type="find_all"
)
return key_info
elif (
if (
start_date is not None
and isinstance(start_date, str)
and end_date is not None
@ -3076,6 +3071,11 @@ async def spend_key_fn(
"message": "This is your SQL query",
"response": response,
}
else:
key_info = await prisma_client.get_data(
table_name="key", query_type="find_all"
)
return key_info
except Exception as e:
raise HTTPException(