Drop RootModel from JSONRPCMessage (#1908)
This commit is contained in:
parent
dcc9b4f8e0
commit
f4672c5084
1 changed files with 314 additions and 476 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue