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

use metadata from SessionMessage to propagate related_request_id (#591)

Browse Source
This commit is contained in:
ihrpr
2025-05-02 14:35:17 +01:00
committed by GitHub
parent da0cf22355
commit cf8b66b82f
3 changed files with 18 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/mcp/server/streamable_http.py
View File

@@ -24,7 +24,7 @@ from starlette.requests import Request
from starlette.responses import Response
from starlette.types import Receive, Scope, Send
from mcp.shared.message import SessionMessage
from mcp.shared.message import ServerMessageMetadata, SessionMessage
from mcp.types import (
INTERNAL_ERROR,
INVALID_PARAMS,
@@ -520,7 +520,7 @@ class StreamableHTTPServerTransport:
)
await response(scope, receive, send)
if writer:
await writer.send(err)
await writer.send(Exception(err))
return
async def _handle_get_request(self, request: Request, send: Send) -> None:
@@ -834,12 +834,17 @@ class StreamableHTTPServerTransport:
):
# Extract related_request_id from meta if it exists
if (
(params := getattr(message.root, "params", None))
and (meta := params.get("_meta"))
and (related_id := meta.get("related_request_id"))
session_message.metadata is not None
and isinstance(
session_message.metadata,
ServerMessageMetadata,
)
and session_message.metadata.related_request_id
is not None
):
target_request_id = str(related_id)
target_request_id = str(
session_message.metadata.related_request_id
)
else:
target_request_id = str(message.root.id)