Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/agents/result.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,6 +213,10 @@ async def stream_events(self) -> AsyncIterator[StreamEvent]:
yield item
self._event_queue.task_done()

# Ensure main execution completes before cleanup to avoid race conditions
# with session operations
await self._await_task_safely(self._run_impl_task)
# Safely terminate all background tasks after main execution has finished
self._cleanup_tasks()

if self._stored_exception:
Expand Down