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
1 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():