Refactor 'check_view_exists' logic (BerriAI#5659)

* fix(proxy/utils.py): comment out auto-upsert logic in check_view_exists

Prevents proxy from failing on startup due to faulty logic

* fix(db/migration_scripts/create_views.py): fix 'DailyTagSpend' quotation on check

* fix(create_views.py): mongly global spend time period should be 30d not 20d

* fix(schema.prisma): index on startTime and endUser for efficient UI querying

Author: krrishdholakia

Diff for: litellm/proxy/db/create_views.py renamed to db_scripts/create_views.py

@@ -1,7 +1,5 @@
 """
 python script to pre-create all views required by LiteLLM Proxy Server
-
-
 """
 
 import asyncio
@@ -27,30 +25,6 @@ async def check_view_exists():
 
     # connect to dB
     await db.connect()
-    # 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 db.query_raw(
-            """
-                SELECT SUM(1) FROM pg_views
-                WHERE schemaname = 'public' AND viewname IN (
-                    'LiteLLM_VerificationTokenView',
-                    'MonthlyGlobalSpend',
-                    'Last30dKeysBySpend',
-                    'Last30dModelsBySpend',
-                    'MonthlyGlobalSpendPerKey',
-                    'MonthlyGlobalSpendPerUserPerKey',
-                    'Last30dTopEndUsersSpend'
-                )
-                """
-        )
-        if ret[0]["sum"] == 8:
-            print("All necessary views exist!")  # noqa
-            return
-    except Exception:
-        pass
-
     try:
         # Try to select one row from the view
         await db.query_raw("""SELECT 1 FROM "LiteLLM_VerificationTokenView" LIMIT 1""")
@@ -180,7 +154,7 @@ async def check_view_exists():
             FROM 
             "LiteLLM_SpendLogs" 
             WHERE 
-            "startTime" >= (CURRENT_DATE - INTERVAL '20 days')
+            "startTime" >= (CURRENT_DATE - INTERVAL '30 days')
             GROUP BY 
             DATE("startTime"),
             "user",
@@ -191,7 +165,7 @@ async def check_view_exists():
         print("MonthlyGlobalSpendPerUserPerKey Created!")  # noqa
 
     try:
-        await db.query_raw("""SELECT 1 FROM "DailyTagSpend" LIMIT 1""")
+        await db.query_raw("""SELECT 1 FROM DailyTagSpend LIMIT 1""")
         print("DailyTagSpend Exists!")  # noqa
     except Exception as e:
         sql_query = """

Diff for: litellm/proxy/db/dynamo_db.py

@@ -1,3 +1,7 @@
+"""
+Deprecated. Only PostgresSQL is supported.
+"""
+
 import json
 from datetime import datetime
 from typing import Any, List, Literal, Optional, Union

Diff for: litellm/proxy/schema.prisma

@@ -168,7 +168,7 @@ model LiteLLM_Config {
   param_value Json?
 }
 
-// View spend, model, hashed api_key per request
+// View spend, model, api_key per request
 model LiteLLM_SpendLogs {
   request_id          String @id
   call_type           String
@@ -192,6 +192,8 @@ model LiteLLM_SpendLogs {
   team_id             String? 
   end_user            String?
   requester_ip_address String?
+  @@index([startTime])
+  @@index([end_user])
 }
 
 // View spend, model, api_key per request