From 1134421a3af7950e609f2b595bf18296d4626b16 Mon Sep 17 00:00:00 2001 From: Allen Porter Date: Mon, 30 Dec 2024 10:08:57 -0800 Subject: [PATCH] Fix inconsistencies in examples --- examples/servers/simple-prompt/mcp_simple_prompt/server.py | 2 +- src/mcp/server/sse.py | 5 +---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/servers/simple-prompt/mcp_simple_prompt/server.py b/examples/servers/simple-prompt/mcp_simple_prompt/server.py index 351099f..89b0eea 100644 --- a/examples/servers/simple-prompt/mcp_simple_prompt/server.py +++ b/examples/servers/simple-prompt/mcp_simple_prompt/server.py @@ -106,7 +106,7 @@ def main(port: int, transport: str) -> int: debug=True, routes=[ Route("/sse", endpoint=handle_sse), - Mount("/messages/", app=sse.handle_post_message) + Mount("/messages/", app=sse.handle_post_message), ], ) diff --git a/src/mcp/server/sse.py b/src/mcp/server/sse.py index 0f50023..96410bb 100644 --- a/src/mcp/server/sse.py +++ b/src/mcp/server/sse.py @@ -11,7 +11,7 @@ Example usage: # Create Starlette routes for SSE and message handling routes = [ Route("/sse", endpoint=handle_sse), - Mount("/messages", endpoint=handle_messages, methods=["POST"]) + Mount("/messages", app=sse.handle_post_message), ] # Define handler functions @@ -23,9 +23,6 @@ Example usage: streams[0], streams[1], app.create_initialization_options() ) - async def handle_messages(request): - await sse.handle_post_message(request.scope, request.receive, request._send) - # Create and run Starlette app starlette_app = Starlette(routes=routes) uvicorn.run(starlette_app, host="0.0.0.0", port=port)