Handle asyncio.cancellederror and raise it so it propagates back

Author: rajeshvelicheti

src/a2a/utils/telemetry.py

@@ -53,6 +53,7 @@ def internal_method(self):
     ```
 """
 
+import asyncio
 import functools
 import inspect
 import logging
@@ -167,6 +168,12 @@ async def async_wrapper(*args, **kwargs) -> Any:
                 span.record_exception(e)
                 span.set_status(StatusCode.ERROR, description=str(e))
                 raise
+            # asyncio.CancelledError extends from BaseException
+            except asyncio.CancelledError as ce:
+                logger.debug(f'CancelledError in span {actual_span_name}')
+                span.record_exception(ce)
+                span.set_status(StatusCode.ERROR, description=str(ce))
+                raise
             finally:
                 if attribute_extractor:
                     try: