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)