Clean up verbose comments and docstrings

- Remove SEP-1699 header comment from tests
- Remove verbose Args docstrings from StreamableHTTPTransport.__init__ and streamablehttp_client
- Remove Note comment about _handle_sse_response return value

Author: felixweinberger

src/mcp/client/streamable_http.py

@@ -100,16 +100,7 @@ def __init__(
         auth: httpx.Auth | None = None,
         reconnection_options: StreamableHTTPReconnectionOptions | None = None,
     ) -> None:
-        """Initialize the StreamableHTTP transport.
-
-        Args:
-            url: The endpoint URL.
-            headers: Optional headers to include in requests.
-            timeout: HTTP timeout for regular operations.
-            sse_read_timeout: Timeout for SSE read operations.
-            auth: Optional HTTPX authentication handler.
-            reconnection_options: Options for configuring reconnection behavior.
-        """
+        """Initialize the StreamableHTTP transport."""
         self.url = url
         self.headers = headers or {}
         self.timeout = timeout.total_seconds() if isinstance(timeout, timedelta) else timeout
@@ -344,11 +335,7 @@ async def _handle_post_request(self, ctx: RequestContext) -> None:
                 if content_type.startswith(JSON):
                     await self._handle_json_response(response, ctx.read_stream_writer, is_initialization)
                 elif content_type.startswith(SSE):
-                    # Note: _handle_sse_response returns (has_priming_event, last_event_id)
-                    # which can be used for reconnection logic if needed
-                    _has_priming_event, _last_event_id = await self._handle_sse_response(
-                        response, ctx, is_initialization
-                    )
+                    await self._handle_sse_response(response, ctx, is_initialization)
                 else:
                     await self._handle_unexpected_content_type(  # pragma: no cover
                         content_type,  # pragma: no cover
@@ -598,22 +585,6 @@ async def streamablehttp_client(
 
     `sse_read_timeout` determines how long (in seconds) the client will wait for a new
     event before disconnecting. All other HTTP operations are controlled by `timeout`.
-
-    Args:
-        url: The endpoint URL.
-        headers: Optional headers to include in requests.
-        timeout: HTTP timeout for regular operations.
-        sse_read_timeout: Timeout for SSE read operations.
-        terminate_on_close: Whether to terminate the session on close.
-        httpx_client_factory: Factory function to create the HTTP client.
-        auth: Optional HTTPX authentication handler.
-        reconnection_options: Options for configuring reconnection behavior.
-
-    Yields:
-        Tuple containing:
-            - read_stream: Stream for reading messages from the server
-            - write_stream: Stream for sending messages to the server
-            - get_session_id_callback: Function to retrieve the current session ID
     """
     transport = StreamableHTTPTransport(url, headers, timeout, sse_read_timeout, auth, reconnection_options)
 

tests/shared/test_streamable_http.py

@@ -1608,11 +1608,6 @@ async def bad_client():
             assert tools.tools
 
 
-# =============================================================================
-# SEP-1699: SSE Polling Support Tests
-# =============================================================================
-
-
 @pytest.mark.anyio
 async def test_reconnection_delay_with_server_retry():
     """Test _get_next_reconnection_delay uses server-provided retry value."""
@@ -1769,9 +1764,7 @@ async def test_resume_stream_without_session_id():
     assert transport.session_id is None
 
     # Create a dummy stream writer with type annotation
-    read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[
-        SessionMessage | Exception
-    ](0)
+    read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[SessionMessage | Exception](0)
 
     async with httpx.AsyncClient() as client:
         # Should return early without making request
@@ -1806,9 +1799,7 @@ async def test_resume_stream_with_405_response(basic_server: None, basic_server_
             transport.session_id = get_session_id()
 
     # Now try to resume with the session - server might return 405
-    read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[
-        SessionMessage | Exception
-    ](0)
+    read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[SessionMessage | Exception](0)
 
     async with httpx.AsyncClient() as client:
         # This should handle the 405 gracefully