Update dependencies and fix type issues (#1268)

Co-authored-by: Marcelo Trylesinski <marcelotryle@gmail.com>
This commit is contained in:
David Soria Parra 2025-08-14 10:40:47 +01:00 • committed by GitHub
commit 0926613714
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 303 additions and 286 deletions

View file

@ -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

View file

@ -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()