Backport: Add missing TasksCallCapability to v1.x (#2137)
Co-authored-by: Max Isbey <224885523+maxisbey@users.noreply.github.com> Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
parent
c68e254bad
commit
f8d98b63a7
2 changed files with 6 additions and 1 deletions
|
|
@ -31,6 +31,7 @@ from mcp.types import (
|
||||||
ServerResult,
|
ServerResult,
|
||||||
ServerTasksCapability,
|
ServerTasksCapability,
|
||||||
ServerTasksRequestsCapability,
|
ServerTasksRequestsCapability,
|
||||||
|
TasksCallCapability,
|
||||||
TasksCancelCapability,
|
TasksCancelCapability,
|
||||||
TasksListCapability,
|
TasksListCapability,
|
||||||
TasksToolsCapability,
|
TasksToolsCapability,
|
||||||
|
|
@ -79,7 +80,7 @@ class ExperimentalHandlers:
|
||||||
capabilities.tasks.cancel = TasksCancelCapability()
|
capabilities.tasks.cancel = TasksCancelCapability()
|
||||||
|
|
||||||
capabilities.tasks.requests = ServerTasksRequestsCapability(
|
capabilities.tasks.requests = ServerTasksRequestsCapability(
|
||||||
tools=TasksToolsCapability()
|
tools=TasksToolsCapability(call=TasksCallCapability())
|
||||||
) # assuming always supported for now
|
) # assuming always supported for now
|
||||||
|
|
||||||
def enable_tasks(
|
def enable_tasks(
|
||||||
|
|
|
||||||
|
|
@ -227,6 +227,10 @@ async def test_enable_tasks_auto_registers_handlers() -> None:
|
||||||
assert caps_after.tasks is not None
|
assert caps_after.tasks is not None
|
||||||
assert caps_after.tasks.list is not None
|
assert caps_after.tasks.list is not None
|
||||||
assert caps_after.tasks.cancel is not None
|
assert caps_after.tasks.cancel is not None
|
||||||
|
# Verify nested call capability is present
|
||||||
|
assert caps_after.tasks.requests is not None
|
||||||
|
assert caps_after.tasks.requests.tools is not None
|
||||||
|
assert caps_after.tasks.requests.tools.call is not None
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.anyio
|
@pytest.mark.anyio
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue