scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

fix(workflows): validate internal auth context types

Browse source
This commit is contained in:
ishaan-j-berri 2026-11-06 18:49:54 +08:00
parent bb147ec202
commit b463dfa254
1 changed files with 15 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
litellm/proxy/auth/user_api_key_auth.py
View file

@ -4891,8 +4891,7 @@ async def authorize_internal_virtual_key(
execution_route: Final = route if route is not None else get_request_route(request=request)
await pre_db_read_auth_checks(request_data=request_data, request=request, route=execution_route)
identity.end_user_id = await resolve_and_validate_end_user_id(
raw_end_user_id=identity.end_user_id
or get_end_user_id_from_request_body(request_data, safe_get_request_headers(request)),
raw_end_user_id=identity.end_user_id or get_end_user_id_from_request_body(request_data, request.headers),
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
parent_otel_span=None,
@ -4900,18 +4899,20 @@ async def authorize_internal_virtual_key(
route=execution_route,
key_end_user_budget_id=get_key_end_user_budget_id(identity.metadata),
)
end_user_object: Final[LiteLLM_EndUserTable | None]
if identity.end_user_id is not None:
_, end_user_object = await _lookup_end_user_and_apply_budget(
valid_token=identity,
route=execution_route,
parent_otel_span=None,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
proxy_logging_obj=proxy_logging_obj,
)
else:
end_user_object = None
end_user_object: Final[LiteLLM_EndUserTable | None] = (
(
await _lookup_end_user_and_apply_budget(
valid_token=identity,
route=execution_route,
parent_otel_span=None,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
proxy_logging_obj=proxy_logging_obj,
)
)[1]
if identity.end_user_id is not None
else None
)
auth: Final = await validate_resolved_virtual_key(
request=request,
request_data=request_data,