From 73c055f136be4f33591ceb69091cb750135e7bcb Mon Sep 17 00:00:00 2001 From: David Soria Parra Date: Thu, 2 Jan 2025 16:59:26 +0000 Subject: [PATCH] fix: Update AnyIO stream creation to use type annotation syntax --- tests/server/test_session.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/server/test_session.py b/tests/server/test_session.py index ead18f7..67d7d48 100644 --- a/tests/server/test_session.py +++ b/tests/server/test_session.py @@ -18,12 +18,8 @@ from mcp.types import ( @pytest.mark.anyio async def test_server_session_initialize(): - server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream( - 1, JSONRPCMessage - ) - client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream( - 1, JSONRPCMessage - ) + server_to_client_send, server_to_client_receive = anyio.create_memory_object_stream[JSONRPCMessage](1) + client_to_server_send, client_to_server_receive = anyio.create_memory_object_stream[JSONRPCMessage](1) async def run_client(client: ClientSession): async for message in client_session.incoming_messages: