refactor: clean up code formatting and remove unused imports across server and test modules

Author: tercel

src/apcore_mcp/server/router.py

@@ -450,9 +450,7 @@ async def _handle_stream(
 
             async for chunk in stream_iter:
                 if not isinstance(chunk, dict):
-                    raise TypeError(
-                        f"stream chunk must be dict, got {type(chunk).__name__}: {chunk!r}"
-                    )
+                    raise TypeError(f"stream chunk must be dict, got {type(chunk).__name__}: {chunk!r}")
                 # Redact each chunk before sending so sensitive fields never reach
                 # the client via progress notifications, even before final accumulation.
                 safe_chunk = self._maybe_redact(tool_name, chunk)
@@ -481,8 +479,6 @@ async def _handle_stream(
         except Exception as error:
             logger.error("handle_call stream error for %s: %s", tool_name, error)
             error_info = self._error_mapper.to_mcp_error(error)
-            stream_error_content: list[dict[str, Any]] = [
-                {"type": "text", "text": self._build_error_text(error_info)}
-            ]
+            stream_error_content: list[dict[str, Any]] = [{"type": "text", "text": self._build_error_text(error_info)}]
             self._attach_traceparent(stream_error_content, context)
             return (stream_error_content, True, context.trace_id if context is not None else None)

src/apcore_mcp/server/server.py

@@ -49,9 +49,7 @@ def __init__(
     ) -> None:
         transport_lower = transport.lower()
         if transport_lower not in _VALID_TRANSPORTS:
-            raise ValueError(
-                f"Unknown transport: {transport!r}. Expected one of {sorted(_VALID_TRANSPORTS)}"
-            )
+            raise ValueError(f"Unknown transport: {transport!r}. Expected one of {sorted(_VALID_TRANSPORTS)}")
         self._registry_or_executor = registry_or_executor
         self._transport = transport_lower
         self._host = host

tests/server/test_async_task_bridge.py

@@ -13,14 +13,12 @@
 from typing import Any
 
 import pytest
-
 from apcore.async_task import AsyncTaskManager, TaskStatus
 from apcore.errors import TaskLimitExceededError
 
 from apcore_mcp.server.async_task_bridge import META_TOOL_NAMES, AsyncTaskBridge
 from apcore_mcp.server.factory import MCPServerFactory
 
-
 # ---------------------------------------------------------------------------
 # Stubs
 # ---------------------------------------------------------------------------
@@ -119,9 +117,7 @@ async def test_status_tool_returns_result_when_completed() -> None:
             break
         await asyncio.sleep(0.005)
 
-    content, is_error, _ = await bridge.handle_meta_tool(
-        "__apcore_task_status", {"task_id": task_id}
-    )
+    content, is_error, _ = await bridge.handle_meta_tool("__apcore_task_status", {"task_id": task_id})
     assert is_error is False
     body = json.loads(content[0]["text"])
     assert body["status"] == "completed"
@@ -132,9 +128,7 @@ async def test_status_tool_returns_result_when_completed() -> None:
 async def test_status_tool_unknown_task_id() -> None:
     mgr = AsyncTaskManager(_SlowExecutor())
     bridge = AsyncTaskBridge(mgr)
-    content, is_error, _ = await bridge.handle_meta_tool(
-        "__apcore_task_status", {"task_id": "missing"}
-    )
+    content, is_error, _ = await bridge.handle_meta_tool("__apcore_task_status", {"task_id": "missing"})
     assert is_error is True
     body = json.loads(content[0]["text"])
     assert body["error"] == "ASYNC_TASK_NOT_FOUND"
@@ -154,9 +148,7 @@ async def test_cancel_tool_cancels_running_task() -> None:
     envelope = await bridge.submit("m", {}, None)
     task_id = envelope["task_id"]
 
-    content, is_error, _ = await bridge.handle_meta_tool(
-        "__apcore_task_cancel", {"task_id": task_id}
-    )
+    content, is_error, _ = await bridge.handle_meta_tool("__apcore_task_cancel", {"task_id": task_id})
     assert is_error is False
     body = json.loads(content[0]["text"])
     assert body["task_id"] == task_id
@@ -246,9 +238,7 @@ async def send(notification: dict[str, Any]) -> None:
     from apcore import Context
 
     ctx = Context.create()
-    await bridge._submit_with_progress(
-        "m", {}, ctx, progress_token="tok-1", send_notification=send
-    )
+    await bridge._submit_with_progress("m", {}, ctx, progress_token="tok-1", send_notification=send)
     # Simulate the module invoking the installed progress callback.
     cb = ctx.data["_mcp_progress"]
     await cb(0.5, 1.0, "halfway")

tests/test_observability.py

@@ -8,7 +8,6 @@
 from __future__ import annotations
 
 from typing import Any
-from unittest.mock import patch
 
 import pytest