Update protocol version handling

This commit is contained in:
Justin Spahr-Summers
2024-10-21 14:50:44 +01:00
parent 2d55eabb2f
commit eb1024c654
5 changed files with 15 additions and 11 deletions

View File

@@ -3,6 +3,7 @@ import pytest
from mcp_python.client.session import ClientSession
from mcp_python.types import (
LATEST_PROTOCOL_VERSION,
ClientNotification,
ClientRequest,
Implementation,
@@ -41,7 +42,7 @@ async def test_client_session_initialize():
result = ServerResult(
InitializeResult(
protocolVersion=1,
protocolVersion=LATEST_PROTOCOL_VERSION,
capabilities=ServerCapabilities(
logging=None,
resources=None,
@@ -88,7 +89,7 @@ async def test_client_session_initialize():
# Assert the result
assert isinstance(result, InitializeResult)
assert result.protocolVersion == 1
assert result.protocolVersion == LATEST_PROTOCOL_VERSION
assert isinstance(result.capabilities, ServerCapabilities)
assert result.serverInfo == Implementation(name="mock-server", version="0.1.0")