Exclude Nones

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

View file

@ -74,7 +74,7 @@ class SseServerTransport:
await sse_stream_writer.send(
{
"event": "message",
"data": message.model_dump_json(by_alias=True),
"data": message.model_dump_json(by_alias=True, exclude_none=True),
}
)

View file

@ -48,7 +48,7 @@ async def stdio_server(
try:
async with write_stream_reader:
async for message in write_stream_reader:
json = message.model_dump_json(by_alias=True)
json = message.model_dump_json(by_alias=True, exclude_none=True)
await stdout.write(json + "\n")
await stdout.flush()
except anyio.ClosedResourceError:

View file

@ -47,7 +47,7 @@ async def websocket_server(scope: Scope, receive: Receive, send: Send):
try:
async with write_stream_reader:
async for message in write_stream_reader:
obj = message.model_dump(by_alias=True, mode="json")
obj = message.model_dump(by_alias=True, mode="json", exclude_none=True)
await websocket.send_json(obj)
except anyio.ClosedResourceError:
await websocket.close()