Completely drop RootModel from types module (#1910)
This commit is contained in:
parent
f4672c5084
commit
5fdd48a0d3
2 changed files with 640 additions and 800 deletions
|
|
@ -26,8 +26,8 @@ async def test_resource_templates():
|
|||
result = await mcp._mcp_server.request_handlers[types.ListResourceTemplatesRequest](
|
||||
types.ListResourceTemplatesRequest(params=None)
|
||||
)
|
||||
assert isinstance(result.root, types.ListResourceTemplatesResult)
|
||||
templates = result.root.resource_templates
|
||||
assert isinstance(result, types.ListResourceTemplatesResult)
|
||||
templates = result.resource_templates
|
||||
|
||||
# Verify we get both templates back
|
||||
assert len(templates) == 2
|
||||
|
|
|
|||
|
|
@ -60,7 +60,7 @@ async def test_server_base64_encoding_issue():
|
|||
result: ServerResult = await handler(request)
|
||||
|
||||
# After (fixed code):
|
||||
read_result: ReadResourceResult = cast(ReadResourceResult, result.root)
|
||||
read_result: ReadResourceResult = cast(ReadResourceResult, result)
|
||||
blob_content = read_result.contents[0]
|
||||
|
||||
# First verify our test data actually produces different encodings
|
||||
|
|
|
|||
Loading…
Reference in a new issue