phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(handle_jwt.py): cache public keys

Browse source 
caches jwt public keys - reducing need for making http calls on every request
This commit is contained in:
Krrish Dholakia 2024-03-25 12:36:32 -07:00
parent f1b1640a31
commit c2ffb83c71
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -155,9 +155,19 @@ class JWTHandler:
if keys_url is None: if keys_url is None:
raise Exception("Missing JWT Public Key URL from environment.") raise Exception("Missing JWT Public Key URL from environment.")
response = await self.http_handler.get(keys_url) cached_keys = await self.user_api_key_cache.async_get_cache(
"litellm_jwt_auth_keys"
)
if cached_keys is None:
response = await self.http_handler.get(keys_url)
keys = response.json()["keys"] keys = response.json()["keys"]
await self.user_api_key_cache.async_set_cache(
key="litellm_jwt_auth_keys", value=keys, ttl=600 # cache for 10 mins
)
else:
keys = cached_keys
header = jwt.get_unverified_header(token) header = jwt.get_unverified_header(token)