Attempt to fix flakiness

Author: ishymko

src/a2a/server/request_handlers/request_handler.py

@@ -241,8 +241,17 @@ async def async_gen_wrapper(
         ) -> Any:
             if params is not None:
                 validate_proto_required_fields(params)
-            async for item in method(self, params, context, *args, **kwargs):
-                yield item
+            # Explicitly close the inner async generator in a finally block
+            # so that its cleanup (except/finally) runs deterministically
+            # during aclose(). On Python < 3.13, async-for does NOT call
+            # aclose() on the iterator when an exception (e.g. GeneratorExit)
+            # propagates through the loop body, leaving cleanup to the GC.
+            inner = method(self, params, context, *args, **kwargs)
+            try:
+                async for item in inner:
+                    yield item
+            finally:
+                await inner.aclose()
 
         return async_gen_wrapper