Fix example servers SSE invocation

Reported-by: Simon Willison
This commit is contained in:
David Soria Parra
2024-11-22 10:45:24 +00:00
parent 6c46bd283a
commit bbd9d05851
3 changed files with 60 additions and 52 deletions

View File

@@ -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