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

Wrap JSONRPC messages with SessionMessage for metadata support (#590)

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

10
src/mcp/shared/memory.py
View file

@ -19,11 +19,11 @@ from mcp.client.session import (
SamplingFnT,
)
from mcp.server import Server
from mcp.types import JSONRPCMessage
from mcp.shared.message import SessionMessage
MessageStream = tuple[
MemoryObjectReceiveStream[JSONRPCMessage | Exception],
MemoryObjectSendStream[JSONRPCMessage],
MemoryObjectReceiveStream[SessionMessage | Exception],
MemoryObjectSendStream[SessionMessage],
]
@ -40,10 +40,10 @@ async def create_client_server_memory_streams() -> (
"""
# Create streams for both directions
server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[
JSONRPCMessage | Exception
SessionMessage | Exception
](1)
client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[
JSONRPCMessage | Exception
SessionMessage | Exception
](1)
client_streams = (server_to_client_receive, client_to_server_send)