Cosmetics

Author: ishymko

src/a2a/client/transports/grpc.py

@@ -47,7 +47,7 @@
     TaskPushNotificationConfig,
 )
 from a2a.utils.constants import PROTOCOL_VERSION_CURRENT, VERSION_HEADER
-from a2a.utils.errors import InvalidParamsError
+from a2a.utils.errors import A2A_REASON_TO_ERROR, A2AError, InvalidParamsError
 from a2a.utils.telemetry import SpanKind, trace_class
 
 
@@ -64,7 +64,7 @@ def _map_grpc_error(e: grpc.aio.AioRpcError) -> NoReturn:
     data = None
 
     if status is not None:
-        exception_cls = None
+        exception_cls: type[A2AError] | None = None
         for detail in status.details:
             if detail.Is(error_details_pb2.BadRequest.DESCRIPTOR):
                 bad_request = error_details_pb2.BadRequest()

src/a2a/client/transports/http_helpers.py

@@ -40,6 +40,11 @@ def handle_http_exceptions(
         raise A2AClientError(f'Network communication error: {e}') from e
     except json.JSONDecodeError as e:
         raise A2AClientError(f'JSON Decode Error: {e}') from e
+    except Exception as e:
+        # ASGITransport propagates local server-side generator crashes as ExceptionGroups
+        if hasattr(e, 'exceptions') and len(e.exceptions) == 1:
+            raise e.exceptions[0] from e
+        raise e
 
 
 def get_http_args(context: ClientCallContext | None) -> dict[str, Any]:

tests/integration/test_end_to_end.py

@@ -555,37 +555,26 @@ async def test_end_to_end_input_required(transport_setups):
     [
         (
             SendMessageRequest(),
-            ['message'],
+            {'message'},
         ),
         (
             SendMessageRequest(message=Message()),
-            ['message.message_id', 'message.role', 'message.parts'],
+            {'message.message_id', 'message.role', 'message.parts'},
         ),
     ],
 )
 async def test_end_to_end_validation_errors(
     transport_setups,
     empty_request: SendMessageRequest,
-    expected_fields: list[str],
+    expected_fields: set[str],
 ) -> None:
     client = transport_setups.client
 
-    try:
+    with pytest.raises(InvalidParamsError) as exc_info:
         async for _ in client.send_message(request=empty_request):
             pass
-    except Exception as e:
-        # ASGITransport propagates server-side generator crashes as ExceptionGroups
-        exc = e
-        if hasattr(e, 'exceptions') and len(e.exceptions) == 1:
-            exc = e.exceptions[0]
-        
-        if not isinstance(exc, InvalidParamsError):
-            raise e
-            
-        errors = exc.data.get('errors', []) if exc.data else []
-        assert {e['field'] for e in errors} == set(expected_fields)
-        return
-
-    pytest.fail('InvalidParamsError was not raised')
+
+    errors = exc_info.value.data.get('errors', [])
+    assert {e['field'] for e in errors} == expected_fields
 
     await client.close()