Skip to content

Commit 3169d76

Browse files
committed
fix validate function
1 parent c2de945 commit 3169d76

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

src/a2a/utils/helpers.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,8 +223,12 @@ async def async_gen_wrapper(self: Any, *args, **kwargs) -> Any:
223223
if final_message
224224
else error_type
225225
)
226-
async for item in function(self, *args, **kwargs):
227-
yield item
226+
inner = function(self, *args, **kwargs)
227+
try:
228+
async for item in inner:
229+
yield item
230+
finally:
231+
await inner.aclose()
228232

229233
return async_gen_wrapper
230234

0 commit comments

Comments
 (0)