refactor: inline _register_default_task_handlers into enable_tasks

Author: maxisbey

src/mcp/server/lowlevel/experimental.py

@@ -110,22 +110,12 @@ def enable_tasks(
 
         self._task_support = TaskSupport(store=store, queue=queue)
 
-        # Auto-register default handlers
-        self._register_default_task_handlers()
-
-        return self._task_support
-
-    def _register_default_task_handlers(self) -> None:
-        """Register default handlers for task operations."""
-        assert self._task_support is not None
-        support = self._task_support
-
         if not self._has_handler("tasks/get"):
 
             async def _default_get_task(
                 ctx: ServerRequestContext[Any, Any], params: GetTaskRequestParams
             ) -> GetTaskResult:
-                task = await support.store.get_task(params.task_id)
+                task = await self._task_support.store.get_task(params.task_id)
                 if task is None:
                     raise MCPError(code=INVALID_PARAMS, message=f"Task not found: {params.task_id}")
                 return GetTaskResult(
@@ -147,7 +137,7 @@ async def _default_get_task_result(
             ) -> GetTaskPayloadResult:
                 assert ctx.request_id is not None
                 req = GetTaskPayloadRequest(params=params)
-                result = await support.handler.handle(req, ctx.session, ctx.request_id)
+                result = await self._task_support.handler.handle(req, ctx.session, ctx.request_id)
                 return result
 
             self._add_request_handler("tasks/result", _default_get_task_result)
@@ -158,7 +148,7 @@ async def _default_list_tasks(
                 ctx: ServerRequestContext[Any, Any], params: PaginatedRequestParams | None
             ) -> ListTasksResult:
                 cursor = params.cursor if params else None
-                tasks, next_cursor = await support.store.list_tasks(cursor)
+                tasks, next_cursor = await self._task_support.store.list_tasks(cursor)
                 return ListTasksResult(tasks=tasks, next_cursor=next_cursor)
 
             self._add_request_handler("tasks/list", _default_list_tasks)
@@ -168,7 +158,7 @@ async def _default_list_tasks(
             async def _default_cancel_task(
                 ctx: ServerRequestContext[Any, Any], params: CancelTaskRequestParams
             ) -> CancelTaskResult:
-                result = await cancel_task(support.store, params.task_id)
+                result = await cancel_task(self._task_support.store, params.task_id)
                 return result
 
             self._add_request_handler("tasks/cancel", _default_cancel_task)