test: Migrate load test framework to anyio as well

This commit is contained in:
Chris Coutinho
2025-10-18 22:02:25 +02:00
parent 1459fe9bc8
commit 240ceb3808
4 changed files with 8 additions and 27 deletions
+2 -1
View File
@@ -18,6 +18,7 @@ from collections import Counter
from contextlib import asynccontextmanager
from typing import Any
import anyio
import click
from mcp import ClientSession
from mcp.client.streamable_http import streamablehttp_client
@@ -494,7 +495,7 @@ def main(
print(f"Results exported to: {output}")
try:
asyncio.run(run())
anyio.run(run)
except KeyboardInterrupt:
print("\nBenchmark interrupted by user")
sys.exit(130)