mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-13 00:59:10 +02:00
66aaf420c9* Implement MCP background tasks (SEP-1686) using Docket Adds support for background task execution via the MCP task protocol, powered by Docket for task queue management. - Tools, resources, and prompts can be marked with `task=True` to run async - Progress dependency for tracking task progress - CurrentDocket and CurrentWorker dependencies for advanced use cases - Client API with `.call_tool(..., task=True)` returns task handles - Task status notifications via subscriptions - CLI worker command for distributed task processing Configuration via environment: - FASTMCP_ENABLE_DOCKET=true - FASTMCP_ENABLE_TASKS=true - FASTMCP_DOCKET_URL=redis://... (or memory:// for single-process) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * Fix tasks example import (TaskStatusResponse → GetTaskResult) The example was using a non-existent TaskStatusResponse type. Updated to use mcp.types.GetTaskResult which is what the on_status_change callback actually receives. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * Fix env var name in Docket error messages The error messages referenced FASTMCP_EXPERIMENTAL_ENABLE_DOCKET but the actual setting is FASTMCP_ENABLE_DOCKET. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * Remove deprecated code re-added from pre-#2329 branch - Remove ExtendedEnvSettingsSource (FASTMCP_SERVER_ prefix support) - Remove dependencies parameter from FastMCP.__init__ * Replace fakeredis git pin with PyPI release * Remove redundant fakeredis dev dep (pulled via pydocket) --------- Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: Jeremiah Lowin <153965+jlowin@users.noreply.github.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
|
|
modelcontextprotocol.io/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 modelcontextprotocol.io/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 modelcontextprotocol.io/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 modelcontextprotocol.io/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)
|