Spaces:
Running
Running
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 |
|