Fix SSE server bug with uncaught TypeError

This commit is contained in:
Allen Porter
2024-12-30 10:04:19 -08:00
parent 59fff69c38
commit 0970ef4ae0
5 changed files with 13 additions and 25 deletions

View File

@@ -47,9 +47,9 @@ def main(port: int, transport: str) -> int:
if transport == "sse":
from mcp.server.sse import SseServerTransport
from starlette.applications import Starlette
from starlette.routing import Route
from starlette.routing import Route, Mount
sse = SseServerTransport("/messages")
sse = SseServerTransport("/messages/")
async def handle_sse(request):
async with sse.connect_sse(
@@ -59,14 +59,11 @@ def main(port: int, transport: str) -> int:
streams[0], streams[1], app.create_initialization_options()
)
async def handle_messages(request):
await sse.handle_post_message(request.scope, request.receive, request._send)
starlette_app = Starlette(
debug=True,
routes=[
Route("/sse", endpoint=handle_sse),
Route("/messages", endpoint=handle_messages, methods=["POST"]),
Mount("/messages/", app=sse.handle_post_message),
],
)