ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

can_user_call_model (#10170)

Browse Source
This commit is contained in:
Ishaan Jaff 2025-04-19 16:46:00 -07:00 committed by GitHub
parent 431b230f07
commit 7c3df984da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/proxy/auth/auth_checks.py
View File

@ -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,