phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): rename proxy roles param to litellm_jwtauth

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 15:04:30 -07:00
parent 4281f1545b
commit 7b37f4c05f
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_server.py
View file

@ -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 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 ## ## JWT AUTH ##
if general_settings.get("litellm_proxy_roles", None) is not None: if general_settings.get("litellm_jwtauth", None) is not None:
litellm_proxy_roles = LiteLLM_JWTAuth(**general_settings["litellm_proxy_roles"]) litellm_jwtauth = LiteLLM_JWTAuth(**general_settings["litellm_jwtauth"])
else: else:
litellm_proxy_roles = LiteLLM_JWTAuth() litellm_jwtauth = LiteLLM_JWTAuth()
jwt_handler.update_environment( jwt_handler.update_environment(
prisma_client=prisma_client, prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache, user_api_key_cache=user_api_key_cache,
litellm_proxy_roles=litellm_proxy_roles, litellm_jwtauth=litellm_jwtauth,
) )
if use_background_health_checks: if use_background_health_checks: