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

fix: omit null optional fields from task result payloads (#2809)

Browse source 
Co-authored-by: Sheldon <neooosky@gmail.com>
This commit is contained in:
Sheldon 2026-06-10 17:37:52 +08:00 • committed by GitHub
commit 2309e5ef97
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 31 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/experimental/tasks/server/test_task_result_handler.py
View file

@ -67,6 +67,29 @@ async def test_handle_returns_result_for_completed_task(
assert "io.modelcontextprotocol/related-task" in response.meta
@pytest.mark.anyio
async def test_handle_omits_none_fields_from_completed_task_payload(
store: InMemoryTaskStore, queue: InMemoryTaskMessageQueue, handler: TaskResultHandler
) -> None:
"""Test task result payloads omit optional None fields instead of serializing null."""
task = await store.create_task(TaskMetadata(ttl=60000), task_id="test-task")
result = CallToolResult(content=[TextContent(type="text", text="Done!")])
await store.store_result(task.taskId, result)
await store.update_task(task.taskId, status="completed")
mock_session = Mock()
mock_session.send_message = AsyncMock()
request = GetTaskPayloadRequest(params=GetTaskPayloadRequestParams(taskId=task.taskId))
response = await handler.handle(request, mock_session, "req-1")
payload = response.model_dump(by_alias=True, mode="json")
assert payload["content"] == [{"type": "text", "text": "Done!"}]
assert "annotations" not in payload["content"][0]
assert "_meta" not in payload["content"][0]
assert "io.modelcontextprotocol/related-task" in payload["_meta"]
@pytest.mark.anyio
async def test_handle_raises_for_nonexistent_task(
store: InMemoryTaskStore, queue: InMemoryTaskMessageQueue, handler: TaskResultHandler