Remove usages of asyncio in tests

Author: LucaButBoring

examples/clients/async-reconnect-client/mcp_async_reconnect_client/client.py

@@ -1,5 +1,4 @@
-import asyncio
-
+import anyio
 import click
 from mcp import ClientSession, types
 from mcp.client.streamable_http import streamablehttp_client
@@ -30,7 +29,7 @@ async def call_async_tool(session: ClientSession, token: str | None):
             print(f"Operation failed: {status.error}")
             break
 
-        await asyncio.sleep(0.5)
+        await anyio.sleep(0.5)
 
 
 async def run_session(endpoint: str, token: str | None):
@@ -44,4 +43,4 @@ async def run_session(endpoint: str, token: str | None):
 @click.option("--endpoint", default="http://127.0.0.1:8000/mcp", help="Endpoint to connect to")
 @click.option("--token", default=None, help="Operation token to resume with")
 def main(endpoint: str, token: str | None):
-    asyncio.run(run_session(endpoint, token))
+    anyio.run(run_session, endpoint, token)

examples/servers/simple-tool-async/mcp_simple_tool_async/server.py

@@ -1,5 +1,4 @@
-import asyncio
-
+import anyio
 import click
 import mcp.types as types
 import uvicorn
@@ -15,7 +14,7 @@ async def fetch_website(
 ) -> list[types.ContentBlock]:
     headers = {"User-Agent": "MCP Test Server (github.com/modelcontextprotocol/python-sdk)"}
     async with create_mcp_http_client(headers=headers) as client:
-        await asyncio.sleep(5)
+        await anyio.sleep(5)
         response = await client.get(url)
         response.raise_for_status()
         return [types.TextContent(type="text", text=response.text)]

examples/snippets/clients/async_elicitation_client.py

@@ -5,9 +5,10 @@
     uv run async-elicitation-client
 """
 
-import asyncio
 import os
 
+import anyio
+
 from mcp import ClientSession, StdioServerParameters, types
 from mcp.client.stdio import stdio_client
 from mcp.shared.context import RequestContext
@@ -70,7 +71,7 @@ async def test_process_with_confirmation(session: ClientSession):
                 print(f"Operation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
 
 async def test_file_operation(session: ClientSession):
@@ -97,7 +98,7 @@ async def test_file_operation(session: ClientSession):
                 print(f"File operation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
 
 async def run():
@@ -115,4 +116,4 @@ async def run():
 
 
 if __name__ == "__main__":
-    asyncio.run(run())
+    anyio.run(run)

examples/snippets/clients/async_progress_client.py

@@ -5,9 +5,10 @@
     uv run async-progress-client
 """
 
-import asyncio
 import os
 
+import anyio
+
 from mcp import ClientSession, StdioServerParameters, types
 from mcp.client.stdio import stdio_client
 
@@ -58,7 +59,7 @@ async def progress_callback(progress: float, total: float | None, message: str |
                 print(f"Operation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
     print(f"Received {len(progress_updates)} progress updates")
 
@@ -91,7 +92,7 @@ async def test_data_pipeline(session: ClientSession):
                 print(f"Pipeline failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
 
 async def run():
@@ -107,4 +108,4 @@ async def run():
 
 
 if __name__ == "__main__":
-    asyncio.run(run())
+    anyio.run(run)

examples/snippets/clients/async_sampling_client.py

@@ -5,9 +5,10 @@
     uv run async-sampling-client
 """
 
-import asyncio
 import os
 
+import anyio
+
 from mcp import ClientSession, StdioServerParameters, types
 from mcp.client.stdio import stdio_client
 from mcp.shared.context import RequestContext
@@ -77,7 +78,7 @@ async def test_content_generation(session: ClientSession):
                 print(f"Generation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
 
 async def test_multi_step_generation(session: ClientSession):
@@ -107,7 +108,7 @@ async def test_multi_step_generation(session: ClientSession):
                 print(f"Multi-step generation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.3)
+            await anyio.sleep(0.3)
 
 
 async def run():
@@ -123,4 +124,4 @@ async def run():
 
 
 if __name__ == "__main__":
-    asyncio.run(run())
+    anyio.run(run)

examples/snippets/clients/async_tool_client.py

@@ -5,9 +5,10 @@
     uv run async-tool-client
 """
 
-import asyncio
 import os
 
+import anyio
+
 from mcp import ClientSession, StdioServerParameters, types
 from mcp.client.stdio import stdio_client
 
@@ -44,7 +45,7 @@ async def call_async_tool(session: ClientSession):
                 print(f"Operation failed: {status.error}")
                 break
 
-            await asyncio.sleep(0.5)
+            await anyio.sleep(0.5)
 
 
 async def run():
@@ -56,4 +57,4 @@ async def run():
 
 
 if __name__ == "__main__":
-    asyncio.run(run())
+    anyio.run(run)

examples/snippets/servers/async_tool_basic.py

@@ -5,21 +5,22 @@
     uv run server async_tool_basic stdio
 """
 
-import asyncio
+import anyio
 
 from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSession
 
 mcp = FastMCP("Async Tool Basic")
 
 
 @mcp.tool(invocation_modes=["async"])
-async def analyze_data(dataset: str, ctx: Context) -> str:  # type: ignore[type-arg]
+async def analyze_data(dataset: str, ctx: Context[ServerSession, None]) -> str:
     """Analyze a dataset asynchronously with progress updates."""
     await ctx.info(f"Starting analysis of {dataset}")
 
     # Simulate analysis with progress updates
     for i in range(5):
-        await asyncio.sleep(0.5)
+        await anyio.sleep(0.5)
         progress = (i + 1) / 5
         await ctx.report_progress(progress, 1.0, f"Processing step {i + 1}/5")
 
@@ -28,21 +29,11 @@ async def analyze_data(dataset: str, ctx: Context) -> str:  # type: ignore[type-
 
 
 @mcp.tool(invocation_modes=["sync", "async"])
-def process_text(text: str, ctx: Context | None = None) -> str:  # type: ignore[type-arg]
+async def process_text(text: str, ctx: Context[ServerSession, None]) -> str:
     """Process text in sync or async mode."""
-    if ctx:
-        # Async mode with context
-        import asyncio
-
-        async def async_processing():
-            await ctx.info(f"Processing text asynchronously: {text[:20]}...")
-            await asyncio.sleep(0.3)
-
-        try:
-            loop = asyncio.get_event_loop()
-            loop.create_task(async_processing())
-        except RuntimeError:
-            pass
+
+    await ctx.info(f"Processing text asynchronously: {text[:20]}...")
+    await anyio.sleep(0.3)
 
     return f"Processed: {text.upper()}"
 

examples/snippets/servers/async_tool_elicitation.py

@@ -5,11 +5,11 @@
     uv run server async_tool_elicitation stdio
 """
 
-import asyncio
-
+import anyio
 from pydantic import BaseModel, Field
 
 from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSession
 
 mcp = FastMCP("Async Tool Elicitation")
 
@@ -32,12 +32,12 @@ class FileOperationChoice(BaseModel):
 
 
 @mcp.tool(invocation_modes=["async"])
-async def process_with_confirmation(operation: str, ctx: Context) -> str:  # type: ignore[type-arg]
+async def process_with_confirmation(operation: str, ctx: Context[ServerSession, None]) -> str:
     """Process an operation that requires user confirmation."""
     await ctx.info(f"Starting operation: {operation}")
 
     # Simulate some initial processing
-    await asyncio.sleep(0.5)
+    await anyio.sleep(0.5)
     await ctx.report_progress(0.3, 1.0, "Initial processing complete")
 
     # Ask user for preferences
@@ -51,7 +51,7 @@ async def process_with_confirmation(operation: str, ctx: Context) -> str:  # typ
             await ctx.info(f"Continuing with {result.data.priority_level} priority")
             # Simulate processing based on user choice
             processing_time = {"low": 0.5, "normal": 1.0, "high": 1.5}.get(result.data.priority_level, 1.0)
-            await asyncio.sleep(processing_time)
+            await anyio.sleep(processing_time)
             await ctx.report_progress(1.0, 1.0, "Operation complete")
             return f"Operation '{operation}' completed successfully with {result.data.priority_level} priority"
         else:
@@ -63,12 +63,12 @@ async def process_with_confirmation(operation: str, ctx: Context) -> str:  # typ
 
 
 @mcp.tool(invocation_modes=["async"])
-async def file_operation(file_path: str, operation_type: str, ctx: Context) -> str:  # type: ignore[type-arg]
+async def file_operation(file_path: str, operation_type: str, ctx: Context[ServerSession, None]) -> str:
     """Perform file operation with user confirmation."""
     await ctx.info(f"Analyzing file: {file_path}")
 
     # Simulate initial analysis
-    await asyncio.sleep(1)
+    await anyio.sleep(1)
     await ctx.report_progress(0.3, 1.0, "File analysis complete")
 
     # Simulate finding something that requires user confirmation
@@ -84,11 +84,11 @@ async def file_operation(file_path: str, operation_type: str, ctx: Context) -> s
         if result.data.confirm_operation:
             if result.data.backup_first:
                 await ctx.info("Creating backup first...")
-                await asyncio.sleep(0.5)
+                await anyio.sleep(0.5)
                 await ctx.report_progress(0.7, 1.0, "Backup created")
 
             await ctx.info(f"Performing {operation_type} operation...")
-            await asyncio.sleep(1)
+            await anyio.sleep(1)
             await ctx.report_progress(1.0, 1.0, "Operation complete")
 
             backup_msg = " (with backup)" if result.data.backup_first else " (no backup)"

examples/snippets/servers/async_tool_immediate.py

@@ -5,10 +5,11 @@
     uv run server async_tool_immediate stdio
 """
 
-import asyncio
+import anyio
 
 from mcp import types
 from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSession
 
 mcp = FastMCP("Async Tool Immediate")
 
@@ -19,13 +20,13 @@ async def provide_immediate_feedback(operation: str) -> list[types.ContentBlock]
 
 
 @mcp.tool(invocation_modes=["async"], immediate_result=provide_immediate_feedback)
-async def long_analysis(operation: str, ctx: Context) -> str:  # type: ignore[type-arg]
+async def long_analysis(operation: str, ctx: Context[ServerSession, None]) -> str:
     """Perform long-running analysis with immediate user feedback."""
     await ctx.info(f"Beginning {operation} analysis")
 
     # Simulate long-running work
     for i in range(4):
-        await asyncio.sleep(1)
+        await anyio.sleep(1)
         progress = (i + 1) / 4
         await ctx.report_progress(progress, 1.0, f"Analysis step {i + 1}/4")
 

examples/snippets/servers/async_tool_progress.py

@@ -5,15 +5,16 @@
     uv run server async_tool_progress stdio
 """
 
-import asyncio
+import anyio
 
 from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSession
 
 mcp = FastMCP("Async Tool Progress")
 
 
 @mcp.tool(invocation_modes=["async"])
-async def batch_process(items: list[str], ctx: Context) -> list[str]:  # type: ignore[type-arg]
+async def batch_process(items: list[str], ctx: Context[ServerSession, None]) -> list[str]:
     """Process a batch of items with detailed progress reporting."""
     await ctx.info(f"Starting batch processing of {len(items)} items")
 
@@ -24,7 +25,7 @@ async def batch_process(items: list[str], ctx: Context) -> list[str]:  # type: i
 
         # Simulate variable processing time
         processing_time = 0.3 + (len(item) * 0.1)
-        await asyncio.sleep(processing_time)
+        await anyio.sleep(processing_time)
 
         # Report progress for this item
         progress = (i + 1) / len(items)
@@ -41,7 +42,7 @@ async def batch_process(items: list[str], ctx: Context) -> list[str]:  # type: i
 
 
 @mcp.tool(invocation_modes=["async"])
-async def data_pipeline(dataset: str, operations: list[str], ctx: Context) -> dict[str, str]:  # type: ignore[type-arg]
+async def data_pipeline(dataset: str, operations: list[str], ctx: Context[ServerSession, None]) -> dict[str, str]:
     """Execute a data processing pipeline with progress updates."""
     await ctx.info(f"Starting data pipeline for {dataset}")
 
@@ -53,7 +54,7 @@ async def data_pipeline(dataset: str, operations: list[str], ctx: Context) -> di
 
         # Simulate processing time that increases with complexity
         processing_time = 0.5 + (i * 0.2)
-        await asyncio.sleep(processing_time)
+        await anyio.sleep(processing_time)
 
         # Report progress
         progress = (i + 1) / total_ops