ruff fixes

Author: AishwaryaKalloli

src/mcp/client/session_group.py

@@ -16,7 +16,7 @@
 from typing import Any, TypeAlias
 
 import anyio
-from pydantic import BaseModel, AnyUrl
+from pydantic import AnyUrl, BaseModel
 from typing_extensions import Self
 
 import mcp
@@ -303,7 +303,7 @@ async def _aggregate_components(
         resources_temp: dict[str, types.Resource] = {}
         tools_temp: dict[str, types.Tool] = {}
         tool_to_session_temp: dict[str, mcp.ClientSession] = {}
-        resource_to_session_temp: dict[str, mcp.ClientSession] = {} 
+        resource_to_session_temp: dict[str, mcp.ClientSession] = {}
 
         try:
             prompts = (await session.list_prompts()).prompts

tests/client/test_session_group.py

@@ -405,22 +405,25 @@ async def test_read_resource_not_found(self):
         test_resource = types.Resource(
             name="test_resource",
             uri=AnyUrl("test://resource/1"),
-            description="Test resource"
+            description="Test resource",
         )
-        
+
         # Mock all list methods
-        mock_session.list_resources.return_value = types.ListResourcesResult(resources=[test_resource])
+        mock_session.list_resources.return_value = types.ListResourcesResult(
+            resources=[test_resource]
+        )
         mock_session.list_prompts.return_value = types.ListPromptsResult(prompts=[])
         mock_session.list_tools.return_value = types.ListToolsResult(tools=[])
-        
+
         # --- Test Setup ---
         group = ClientSessionGroup()
-        group._session_exit_stacks[mock_session] = mock.AsyncMock(spec=contextlib.AsyncExitStack)
+        group._session_exit_stacks[mock_session] = mock.AsyncMock(
+            spec=contextlib.AsyncExitStack
+        )
         await group.connect_with_session(
-            types.Implementation(name="test_server", version="1.0.0"),
-            mock_session
+            types.Implementation(name="test_server", version="1.0.0"), mock_session
         )
-        
+
         # --- Test Execution & Assertions ---
         with pytest.raises(ValueError, match="Resource not found: test://nonexistent"):
             await group.read_resource(AnyUrl("test://nonexistent"))
@@ -433,36 +436,43 @@ async def test_read_resource_success(self):
         test_resource = types.Resource(
             name="test_resource",
             uri=AnyUrl("test://resource/1"),
-            description="Test resource"
+            description="Test resource",
         )
-        
+
         # Mock all list methods
-        mock_session.list_resources.return_value = types.ListResourcesResult(resources=[test_resource])
+        mock_session.list_resources.return_value = types.ListResourcesResult(
+            resources=[test_resource]
+        )
         mock_session.list_prompts.return_value = types.ListPromptsResult(prompts=[])
         mock_session.list_tools.return_value = types.ListToolsResult(tools=[])
-        
+
         # Mock the session's read_resource method
         mock_read_result = mock.AsyncMock(spec=types.ReadResourceResult)
-        mock_read_result.content = [types.TextContent(type="text", text="Resource content")]
+        mock_read_result.content = [
+            types.TextContent(type="text", text="Resource content")
+        ]
         mock_session.read_resource.return_value = mock_read_result
-        
+
         # --- Test Setup ---
         group = ClientSessionGroup()
-        group._session_exit_stacks[mock_session] = mock.AsyncMock(spec=contextlib.AsyncExitStack)
+        group._session_exit_stacks[mock_session] = mock.AsyncMock(
+            spec=contextlib.AsyncExitStack
+        )
         await group.connect_with_session(
-            types.Implementation(name="test_server", version="1.0.0"),
-            mock_session
+            types.Implementation(name="test_server", version="1.0.0"), mock_session
         )
-        
+
         # Verify resource was added
         assert "test_resource" in group._resources
         assert group._resources["test_resource"] == test_resource
         assert "test_resource" in group._resource_to_session
         assert group._resource_to_session["test_resource"] == mock_session
-        
+
         # --- Test Execution ---
         result = await group.read_resource(AnyUrl("test://resource/1"))
-        
+
         # --- Assertions ---
-        assert result.content == [types.TextContent(type="text", text="Resource content")]
+        assert result.content == [
+            types.TextContent(type="text", text="Resource content")
+        ]
         mock_session.read_resource.assert_called_once_with(AnyUrl("test://resource/1"))