Fix type: use TasksCallCapability instead of dict

This commit is contained in:
Jeremiah Lowin 2026-01-13 22:51:47 -05:00
commit 3c3a62ee7d

View file

@ -5,6 +5,7 @@ from importlib.util import find_spec
from mcp.types import (
ServerTasksCapability,
ServerTasksRequestsCapability,
TasksCallCapability,
TasksCancelCapability,
TasksListCapability,
TasksToolsCapability,
@ -34,7 +35,7 @@ def get_task_capabilities() -> ServerTasksCapability | None:
list=TasksListCapability(),
cancel=TasksCancelCapability(),
requests=ServerTasksRequestsCapability(
tools=TasksToolsCapability(call={}),
tools=TasksToolsCapability(call=TasksCallCapability()),
prompts={"get": {}},
resources={"read": {}},
),