misc fixes

Author: akash329d

examples/servers/simple-prompt/mcp_simple_prompt/server.py

@@ -2,8 +2,6 @@
 import click
 import mcp.types as types
 from mcp.server.lowlevel import Server
-from starlette.responses import Response
-from starlette.routing import Route
 
 
 def create_messages(
@@ -92,7 +90,8 @@ async def get_prompt(
     if transport == "sse":
         from mcp.server.sse import SseServerTransport
         from starlette.applications import Starlette
-        from starlette.routing import Mount
+        from starlette.responses import Response
+        from starlette.routing import Mount, Route
 
         sse = SseServerTransport("/messages/")
 

src/mcp/server/sse.py

@@ -31,8 +31,8 @@ async def handle_sse(request):
 ```
 
 Note: The handle_sse function must return a Response to avoid a "TypeError: 'NoneType'
-object is not callable" error when client disconnects. The example above shows the
-correct approach by returning an empty Response() after the SSE connection ends.
+object is not callable" error when client disconnects. The example above returns
+an empty Response() after the SSE connection ends to fix this.
 
 See SseServerTransport class documentation for more details.
 """

tests/shared/test_sse.py

@@ -9,6 +9,7 @@
 import uvicorn
 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
 
@@ -83,7 +84,7 @@ def make_server_app() -> Starlette:
     sse = SseServerTransport("/messages/")
     server = ServerTest()
 
-    async def handle_sse(request) -> None:
+    async def handle_sse(request: Request) -> Response:
         async with sse.connect_sse(
             request.scope, request.receive, request._send
         ) as streams:
@@ -189,7 +190,7 @@ async def connection_test() -> None:
 
 @pytest.mark.anyio
 async def test_sse_client_basic_connection(server: None, server_url: str) -> None:
-    async with sse_client(server_url + "/sse/") as streams:
+    async with sse_client(server_url + "/sse") as streams:
         async with ClientSession(*streams) as session:
             # Test initialization
             result = await session.initialize()
@@ -205,7 +206,7 @@ async def test_sse_client_basic_connection(server: None, server_url: str) -> Non
 async def initialized_sse_client_session(
     server, server_url: str
 ) -> AsyncGenerator[ClientSession, None]:
-    async with sse_client(server_url + "/sse/", sse_read_timeout=0.5) as streams:
+    async with sse_client(server_url + "/sse", sse_read_timeout=0.5) as streams:
         async with ClientSession(*streams) as session:
             await session.initialize()
             yield session