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

Support for http request injection propagation in StreamableHttp (#833)
Some checks failed
Main branch checks / checks (push) Has been cancelled
Details

Browse source
This commit is contained in:
ihrpr 2025-05-29 15:21:06 +01:00 • committed by GitHub
commit 05b7156ea8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 259 additions and 172 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -397,7 +397,8 @@ class StreamableHTTPServerTransport:
await response(scope, receive, send)
# Process the message after sending the response
session_message = SessionMessage(message)
metadata = ServerMessageMetadata(request_context=request)
session_message = SessionMessage(message, metadata=metadata)
await writer.send(session_message)
return
@ -412,7 +413,8 @@ class StreamableHTTPServerTransport:
if self.is_json_response_enabled:
# Process the message
session_message = SessionMessage(message)
metadata = ServerMessageMetadata(request_context=request)
session_message = SessionMessage(message, metadata=metadata)
await writer.send(session_message)
try:
# Process messages from the request-specific stream
@ -511,7 +513,8 @@ class StreamableHTTPServerTransport:
async with anyio.create_task_group() as tg:
tg.start_soon(response, scope, receive, send)
# Then send the message to be processed by the server
session_message = SessionMessage(message)
metadata = ServerMessageMetadata(request_context=request)
session_message = SessionMessage(message, metadata=metadata)
await writer.send(session_message)
except Exception:
logger.exception("SSE response error")