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

Streamable Http - clean up server memory streams (#604)

Browse source
This commit is contained in:
ihrpr 2025-05-02 14:59:17 +01:00 • committed by GitHub
commit 5d8eaf77be
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 108 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/shared/test_streamable_http.py
View file

@ -234,29 +234,30 @@ def create_app(
event_store=event_store,
)
async with http_transport.connect() as streams:
read_stream, write_stream = streams
async def run_server():
async def run_server(task_status=None):
async with http_transport.connect() as streams:
read_stream, write_stream = streams
if task_status:
task_status.started()
await server.run(
read_stream,
write_stream,
server.create_initialization_options(),
)
if task_group is None:
response = Response(
"Internal Server Error: Task group is not initialized",
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
)
await response(scope, receive, send)
return
if task_group is None:
response = Response(
"Internal Server Error: Task group is not initialized",
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
)
await response(scope, receive, send)
return
# Store the instance before starting the task to prevent races
server_instances[http_transport.mcp_session_id] = http_transport
task_group.start_soon(run_server)
# Store the instance before starting the task to prevent races
server_instances[http_transport.mcp_session_id] = http_transport
await task_group.start(run_server)
await http_transport.handle_request(scope, receive, send)
await http_transport.handle_request(scope, receive, send)
else:
response = Response(
"Bad Request: No valid session ID provided",