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

fix create view - MonthlyGlobalSpendPerUserPerKey

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-05 12:11:59 -07:00
parent cc51fd0df9
commit e0400accca
1 changed files with 27 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

28
litellm/proxy/utils.py
View file

@ -969,11 +969,12 @@ class PrismaClient:
'Last30dKeysBySpend',
'Last30dModelsBySpend',
'MonthlyGlobalSpendPerKey',
'MonthlyGlobalSpendPerUserPerKey',
'Last30dTopEndUsersSpend'
)
"""
)
if ret[0]["sum"] == 6:
if ret[0]["sum"] == 7:
print("All necessary views exist!") # noqa
return
except Exception:
@ -1097,6 +1098,31 @@ class PrismaClient:
await self.db.execute_raw(query=sql_query)
print("MonthlyGlobalSpendPerKey Created!") # noqa
try:
await self.db.query_raw(
"""SELECT 1 FROM "MonthlyGlobalSpendPerUserPerKey" LIMIT 1"""
)
print("MonthlyGlobalSpendPerUserPerKey Exists!") # noqa
except Exception as e:
sql_query = """
CREATE OR REPLACE VIEW "MonthlyGlobalSpendPerUserPerKey" AS
SELECT
DATE("startTime") AS date,
SUM("spend") AS spend,
api_key as api_key,
"user" as "user"
FROM
"LiteLLM_SpendLogs"
WHERE
"startTime" >= (CURRENT_DATE - INTERVAL '20 days')
GROUP BY
DATE("startTime"),
"user",
api_key;
"""
await self.db.execute_raw(query=sql_query)
print("MonthlyGlobalSpendPerUserPerKey Created!") # noqa
try:
await self.db.query_raw(