Skip to content

Commit 3633bd2

Browse files
committed
fix: make streamable-http tests pass CI
1 parent 23caf52 commit 3633bd2

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
@@ -440,7 +440,7 @@ async def _handle_reconnection(
440440
# Stream ended again without response - reconnect again (reset attempt counter)
441441
logger.info("SSE stream disconnected, reconnecting...")
442442
await self._handle_reconnection(ctx, reconnect_last_event_id, reconnect_retry_ms, 0)
443-
except Exception as e: # pragma: no cover
443+
except Exception as e:
444444
logger.debug(f"Reconnection failed: {e}")
445445
# Try to reconnect again if we still have an event ID
446446
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
@@ -27,6 +27,7 @@ async def test_sse_response_disconnect_before_any_event_id_fails_request() -> No
2727
await transport._handle_sse_response(response, ctx)
2828
message = await read_stream.receive()
2929

30+
assert isinstance(message, SessionMessage)
3031
assert isinstance(message.message, JSONRPCError)
3132
assert message.message.id == 1
3233
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
@@ -346,7 +346,7 @@ async def record_request(request: httpx.Request) -> None:
346346
):
347347
manager_ref = manager
348348
with anyio.fail_after(5): # pragma: no branch
349-
async with (
349+
async with ( # pragma: no branch
350350
streamable_http_client(f"{BASE_URL}/mcp", http_client=http, terminate_on_close=False) as (r, w),
351351
ClientSession(r, w) as session,
352352
anyio.create_task_group() as tg,

0 commit comments

Comments
 (0)