Skip to content

Commit c07b7b9

Browse files
committed
- Tried to fix the ruff error.
1 parent 1ea72c5 commit c07b7b9

File tree

2 files changed

+6
-16
lines changed

2 files changed

+6
-16
lines changed

src/mcp/client/auth/extensions/enterprise_managed_auth.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -241,9 +241,7 @@ async def exchange_token_for_id_jag(
241241

242242
if response.status_code != 200:
243243
error_data = (
244-
response.json()
245-
if response.headers.get("content-type", "").startswith("application/json")
246-
else {}
244+
response.json() if response.headers.get("content-type", "").startswith("application/json") else {}
247245
)
248246
error = error_data.get("error", "unknown_error")
249247
error_description = error_data.get("error_description", "Token exchange failed")
@@ -313,9 +311,7 @@ async def exchange_id_jag_for_access_token(
313311

314312
if response.status_code != 200:
315313
error_data = (
316-
response.json()
317-
if response.headers.get("content-type", "").startswith("application/json")
318-
else {}
314+
response.json() if response.headers.get("content-type", "").startswith("application/json") else {}
319315
)
320316
error = error_data.get("error", "unknown_error")
321317
error_description = error_data.get("error_description", "JWT bearer grant failed")

src/mcp/server/auth/extensions/enterprise_managed_auth.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -117,9 +117,7 @@ def is_used(self, jti: str) -> bool:
117117
def _cleanup(self) -> None:
118118
"""Remove expired entries."""
119119
now = time.time()
120-
self._used_jtis = {
121-
jti: timestamp for jti, timestamp in self._used_jtis.items() if now - timestamp < self._ttl
122-
}
120+
self._used_jtis = {jti: timestamp for jti, timestamp in self._used_jtis.items() if now - timestamp < self._ttl}
123121

124122

125123
# ============================================================================
@@ -205,22 +203,18 @@ def validate_id_jag(
205203
# Step 5: Validate audience
206204
if claims.get("aud") != self.config.server_auth_issuer:
207205
raise ValueError(
208-
f"Invalid audience: expected '{self.config.server_auth_issuer}', "
209-
f"got '{claims.get('aud')}'"
206+
f"Invalid audience: expected '{self.config.server_auth_issuer}', got '{claims.get('aud')}'"
210207
)
211208

212209
# Step 6: Validate resource
213210
if claims.get("resource") != self.config.server_resource_id:
214211
raise ValueError(
215-
f"Invalid resource: expected '{self.config.server_resource_id}', "
216-
f"got '{claims.get('resource')}'"
212+
f"Invalid resource: expected '{self.config.server_resource_id}', got '{claims.get('resource')}'"
217213
)
218214

219215
# Step 7: Validate client_id
220216
if claims.get("client_id") != expected_client_id:
221-
raise ValueError(
222-
f"client_id mismatch: expected '{expected_client_id}', " f"got '{claims.get('client_id')}'"
223-
)
217+
raise ValueError(f"client_id mismatch: expected '{expected_client_id}', got '{claims.get('client_id')}'")
224218

225219
# Step 8: Check for replay (if enabled)
226220
jti = claims.get("jti")

0 commit comments

Comments
 (0)