Update dependencies and fix type issues (#1268)

Co-authored-by: Marcelo Trylesinski <marcelotryle@gmail.com>

tests/client/test_list_roots_callback.py

@@ -3,6 +3,7 @@ from pydantic import FileUrl
 
 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 +36,7 @@ async def test_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, None], message: str):
         roots = await context.session.list_roots()
         assert roots == callback_return
         return True

tests/client/test_session.py

@@ -32,6 +32,7 @@ async def test_client_session_initialize():
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](1)
 
     initialized_notification = None
+    result = None
 
     async def mock_server():
         nonlocal initialized_notification
@@ -239,6 +240,7 @@ async def test_client_session_version_negotiation_success():
     """Test successful version negotiation with supported version"""
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](1)
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](1)
+    result = None
 
     async def mock_server():
         session_message = await client_to_server_receive.receive()

tests/server/test_lowlevel_input_validation.py

@@ -21,7 +21,7 @@ async def run_tool_test(
     tools: list[Tool],
     call_tool_handler: Callable[[str, dict[str, Any]], Awaitable[list[TextContent]]],
     test_callback: Callable[[ClientSession], Awaitable[CallToolResult]],
-) -> CallToolResult:
+) -> CallToolResult | None:
     """Helper to run a tool test with minimal boilerplate.
 
     Args:
@@ -33,6 +33,7 @@ async def run_tool_test(
         The result of the tool call
     """
     server = Server("test")
+    result = None
 
     @server.list_tools()
     async def list_tools():

tests/server/test_lowlevel_output_validation.py

@@ -21,7 +21,7 @@ async def run_tool_test(
     tools: list[Tool],
     call_tool_handler: Callable[[str, dict[str, Any]], Awaitable[Any]],
     test_callback: Callable[[ClientSession], Awaitable[CallToolResult]],
-) -> CallToolResult:
+) -> CallToolResult | None:
     """Helper to run a tool test with minimal boilerplate.
 
     Args:
@@ -34,6 +34,8 @@ async def run_tool_test(
     """
     server = Server("test")
 
+    result = None
+
     @server.list_tools()
     async def list_tools():
         return tools

tests/server/test_lowlevel_tool_annotations.py

@@ -39,6 +39,7 @@ async def test_lowlevel_server_tool_annotations():
             )
         ]
 
+    tools_result = None
     server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[SessionMessage](10)
     client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[SessionMessage](10)
 

tests/shared/test_progress_notifications.py

@@ -217,6 +217,8 @@ async def test_progress_context_manager():
 
     server = Server(name="ProgressContextTestServer")
 
+    progress_token = None
+
     # Register progress handler
     @server.progress_notification()
     async def handle_progress(