From 88ae1f108f7fedb6f160486c5f9c8ad1b7f70851 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 8 Jan 2024 07:43:09 +0530
Subject: [PATCH] v0 commit

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

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 55518141d5..4402764a65 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1103,6 +1103,11 @@ 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 os.getenv("DATABASE_URL", None) is not None and prisma_client is None:
+        prisma_setup(database_url=os.getenv("DATABASE_URL"))
+
     ### LOAD CONFIG ###
     worker_config = litellm.get_secret("WORKER_CONFIG")
     print_verbose(f"worker_config: {worker_config}")