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,
|
||||
ServerTasksCapability,
|
||||
ServerTasksRequestsCapability,
|
||||
TasksCallCapability,
|
||||
TasksCancelCapability,
|
||||
TasksListCapability,
|
||||
TasksToolsCapability,
|
||||
|
|
@ -79,7 +80,7 @@ class ExperimentalHandlers:
|
|||
capabilities.tasks.cancel = TasksCancelCapability()
|
||||
|
||||
capabilities.tasks.requests = ServerTasksRequestsCapability(
|
||||
tools=TasksToolsCapability()
|
||||
tools=TasksToolsCapability(call=TasksCallCapability())
|
||||
) # assuming always supported for now
|
||||
|
||||
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.list 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
|
||||
|
|
|
|||
Loading…
Reference in a new issue