Skip to content

Commit dce39ec

Browse files
committed
fix(test): update message_handler to the current IncomingMessage type
RequestResponder and mcp.shared.session were removed upstream since this test was written; message_handler callbacks now receive IncomingMessage (ServerNotification | Exception), matching every other message_handler in this file.
1 parent 333250b commit dce39ec

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

tests/client/test_session.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -205,9 +205,7 @@ async def mock_server():
205205
)
206206

207207
# Create a message handler to catch exceptions
208-
async def message_handler( # pragma: no cover
209-
message: RequestResponder[types.ServerRequest, types.ClientResult] | types.ServerNotification | Exception,
210-
) -> None:
208+
async def message_handler(message: IncomingMessage) -> None: # pragma: no cover
211209
if isinstance(message, Exception):
212210
raise message
213211

0 commit comments

Comments
 (0)