From 7b37f4c05fc4d53ce11692ff9c40f67447a04033 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 26 Mar 2024 15:04:30 -0700 Subject: [PATCH] fix(proxy_server.py): rename proxy roles param to litellm_jwtauth --- 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 c2a5c5372..656abe474 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2740,14 +2740,14 @@ async def startup_event(): proxy_logging_obj._init_litellm_callbacks() # INITIALIZE LITELLM CALLBACKS ON SERVER STARTUP <- do this to catch any logging errors on startup, not when calls are being made ## JWT AUTH ## - if general_settings.get("litellm_proxy_roles", None) is not None: - litellm_proxy_roles = LiteLLM_JWTAuth(**general_settings["litellm_proxy_roles"]) + if general_settings.get("litellm_jwtauth", None) is not None: + litellm_jwtauth = LiteLLM_JWTAuth(**general_settings["litellm_jwtauth"]) else: - litellm_proxy_roles = LiteLLM_JWTAuth() + litellm_jwtauth = LiteLLM_JWTAuth() jwt_handler.update_environment( prisma_client=prisma_client, user_api_key_cache=user_api_key_cache, - litellm_proxy_roles=litellm_proxy_roles, + litellm_jwtauth=litellm_jwtauth, ) if use_background_health_checks: