From b09c9820081aa0cd4aa27f122de9aa5d3d27a097 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 13:27:50 -0800 Subject: [PATCH] (feat) proxy - auth - raise Value Error when master key is None --- litellm/proxy/proxy_server.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 72c0e25bfa..8b4b2a0968 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1285,7 +1285,11 @@ async def startup_event(): verbose_proxy_logger.debug(f"custom_db_client connecting - {custom_db_client}") await custom_db_client.connect() - if prisma_client is not None and master_key is not None: + if prisma_client is not None: + if master_key is None: + raise ValueError( + "Using Proxy Auth, but Master Key not set, please set `LITELLM_MASTER_KEY` in your environment or `master_key` in your config.yaml" + ) # add master key to db await generate_key_helper_fn( duration=None, models=[], aliases={}, config={}, spend=0, token=master_key @@ -1293,7 +1297,11 @@ async def startup_event(): verbose_proxy_logger.debug( f"custom_db_client client - Inserting master key {custom_db_client}. Master_key: {master_key}" ) - if custom_db_client is not None and master_key is not None: + if custom_db_client is not None: + if master_key is None: + raise ValueError( + "Using Proxy Auth, but Master Key not set, please set `LITELLM_MASTER_KEY` in your environment or `master_key` in your config.yaml" + ) # add master key to db await generate_key_helper_fn( duration=None, models=[], aliases={}, config={}, spend=0, token=master_key