From fccdb92c6b2ffd34b8e7aef78d492055f3a531b9 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 3 May 2024 21:20:19 -0700
Subject: [PATCH] fix - select startTime and endTime on UI

---
 litellm/proxy/proxy_server.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 27a05ca21..4bb8dee7f 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -7552,7 +7552,7 @@ async def model_metrics(
         FROM
             "LiteLLM_SpendLogs"
         WHERE
-            "startTime" >= NOW() - INTERVAL '30 days'
+            "startTime" BETWEEN $2::timestamp AND $3::timestamp
             AND "model" = $1 AND "cache_hit" != 'True'
         GROUP BY
             api_base,
@@ -7653,6 +7653,8 @@ FROM
 WHERE
     "model" = $2
     AND "cache_hit" != 'True'
+    AND "startTime" >= $3::timestamp
+    AND "startTime" <= $4::timestamp
 GROUP BY
     api_base
 ORDER BY
@@ -7660,7 +7662,7 @@ ORDER BY
     """
 
     db_response = await prisma_client.db.query_raw(
-        sql_query, alerting_threshold, _selected_model_group
+        sql_query, alerting_threshold, _selected_model_group, startTime, endTime
     )
 
     if db_response is not None: