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

set timeout for sse in httpx_client_factory (#943)

Browse source
This commit is contained in:
ihrpr 2025-06-12 09:09:13 +01:00 • committed by GitHub
commit 0bcecffc4c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -54,12 +54,13 @@ async def sse_client(
async with anyio.create_task_group() as tg: async with anyio.create_task_group() as tg:
try: try:
logger.debug(f"Connecting to SSE endpoint: {remove_request_params(url)}") logger.debug(f"Connecting to SSE endpoint: {remove_request_params(url)}")
async with httpx_client_factory(headers=headers, auth=auth) as client: async with httpx_client_factory(
headers=headers, auth=auth, timeout=httpx.Timeout(timeout, read=sse_read_timeout)
) as client:
async with aconnect_sse( async with aconnect_sse(
client, client,
"GET", "GET",
url, url,
timeout=httpx.Timeout(timeout, read=sse_read_timeout),
) as event_source: ) as event_source:
event_source.response.raise_for_status() event_source.response.raise_for_status()
logger.debug("SSE connection established") logger.debug("SSE connection established")