refactor: address additional PR review comments

- Add imports to docstring example for self-contained rendering
- Move TODO from docstring to regular comment
- Rename get_server_capabilities() to server_capabilities property
- Use Google-style docstring (opening on same line)

Author: felixweinberger

src/mcp/client/client.py

@@ -26,13 +26,16 @@
 
 
 class Client:
-    """
-    A high-level MCP client for connecting to MCP servers.
+    """A high-level MCP client for connecting to MCP servers.
 
     Currently supports in-memory transport for testing. Pass a Server or
     FastMCP instance directly to the constructor.
 
     Example:
+        ```python
+        from mcp.client import Client
+        from mcp.server.fastmcp import FastMCP
+
         server = FastMCP("test")
 
         @server.tool()
@@ -41,15 +44,16 @@ def add(a: int, b: int) -> int:
 
         async with Client(server) as client:
             result = await client.call_tool("add", {"a": 1, "b": 2})
-
-    TODO(felixweinberger): Expand to support all transport types (like FastMCP 2):
-    - Add ClientTransport base class with connect_session() method
-    - Add StreamableHttpTransport, SSETransport, StdioTransport
-    - Add infer_transport() to auto-detect transport from input type
-    - Accept URL strings, Path objects, config dicts in constructor
-    - Add auth support (OAuth, bearer tokens)
+        ```
     """
 
+    # TODO(felixweinberger): Expand to support all transport types (like FastMCP 2):
+    # - Add ClientTransport base class with connect_session() method
+    # - Add StreamableHttpTransport, SSETransport, StdioTransport
+    # - Add infer_transport() to auto-detect transport from input type
+    # - Accept URL strings, Path objects, config dicts in constructor
+    # - Add auth support (OAuth, bearer tokens)
+
     def __init__(
         self,
         server: Server[Any] | FastMCP,
@@ -147,13 +151,9 @@ def session(self) -> ClientSession:
             raise RuntimeError("Client must be used within an async context manager")
         return self._session
 
-    def get_server_capabilities(self) -> types.ServerCapabilities | None:
-        """
-        Return the server capabilities received during initialization.
-
-        Returns:
-            The server capabilities, or None if not yet initialized
-        """
+    @property
+    def server_capabilities(self) -> types.ServerCapabilities | None:
+        """The server capabilities received during initialization, or None if not yet initialized."""
         return self.session.get_server_capabilities()
 
     async def send_ping(self) -> types.EmptyResult:

tests/client/test_client.py

@@ -68,7 +68,7 @@ async def test_creates_client(app: FastMCP):
 async def test_client_is_initialized(app: FastMCP):
     """Test that the client is initialized after entering context."""
     async with Client(app) as client:
-        caps = client.get_server_capabilities()
+        caps = client.server_capabilities
         assert caps is not None
         assert caps.tools is not None
 
@@ -77,7 +77,7 @@ async def test_with_simple_server(simple_server: Server):
     """Test that from_server works with a basic Server instance."""
     async with Client(simple_server) as client:
         assert client is not None
-        caps = client.get_server_capabilities()
+        caps = client.server_capabilities
         assert caps is not None
         # Verify list_resources works and returns expected resource
         resources = await client.list_resources()
@@ -204,10 +204,10 @@ async def test_aexit_without_aenter(app: FastMCP):
     assert client._session is None
 
 
-async def test_get_server_capabilities_after_init(app: FastMCP):
-    """Test getting server capabilities after initialization."""
+async def test_server_capabilities_after_init(app: FastMCP):
+    """Test server_capabilities property after initialization."""
     async with Client(app) as client:
-        caps = client.get_server_capabilities()
+        caps = client.server_capabilities
         assert caps is not None
         # FastMCP should advertise tools capability
         assert caps.tools is not None