fix pyright

Author: PWZER

src/mcp/server/sse.py

@@ -44,7 +44,7 @@ async def handle_sse(request):
 from sse_starlette import EventSourceResponse
 from starlette.requests import Request
 from starlette.responses import Response
-from starlette.types import Receive, Scope, Send, Message
+from starlette.types import Message, Receive, Scope, Send
 
 import mcp.types as types
 
@@ -139,7 +139,7 @@ async def client_close_handler(message: Message) -> None:
             response = EventSourceResponse(
                 content=sse_stream_reader,
                 data_sender_callable=sse_writer,
-                client_close_handler_callable=client_close_handler,
+                client_close_handler_callable=client_close_handler,  # type: ignore[arg-type]
             )
             logger.debug("Starting SSE response task")
             tg.start_soon(response, scope, receive, send)

tests/shared/test_sse.py

@@ -10,6 +10,7 @@
 from pydantic import AnyUrl
 from starlette.applications import Starlette
 from starlette.requests import Request
+from starlette.responses import Response
 from starlette.routing import Mount, Route
 
 from mcp.client.session import ClientSession
@@ -83,13 +84,18 @@ def make_server_app() -> Starlette:
     sse = SseServerTransport("/messages/")
     server = ServerTest()
 
-    async def handle_sse(request: Request) -> None:
-        async with sse.connect_sse(
-            request.scope, request.receive, request._send
-        ) as streams:
-            await server.run(
-                streams[0], streams[1], server.create_initialization_options()
-            )
+    async def handle_sse(request: Request) -> Response:
+        with anyio.CancelScope() as cancel_scope:
+            async with sse.connect_sse(
+                request.scope,
+                request.receive,
+                request._send,
+                lambda: cancel_scope.cancel(),
+            ) as streams:
+                await server.run(
+                    streams[0], streams[1], server.create_initialization_options()
+                )
+        return Response(status_code=200)
 
     app = Starlette(
         routes=[