1818from mcp .shared .message import SessionMessage
1919from mcp .shared .session import RequestResponder
2020from mcp .shared .transport_context import TransportContext
21- from mcp .shared .version import HANDSHAKE_PROTOCOL_VERSIONS
21+ from mcp .shared .version import HANDSHAKE_PROTOCOL_VERSIONS , LATEST_HANDSHAKE_VERSION
2222from mcp .types import (
2323 CONNECTION_CLOSED ,
2424 INTERNAL_ERROR ,
@@ -87,7 +87,7 @@ async def mock_server():
8787 assert isinstance (request , InitializeRequest )
8888
8989 result = InitializeResult (
90- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
90+ protocol_version = LATEST_HANDSHAKE_VERSION ,
9191 capabilities = ServerCapabilities (
9292 logging = None ,
9393 resources = None ,
@@ -140,7 +140,7 @@ async def message_handler( # pragma: no cover
140140
141141 # Assert the result
142142 assert isinstance (result , InitializeResult )
143- assert result .protocol_version == HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ]
143+ assert result .protocol_version == LATEST_HANDSHAKE_VERSION
144144 assert isinstance (result .capabilities , ServerCapabilities )
145145 assert result .server_info == Implementation (name = "mock-server" , version = "0.1.0" )
146146 assert result .instructions == "The server instructions."
@@ -171,7 +171,7 @@ async def mock_server():
171171 received_client_info = request .params .client_info
172172
173173 result = InitializeResult (
174- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
174+ protocol_version = LATEST_HANDSHAKE_VERSION ,
175175 capabilities = ServerCapabilities (),
176176 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
177177 )
@@ -228,7 +228,7 @@ async def mock_server():
228228 received_client_info = request .params .client_info
229229
230230 result = InitializeResult (
231- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
231+ protocol_version = LATEST_HANDSHAKE_VERSION ,
232232 capabilities = ServerCapabilities (),
233233 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
234234 )
@@ -278,7 +278,7 @@ async def mock_server():
278278 assert isinstance (request , InitializeRequest )
279279
280280 # Verify client offers the newest handshake protocol version
281- assert request .params .protocol_version == HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ]
281+ assert request .params .protocol_version == LATEST_HANDSHAKE_VERSION
282282
283283 # Server responds with a supported older version
284284 result = InitializeResult (
@@ -386,7 +386,7 @@ async def mock_server():
386386 received_capabilities = request .params .capabilities
387387
388388 result = InitializeResult (
389- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
389+ protocol_version = LATEST_HANDSHAKE_VERSION ,
390390 capabilities = ServerCapabilities (),
391391 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
392392 )
@@ -457,7 +457,7 @@ async def mock_server():
457457 received_capabilities = request .params .capabilities
458458
459459 result = InitializeResult (
460- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
460+ protocol_version = LATEST_HANDSHAKE_VERSION ,
461461 capabilities = ServerCapabilities (),
462462 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
463463 )
@@ -536,7 +536,7 @@ async def mock_server():
536536 received_capabilities = request .params .capabilities
537537
538538 result = InitializeResult (
539- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
539+ protocol_version = LATEST_HANDSHAKE_VERSION ,
540540 capabilities = ServerCapabilities (),
541541 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
542542 )
@@ -604,7 +604,7 @@ async def mock_server():
604604 assert isinstance (request , InitializeRequest )
605605
606606 result = InitializeResult (
607- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
607+ protocol_version = LATEST_HANDSHAKE_VERSION ,
608608 capabilities = expected_capabilities ,
609609 server_info = expected_server_info ,
610610 instructions = expected_instructions ,
@@ -643,12 +643,12 @@ async def mock_server():
643643 assert result .server_info == expected_server_info
644644 assert result .capabilities == expected_capabilities
645645 assert result .instructions == expected_instructions
646- assert result .protocol_version == HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ]
646+ assert result .protocol_version == LATEST_HANDSHAKE_VERSION
647647 # Era-neutral accessors are populated from the InitializeResult.
648648 assert session .server_info == expected_server_info
649649 assert session .server_capabilities == expected_capabilities
650650 assert session .instructions == expected_instructions
651- assert session .protocol_version == HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ]
651+ assert session .protocol_version == LATEST_HANDSHAKE_VERSION
652652
653653
654654@pytest .mark .anyio
@@ -671,7 +671,7 @@ async def mock_server():
671671 assert isinstance (request , InitializeRequest )
672672
673673 result = InitializeResult (
674- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
674+ protocol_version = LATEST_HANDSHAKE_VERSION ,
675675 capabilities = ServerCapabilities (),
676676 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
677677 )
@@ -1354,7 +1354,7 @@ async def send_raw_request(
13541354 self .calls .append ((method , opts or {}))
13551355 if method == "initialize" :
13561356 return InitializeResult (
1357- protocol_version = HANDSHAKE_PROTOCOL_VERSIONS [ - 1 ] ,
1357+ protocol_version = LATEST_HANDSHAKE_VERSION ,
13581358 capabilities = ServerCapabilities (),
13591359 server_info = Implementation (name = "mock-server" , version = "0.1.0" ),
13601360 ).model_dump (by_alias = True , mode = "json" , exclude_none = True )
0 commit comments