From d6b45ba143246d168affaa2808d27fcd57d3bf3d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 19 Apr 2025 16:46:00 -0700 Subject: [PATCH] can_user_call_model (#10170) --- litellm/proxy/auth/auth_checks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index bf8fd2350e..653455b288 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -88,7 +88,7 @@ async def common_checks( 9. Check if request body is safe 10. [OPTIONAL] Organization checks - is user_object.organization_id is set, run these checks """ - _model = request_body.get("model", None) + _model: Optional[str] = cast(Optional[str], request_body.get("model", None)) # 1. If team is blocked if team_object is not None and team_object.blocked is True: @@ -112,7 +112,7 @@ async def common_checks( ) ## 2.1 If user can call model (if personal key) - if team_object is None and user_object is not None: + if _model and team_object is None and user_object is not None: await can_user_call_model( model=_model, llm_router=llm_router,