modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle SSE Disconnects Properly (#612)

Browse source
This commit is contained in:
Akash D 2025-05-02 09:32:46 -07:00 • committed by GitHub
commit 83968b5b2f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 38 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/servers/simple-prompt/mcp_simple_prompt/server.py
View file

@ -90,6 +90,7 @@ def main(port: int, transport: str) -> int:
if transport == "sse":
from mcp.server.sse import SseServerTransport
from starlette.applications import Starlette
from starlette.responses import Response
from starlette.routing import Mount, Route
sse = SseServerTransport("/messages/")
@ -101,6 +102,7 @@ def main(port: int, transport: str) -> int:
await app.run(
streams[0], streams[1], app.create_initialization_options()
)
return Response()
starlette_app = Starlette(
debug=True,

4
examples/servers/simple-resource/mcp_simple_resource/server.py
View file

@ -46,6 +46,7 @@ def main(port: int, transport: str) -> int:
if transport == "sse":
from mcp.server.sse import SseServerTransport
from starlette.applications import Starlette
from starlette.responses import Response
from starlette.routing import Mount, Route
sse = SseServerTransport("/messages/")
@ -57,11 +58,12 @@ def main(port: int, transport: str) -> int:
await app.run(
streams[0], streams[1], app.create_initialization_options()
)
return Response()
starlette_app = Starlette(
debug=True,
routes=[
Route("/sse", endpoint=handle_sse),
Route("/sse", endpoint=handle_sse, methods=["GET"]),
Mount("/messages/", app=sse.handle_post_message),
],
)

4
examples/servers/simple-tool/mcp_simple_tool/server.py
View file

@ -60,6 +60,7 @@ def main(port: int, transport: str) -> int:
if transport == "sse":
from mcp.server.sse import SseServerTransport
from starlette.applications import Starlette
from starlette.responses import Response
from starlette.routing import Mount, Route
sse = SseServerTransport("/messages/")
@ -71,11 +72,12 @@ def main(port: int, transport: str) -> int:
await app.run(
streams[0], streams[1], app.create_initialization_options()
)
return Response()
starlette_app = Starlette(
debug=True,
routes=[
Route("/sse", endpoint=handle_sse),
Route("/sse", endpoint=handle_sse, methods=["GET"]),
Mount("/messages/", app=sse.handle_post_message),
],
)