From 2a7f113fde0a9d62c3ff989f08a8e59ba2ce4104 Mon Sep 17 00:00:00 2001
From: Cole McIntosh <82463175+colesmcintosh@users.noreply.github.com>
Date: Thu, 12 Jun 2025 18:55:34 -0600
Subject: [PATCH] feat(schema): add additional indexes to LiteLLM_SpendLogs for
 improved query performance (#11675)

---
 litellm-proxy-extras/litellm_proxy_extras/schema.prisma | 3 +++
 litellm/proxy/schema.prisma                             | 3 +++
 schema.prisma                                           | 3 +++
 3 files changed, 9 insertions(+)

diff --git a/litellm-proxy-extras/litellm_proxy_extras/schema.prisma b/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
index 8088edf29ed..ac8834018d3 100644
--- a/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
+++ b/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
@@ -266,6 +266,9 @@ model LiteLLM_SpendLogs {
   @@index([startTime])
   @@index([end_user])
   @@index([session_id])
+  @@index([model_group, cache_hit, startTime, api_base])
+  @@index([model_group, cache_hit, api_key, startTime])
+  @@index([model_group, cache_hit, end_user, startTime])
 }
 
 // View spend, model, api_key per request
diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index dc16a0ceaf0..78266fe4388 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -265,6 +265,9 @@ model LiteLLM_SpendLogs {
   @@index([startTime])
   @@index([end_user])
   @@index([session_id])
+  @@index([model_group, cache_hit, startTime, api_base])
+  @@index([model_group, cache_hit, api_key, startTime])
+  @@index([model_group, cache_hit, end_user, startTime])
 }
 
 // View spend, model, api_key per request
diff --git a/schema.prisma b/schema.prisma
index 032b9e0b58d..0b5cecd69d4 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -265,6 +265,9 @@ model LiteLLM_SpendLogs {
   @@index([startTime])
   @@index([end_user])
   @@index([session_id])
+  @@index([model_group, cache_hit, startTime, api_base])
+  @@index([model_group, cache_hit, api_key, startTime])
+  @@index([model_group, cache_hit, end_user, startTime])
 }
 
 // View spend, model, api_key per request