Formatting

Author: holtskinner

tests/client/test_auth_middleware.py

@@ -6,10 +6,18 @@
 
 from a2a.client import A2AClient, ClientCallContext, ClientCallInterceptor
 from a2a.client.auth import AuthInterceptor, InMemoryContextCredentialStore
-from a2a.types import (AgentCapabilities, AgentCard, APIKeySecurityScheme,
-                       AuthorizationCodeOAuthFlow, In, OAuth2SecurityScheme,
-                       OAuthFlows, OpenIdConnectSecurityScheme, SecurityScheme,
-                       SendMessageRequest)
+from a2a.types import (
+    APIKeySecurityScheme,
+    AgentCapabilities,
+    AgentCard,
+    AuthorizationCodeOAuthFlow,
+    In,
+    OAuth2SecurityScheme,
+    OAuthFlows,
+    OpenIdConnectSecurityScheme,
+    SecurityScheme,
+    SendMessageRequest,
+)
 
 
 # A simple mock interceptor for testing basic middleware functionality
@@ -321,9 +329,7 @@ async def test_auth_interceptor_with_oidc_scheme():
         defaultOutputModes=[],
         skills=[],
         capabilities=AgentCapabilities(),
-        security=[
-            {scheme_name: []}
-        ],
+        security=[{scheme_name: []}],
         securitySchemes={
             scheme_name: SecurityScheme(
                 root=OpenIdConnectSecurityScheme(
@@ -375,4 +381,4 @@ async def test_auth_interceptor_with_oidc_scheme():
         assert len(respx.calls) == 1
         request_sent = respx.calls.last.request
         assert 'Authorization' in request_sent.headers
-        assert request_sent.headers['Authorization'] == f'Bearer {id_token}'
+        assert request_sent.headers['Authorization'] == f'Bearer {id_token}'