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

fix(handle_jwt.py): document the raw received jwt token

Browse source
This commit is contained in:
Krrish Dholakia 2025-08-29 14:53:07 -07:00
parent 607f425cc6
commit 86c5de17bc
1 changed files with 26 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
litellm/proxy/auth/handle_jwt.py
View file

@ -414,7 +414,16 @@ class JWTHandler:
if cached_keys is None:
response = await self.http_handler.get(key_url)
response_json = response.json()
try:
response_json = response.json()
except Exception as e:
verbose_proxy_logger.error(
f"Error parsing response: {e}. Original Response: {response.text}"
)
raise Exception(
f"Error parsing response: {e}. Check server logs for original response."
)
if "keys" in response_json:
keys: JWKKeyValue = response.json()["keys"]
else:
@ -904,7 +913,7 @@ class JWTAuthManager:
if end_user_id
else None
)
team_membership_object: Optional[LiteLLM_TeamMembership] = None
if user_id and team_id:
team_membership_object = (
@ -1145,19 +1154,21 @@ class JWTAuthManager:
)
# Get other objects
user_object, org_object, end_user_object, team_membership_object = await JWTAuthManager.get_objects(
user_id=user_id,
user_email=user_email,
org_id=org_id,
end_user_id=end_user_id,
team_id=team_id,
valid_user_email=valid_user_email,
jwt_handler=jwt_handler,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
parent_otel_span=parent_otel_span,
proxy_logging_obj=proxy_logging_obj,
route=route,
user_object, org_object, end_user_object, team_membership_object = (
await JWTAuthManager.get_objects(
user_id=user_id,
user_email=user_email,
org_id=org_id,
end_user_id=end_user_id,
team_id=team_id,
valid_user_email=valid_user_email,
jwt_handler=jwt_handler,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
parent_otel_span=parent_otel_span,
proxy_logging_obj=proxy_logging_obj,
route=route,
)
)
await JWTAuthManager.sync_user_role_and_teams(
@ -1186,8 +1197,6 @@ class JWTAuthManager:
is_proxy_admin = True
else:
is_proxy_admin = False
return JWTAuthBuilderResult(
is_proxy_admin=is_proxy_admin,