diff --git a/examples/servers/simple-prompt/mcp_simple_prompt/server.py b/examples/servers/simple-prompt/mcp_simple_prompt/server.py index c97f870..55b58b5 100644 --- a/examples/servers/simple-prompt/mcp_simple_prompt/server.py +++ b/examples/servers/simple-prompt/mcp_simple_prompt/server.py @@ -4,23 +4,6 @@ import mcp.types as types from mcp.server import Server -@click.group() -def cli(): - pass - - -@cli.command() -@click.option("--port", default=8000, help="Port to listen on for SSE") -@click.option( - "--transport", - type=click.Choice(["stdio", "sse"]), - default="stdio", - help="Transport type", -) -def main(port: int, transport: str) -> int: - return anyio.run(_amain, port, transport) - - def create_messages( context: str | None = None, topic: str | None = None ) -> list[types.PromptMessage]: @@ -54,7 +37,19 @@ def create_messages( return messages -async def _amain(port: int, transport: str) -> int: +@click.group() +def cli(): + pass + + +@click.option("--port", default=8000, help="Port to listen on for SSE") +@click.option( + "--transport", + type=click.Choice(["stdio", "sse"]), + default="stdio", + help="Transport type", +) +def main(port: int, transport: str) -> int: app = Server("mcp-simple-prompt") @app.list_prompts() @@ -63,7 +58,7 @@ async def _amain(port: int, transport: str) -> int: types.Prompt( name="simple", description="A simple prompt that can take optional context and topic " - "arguments", + "arguments", arguments=[ types.PromptArgument( name="context", @@ -103,14 +98,16 @@ async def _amain(port: int, transport: str) -> int: sse = SseServerTransport("/messages") - async def handle_sse(scope, receive, send): - async with sse.connect_sse(scope, receive, send) as streams: + async def handle_sse(request): + async with sse.connect_sse( + request.scope, request.receive, request._send + ) as streams: await app.run( streams[0], streams[1], app.create_initialization_options() ) - async def handle_messages(scope, receive, send): - await sse.handle_post_message(scope, receive, send) + async def handle_messages(request): + await sse.handle_post_message(request.scope, request.receive, request._send) starlette_app = Starlette( debug=True, @@ -126,7 +123,12 @@ async def _amain(port: int, transport: str) -> int: else: from mcp.server.stdio import stdio_server - async with stdio_server() as streams: - await app.run(streams[0], streams[1], app.create_initialization_options()) + async def arun(): + async with stdio_server() as streams: + await app.run( + streams[0], streams[1], app.create_initialization_options() + ) + + anyio.run(arun) return 0 diff --git a/examples/servers/simple-resource/mcp_simple_resource/server.py b/examples/servers/simple-resource/mcp_simple_resource/server.py index 4cd0dca..8db1c71 100644 --- a/examples/servers/simple-resource/mcp_simple_resource/server.py +++ b/examples/servers/simple-resource/mcp_simple_resource/server.py @@ -4,6 +4,13 @@ import mcp.types as types from mcp.server import AnyUrl, Server +SAMPLE_RESOURCES = { + "greeting": "Hello! This is a sample text resource.", + "help": "This server provides a few sample text resources for testing.", + "about": "This is the simple-resource MCP server implementation.", +} + + @click.group() def cli(): pass @@ -18,17 +25,6 @@ def cli(): help="Transport type", ) def main(port: int, transport: str) -> int: - return anyio.run(_amain, port, transport) - - -SAMPLE_RESOURCES = { - "greeting": "Hello! This is a sample text resource.", - "help": "This server provides a few sample text resources for testing.", - "about": "This is the simple-resource MCP server implementation.", -} - - -async def _amain(port: int, transport: str) -> int: app = Server("mcp-simple-resource") @app.list_resources() @@ -60,14 +56,16 @@ async def _amain(port: int, transport: str) -> int: sse = SseServerTransport("/messages") - async def handle_sse(scope, receive, send): - async with sse.connect_sse(scope, receive, send) as streams: + async def handle_sse(request): + async with sse.connect_sse( + request.scope, request.receive, request._send + ) as streams: await app.run( streams[0], streams[1], app.create_initialization_options() ) - async def handle_messages(scope, receive, send): - await sse.handle_post_message(scope, receive, send) + async def handle_messages(request): + await sse.handle_post_message(request.scope, request.receive, request._send) starlette_app = Starlette( debug=True, @@ -83,7 +81,12 @@ async def _amain(port: int, transport: str) -> int: else: from mcp.server.stdio import stdio_server - async with stdio_server() as streams: - await app.run(streams[0], streams[1], app.create_initialization_options()) + async def arun(): + async with stdio_server() as streams: + await app.run( + streams[0], streams[1], app.create_initialization_options() + ) + + anyio.run(arun) return 0 diff --git a/examples/servers/simple-tool/mcp_simple_tool/server.py b/examples/servers/simple-tool/mcp_simple_tool/server.py index 7cff93e..a0f5b7b 100644 --- a/examples/servers/simple-tool/mcp_simple_tool/server.py +++ b/examples/servers/simple-tool/mcp_simple_tool/server.py @@ -31,10 +31,6 @@ def cli(): help="Transport type", ) def main(port: int, transport: str) -> int: - return anyio.run(_amain, port, transport) - - -async def _amain(port: int, transport: str) -> int: app = Server("mcp-website-fetcher") @app.call_tool() @@ -73,14 +69,16 @@ async def _amain(port: int, transport: str) -> int: sse = SseServerTransport("/messages") - async def handle_sse(scope, receive, send): - async with sse.connect_sse(scope, receive, send) as streams: + async def handle_sse(request): + async with sse.connect_sse( + request.scope, request.receive, request._send + ) as streams: await app.run( streams[0], streams[1], app.create_initialization_options() ) - async def handle_messages(scope, receive, send): - await sse.handle_post_message(scope, receive, send) + async def handle_messages(request): + await sse.handle_post_message(request.scope, request.receive, request._send) starlette_app = Starlette( debug=True, @@ -96,7 +94,12 @@ async def _amain(port: int, transport: str) -> int: else: from mcp.server.stdio import stdio_server - async with stdio_server() as streams: - await app.run(streams[0], streams[1], app.create_initialization_options()) + async def arun(): + async with stdio_server() as streams: + await app.run( + streams[0], streams[1], app.create_initialization_options() + ) + + anyio.run(arun) return 0