mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-09 15:19: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>
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
"""
|
|
Tests for client-side task protocol.
|
|
|
|
Generic protocol tests that use tools as test fixtures.
|
|
"""
|
|
|
|
import asyncio
|
|
|
|
from fastmcp import FastMCP
|
|
from fastmcp.client import Client
|
|
|
|
|
|
async def test_end_to_end_task_flow():
|
|
"""Complete end-to-end flow: submit, poll, retrieve."""
|
|
start_signal = asyncio.Event()
|
|
complete_signal = asyncio.Event()
|
|
|
|
mcp = FastMCP("protocol-test")
|
|
|
|
@mcp.tool(task=True)
|
|
async def controlled_tool(message: str) -> str:
|
|
"""Tool with controlled execution."""
|
|
start_signal.set()
|
|
await complete_signal.wait()
|
|
return f"Processed: {message}"
|
|
|
|
async with Client(mcp) as client:
|
|
# Submit task
|
|
task = await client.call_tool(
|
|
"controlled_tool", {"message": "integration test"}, task=True
|
|
)
|
|
|
|
# Wait for execution to start
|
|
await asyncio.wait_for(start_signal.wait(), timeout=2.0)
|
|
|
|
# Check status while running
|
|
status = await task.status()
|
|
assert status.status in ["working"]
|
|
|
|
# Signal completion
|
|
complete_signal.set()
|
|
|
|
# Wait for task to finish and retrieve result
|
|
result = await task.result()
|
|
assert result.data == "Processed: integration test"
|
|
|
|
|
|
async def test_multiple_concurrent_tasks():
|
|
"""Multiple tasks can run concurrently."""
|
|
mcp = FastMCP("concurrent-test")
|
|
|
|
@mcp.tool(task=True)
|
|
async def multiply(a: int, b: int) -> int:
|
|
return a * b
|
|
|
|
async with Client(mcp) as client:
|
|
# Submit multiple tasks
|
|
tasks = []
|
|
for i in range(5):
|
|
task = await client.call_tool("multiply", {"a": i, "b": 2}, task=True)
|
|
tasks.append((task, i * 2))
|
|
|
|
# Wait for all to complete and verify results
|
|
for task, expected in tasks:
|
|
result = await task.result()
|
|
assert result.data == expected
|
|
|
|
|
|
async def test_task_id_auto_generation():
|
|
"""Task IDs are auto-generated if not provided."""
|
|
mcp = FastMCP("id-test")
|
|
|
|
@mcp.tool(task=True)
|
|
async def echo(message: str) -> str:
|
|
return f"Echo: {message}"
|
|
|
|
async with Client(mcp) as client:
|
|
# Submit without custom task ID
|
|
task_1 = await client.call_tool("echo", {"message": "first"}, task=True)
|
|
task_2 = await client.call_tool("echo", {"message": "second"}, task=True)
|
|
|
|
# Should generate different IDs
|
|
assert task_1.task_id != task_2.task_id
|
|
assert len(task_1.task_id) > 0
|
|
assert len(task_2.task_id) > 0
|