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: catch and rethrow SSEError during SSE connection establishment (#975)

Browse source 
Co-authored-by: zhangchuanhui <zhangchal@digitalchina.com>
Co-authored-by: Felix Weinberger <fweinberger@anthropic.com>
This commit is contained in:
ZhangChuanhui 2025-09-30 22:52:32 +08:00 • committed by GitHub
commit f676f6c0f0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -8,6 +8,7 @@ import httpx
from anyio.abc import TaskStatus from anyio.abc import TaskStatus
from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
from httpx_sse import aconnect_sse from httpx_sse import aconnect_sse
from httpx_sse._exceptions import SSEError
import mcp.types as types import mcp.types as types
from mcp.shared._httpx_utils import McpHttpClientFactory, create_mcp_http_client from mcp.shared._httpx_utils import McpHttpClientFactory, create_mcp_http_client
@ -105,6 +106,9 @@ async def sse_client(
await read_stream_writer.send(session_message) await read_stream_writer.send(session_message)
case _: case _:
logger.warning(f"Unknown SSE event: {sse.event}") logger.warning(f"Unknown SSE event: {sse.event}")
except SSEError as sse_exc:
logger.exception("Encountered SSE exception")
raise sse_exc
except Exception as exc: except Exception as exc:
logger.exception("Error in sse_reader") logger.exception("Error in sse_reader")
await read_stream_writer.send(exc) await read_stream_writer.send(exc)