[tests] Fix generic parameters for Context

Author: dsp-ant

tests/client/test_list_roots_callback.py

@@ -1,8 +1,10 @@
 import pytest
 from pydantic import FileUrl
 
+from typing import Any
 from mcp.client.session import ClientSession
 from mcp.server.fastmcp.server import Context
+from mcp.server.session import ServerSession
 from mcp.shared.context import RequestContext
 from mcp.shared.memory import (
     create_connected_server_and_client_session as create_session,
@@ -35,7 +37,7 @@ async def list_roots_callback(
         return callback_return
 
     @server.tool("test_list_roots")
-    async def test_list_roots(context: Context, message: str):  # type: ignore[reportUnknownMemberType]
+    async def test_list_roots(context: Context[ServerSession, Any, Any], message: str):
         roots = await context.session.list_roots()
         assert roots == callback_return
         return True