Skip to content

Commit 791496a

Browse files
author
Jianke LIN
committed
fix: make streamable-http tests pass CI
1 parent 9f92735 commit 791496a

3 files changed

Lines changed: 3 additions & 2 deletions

File tree

src/mcp/client/streamable_http.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ async def _handle_reconnection(
451451
# Stream ended again without response - reconnect again (reset attempt counter)
452452
logger.info("SSE stream disconnected, reconnecting...")
453453
await self._handle_reconnection(ctx, reconnect_last_event_id, reconnect_retry_ms, 0)
454-
except Exception as e: # pragma: no cover
454+
except Exception as e:
455455
logger.debug(f"Reconnection failed: {e}")
456456
# Try to reconnect again if we still have an event ID
457457
await self._handle_reconnection(ctx, last_event_id, retry_interval_ms, attempt + 1)

tests/client/test_streamable_http.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
7272
await transport._handle_sse_response(response, ctx)
7373
message = await read_stream.receive()
7474

75+
assert isinstance(message, SessionMessage)
7576
assert isinstance(message.message, JSONRPCError)
7677
assert message.message.id == 1
7778
assert message.message.error.code == CONNECTION_CLOSED

tests/interaction/transports/test_hosting_resume.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ async def record_request(request: httpx.Request) -> None:
423423
):
424424
manager_ref = manager
425425
with anyio.fail_after(5): # pragma: no branch
426-
async with (
426+
async with ( # pragma: no branch
427427
streamable_http_client(f"{BASE_URL}/mcp", http_client=http, terminate_on_close=False) as (r, w),
428428
ClientSession(r, w) as session,
429429
anyio.create_task_group() as tg,

0 commit comments

Comments
 (0)