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

fix(proxy_server.py): linting fix

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-11 22:05:01 -07:00
parent 094f20121a
commit b4684d5132
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/proxy_server.py
View file

@ -425,7 +425,7 @@ async def user_api_key_auth(
litellm_proxy_roles=jwt_handler.litellm_jwtauth, litellm_proxy_roles=jwt_handler.litellm_jwtauth,
) )
if is_allowed == False: if is_allowed == False:
allowed_routes = jwt_handler.litellm_jwtauth.team_allowed_routes allowed_routes = jwt_handler.litellm_jwtauth.team_allowed_routes # type: ignore
actual_routes = get_actual_routes(allowed_routes=allowed_routes) actual_routes = get_actual_routes(allowed_routes=allowed_routes)
raise Exception( raise Exception(
f"Team not allowed to access this route. Route={route}, Allowed Routes={actual_routes}" f"Team not allowed to access this route. Route={route}, Allowed Routes={actual_routes}"
@ -2263,10 +2263,18 @@ class ProxyConfig:
_PROXY_AzureContentSafety, _PROXY_AzureContentSafety,
) )
azure_content_safety_params = litellm_settings["azure_content_safety_params"] azure_content_safety_params = litellm_settings[
"azure_content_safety_params"
]
for k, v in azure_content_safety_params.items(): for k, v in azure_content_safety_params.items():
if v is not None and isinstance(v, str) and v.startswith("os.environ/"): if (
azure_content_safety_params[k] = litellm.get_secret(v) v is not None
and isinstance(v, str)
and v.startswith("os.environ/")
):
azure_content_safety_params[k] = (
litellm.get_secret(v)
)
azure_content_safety_obj = _PROXY_AzureContentSafety( azure_content_safety_obj = _PROXY_AzureContentSafety(
**azure_content_safety_params, **azure_content_safety_params,