Skip to content

Commit 4bee402

Browse files
author
Jianke LIN
committed
fix: make streamable-http tests pass CI
1 parent 6537096 commit 4bee402

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
@@ -511,7 +511,7 @@ async def _handle_reconnection(
511511
# Stream ended again without response - reconnect again (reset attempt counter)
512512
logger.info("SSE stream disconnected, reconnecting...")
513513
await self._handle_reconnection(ctx, reconnect_last_event_id, reconnect_retry_ms, 0)
514-
except Exception as e: # pragma: no cover
514+
except Exception as e:
515515
logger.debug(f"Reconnection failed: {e}")
516516
# Try to reconnect again if we still have an event ID
517517
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
@@ -94,6 +94,7 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
9494
await transport._handle_sse_response(response, ctx)
9595
message = await read_stream.receive()
9696

97+
assert isinstance(message, SessionMessage)
9798
assert isinstance(message.message, JSONRPCError)
9899
assert message.message.id == 1
99100
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)