autofix code style issues

Author: kantord

mcp_client.py

@@ -170,7 +170,9 @@ async def list_tools_from_server(workload: dict[str, Any]) -> dict[str, Any]:
         # ToolHive can proxy servers via SSE even if the original transport is stdio
         if proxy_mode == "sse":
             # Use SSE client for SSE proxy
-            async with sse_client(url, httpx_client_factory=_create_tolerant_httpx_client) as (read, write):
+            async with sse_client(
+                url, httpx_client_factory=_create_tolerant_httpx_client
+            ) as (read, write):
                 async with ClientSession(read, write) as session:
                     await session.initialize()
                     tools_response = await session.list_tools()
@@ -309,7 +311,9 @@ async def get_tool_details_from_server(
 
         # Connect and list tools to find the requested tool
         if proxy_mode == "sse":
-            async with sse_client(url, httpx_client_factory=_create_tolerant_httpx_client) as (read, write):
+            async with sse_client(
+                url, httpx_client_factory=_create_tolerant_httpx_client
+            ) as (read, write):
                 async with ClientSession(read, write) as session:
                     await session.initialize()
                     tools_response = await session.list_tools()
@@ -416,7 +420,9 @@ async def call_tool(
 
     # Connect and call the tool with timeout
     if proxy_mode == "sse":
-        async with sse_client(url, httpx_client_factory=_create_tolerant_httpx_client) as (read, write):
+        async with sse_client(
+            url, httpx_client_factory=_create_tolerant_httpx_client
+        ) as (read, write):
             async with ClientSession(read, write) as session:
                 await session.initialize()
                 result = await asyncio.wait_for(
@@ -544,7 +550,9 @@ async def execute_calls(session):
                 raise RuntimeError("\n".join(error_parts)) from e
 
     if proxy_mode == "sse":
-        async with sse_client(url, httpx_client_factory=_create_tolerant_httpx_client) as (read, write):
+        async with sse_client(
+            url, httpx_client_factory=_create_tolerant_httpx_client
+        ) as (read, write):
             async with ClientSession(read, write) as session:
                 await session.initialize()
                 await execute_calls(session)

toolhive_client.py

@@ -65,15 +65,11 @@ async def _is_toolhive_available(
     """
     try:
         async with httpx.AsyncClient(timeout=timeout) as client:
-            response = await client.get(
-                f"http://{host}:{port}/api/v1beta/version"
-            )
+            response = await client.get(f"http://{host}:{port}/api/v1beta/version")
             response.raise_for_status()
             data = response.json()
     except (httpx.HTTPError, OSError) as e:
-        raise ConnectionError(
-            f"ToolHive not available at {host}:{port}: {e}"
-        )
+        raise ConnectionError(f"ToolHive not available at {host}:{port}: {e}")
 
     if not isinstance(data, dict) or "version" not in data:
         raise ConnectionError(
@@ -117,9 +113,7 @@ async def _scan_for_toolhive_async(
         )
 
     found_ports = [
-        result
-        for result in task_outcomes
-        if not isinstance(result, BaseException)
+        result for result in task_outcomes if not isinstance(result, BaseException)
     ]
 
     if found_ports:
@@ -186,9 +180,7 @@ async def discover_toolhive_async(
         if not found:
             # Fall back to scanning
             print(f"Port {port} not available, scanning for ToolHive...")
-            port = await _scan_for_toolhive_async(
-                host, scan_port_start, scan_port_end
-            )
+            port = await _scan_for_toolhive_async(host, scan_port_start, scan_port_end)
     else:
         # Scan for ToolHive with host fallbacks
         scan_hosts = [host]