fix: update remaining token storage implementations

Author: daleselaji-dev

docs_src/identity_assertion/tutorial001.py

@@ -20,13 +20,13 @@ def __init__(self) -> None:
     async def get_tokens(self) -> OAuthToken | None:
         return self.tokens
 
-    async def set_tokens(self, tokens: OAuthToken) -> None:
+    async def set_tokens(self, tokens: OAuthToken | None) -> None:
         self.tokens = tokens
 
     async def get_client_info(self) -> OAuthClientInformationFull | None:
         return self.client_info
 
-    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+    async def set_client_info(self, client_info: OAuthClientInformationFull | None) -> None:
         self.client_info = client_info
 
 

examples/clients/simple-auth-client/mcp_simple_auth_client/main.py

@@ -37,13 +37,13 @@ def __init__(self):
     async def get_tokens(self) -> OAuthToken | None:
         return self._tokens
 
-    async def set_tokens(self, tokens: OAuthToken) -> None:
+    async def set_tokens(self, tokens: OAuthToken | None) -> None:
         self._tokens = tokens
 
     async def get_client_info(self) -> OAuthClientInformationFull | None:
         return self._client_info
 
-    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+    async def set_client_info(self, client_info: OAuthClientInformationFull | None) -> None:
         self._client_info = client_info
 
 

examples/snippets/clients/identity_assertion_client.py

@@ -34,13 +34,13 @@ def __init__(self) -> None:
     async def get_tokens(self) -> OAuthToken | None:
         return self.tokens
 
-    async def set_tokens(self, tokens: OAuthToken) -> None:
+    async def set_tokens(self, tokens: OAuthToken | None) -> None:
         self.tokens = tokens
 
     async def get_client_info(self) -> OAuthClientInformationFull | None:
         return self.client_info
 
-    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+    async def set_client_info(self, client_info: OAuthClientInformationFull | None) -> None:
         self.client_info = client_info
 
 

examples/snippets/clients/oauth_client.py

@@ -29,15 +29,15 @@ async def get_tokens(self) -> OAuthToken | None:
         """Get stored tokens."""
         return self.tokens
 
-    async def set_tokens(self, tokens: OAuthToken) -> None:
+    async def set_tokens(self, tokens: OAuthToken | None) -> None:
         """Store tokens."""
         self.tokens = tokens
 
     async def get_client_info(self) -> OAuthClientInformationFull | None:
         """Get stored client information."""
         return self.client_info
 
-    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+    async def set_client_info(self, client_info: OAuthClientInformationFull | None) -> None:
         """Store client information."""
         self.client_info = client_info
 

examples/stories/_shared/auth.py

@@ -38,13 +38,13 @@ class InMemoryTokenStorage:
     async def get_tokens(self) -> OAuthToken | None:
         return self.tokens
 
-    async def set_tokens(self, tokens: OAuthToken) -> None:
+    async def set_tokens(self, tokens: OAuthToken | None) -> None:
         self.tokens = tokens
 
     async def get_client_info(self) -> OAuthClientInformationFull | None:
         return self.client_info
 
-    async def set_client_info(self, client_info: OAuthClientInformationFull) -> None:
+    async def set_client_info(self, client_info: OAuthClientInformationFull | None) -> None:
         self.client_info = client_info
 
 

src/mcp/client/auth/oauth2.py

@@ -119,7 +119,7 @@ def _is_expired_client_secret(client_info: OAuthClientInformationFull) -> bool:
 def _is_invalid_client_response(body: bytes) -> bool:
     """Identify RFC 6749 ``invalid_client`` responses independent of HTTP status."""
     try:
-        payload = json.loads(body)
+        payload: Any = json.loads(body)
     except (json.JSONDecodeError, UnicodeDecodeError):
         return False
     return isinstance(payload, dict) and payload.get("error") == "invalid_client"