Skip to content

Commit e628977

Browse files
committed
cancel lingering MCP session task to prevent event loop spin on exit
1 parent 5dd6ec2 commit e628977

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

src/seclab_taskflow_agent/runner.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -475,6 +475,14 @@ async def _run_streamed() -> None:
475475
continue
476476
except Exception:
477477
logging.exception("Exception in mcp server cleanup task")
478+
# Cancel the MCP session task if it's still running to prevent
479+
# the asyncio event loop from spinning on a dangling task.
480+
if not mcp_sessions.done():
481+
mcp_sessions.cancel()
482+
try:
483+
await mcp_sessions
484+
except (asyncio.CancelledError, Exception):
485+
pass
478486

479487

480488
async def run_main(

0 commit comments

Comments
 (0)