From 4d411fc9b41d41c4a719cf7b430636fa7013cb93 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 5 Nov 2024 19:46:05 -0800
Subject: [PATCH] fix ProxyStartupEvent._setup_prisma_client

---
 litellm/proxy/proxy_server.py | 4 ++++
 litellm/proxy/utils.py        | 3 ---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index aeb175edcd4..8295364ba8d 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -3022,6 +3022,7 @@ class ProxyStartupEvent:
         - Sets up prisma client
         - Adds necessary views to proxy
         """
+        verbose_proxy_logger.debug("Setting up Prisma Client...")
         prisma_client: Optional[PrismaClient] = None
         if database_url is not None:
             try:
@@ -3033,6 +3034,8 @@ class ProxyStartupEvent:
 
             await prisma_client.connect()
 
+            verbose_proxy_logger.debug("Success - Connected Prisma Client to DB")
+
             ## Add necessary views to proxy ##
             asyncio.create_task(
                 prisma_client.check_view_exists()
@@ -3047,6 +3050,7 @@ async def startup_event():
     import json
 
     init_verbose_loggers()
+    verbose_proxy_logger.debug("Starting up LiteLLM Proxy...")
 
     ### LOAD MASTER KEY ###
     # check if master key set in environment - load from there
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 53923106c30..7418f3f8ca5 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -2289,9 +2289,6 @@ class PrismaClient:
             verbose_proxy_logger.debug(
                 "PrismaClient: connect() called Attempting to Connect to DB"
             )
-            verbose_proxy_logger.debug(
-                "PrismaClient: DB not connected, Attempting to Connect to DB"
-            )
             await self.db.connect()
         except Exception as e:
             import traceback