Drop RootModel from JSONRPCMessage (#1908)

This commit is contained in:
Marcelo Trylesinski 2026-01-19 14:04:15 +01:00 • committed by GitHub
commit f4672c5084
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -12,7 +12,6 @@ from mcp.types import (
Implementation,
InitializeRequest,
InitializeRequestParams,
JSONRPCMessage,
JSONRPCRequest,
ListToolsResult,
SamplingCapability,
@ -22,6 +21,7 @@ from mcp.types import (
ToolChoice,
ToolResultContent,
ToolUseContent,
jsonrpc_message_adapter,
)
@ -38,15 +38,15 @@ async def test_jsonrpc_request():
},
}
request = JSONRPCMessage.model_validate(json_data)
assert isinstance(request.root, JSONRPCRequest)
request = jsonrpc_message_adapter.validate_python(json_data)
assert isinstance(request, JSONRPCRequest)
ClientRequest.model_validate(request.model_dump(by_alias=True, exclude_none=True))
assert request.root.jsonrpc == "2.0"
assert request.root.id == 1
assert request.root.method == "initialize"
assert request.root.params is not None
assert request.root.params["protocolVersion"] == LATEST_PROTOCOL_VERSION
assert request.jsonrpc == "2.0"
assert request.id == 1
assert request.method == "initialize"
assert request.params is not None
assert request.params["protocolVersion"] == LATEST_PROTOCOL_VERSION
@pytest.mark.anyio