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

Fix streamable http sampling (#693)
Some checks failed
Main branch checks / checks (push) Has been cancelled
Details

Browse source
This commit is contained in:
ihrpr 2025-05-12 18:31:35 +01:00 • committed by GitHub
commit c6fb822c86
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 152 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/mcp/server/streamable_http.py
View file

@ -33,7 +33,6 @@ from mcp.types import (
ErrorData,
JSONRPCError,
JSONRPCMessage,
JSONRPCNotification,
JSONRPCRequest,
JSONRPCResponse,
RequestId,
@ -849,9 +848,15 @@ class StreamableHTTPServerTransport:
# Determine which request stream(s) should receive this message
message = session_message.message
target_request_id = None
if isinstance(
message.root, JSONRPCNotification | JSONRPCRequest
):
# Check if this is a response
if isinstance(message.root, JSONRPCResponse | JSONRPCError):
response_id = str(message.root.id)
# If this response is for an existing request stream,
# send it there
if response_id in self._request_streams:
target_request_id = response_id
else:
# Extract related_request_id from meta if it exists
if (
session_message.metadata is not None
@ -865,10 +870,12 @@ class StreamableHTTPServerTransport:
target_request_id = str(
session_message.metadata.related_request_id
)
else:
target_request_id = str(message.root.id)
request_stream_id = target_request_id or GET_STREAM_KEY
request_stream_id = (
target_request_id
if target_request_id is not None
else GET_STREAM_KEY
)
# Store the event if we have an event store,
# regardless of whether a client is connected