scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix: add break on match and guard empty normalized_route in mapped route checks

Browse source 
- Add break after match in user_api_key_auth.py loop to avoid unnecessary
  iterations over remaining mapped routes
- Guard against normalized_route being empty when route == root_path exactly,
  which would otherwise match every mapped route via startswith("")
- Apply same empty-string guard in pass_through_endpoints.py for consistency
This commit is contained in:
joereyna 2026-03-11 23:50:34 -07:00
parent 47e41deab6
commit 791e598ad5
2 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -390,13 +390,16 @@ async def check_api_key_for_custom_headers_or_pass_through_endpoints(
if root_path and root_path != "/":
if route.startswith(root_path):
normalized_route = route[len(root_path):]
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value: # type: ignore
if normalized_route.startswith(mapped_route):
is_mapped_pass_through_route = True
if normalized_route: # guard against route == root_path exactly
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value: # type: ignore
if normalized_route.startswith(mapped_route):
is_mapped_pass_through_route = True
break
else:
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value: # type: ignore
if route.startswith(mapped_route):
is_mapped_pass_through_route = True
break
if is_mapped_pass_through_route:
if request.headers.get("litellm_user_api_key") is not None:
api_key = request.headers.get("litellm_user_api_key") or ""

9
litellm/proxy/pass_through_endpoints/pass_through_endpoints.py
View file

@ -2065,10 +2065,11 @@ class InitPassThroughEndpointHelpers:
if root_path and root_path != "/":
if route.startswith(root_path):
normalized_route = route[len(root_path):]
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value:
if normalized_route.startswith(mapped_route):
return True
# Route lacks expected prefix — not a mapped pass-through route
if normalized_route: # guard against route == root_path exactly
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value:
if normalized_route.startswith(mapped_route):
return True
# Route lacks expected prefix (or is exactly root_path) — not a mapped pass-through route
else:
for mapped_route in LiteLLMRoutes.mapped_pass_through_routes.value:
if route.startswith(mapped_route):