From c1f97037345d87e427aa8cc1e75c57a55ff262c7 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 26 Feb 2024 17:54:10 -0800
Subject: [PATCH] (feat) use clickhouse for /spend/logs

---
 litellm/proxy/proxy_server.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 56c49ae32..cdf57fe13 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -3782,6 +3782,17 @@ async def view_spend_logs(
 -H "Authorization: Bearer sk-1234"
     ```
     """
+    if os.getenv("CLICKHOUSE_HOST") is not None:
+        # gettting spend logs from clickhouse
+        from litellm.proxy.enterprise.utils import view_spend_logs_from_clickhouse
+
+        return await view_spend_logs_from_clickhouse(
+            api_key=api_key,
+            user_id=user_id,
+            request_id=request_id,
+            start_date=start_date,
+            end_date=end_date,
+        )
     global prisma_client
     try:
         verbose_proxy_logger.debug("inside view_spend_logs")