aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-19 14:54:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 5d8eaf77be
commit 83968b5b2f
7 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),
],
)