refactor(handle_jwt.py): print jwt version

This commit is contained in:
Krrish Dholakia 2024-03-19 17:17:11 -07:00
parent 9ce7aeab67
commit 1c68a7fd1d

View file

@ -8,6 +8,8 @@ JWT token must have 'litellm_proxy_admin' in scope.
import httpx import httpx
import jwt import jwt
print(jwt.__version__) # noqa
from jwt.algorithms import RSAAlgorithm from jwt.algorithms import RSAAlgorithm
import json import json
import os import os
@ -74,7 +76,7 @@ class JWTHandler:
return scopes return scopes
async def auth_jwt(self, token: str) -> dict: async def auth_jwt(self, token: str) -> dict:
keys_url = os.getenv("OPENID_PUBLIC_KEY_URL") keys_url = os.getenv("JWT_PUBLIC_KEY_URL")
async with self.http_handler as http: async with self.http_handler as http:
response = await http.get(keys_url) response = await http.get(keys_url)