fix: close run budget and MCP migration gaps

tests/wf_sources_mcp/test_sdk_adapter.py

@@ -5,11 +5,10 @@ from contextlib import asynccontextmanager
 from typing import Any
 
 import pytest
-from mcp import GetPromptResult, ReadResourceResult, ServerResult
+from mcp import GetPromptResult, ReadResourceResult
 from mcp.types import (
     CallToolResult,
     ClientNotification,
-    ClientRequest,
     ListPromptsResult,
     ListResourcesResult,
     ListToolsResult,
@@ -18,10 +17,10 @@ from mcp.types import (
     TextContent,
     Tool,
 )
-from pydantic import TypeAdapter
 
 from wf_sources_mcp.client import McpSourceClient
 from wf_sources_mcp.connections import McpSourceConnection
+from wf_sources_mcp.raw_messages import RawRequest, RawResult
 from wf_sources_mcp.sdk import BackendAdapter, McpSdkAdapter
 from wf_sources_mcp.transports import StdioSourceTransport
 
@@ -38,7 +37,7 @@ def _connection() -> McpSourceConnection:
 class _FakeSession:
     def __init__(self) -> None:
         self.notifications: list[ClientNotification] = []
-        self.requests: list[ClientRequest] = []
+        self.requests: list[RawRequest] = []
 
     async def list_tools(self) -> ListToolsResult:
         return ListToolsResult(
@@ -103,16 +102,11 @@ class _FakeSession:
 
     async def send_request(
         self,
-        request: ClientRequest,
-        result_type: type[ServerResult] | TypeAdapter[ServerResult],
-    ) -> Any:
-        assert isinstance(result_type, TypeAdapter)
+        request: RawRequest,
+        result_type: type[RawResult],
+    ) -> RawResult:
         self.requests.append(request)
-        return type(
-            "ServerResultModel",
-            (),
-            {"model_dump": lambda _self, **_kwargs: {"ok": True}},
-        )()
+        return result_type.model_validate({"ok": True})
 
     async def send_notification(self, notification: ClientNotification) -> None:
         self.notifications.append(notification)