From d45101b652d7f32571ed102ade646fad428aa0bb Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 3 Jan 2024 16:10:10 +0530
Subject: [PATCH] fix(proxy_server.py): fix master key reset, to preserve key
 from env

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

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 4d5730ec8..64f799dce 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -856,10 +856,6 @@ def initialize(
         llm_router, llm_model_list, general_settings = load_router_config(
             router=llm_router, config_file_path=config
         )
-    else:
-        # reset auth if config not passed, needed for consecutive tests on proxy
-        master_key = None
-        user_custom_auth = None
     if headers:  # model-specific param
         user_headers = headers
         dynamic_config[user_model]["headers"] = headers
@@ -991,6 +987,10 @@ async def startup_event():
     global prisma_client, master_key, use_background_health_checks
     import json
 
+    ### LOAD MASTER KEY ###
+    # check if master key set in environment - load from there
+    master_key = litellm.get_secret("LITELLM_MASTER_KEY", None)
+
     ### LOAD CONFIG ###
     worker_config = litellm.get_secret("WORKER_CONFIG")
     print_verbose(f"worker_config: {worker_config}")