phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Expanding jwt access to other RS and PS algos. Updated to resolve merge conflicts.

Browse source
This commit is contained in:
Jean-Luc Duckworth 2024-05-07 15:42:06 -04:00
parent 7d27e4a85d
commit a04281a478
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -156,6 +156,11 @@ class JWTHandler:
return public_key return public_key
async def auth_jwt(self, token: str) -> dict: async def auth_jwt(self, token: str) -> dict:
# Supported algos: https://pyjwt.readthedocs.io/en/stable/algorithms.html
# "Warning: Make sure not to mix symmetric and asymmetric algorithms that interpret
# the key in different ways (e.g. HS* and RS*)."
algorithms = ["RS256", "RS384", "RS512", "PS256", "PS384", "PS512"],
audience = os.getenv("JWT_AUDIENCE") audience = os.getenv("JWT_AUDIENCE")
decode_options = None decode_options = None
if audience is None: if audience is None:
@ -189,7 +194,7 @@ class JWTHandler:
payload = jwt.decode( payload = jwt.decode(
token, token,
public_key_rsa, # type: ignore public_key_rsa, # type: ignore
algorithms=["RS256"], algorithms=algorithms,
options=decode_options, options=decode_options,
audience=audience, audience=audience,
) )
@ -214,7 +219,7 @@ class JWTHandler:
payload = jwt.decode( payload = jwt.decode(
token, token,
key, key,
algorithms=["RS256"], algorithms=algorithms,
audience=audience, audience=audience,
options=decode_options options=decode_options
) )