Jeremiah Lowin commited on
Commit
cc579d4
·
1 Parent(s): 904d4e5

update typing

Browse files
src/fastmcp/server/auth/providers/bearer.py CHANGED
@@ -264,7 +264,7 @@ class BearerAuthProvider(OAuthProvider):
264
  for key_data in jwks_data.get("keys", []):
265
  key_kid = key_data.get("kid")
266
  jwk = JsonWebKey.import_key(key_data)
267
- public_key = jwk.get_public_key()
268
 
269
  if key_kid:
270
  self._jwks_cache[key_kid] = public_key
 
264
  for key_data in jwks_data.get("keys", []):
265
  key_kid = key_data.get("kid")
266
  jwk = JsonWebKey.import_key(key_data)
267
+ public_key = jwk.get_public_key() # type: ignore
268
 
269
  if key_kid:
270
  self._jwks_cache[key_kid] = public_key
tests/auth/providers/test_bearer.py CHANGED
@@ -9,6 +9,7 @@ from fastmcp import Client, FastMCP
9
  from fastmcp.client.auth import BearerAuth
10
  from fastmcp.server.auth.providers.bearer import (
11
  BearerAuthProvider,
 
12
  JWKSData,
13
  RSAKeyPair,
14
  )
@@ -126,8 +127,8 @@ class TestBearerTokenJWKS:
126
  from authlib.jose import JsonWebKey
127
 
128
  # Create JWK from the RSA public key
129
- jwk = JsonWebKey.import_key(rsa_key_pair.public_key)
130
- jwk_data = jwk.as_dict()
131
  jwk_data["kid"] = "test-key-1"
132
  jwk_data["alg"] = "RS256"
133
 
 
9
  from fastmcp.client.auth import BearerAuth
10
  from fastmcp.server.auth.providers.bearer import (
11
  BearerAuthProvider,
12
+ JWKData,
13
  JWKSData,
14
  RSAKeyPair,
15
  )
 
127
  from authlib.jose import JsonWebKey
128
 
129
  # Create JWK from the RSA public key
130
+ jwk = JsonWebKey.import_key(rsa_key_pair.public_key) # type: ignore
131
+ jwk_data: JWKData = jwk.as_dict() # type: ignore
132
  jwk_data["kid"] = "test-key-1"
133
  jwk_data["alg"] = "RS256"
134