aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-28 11:14:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 20dc0fbabb
commit 0bcecffc4c
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:
try:
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(
client,
"GET",
url,
timeout=httpx.Timeout(timeout, read=sse_read_timeout),
) as event_source:
event_source.response.raise_for_status()
logger.debug("SSE connection established")