update examples

Author: PWZER

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

@@ -95,12 +95,16 @@ async def get_prompt(
         sse = SseServerTransport("/messages/")
 
         async def handle_sse(request):
-            async with sse.connect_sse(
-                request.scope, request.receive, request._send
-            ) as streams:
-                await app.run(
-                    streams[0], streams[1], app.create_initialization_options()
-                )
+            with anyio.CancelScope() as cancel_scope:
+                async with sse.connect_sse(
+                    request.scope,
+                    request.receive,
+                    request._send,
+                    lambda: cancel_scope.cancel(),
+                ) as streams:
+                    await app.run(
+                        streams[0], streams[1], app.create_initialization_options()
+                    )
 
         starlette_app = Starlette(
             debug=True,

examples/servers/simple-resource/mcp_simple_resource/server.py

@@ -51,12 +51,16 @@ async def read_resource(uri: FileUrl) -> str | bytes:
         sse = SseServerTransport("/messages/")
 
         async def handle_sse(request):
-            async with sse.connect_sse(
-                request.scope, request.receive, request._send
-            ) as streams:
-                await app.run(
-                    streams[0], streams[1], app.create_initialization_options()
-                )
+            with anyio.CancelScope() as cancel_scope:
+                async with sse.connect_sse(
+                    request.scope,
+                    request.receive,
+                    request._send,
+                    lambda: cancel_scope.cancel(),
+                ) as streams:
+                    await app.run(
+                        streams[0], streams[1], app.create_initialization_options()
+                    )
 
         starlette_app = Starlette(
             debug=True,