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

Check existence of multiple views in 1 query

Browse source 
instead of multiple queries. This is more efficient because it lets us check for
all views in one query instead of multiple queries.
This commit is contained in:
Marc Abramowitz 2024-07-23 21:45:01 -07:00
parent d5d2ffffdf
commit 642f1a7bcc
1 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/utils.py
View file

@ -844,6 +844,30 @@ class PrismaClient:
If the view doesn't exist, one will be created.
"""
# Check to see if all of the necessary views exist and if they do, simply return
# This is more efficient because it lets us check for all views in one
# query instead of multiple queries.
try:
ret = await self.db.query_raw(
"""
SELECT SUM(1) FROM pg_views
WHERE schemaname = 'public' AND viewname IN (
'LiteLLM_VerificationTokenView',
'MonthlyGlobalSpend',
'Last30dKeysBySpend',
'Last30dModelsBySpend',
'MonthlyGlobalSpendPerKey',
'Last30dTopEndUsersSpend'
)
"""
)
if ret[0]['sum'] == 6:
print("All necessary views exist!") # noqa
return
except Exception:
pass
try:
# Try to select one row from the view
await self.db.query_raw(