mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-26 07:24:18 +02:00
6fce4e538fEngine modules (keys, context snapshot, docket lifespan, worker CLI, client handles) move intact; SEP-1686 wire modules park in _legacy_wire for adaptation to SEP-2663. Core keeps task=True declaration on tools only and raises at serve time until the tasks extension is registered. Co-Authored-By: Claude <noreply@anthropic.com>
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
"""
|
|
Tests for SEP-1686 task capabilities declaration.
|
|
|
|
Verifies that the server correctly advertises task support.
|
|
Task protocol is now always enabled.
|
|
"""
|
|
|
|
import pytest
|
|
from fastmcp_tasks._legacy_wire.capabilities import get_task_capabilities
|
|
|
|
from fastmcp import FastMCP
|
|
from fastmcp.client import Client
|
|
|
|
pytestmark = pytest.mark.skip(
|
|
reason="Phase 3: requires TasksExtension (SEP-2663 adapter)"
|
|
)
|
|
|
|
|
|
async def test_capabilities_include_tasks():
|
|
"""Server capabilities always include tasks in first-class field (SEP-1686)."""
|
|
mcp = FastMCP("capability-test")
|
|
|
|
@mcp.tool()
|
|
async def test_tool() -> str:
|
|
return "test"
|
|
|
|
async with Client(mcp, mode="legacy") as client:
|
|
# Get server initialization result which includes capabilities
|
|
init_result = client.initialize_result
|
|
|
|
# Verify tasks capability is present as a first-class field (not experimental)
|
|
assert init_result.capabilities.tasks is not None
|
|
assert init_result.capabilities.tasks == get_task_capabilities()
|
|
# Verify it's NOT in experimental
|
|
assert "tasks" not in (init_result.capabilities.experimental or {})
|
|
|
|
|
|
def test_only_tools_advertise_task_support():
|
|
"""Task requests advertise tools only, not prompts/resources (sdk-feedback #3).
|
|
|
|
SDK v2 b1 ``ReadResourceRequestParams`` / ``GetPromptRequestParams`` have no
|
|
``task`` field, so resource/prompt task submissions always graceful-degrade
|
|
to synchronous execution. Advertising those capabilities would mislead
|
|
clients into sending task-augmented reads/gets, so the honest contract is
|
|
tools-only.
|
|
"""
|
|
capabilities = get_task_capabilities()
|
|
assert capabilities is not None
|
|
requests = capabilities.requests
|
|
assert requests is not None
|
|
assert requests.tools is not None
|
|
assert requests.tools.call is not None
|
|
# No prompt/resource task capability of any form is advertised.
|
|
assert getattr(requests, "prompts", None) is None
|
|
assert getattr(requests, "resources", None) is None
|
|
dumped = requests.model_dump(exclude_none=True)
|
|
assert set(dumped) == {"tools"}
|
|
|
|
|
|
async def test_client_uses_task_capable_session():
|
|
"""Client uses task-capable initialization."""
|
|
mcp = FastMCP("client-cap-test")
|
|
|
|
@mcp.tool()
|
|
async def test_tool() -> str:
|
|
return "test"
|
|
|
|
async with Client(mcp, mode="legacy") as client:
|
|
# Client should have connected successfully with task capabilities
|
|
assert client.initialize_result is not None
|
|
# Session should be a ClientSession (task-capable init uses standard session)
|
|
assert type(client.session).__name__ == "ClientSession"
|
|
|
|
|
|
def test_capabilities_hidden_when_pydocket_too_old(monkeypatch):
|
|
"""Capability advertisement and handler registration must agree.
|
|
|
|
If ``is_docket_available()`` returns False (e.g. an old transitive
|
|
pydocket), the server skips registering task handlers — so it must
|
|
also stop advertising task capabilities, or clients would discover
|
|
task support and then hit "method not found" at runtime.
|
|
"""
|
|
import importlib.metadata
|
|
|
|
from fastmcp.server import dependencies
|
|
|
|
original_version = importlib.metadata.version
|
|
|
|
def fake_version(name: str) -> str:
|
|
if name == "pydocket":
|
|
return "0.16.6"
|
|
return original_version(name)
|
|
|
|
monkeypatch.setattr(dependencies, "_DOCKET_AVAILABLE", None)
|
|
monkeypatch.setattr(importlib.metadata, "version", fake_version)
|
|
|
|
assert get_task_capabilities() is None
|