From 543784c362308038f75bfba8f94d4c30969548df Mon Sep 17 00:00:00 2001 From: Paulchen Date: Tue, 1 Jul 2025 11:56:57 +0900 Subject: [PATCH 1/3] chore: sort imports in `tests/client/test_auth_middleware.py` --- tests/client/test_auth_middleware.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/client/test_auth_middleware.py b/tests/client/test_auth_middleware.py index e03b97596..38faaf1c6 100644 --- a/tests/client/test_auth_middleware.py +++ b/tests/client/test_auth_middleware.py @@ -7,9 +7,9 @@ from a2a.client import A2AClient, ClientCallContext, ClientCallInterceptor from a2a.client.auth import AuthInterceptor, InMemoryContextCredentialStore from a2a.types import ( - APIKeySecurityScheme, AgentCapabilities, AgentCard, + APIKeySecurityScheme, AuthorizationCodeOAuthFlow, In, OAuth2SecurityScheme, From 163f89817cc2e96bc1b6bdc2165d4bf1ea7c6f84 Mon Sep 17 00:00:00 2001 From: Paulchen Date: Tue, 1 Jul 2025 11:59:05 +0900 Subject: [PATCH 2/3] chore: update message types used in `tests/client/test_auth_middleware.py` --- tests/client/test_auth_middleware.py | 59 +++++++++++++++------------- 1 file changed, 31 insertions(+), 28 deletions(-) diff --git a/tests/client/test_auth_middleware.py b/tests/client/test_auth_middleware.py index 38faaf1c6..cc77ca35d 100644 --- a/tests/client/test_auth_middleware.py +++ b/tests/client/test_auth_middleware.py @@ -12,9 +12,12 @@ APIKeySecurityScheme, AuthorizationCodeOAuthFlow, In, + Message, + MessageSendParams, OAuth2SecurityScheme, OAuthFlows, OpenIdConnectSecurityScheme, + Role, SecurityScheme, SendMessageRequest, ) @@ -76,13 +79,13 @@ async def test_client_with_simple_interceptor(): await client.send_message( request=SendMessageRequest( id='1', - params={ - 'message': { - 'messageId': 'msg1', - 'role': 'user', - 'parts': [], - } - }, + params=MessageSendParams( + message=Message( + messageId='msg1', + role=Role.user, + parts=[], + ) + ), ) ) @@ -198,13 +201,13 @@ async def test_auth_interceptor_with_api_key(): await client.send_message( request=SendMessageRequest( id='1', - params={ - 'message': { - 'messageId': 'msg1', - 'role': 'user', - 'parts': [], - } - }, + params=MessageSendParams( + message=Message( + messageId='msg1', + role=Role.user, + parts=[], + ) + ), ), context=context, ) @@ -284,13 +287,13 @@ async def test_auth_interceptor_with_oauth2_scheme(): await client.send_message( request=SendMessageRequest( id='oauth_test_1', - params={ - 'message': { - 'messageId': 'msg-oauth', - 'role': 'user', - 'parts': [], - } - }, + params=MessageSendParams( + message=Message( + messageId='msg-oauth', + role=Role.user, + parts=[], + ) + ), ), context=context, ) @@ -366,13 +369,13 @@ async def test_auth_interceptor_with_oidc_scheme(): await client.send_message( request=SendMessageRequest( id='oidc_test_1', - params={ - 'message': { - 'messageId': 'msg-oidc', - 'role': 'user', - 'parts': [], - } - }, + params=MessageSendParams( + message=Message( + messageId='msg-oidc', + role=Role.user, + parts=[], + ) + ), ), context=context, ) From 1514eb4364f2fa5c03cfd2af8599bde7852a7ecc Mon Sep 17 00:00:00 2001 From: Paulchen Date: Tue, 1 Jul 2025 14:00:23 +0900 Subject: [PATCH 3/3] run `nox -s format ` --- tests/client/test_auth_middleware.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/client/test_auth_middleware.py b/tests/client/test_auth_middleware.py index cc77ca35d..94c0abdef 100644 --- a/tests/client/test_auth_middleware.py +++ b/tests/client/test_auth_middleware.py @@ -7,9 +7,9 @@ from a2a.client import A2AClient, ClientCallContext, ClientCallInterceptor from a2a.client.auth import AuthInterceptor, InMemoryContextCredentialStore from a2a.types import ( + APIKeySecurityScheme, AgentCapabilities, AgentCard, - APIKeySecurityScheme, AuthorizationCodeOAuthFlow, In, Message,