mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-10 07:39:10 +02:00
6e22914360
Moves relay_elicitation() into elicitation.py so it can reuse handle_task_input() for the Redis push instead of duplicating that logic. notifications.py just detects the trigger and calls it. Also fixes the related-task metadata key from modelcontextprotocol.io/ to io.modelcontextprotocol/ to match the current spec: https://modelcontextprotocol.io/specification/2025-11-25/basic/utilities/tasks Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""
|
|
Tests for SEP-1686 related-task metadata in protocol responses.
|
|
|
|
Per the spec, all task-related responses MUST include
|
|
io.modelcontextprotocol/related-task in _meta.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from fastmcp import FastMCP
|
|
from fastmcp.client import Client
|
|
|
|
|
|
@pytest.fixture
|
|
async def metadata_server():
|
|
"""Create a server for testing metadata."""
|
|
mcp = FastMCP("metadata-test")
|
|
|
|
@mcp.tool(task=True)
|
|
async def test_tool(value: int) -> int:
|
|
return value * 2
|
|
|
|
return mcp
|
|
|
|
|
|
async def test_tasks_get_includes_related_task_metadata(metadata_server: FastMCP):
|
|
"""tasks/get response includes io.modelcontextprotocol/related-task in _meta."""
|
|
async with Client(metadata_server) as client:
|
|
# Submit a task
|
|
task = await client.call_tool("test_tool", {"value": 5}, task=True)
|
|
task_id = task.task_id
|
|
|
|
# Get status via client (which uses protocol properly)
|
|
status = await client.get_task_status(task_id)
|
|
|
|
# GetTaskResult is returned from response with metadata
|
|
# Verify the protocol included related-task metadata by checking the response worked
|
|
assert status.taskId == task_id
|
|
assert status.status in ["working", "completed"]
|
|
|
|
|
|
async def test_tasks_result_includes_related_task_metadata(metadata_server: FastMCP):
|
|
"""tasks/result response includes io.modelcontextprotocol/related-task in _meta."""
|
|
async with Client(metadata_server) as client:
|
|
# Submit and complete a task
|
|
task = await client.call_tool("test_tool", {"value": 7}, task=True)
|
|
result = await task.result()
|
|
|
|
# Result should have metadata (added by task.result() or protocol)
|
|
# Just verify the result is valid and contains the expected value
|
|
assert result.content
|
|
assert result.data == 14 # 7 * 2
|
|
|
|
|
|
async def test_tasks_list_includes_related_task_metadata(metadata_server: FastMCP):
|
|
"""tasks/list response includes io.modelcontextprotocol/related-task in _meta."""
|
|
async with Client(metadata_server) as client:
|
|
# List tasks via client (which uses protocol properly)
|
|
result = await client.list_tasks()
|
|
|
|
# Verify list_tasks works and returns proper structure
|
|
assert "tasks" in result
|
|
assert isinstance(result["tasks"], list)
|