WIP

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 A2A_REASON_TO_ERROR
+from a2a.utils.errors import InvalidParamsError
 from a2a.utils.telemetry import SpanKind, trace_class
 
 
@@ -74,8 +74,7 @@ def _map_grpc_error(e: grpc.aio.AioRpcError) -> NoReturn:
                     for v in bad_request.field_violations
                 ]
                 data = {'errors': errors}
-                # Infer InvalidParamsError from BadRequest details
-                exception_cls = A2A_REASON_TO_ERROR.get('INVALID_PARAMS')
+                exception_cls = InvalidParamsError
             elif detail.Is(error_details_pb2.ErrorInfo.DESCRIPTOR):
                 error_info = error_details_pb2.ErrorInfo()
                 detail.Unpack(error_info)

src/a2a/server/request_handlers/grpc_handler.py

@@ -441,8 +441,7 @@ async def abort_context(
             # Create standard Status
             status = status_pb2.Status(code=status_code, message=error_msg)
 
-            # Exclusive details based on error type:
-            if error.data and error.data.get('errors'):
+            if isinstance(error, types.InvalidParamsError) and error.data and error.data.get('errors'):
                 bad_request = error_details_pb2.BadRequest()
                 for err_dict in error.data['errors']:
                     violation = bad_request.field_violations.add()

tests/integration/test_end_to_end.py

@@ -570,11 +570,22 @@ async def test_end_to_end_validation_errors(
 ) -> None:
     client = transport_setups.client
 
-    with pytest.raises(InvalidParamsError) as exc_info:
+    try:
         async for _ in client.send_message(request=empty_request):
             pass
-
-    errors = exc_info.value.data.get('errors', [])
-    assert {e['field'] for e in errors} == set(expected_fields)
+    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')
 
     await client.close()