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

Revert "Add message queue for SSE messages POST endpoint (#459)" (#649)

Browse source
This commit is contained in:
ihrpr 2025-05-07 16:35:20 +01:00 • committed by GitHub
commit 9d99aee014
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 51 additions and 1247 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/mcp/client/sse.py
View file

@ -98,9 +98,7 @@ async def sse_client(
await read_stream_writer.send(exc)
continue
session_message = SessionMessage(
message=message
)
session_message = SessionMessage(message)
await read_stream_writer.send(session_message)
case _:
logger.warning(
@ -150,5 +148,3 @@ async def sse_client(
finally:
await read_stream_writer.aclose()
await write_stream.aclose()
await read_stream.aclose()
await write_stream_reader.aclose()

2
src/mcp/client/stdio/__init__.py
View file

@ -144,7 +144,7 @@ async def stdio_client(server: StdioServerParameters, errlog: TextIO = sys.stder
await read_stream_writer.send(exc)
continue
session_message = SessionMessage(message=message)
session_message = SessionMessage(message)
await read_stream_writer.send(session_message)
except anyio.ClosedResourceError:
await anyio.lowlevel.checkpoint()

6
src/mcp/client/streamable_http.py
View file

@ -153,7 +153,7 @@ class StreamableHTTPTransport:
):
message.root.id = original_request_id
session_message = SessionMessage(message=message)
session_message = SessionMessage(message)
await read_stream_writer.send(session_message)
# Call resumption token callback if we have an ID
@ -286,7 +286,7 @@ class StreamableHTTPTransport:
try:
content = await response.aread()
message = JSONRPCMessage.model_validate_json(content)
session_message = SessionMessage(message=message)
session_message = SessionMessage(message)
await read_stream_writer.send(session_message)
except Exception as exc:
logger.error(f"Error parsing JSON response: {exc}")
@ -333,7 +333,7 @@ class StreamableHTTPTransport:
id=request_id,
error=ErrorData(code=32600, message="Session terminated"),
)
session_message = SessionMessage(message=JSONRPCMessage(jsonrpc_error))
session_message = SessionMessage(JSONRPCMessage(jsonrpc_error))
await read_stream_writer.send(session_message)
async def post_writer(

2
src/mcp/client/websocket.py
View file

@ -60,7 +60,7 @@ async def websocket_client(
async for raw_text in ws:
try:
message = types.JSONRPCMessage.model_validate_json(raw_text)
session_message = SessionMessage(message=message)
session_message = SessionMessage(message)
await read_stream_writer.send(session_message)
except ValidationError as exc:
# If JSON parse or model validation fails, send the exception