ruff format

Author: kantord

shell_engine.py

@@ -407,7 +407,9 @@ async def tool_stage(
                 # Use batch caller for connection reuse (much faster)
                 args_only = [ca[1] for ca in all_call_args]
                 try:
-                    batch_results = await self.batch_tool_caller(server, tool, args_only)
+                    batch_results = await self.batch_tool_caller(
+                        server, tool, args_only
+                    )
                     for result in batch_results:
                         if hasattr(result, "content"):
                             for content_item in result.content:

tests/test_mcp_client.py

@@ -788,7 +788,9 @@ def tracking_streamablehttp_client(url):
             nonlocal connection_open_count
             connection_open_count += 1
             mock_http = MagicMock()
-            mock_http.__aenter__ = AsyncMock(return_value=("read", "write", lambda: None))
+            mock_http.__aenter__ = AsyncMock(
+                return_value=("read", "write", lambda: None)
+            )
             mock_http.__aexit__ = AsyncMock()
             return mock_http
 
@@ -890,9 +892,7 @@ async def test_batch_call_tool_workload_not_found(self, mocker):
         )
 
         with pytest.raises(ValueError, match="not found"):
-            await mcp_client.batch_call_tool(
-                "nonexistent", "fetch", [{"id": 1}]
-            )
+            await mcp_client.batch_call_tool("nonexistent", "fetch", [{"id": 1}])
 
 
 @pytest.mark.asyncio
@@ -937,7 +937,9 @@ async def failing_on_third_call(*args, **kwargs):
         mock_http.__aexit__ = AsyncMock(return_value=None)
 
         mocker.patch("mcp_client.streamablehttp_client", return_value=mock_http)
-        mocker.patch("mcp_client.ClientSession", return_value=mock_client_session_instance)
+        mocker.patch(
+            "mcp_client.ClientSession", return_value=mock_client_session_instance
+        )
 
         call_args_list = [{"id": i} for i in range(5)]
 
@@ -952,9 +954,9 @@ async def failing_on_third_call(*args, **kwargs):
         )
 
         # Should report how many completed successfully
-        assert "2 successful" in error_msg.lower() or "2 completed" in error_msg.lower(), (
-            f"Error should mention 2 items completed successfully. Got: {error_msg}"
-        )
+        assert (
+            "2 successful" in error_msg.lower() or "2 completed" in error_msg.lower()
+        ), f"Error should mention 2 items completed successfully. Got: {error_msg}"
 
         # Should report how many are still pending
         assert "2 pending" in error_msg.lower(), (
@@ -999,7 +1001,9 @@ async def failing_on_third_call(*args, **kwargs):
         mock_http.__aexit__ = AsyncMock(return_value=None)
 
         mocker.patch("mcp_client.streamablehttp_client", return_value=mock_http)
-        mocker.patch("mcp_client.ClientSession", return_value=mock_client_session_instance)
+        mocker.patch(
+            "mcp_client.ClientSession", return_value=mock_client_session_instance
+        )
 
         call_args_list = [{"url": f"http://example.com/{i}"} for i in range(5)]
 
@@ -1057,7 +1061,9 @@ async def hanging_call_tool(*args, **kwargs):
         mock_http.__aexit__ = AsyncMock(return_value=None)
 
         mocker.patch("mcp_client.streamablehttp_client", return_value=mock_http)
-        mocker.patch("mcp_client.ClientSession", return_value=mock_client_session_instance)
+        mocker.patch(
+            "mcp_client.ClientSession", return_value=mock_client_session_instance
+        )
 
         # Verify DEFAULT_TOOL_TIMEOUT constant exists
         assert hasattr(mcp_client, "DEFAULT_TOOL_TIMEOUT"), (
@@ -1068,6 +1074,7 @@ async def hanging_call_tool(*args, **kwargs):
         test_timeout = 0.5
 
         import time
+
         start = time.time()
 
         with pytest.raises(asyncio.TimeoutError):
@@ -1118,12 +1125,15 @@ async def hanging_call_tool(*args, **kwargs):
         mock_http.__aexit__ = AsyncMock(return_value=None)
 
         mocker.patch("mcp_client.streamablehttp_client", return_value=mock_http)
-        mocker.patch("mcp_client.ClientSession", return_value=mock_client_session_instance)
+        mocker.patch(
+            "mcp_client.ClientSession", return_value=mock_client_session_instance
+        )
 
         # Use a short timeout for testing (0.5 seconds)
         test_timeout = 0.5
 
         import time
+
         start = time.time()
 
         # batch_call_tool wraps timeout errors in RuntimeError with progress info