fix(proxy_cli.py): only run prisma db push if prisma in environment

litellm/proxy/proxy_cli.py

@@ -343,6 +343,7 @@ def run_server(
         )
         try:
             import uvicorn
+
             if os.name == "nt":
                 pass
             else:
@@ -384,6 +385,13 @@ def run_server(
                 os.environ["DATABASE_URL"] = database_url
 
         if os.getenv("DATABASE_URL", None) is not None:
+            try:
+                subprocess.run(["prisma"], capture_output=True)
+                is_prisma_runnable = True
+            except FileNotFoundError:
+                is_prisma_runnable = False
+
+            if is_prisma_runnable:
                 # run prisma db push, before starting server
                 # Save the current working directory
                 original_dir = os.getcwd()
@@ -397,9 +405,29 @@ def run_server(
                     )  # this looks like a weird edge case when prisma just wont start on render. we need to have the --accept-data-loss
                 finally:
                     os.chdir(original_dir)
+            else:
+                print(
+                    f"Unable to connect to DB. DATABASE_URL found in environment, but prisma package not found."
+                )
         if port == 8000 and is_port_in_use(port):
             port = random.randint(1024, 49152)
-        _endpoint_str = f"curl --location 'http://0.0.0.0:{port}/chat/completions' \\"
+        from litellm.proxy.proxy_server import app
+
+        if os.name == "nt":
+            uvicorn.run(app, host=host, port=port)  # run uvicorn
+        else:
+            import gunicorn.app.base
+
+            # Gunicorn Application Class
+            class StandaloneApplication(gunicorn.app.base.BaseApplication):
+                def __init__(self, app, options=None):
+                    self.options = options or {}  # gunicorn options
+                    self.application = app  # FastAPI app
+                    super().__init__()
+
+                    _endpoint_str = (
+                        f"curl --location 'http://0.0.0.0:{port}/chat/completions' \\"
+                    )
                     curl_command = (
                         _endpoint_str
                         + """
@@ -430,18 +458,6 @@ def run_server(
                         f"\033[1;34mSee all Router/Swagger docs on http://0.0.0.0:{port} \033[0m\n"
                     )  # noqa
 
-        from litellm.proxy.proxy_server import app
-        if os.name == "nt":
-            uvicorn.run(app, host=host, port=port)  # run uvicorn
-        else:
-            import gunicorn.app.base
-            # Gunicorn Application Class
-            class StandaloneApplication(gunicorn.app.base.BaseApplication):
-                def __init__(self, app, options=None):
-                    self.options = options or {}  # gunicorn options
-                    self.application = app  # FastAPI app
-                    super().__init__()
-
                 def load_config(self):
                     # note: This Loads the gunicorn config - has nothing to do with LiteLLM Proxy config
                     config = {
@@ -462,7 +478,9 @@ def run_server(
                 "worker_class": "uvicorn.workers.UvicornWorker",
                 "preload": True,  # Add the preload flag
             }
-            StandaloneApplication(app=app, options=gunicorn_options).run()  # Run gunicorn
+            StandaloneApplication(
+                app=app, options=gunicorn_options
+            ).run()  # Run gunicorn
 
 
 if __name__ == "__main__":

litellm/proxy/proxy_server.py

@@ -1210,11 +1210,6 @@ async def startup_event():
     # check if master key set in environment - load from there
     master_key = litellm.get_secret("LITELLM_MASTER_KEY", None)
 
-    ### CONNECT TO DB ###
-    # check if DATABASE_URL in environment - load from there
-    if prisma_client is None:
-        prisma_setup(database_url=os.getenv("DATABASE_URL"))
-
     ### LOAD CONFIG ###
     worker_config = litellm.get_secret("WORKER_CONFIG")
     verbose_proxy_logger.debug(f"worker_config: {worker_config}")

litellm/proxy/utils.py

@@ -259,7 +259,7 @@ class PrismaClient:
         self.proxy_logging_obj = proxy_logging_obj
         try:
             from prisma import Prisma  # type: ignore
-        except:
+        except Exception as e:
             os.environ["DATABASE_URL"] = database_url
             # Save the current working directory
             original_dir = os.getcwd()
@@ -273,6 +273,10 @@ class PrismaClient:
                 subprocess.run(
                     ["prisma", "db", "push", "--accept-data-loss"]
                 )  # this looks like a weird edge case when prisma just wont start on render. we need to have the --accept-data-loss
+            except:
+                raise Exception(
+                    f"Unable to run prisma commands. Run `pip install prisma`"
+                )
             finally:
                 os.chdir(original_dir)
             # Now you can import the Prisma Client