From 56fe8726abe68d6fc51bccd37522c0f159339aaa Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 29 Feb 2024 17:46:39 -0800
Subject: [PATCH] (feat) use view to see monthly spend per key

---
 litellm/proxy/utils.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 6a01094433..f3a552fc1a 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -592,6 +592,29 @@ class PrismaClient:
             await self.db.execute_raw(query=sql_query)
 
             print("Last30dModelsBySpend Created!")  # noqa
+        try:
+            await self.db.query_raw(
+                """SELECT 1 FROM "MonthlyGlobalSpendPerKey" LIMIT 1"""
+            )
+            print("MonthlyGlobalSpendPerKey Exists!")  # noqa
+        except Exception as e:
+            sql_query = """
+                CREATE OR REPLACE VIEW "MonthlyGlobalSpendPerKey" AS 
+                SELECT
+                DATE("startTime") AS date, 
+                SUM("spend") AS spend,
+                api_key as api_key
+                FROM 
+                "LiteLLM_SpendLogs" 
+                WHERE 
+                "startTime" >= (CURRENT_DATE - INTERVAL '30 days')
+                GROUP BY 
+                DATE("startTime"),
+                api_key;
+            """
+            await self.db.execute_raw(query=sql_query)
+
+            print("MonthlyGlobalSpendPerKey Created!")  # noqa
 
         return