modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Exclude Nones

Browse source
This commit is contained in:
Justin Spahr-Summers 2024-10-02 21:45:37 +01:00
commit 4040945cdf
No known key found for this signature in database
GPG key ID: E78498BF3C053CA7
3 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/client/test_session.py
View file

@ -35,7 +35,7 @@ async def test_client_session_initialize():
jsonrpc_request = await client_to_server_receive.receive()
assert isinstance(jsonrpc_request.root, JSONRPCRequest)
request = ClientRequest.model_validate(
jsonrpc_request.model_dump(by_alias=True, mode="json")
jsonrpc_request.model_dump(by_alias=True, mode="json", exclude_none=True)
)
assert isinstance(request.root, InitializeRequest)
@ -59,14 +59,14 @@ async def test_client_session_initialize():
JSONRPCResponse(
jsonrpc="2.0",
id=jsonrpc_request.root.id,
result=result.model_dump(by_alias=True, mode="json"),
result=result.model_dump(by_alias=True, mode="json", exclude_none=True),
)
)
)
jsonrpc_notification = await client_to_server_receive.receive()
assert isinstance(jsonrpc_notification.root, JSONRPCNotification)
initialized_notification = ClientNotification.model_validate(
jsonrpc_notification.model_dump(by_alias=True, mode="json")
jsonrpc_notification.model_dump(by_alias=True, mode="json", exclude_none=True)
)
async def listen_session():

2
tests/server/test_stdio.py
View file

@ -18,7 +18,7 @@ async def test_stdio_server():
]
for message in messages:
stdin.write(message.model_dump_json() + "\n")
stdin.write(message.model_dump_json(by_alias=True, exclude_none=True) + "\n")
stdin.seek(0)
async with stdio_server(

2
tests/test_types.py
View file

@ -15,7 +15,7 @@ def test_jsonrpc_request():
request = JSONRPCMessage.model_validate(json_data)
assert isinstance(request.root, JSONRPCRequest)
ClientRequest.model_validate(request.model_dump(by_alias=True))
ClientRequest.model_validate(request.model_dump(by_alias=True, exclude_none=True))
assert request.root.jsonrpc == "2.0"
assert request.root.id == 1