phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix use view for getting tag usage

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-06 16:28:24 -07:00
parent 43acf6d08f
commit 67751d0ecc
1 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/proxy/utils.py
View file

@ -974,7 +974,7 @@ class PrismaClient:
) )
""" """
) )
if ret[0]["sum"] == 7: if ret[0]["sum"] == 8:
print("All necessary views exist!") # noqa print("All necessary views exist!") # noqa
return return
except Exception: except Exception:
@ -1124,6 +1124,24 @@ class PrismaClient:
print("MonthlyGlobalSpendPerUserPerKey Created!") # noqa print("MonthlyGlobalSpendPerUserPerKey Created!") # noqa
try:
await self.db.query_raw("""SELECT 1 FROM "DailyTagSpend" LIMIT 1""")
print("DailyTagSpend Exists!") # noqa
except Exception as e:
sql_query = """
CREATE OR REPLACE VIEW DailyTagSpend AS
SELECT
jsonb_array_elements_text(request_tags) AS individual_request_tag,
DATE(s."startTime") AS spend_date,
COUNT(*) AS log_count,
SUM(spend) AS total_spend
FROM "LiteLLM_SpendLogs" s
GROUP BY individual_request_tag, DATE(s."startTime");
"""
await self.db.execute_raw(query=sql_query)
print("DailyTagSpend Created!") # noqa
try: try:
await self.db.query_raw( await self.db.query_raw(
"""SELECT 1 FROM "Last30dTopEndUsersSpend" LIMIT 1""" """SELECT 1 FROM "Last30dTopEndUsersSpend" LIMIT 1"""