fix type issue

tests/wf_mcp/test_stateful_runtime.py

@@ -2,8 +2,10 @@ from __future__ import annotations
 
 import asyncio
 from dataclasses import dataclass, field
-from typing import Any
+from typing import Any, cast
 
+from mcp.client.session import ClientSession
+from mcp.types import CallToolResult
 from wf_authoring import build_async_registry
 from wf_core import RuntimeContext
 from wf_mcp.capabilities import DiscoveredTool
@@ -46,14 +48,26 @@ class FakeStatefulClient:
     closed: bool = False
     calls: list[tuple[str, dict[str, Any]]] = field(default_factory=list)
 
-    async def call_tool(self, tool_name: str, payload: dict[str, Any]) -> ToolCallResult:
+    async def call_tool(
+        self, tool_name: str, payload: dict[str, Any]
+    ) -> CallToolResult:
         self.calls.append((tool_name, payload))
         if tool_name == "browser_navigate":
             self.page_open = True
-            return ToolCallResult(outcome="ok", output={"content": "opened"})
+            return CallToolResult(
+                content=[],
+                structuredContent={"content": "opened"},
+            )
         if tool_name == "browser_snapshot" and self.page_open:
-            return ToolCallResult(outcome="ok", output={"content": "snapshot"})
-        return ToolCallResult(outcome="error", output={"message": "No open page"})
+            return CallToolResult(
+                content=[],
+                structuredContent={"content": "snapshot"},
+            )
+        return CallToolResult(
+            content=[],
+            structuredContent={"message": "No open page"},
+            isError=True,
+        )
 
     async def close(self) -> None:
         self.closed = True
@@ -129,7 +143,12 @@ def test_runtime_pool_reuses_stateful_session_for_same_connection() -> None:
     ) -> PersistentMcpSession:
         client = FakeStatefulClient()
         created_clients.append(client)
-        return PersistentMcpSession(connection=connection, auth=auth, client=client)
+        return PersistentMcpSession(
+            connection=connection,
+            auth=auth,
+            client=cast(ClientSession, client),
+            close_callback=client.close,
+        )
 
     async def run_calls() -> ToolCallResult:
         pool = McpRuntimePool(factory)
@@ -164,7 +183,12 @@ def test_runtime_pool_replaces_session_when_fingerprint_changes() -> None:
     ) -> PersistentMcpSession:
         client = FakeStatefulClient()
         created_clients.append(client)
-        return PersistentMcpSession(connection=connection, auth=auth, client=client)
+        return PersistentMcpSession(
+            connection=connection,
+            auth=auth,
+            client=cast(ClientSession, client),
+            close_callback=client.close,
+        )
 
     async def run_calls() -> None:
         pool = McpRuntimePool(factory)