@j0hanz/fetch-url-mcp 1.12.0 → 1.12.2

package/dist/tasks/execution.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"execution.d.ts","sourceRoot":"","sources":["../../src/tasks/execution.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,EAGL,KAAK,YAAY,EAClB,MAAM,oCAAoC,CAAC;AAQ5C,OAAO,EAEL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,KAAK,gBAAgB,EAErB,KAAK,SAAS,EACf,MAAM,cAAc,CAAC;AACtB,OAAO,EAGL,KAAK,eAAe,EAErB,MAAM,YAAY,CAAC;AAgCpB,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAGvD;AAED,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,MAAM,EAChB,aAAa,SAA4D,GACxE,MAAM,CAOR;AAED,wBAAgB,sBAAsB,IAAI,IAAI,CAE7C;AAMD,KAAK,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAC5C,KAAK,uBAAuB,GAAG,IAAI,CACjC,SAAS,EACP,QAAQ,GACR,QAAQ,GACR,eAAe,GACf,WAAW,GACX,eAAe,GACf,KAAK,GACL,cAAc,CACjB,CAAC;AAEF,wBAAgB,aAAa,CAAC,IAAI,EAAE,uBAAuB,GAAG,WAAW,CAUxE;AAED,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,SAAS,EACjB,IAAI,EAAE,SAAS,GACd,IAAI,CAeN;AAED,wBAAgB,iBAAiB,IAAI,KAAK,CAEzC;AAmID,wBAAsB,qBAAqB,CACzC,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,uBAAuB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,YAAY,CAAC,CA4CvB"}
+{"version":3,"file":"execution.d.ts","sourceRoot":"","sources":["../../src/tasks/execution.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,EAGL,KAAK,YAAY,EAClB,MAAM,oCAAoC,CAAC;AAc5C,OAAO,EAEL,KAAK,uBAAuB,EAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,KAAK,gBAAgB,EAErB,KAAK,SAAS,EACf,MAAM,cAAc,CAAC;AACtB,OAAO,EAGL,KAAK,eAAe,EAErB,MAAM,YAAY,CAAC;AAkCpB,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAGvD;AAED,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,MAAM,EAChB,aAAa,SAA4D,GACxE,MAAM,CAOR;AAED,wBAAgB,sBAAsB,IAAI,IAAI,CAE7C;AAMD,KAAK,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAC5C,KAAK,uBAAuB,GAAG,IAAI,CACjC,SAAS,EACP,QAAQ,GACR,QAAQ,GACR,eAAe,GACf,UAAU,GACV,OAAO,GACP,WAAW,GACX,eAAe,GACf,KAAK,GACL,cAAc,CACjB,CAAC;AAEF,wBAAgB,aAAa,CAAC,IAAI,EAAE,uBAAuB,GAAG,WAAW,CAYxE;AAED,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,SAAS,EACjB,IAAI,EAAE,SAAS,GACd,IAAI,CAmBN;AAED,wBAAgB,iBAAiB,IAAI,KAAK,CAEzC;AA2JD,wBAAsB,qBAAqB,CACzC,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,uBAAuB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,YAAY,CAAC,CAwFvB"}

package/dist/tasks/execution.js

@@ -1,13 +1,13 @@
 import { ErrorCode, McpError, } from '@modelcontextprotocol/sdk/types.js';
 import { config } from '../lib/core.js';
-import { logWarn, runWithRequestContext } from '../lib/core.js';
+import { logDebug, logError, logInfo, logWarn, runWithRequestContext, } from '../lib/core.js';
+import {} from '../lib/mcp-interop.js';
 import { getErrorMessage } from '../lib/utils.js';
 import { isObject } from '../lib/utils.js';
 import { buildRelatedTaskMeta, } from './call-contract.js';
 import { taskManager, } from './manager.js';
 import { buildToolHandlerExtra, compact, tryReadToolStructuredError, } from './owner.js';
 import { getTaskCapableTool, getTaskCapableToolSupport, } from './registry.js';
-const TASK_NOT_FOUND_ERROR_CODE = -32002;
 /* -------------------------------------------------------------------------------------------------
  * Abort-controller management for in-flight task executions
  * ------------------------------------------------------------------------------------------------- */
@@ -20,7 +20,7 @@ function attachAbortController(taskId) {
         logWarn('Abort controller map reached task capacity — possible leak', {
             size: taskAbortControllers.size,
             maxTotal: config.tasks.maxTotal,
-        });
+        }, 'tasks');
     }
     const controller = new AbortController();
     taskAbortControllers.set(taskId, controller);
@@ -48,6 +48,8 @@ export function toTaskSummary(task) {
         taskId: task.taskId,
         status: task.status,
         ...(task.statusMessage ? { statusMessage: task.statusMessage } : {}),
+        ...(task.progress !== undefined ? { progress: task.progress } : {}),
+        ...(task.total !== undefined ? { total: task.total } : {}),
         createdAt: task.createdAt,
         lastUpdatedAt: task.lastUpdatedAt,
         ttl: task.ttl,
@@ -63,15 +65,15 @@ export function emitTaskStatusNotification(server, task) {
         params: { ...toTaskSummary(task) },
     })
         .catch((error) => {
-        logWarn('Failed to send task status notification', {
+        logError('Failed to send task status notification', {
             taskId: task.taskId,
             status: task.status,
             error: getErrorMessage(error),
-        });
+        }, 'tasks');
     });
 }
 export function throwTaskNotFound() {
-    throw new McpError(TASK_NOT_FOUND_ERROR_CODE, 'Task not found');
+    throw new McpError(ErrorCode.InvalidParams, 'Task not found');
 }
 /* -------------------------------------------------------------------------------------------------
  * Execution pipeline
@@ -87,28 +89,23 @@ function buildTaskFailureState(error) {
         ? (/^MCP error -?\d+:\s*(.*)$/s.exec(error.message)?.[1] ?? error.message)
         : undefined;
     const statusMessage = mcpErrorMessage ?? getErrorMessage(error);
-    const payload = { error: statusMessage };
     if (error instanceof McpError) {
-        payload['code'] = error.code;
-        if (error.data !== undefined) {
-            payload['data'] = error.data;
-        }
         return {
             status: 'failed',
             statusMessage,
-            result: {
-                content: [{ type: 'text', text: JSON.stringify(payload) }],
-                isError: true,
+            error: {
+                code: error.code,
+                ...(error.data !== undefined ? { data: error.data } : {}),
+                message: statusMessage,
             },
         };
     }
-    payload['code'] = ErrorCode.InternalError;
     return {
         status: 'failed',
         statusMessage,
-        result: {
-            content: [{ type: 'text', text: JSON.stringify(payload) }],
-            isError: true,
+        error: {
+            code: ErrorCode.InternalError,
+            message: statusMessage,
         },
     };
 }
@@ -131,30 +128,50 @@ async function runTaskToolExecution(params) {
         ...(sessionId ? { sessionId } : {}),
     }, async () => {
         const controller = attachAbortController(taskId);
+        const progressState = { closed: false };
         try {
+            logInfo('Task execution started', { taskId, tool: tool.name }, 'tasks');
             const relatedMeta = buildRelatedTaskMeta(taskId, meta);
             const result = await tool.execute(args, {
                 signal: controller.signal,
                 requestId: taskId,
                 _meta: relatedMeta,
+                progressState,
                 canReportProgress: () => taskManager.getTask(taskId)?.status === 'working',
                 ...compact({ sendNotification }),
-                onProgress: (_progress, message) => {
+                onProgress: (progress, message, total) => {
                     const current = taskManager.getTask(taskId);
                     if (current?.status === 'working' &&
-                        current.statusMessage !== message) {
+                        (current.statusMessage !== message ||
+                            current.progress !== progress ||
+                            (total !== undefined && current.total !== total))) {
                         updateTaskAndEmitStatus(server, taskId, {
                             statusMessage: message,
+                            progress,
+                            ...(total !== undefined ? { total } : {}),
                         });
                     }
                 },
             });
-            updateTaskAndEmitStatus(server, taskId, buildTaskCompletionUpdate(result, tool));
+            const completionUpdate = buildTaskCompletionUpdate(result, tool);
+            updateTaskAndEmitStatus(server, taskId, completionUpdate);
+            if (completionUpdate.status === 'completed') {
+                logInfo('Task execution completed', { taskId, tool: tool.name }, 'tasks');
+            }
+            else {
+                logWarn('Task execution completed with tool error result', { taskId, tool: tool.name }, 'tasks');
+            }
         }
         catch (error) {
+            logError('Task execution failed', {
+                taskId,
+                tool: tool.name,
+                error: getErrorMessage(error),
+            }, 'tasks');
             updateTaskAndEmitStatus(server, taskId, buildTaskFailureState(error));
         }
         finally {
+            progressState.closed = true;
             taskAbortControllers.delete(taskId);
         }
     });
@@ -162,16 +179,21 @@ async function runTaskToolExecution(params) {
 export async function handleToolCallRequest(server, request, context) {
     const { params } = request;
     // Validate the tool name first so an unknown tool always produces MethodNotFound
-    const tool = getTaskCapableTool(params.name);
+    const tool = getTaskCapableTool(server, params.name);
     if (!tool) {
         throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: '${params.name}'`);
     }
     if (params.task) {
-        if (getTaskCapableToolSupport(params.name) === 'forbidden') {
+        if (getTaskCapableToolSupport(server, params.name) === 'forbidden') {
             throw new McpError(ErrorCode.MethodNotFound, `Task augmentation is forbidden for tool '${params.name}'`);
         }
         const args = tool.parseArguments(params.arguments);
         const task = taskManager.createTask(params.task.ttl !== undefined ? { ttl: params.task.ttl } : undefined, 'Task started', context.ownerKey);
+        logInfo('Task execution queued', {
+            taskId: task.taskId,
+            tool: params.name,
+            ...(params.task.ttl !== undefined ? { ttl: params.task.ttl } : {}),
+        }, 'tasks');
         void runTaskToolExecution({
             server,
             taskId: task.taskId,
@@ -183,8 +205,33 @@ export async function handleToolCallRequest(server, request, context) {
                 sendNotification: context.sendNotification,
             }),
         });
-        return { task: toTaskSummary(task) };
+        return {
+            task: toTaskSummary(task),
+            ...(tool.immediateResponse
+                ? {
+                    _meta: {
+                        'io.modelcontextprotocol/model-immediate-response': tool.immediateResponse,
+                    },
+                }
+                : {}),
+        };
+    }
+    if (getTaskCapableToolSupport(server, params.name) === 'required') {
+        throw new McpError(ErrorCode.MethodNotFound, `Task augmentation is required for tool '${params.name}'`);
     }
     const args = tool.parseArguments(params.arguments);
-    return tool.execute(args, buildToolHandlerExtra(context, params._meta));
+    const progressState = { closed: false };
+    logDebug('Executing task-capable tool inline', {
+        tool: params.name,
+        hasProgressToken: params._meta?.progressToken !== undefined,
+    }, 'tasks');
+    try {
+        return await tool.execute(args, {
+            ...buildToolHandlerExtra(context, params._meta),
+            progressState,
+        });
+    }
+    finally {
+        progressState.closed = true;
+    }
 }

package/dist/tasks/handlers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"handlers.d.ts","sourceRoot":"","sources":["../../src/tasks/handlers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAmCzE,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AA4D7E,UAAU,8BAA8B;IACtC,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,UAAU,6BAA6B;IACrC,oBAAoB,EAAE,OAAO,CAAC;IAC9B,0BAA0B,EAAE,OAAO,CAAC;CACrC;AACD,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,SAAS,EACjB,OAAO,CAAC,EAAE,8BAA8B,GACvC,6BAA6B,CAyJ/B"}
+{"version":3,"file":"handlers.d.ts","sourceRoot":"","sources":["../../src/tasks/handlers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAmCzE,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AA4D7E,UAAU,8BAA8B;IACtC,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,UAAU,6BAA6B;IACrC,oBAAoB,EAAE,OAAO,CAAC;IAC9B,0BAA0B,EAAE,OAAO,CAAC;CACrC;AAkBD,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,SAAS,EACjB,OAAO,CAAC,EAAE,8BAA8B,GACvC,6BAA6B,CAoK/B"}

package/dist/tasks/handlers.js

@@ -1,8 +1,9 @@
 import { randomUUID } from 'node:crypto';
 import {} from '@modelcontextprotocol/sdk/server/mcp.js';
-import { CallToolRequestSchema, ErrorCode, McpError, } from '@modelcontextprotocol/sdk/types.js';
+import { CallToolRequestSchema, ErrorCode, } from '@modelcontextprotocol/sdk/types.js';
+import { McpError } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
-import { logWarn, runWithRequestContext } from '../lib/core.js';
+import { logDebug, logWarn, runWithRequestContext } from '../lib/core.js';
 import { getSdkCallToolHandler } from '../lib/mcp-interop.js';
 import { parseExtendedCallToolRequest, withRelatedTaskMeta, } from './call-contract.js';
 import { abortTaskExecution, emitTaskStatusNotification, handleToolCallRequest, throwTaskNotFound, toTaskSummary, } from './execution.js';
@@ -40,15 +41,21 @@ function resolveOwnerScopedExtra(extra) {
         ownerKey: resolveTaskOwnerKey(parsedExtra),
     };
 }
+function throwStoredTaskError(task) {
+    if (task.error) {
+        throw new McpError(task.error.code, task.error.message, task.error.data);
+    }
+    throw new McpError(ErrorCode.InternalError, task.statusMessage ?? 'Task execution failed', { taskId: task.taskId });
+}
 export function registerTaskHandlers(server, options) {
     const sdkCallToolHandler = getSdkCallToolHandler(server);
-    const taskCapableToolsRegistered = hasRegisteredTaskCapableTools();
+    const taskCapableToolsRegistered = hasRegisteredTaskCapableTools(server);
     const requireInterception = options?.requireInterception ?? true;
     if (!sdkCallToolHandler) {
         if (taskCapableToolsRegistered && requireInterception) {
             throw new Error('Task-capable tools are registered but SDK tools/call interception is unavailable. Upgrade compatibility or disable strict interception with TASKS_REQUIRE_INTERCEPTION=false.');
         }
-        logWarn('Task call interception disabled: SDK tools/call handler unavailable; task-capable tools require MCP SDK compatibility update', { sdkVersion: 'unknown' });
+        logWarn('Task call interception disabled: SDK tools/call handler unavailable; task-capable tools require MCP SDK compatibility update', { sdkVersion: 'unknown' }, 'tasks');
     }
     if (sdkCallToolHandler) {
         server.server.setRequestHandler(CallToolRequestSchema, async (request, extra) => {
@@ -66,11 +73,16 @@ export function registerTaskHandlers(server, options) {
                 const toolName = request.params.name;
                 // Only intercept task-capable tools managed by the local task registry.
                 // Delegate all other tools to the SDK handler to avoid shadowing future tools.
-                if (!hasTaskCapableTool(toolName)) {
+                if (!hasTaskCapableTool(server, toolName)) {
                     return sdkCallToolHandler(request, extra);
                 }
                 const parsed = parseExtendedCallToolRequest(request);
                 const context = resolveToolCallContext(parsedExtra, parsed.params._meta);
+                logDebug('Intercepted task-capable tool call', {
+                    tool: toolName,
+                    taskRequested: parsed.params.task !== undefined,
+                    hasProgressToken: parsed.params._meta?.progressToken !== undefined,
+                }, 'tasks');
                 return handleToolCallRequest(server, parsed, context);
             });
         });
@@ -78,6 +90,7 @@ export function registerTaskHandlers(server, options) {
     server.server.setRequestHandler(TaskGetSchema, (request, extra) => {
         const { taskId } = request.params;
         const { ownerKey } = resolveOwnerScopedExtra(extra);
+        logDebug('tasks/get requested', { taskId }, 'tasks');
         const task = taskManager.getTask(taskId, ownerKey);
         if (!task)
             throwTaskNotFound();
@@ -86,31 +99,23 @@ export function registerTaskHandlers(server, options) {
     server.server.setRequestHandler(TaskResultSchema, async (request, extra) => {
         const { taskId } = request.params;
         const { parsedExtra, ownerKey } = resolveOwnerScopedExtra(extra);
+        logDebug('tasks/result requested', { taskId }, 'tasks');
         const task = await taskManager.waitForTerminalTask(taskId, ownerKey, parsedExtra?.signal);
         if (!task)
             throwTaskNotFound();
         try {
+            if (task.status === 'cancelled') {
+                throwStoredTaskError(task);
+            }
             if (task.status === 'failed') {
+                if (task.error) {
+                    throwStoredTaskError(task);
+                }
                 const failedResult = (task.result ?? null);
-                const fallback = failedResult ?? {
-                    content: [
-                        {
-                            type: 'text',
-                            text: JSON.stringify({
-                                error: task.statusMessage ?? 'Task execution failed',
-                            }),
-                        },
-                    ],
-                    isError: true,
-                };
-                return withRelatedTaskMeta(fallback, task.taskId);
-            }
-            if (task.status === 'cancelled') {
-                throw new McpError(ErrorCode.InvalidRequest, 'Task was cancelled', {
-                    taskId: task.taskId,
-                    status: 'cancelled',
-                    ...(task.statusMessage ? { statusMessage: task.statusMessage } : {}),
-                });
+                if (failedResult) {
+                    return withRelatedTaskMeta(failedResult, task.taskId);
+                }
+                throwStoredTaskError(task);
             }
             const result = isServerResult(task.result)
                 ? task.result
@@ -127,6 +132,7 @@ export function registerTaskHandlers(server, options) {
     server.server.setRequestHandler(TaskListSchema, (request, extra) => {
         const { ownerKey } = resolveOwnerScopedExtra(extra);
         const cursor = request.params?.cursor;
+        logDebug('tasks/list requested', { hasCursor: cursor !== undefined }, 'tasks');
         const { tasks, nextCursor } = taskManager.listTasks(cursor === undefined ? { ownerKey } : { ownerKey, cursor });
         return {
             tasks: tasks.map((task) => toTaskSummary(task)),
@@ -136,6 +142,7 @@ export function registerTaskHandlers(server, options) {
     server.server.setRequestHandler(TaskCancelSchema, (request, extra) => {
         const { taskId } = request.params;
         const { ownerKey } = resolveOwnerScopedExtra(extra);
+        logDebug('tasks/cancel requested', { taskId }, 'tasks');
         const task = taskManager.cancelTask(taskId, ownerKey);
         if (!task)
             throwTaskNotFound();

package/dist/tasks/manager.d.ts

@@ -1,4 +1,4 @@
-export type TaskStatus = 'working' | 'completed' | 'failed' | 'cancelled';
+export type TaskStatus = 'working' | 'input_required' | 'completed' | 'failed' | 'cancelled';
 export interface TaskError {
     code: number;
     message: string;
@@ -9,6 +9,8 @@ export interface TaskState {
     ownerKey: string;
     status: TaskStatus;
     statusMessage?: string;
+    progress?: number;
+    total?: number;
     createdAt: string;
     lastUpdatedAt: string;
     ttl: number;
@@ -25,6 +27,8 @@ export interface CreateTaskResult {
         taskId: string;
         status: TaskStatus;
         statusMessage?: string;
+        progress?: number;
+        total?: number;
         createdAt: string;
         lastUpdatedAt: string;
         ttl: number;
@@ -33,7 +37,6 @@ export interface CreateTaskResult {
 }
 declare class TaskManager {
     private tasks;
-    private activeTaskCount;
     private ownerCounts;
     private readonly waiters;
     private cleanupInterval;

package/dist/tasks/manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/tasks/manager.ts"],"names":[],"mappings":"AAcA,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;AAE1E,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,UAAU,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,SAAS,CAAC;CACnB;AAOD,UAAU,iBAAiB;IACzB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,gBAAgB;IAC/B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;IACvB,IAAI,EAAE;QACJ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,SAAS,EAAE,MAAM,CAAC;QAClB,aAAa,EAAE,MAAM,CAAC;QACtB,GAAG,EAAE,MAAM,CAAC;QACZ,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAmDD,cAAM,WAAW;IACf,OAAO,CAAC,KAAK,CAAwC;IACrD,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,WAAW,CAA6B;IAChD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAEtB;IACF,OAAO,CAAC,eAAe,CAA+C;IAEtE,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,kBAAkB;IAS1B,OAAO,CAAC,UAAU;IAgBlB,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,gBAAgB;IASxB,OAAO,CAAC,mBAAmB;IAgB3B,OAAO,CAAC,mBAAmB;IAqB3B,UAAU,CACR,OAAO,CAAC,EAAE,iBAAiB,EAC3B,aAAa,SAAiB,EAC9B,QAAQ,GAAE,MAA0B,GACnC,SAAS;IAyBZ,OAAO,CAAC,gBAAgB;IAgBxB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAIjE,UAAU,CACR,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,GAAG,WAAW,CAAC,CAAC,GACxD,IAAI;IA2BP,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAepE,kBAAkB,CAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,SAAkE,GAC9E,SAAS,EAAE;IAad,OAAO,CAAC,WAAW;IA2BnB,SAAS,CAAC,OAAO,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG;QACzE,KAAK,EAAE,SAAS,EAAE,CAAC;QACnB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB;IAgBD,OAAO,CAAC,mBAAmB;IAOrB,mBAAmB,CACvB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAcjC,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;CAU7C;AAED,eAAO,MAAM,WAAW,aAAoB,CAAC;AAY7C,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,CAO7D;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,GACb;IAAE,YAAY,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CA0BjC"}
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/tasks/manager.ts"],"names":[],"mappings":"AAcA,MAAM,MAAM,UAAU,GAClB,SAAS,GACT,gBAAgB,GAChB,WAAW,GACX,QAAQ,GACR,WAAW,CAAC;AAEhB,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,UAAU,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,SAAS,CAAC;CACnB;AAMD,UAAU,iBAAiB;IACzB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,gBAAgB;IAC/B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;IACvB,IAAI,EAAE;QACJ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,aAAa,EAAE,MAAM,CAAC;QACtB,GAAG,EAAE,MAAM,CAAC;QACZ,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAiFD,cAAM,WAAW;IACf,OAAO,CAAC,KAAK,CAAwC;IACrD,OAAO,CAAC,WAAW,CAA6B;IAChD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAEtB;IACF,OAAO,CAAC,eAAe,CAA+C;IAEtE,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,kBAAkB;IAkB1B,OAAO,CAAC,UAAU;IAgBlB,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,gBAAgB;IAgBxB,OAAO,CAAC,mBAAmB;IAU3B,OAAO,CAAC,mBAAmB;IAoB3B,UAAU,CACR,OAAO,CAAC,EAAE,iBAAiB,EAC3B,aAAa,SAAiB,EAC9B,QAAQ,GAAE,MAA0B,GACnC,SAAS;IAiCZ,OAAO,CAAC,gBAAgB;IAgBxB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAIjE,UAAU,CACR,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,GAAG,WAAW,CAAC,CAAC,GACxD,IAAI;IA+BP,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAuBpE,kBAAkB,CAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,SAAkE,GAC9E,SAAS,EAAE;IAuBd,OAAO,CAAC,WAAW;IA2BnB,SAAS,CAAC,OAAO,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG;QACzE,KAAK,EAAE,SAAS,EAAE,CAAC;QACnB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB;IAgBD,OAAO,CAAC,mBAAmB;IAOrB,mBAAmB,CACvB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAcjC,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;CAU7C;AAED,eAAO,MAAM,WAAW,aAAoB,CAAC;AAY7C,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,CAO7D;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,GACb;IAAE,YAAY,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CA0BjC"}

package/dist/tasks/manager.js

@@ -2,7 +2,7 @@ import { randomUUID } from 'node:crypto';
 import { createHmac, randomBytes } from 'node:crypto';
 import { setInterval } from 'node:timers';
 import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
-import { config, logWarn } from '../lib/core.js';
+import { config, logInfo, logWarn } from '../lib/core.js';
 import { isObject, timingSafeEqualUtf8 } from '../lib/utils.js';
 import { TaskWaiterRegistry, waitForTerminalTask as waitForTerminalTaskWithDeadline, } from './waiters.js';
 const DEFAULT_TTL_MS = 60_000;
@@ -15,12 +15,18 @@ const CLEANUP_INTERVAL_MS = 60_000;
 const RESULT_DELIVERY_GRACE_MS = 10_000;
 const TASK_STATUS_VALUES = new Set([
     'working',
+    'input_required',
+    'completed',
+    'failed',
+    'cancelled',
+]);
+const TERMINAL_STATUSES = new Set([
     'completed',
     'failed',
     'cancelled',
 ]);
 function isTerminalStatus(status) {
-    return status !== 'working';
+    return TERMINAL_STATUSES.has(status);
 }
 function resolveNextTaskStatus(task, updates) {
     const nextStatus = updates.status;
@@ -39,9 +45,24 @@ function normalizeTaskTtl(ttl) {
         return DEFAULT_TTL_MS;
     return Math.max(MIN_TTL_MS, Math.min(Math.trunc(ttl), MAX_TTL_MS));
 }
+function logTaskStatusTransition(task, previousStatus, nextStatus) {
+    if (previousStatus === nextStatus)
+        return;
+    const meta = {
+        taskId: task.taskId,
+        ownerKey: task.ownerKey,
+        previousStatus,
+        nextStatus,
+        ...(task.statusMessage ? { statusMessage: task.statusMessage } : {}),
+    };
+    if (nextStatus === 'failed') {
+        logWarn('Task status changed to failed', meta, 'tasks');
+        return;
+    }
+    logInfo('Task status changed', meta, 'tasks');
+}
 class TaskManager {
     tasks = new Map();
-    activeTaskCount = 0;
     ownerCounts = new Map();
     waiters = new TaskWaiterRegistry(isTerminalStatus);
     cleanupInterval = null;
@@ -67,6 +88,11 @@ class TaskManager {
         const now = Date.now();
         for (const task of this.tasks.values()) {
             if (this.isTaskExpired(task, now)) {
+                logWarn('Task expired', {
+                    taskId: task.taskId,
+                    ownerKey: task.ownerKey,
+                    status: task.status,
+                }, 'tasks');
                 this.removeTask(task.taskId);
             }
         }
@@ -90,15 +116,18 @@ class TaskManager {
         task.lastUpdatedAt = new Date().toISOString();
     }
     cancelActiveTask(task, statusMessage) {
-        this.applyTaskUpdate(task, { status: 'cancelled', statusMessage });
+        this.applyTaskUpdate(task, {
+            status: 'cancelled',
+            statusMessage,
+            error: {
+                code: ErrorCode.ConnectionClosed,
+                message: statusMessage,
+                data: { code: 'ABORTED' },
+            },
+        });
         this.waiters.notify(task);
-        this.releaseTaskCapacity(task);
     }
     releaseTaskCapacity(task) {
-        const internal = task;
-        if (!internal._capacityCounted)
-            return;
-        internal._capacityCounted = false;
         const { ownerKey } = task;
         const nextCount = (this.ownerCounts.get(ownerKey) ?? 0) - 1;
         if (nextCount > 0) {
@@ -107,19 +136,16 @@ class TaskManager {
         else {
             this.ownerCounts.delete(ownerKey);
         }
-        if (this.activeTaskCount > 0)
-            this.activeTaskCount--;
     }
     reserveTaskCapacity(ownerKey) {
         const { maxPerOwner, maxTotal } = config.tasks;
-        if (this.activeTaskCount >= maxTotal) {
+        if (this.tasks.size >= maxTotal) {
             throw new McpError(ErrorCode.InvalidRequest, `Task capacity reached (${maxTotal} total tasks)`);
         }
         if ((this.ownerCounts.get(ownerKey) ?? 0) >= maxPerOwner) {
             throw new McpError(ErrorCode.InvalidRequest, `Task capacity reached for owner (${maxPerOwner} tasks)`);
         }
         this.ownerCounts.set(ownerKey, (this.ownerCounts.get(ownerKey) ?? 0) + 1);
-        this.activeTaskCount += 1;
     }
     createTask(options, statusMessage = 'Task started', ownerKey = DEFAULT_OWNER_KEY) {
         this.removeExpiredTasks();
@@ -136,10 +162,14 @@ class TaskManager {
             ttl: normalizeTaskTtl(options?.ttl),
             pollInterval: DEFAULT_POLL_INTERVAL_MS,
             _createdAtMs: now.getTime(),
-            _capacityCounted: true,
         };
         this.tasks.set(task.taskId, task);
         this.ensureCleanupLoop();
+        logInfo('Task created', {
+            taskId: task.taskId,
+            ownerKey,
+            ttl: task.ttl,
+        }, 'tasks');
         return task;
     }
     lookupActiveTask(taskId, ownerKey) {
@@ -160,25 +190,24 @@ class TaskManager {
     updateTask(taskId, updates) {
         const task = this.tasks.get(taskId);
         if (!task) {
-            logWarn('updateTask called for unknown task', { taskId });
+            logWarn('updateTask called for unknown task', { taskId }, 'tasks');
             return;
         }
         if (isTerminalStatus(task.status)) {
             logWarn('updateTask called for terminal task', {
                 taskId,
                 currentStatus: task.status,
-            });
+            }, 'tasks');
             return;
         }
         const nextStatus = resolveNextTaskStatus(task, updates);
+        const previousStatus = task.status;
         this.applyTaskUpdate(task, {
             ...updates,
             ...(updates.status === undefined ? {} : { status: nextStatus }),
         });
+        logTaskStatusTransition(task, previousStatus, task.status);
         this.waiters.notify(task);
-        if (isTerminalStatus(nextStatus)) {
-            this.releaseTaskCapacity(task);
-        }
     }
     cancelTask(taskId, ownerKey) {
         const task = this.lookupActiveTask(taskId, ownerKey);
@@ -188,6 +217,10 @@ class TaskManager {
             throw new McpError(ErrorCode.InvalidParams, `Cannot cancel task: already in terminal status '${task.status}'`);
         }
         this.cancelActiveTask(task, 'The task was cancelled by request.');
+        logInfo('Task cancelled by request', {
+            taskId: task.taskId,
+            ownerKey: task.ownerKey,
+        }, 'tasks');
         return task;
     }
     cancelTasksByOwner(ownerKey, statusMessage = 'The task was cancelled because its owner is no longer active.') {
@@ -200,6 +233,12 @@ class TaskManager {
                 cancelled.push(task);
             }
         }
+        if (cancelled.length > 0) {
+            logInfo('Tasks cancelled for owner', {
+                ownerKey,
+                count: cancelled.length,
+            }, 'tasks');
+        }
         return cancelled;
     }
     collectPage(ownerKey, anchorTaskId, pageSize) {

package/dist/tasks/owner.d.ts

@@ -21,6 +21,10 @@ export interface ToolExecutionContext {
     requestMeta?: ToolCallRequestMeta;
 }
 export type ToolCallContext = ToolExecutionContext;
+interface AuthIdentity {
+    clientId?: string;
+    token?: string;
+}
 /** Strip keys whose value is `undefined`, returning an object with only the
  * present keys. Return type correctly omits the `undefined` union so the result
  * is compatible with `exactOptionalPropertyTypes`. */
@@ -29,6 +33,7 @@ type Compacted<T extends object> = {
 };
 export declare function compact<T extends object>(obj: T): Compacted<T>;
 export declare function parseHandlerExtra(extra: unknown): HandlerExtra | undefined;
+export declare function buildAuthenticatedOwnerKey(authInfo?: AuthIdentity): string | undefined;
 export declare function resolveTaskOwnerKey(extra?: HandlerExtra): string;
 export declare function resolveToolCallContext(extra?: HandlerExtra, requestMeta?: ToolCallRequestMeta): ToolCallContext;
 export declare function buildToolHandlerExtra(context: ToolExecutionContext, requestMeta?: ToolCallRequestMeta): ToolHandlerExtra;

package/dist/tasks/owner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"owner.d.ts","sourceRoot":"","sources":["../../src/tasks/owner.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AAIvE,OAAO,KAAK,EACV,oBAAoB,EACpB,gBAAgB,EACjB,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EAEL,KAAK,mBAAmB,EACzB,MAAM,oBAAoB,CAAC;AAM5B,UAAU,YAAY;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACjD,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,CAAC,YAAY,EAAE,oBAAoB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,CAAC,YAAY,EAAE,oBAAoB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,WAAW,CAAC,EAAE,mBAAmB,CAAC;CACnC;AAED,MAAM,MAAM,eAAe,GAAG,oBAAoB,CAAC;AAEnD;;sDAEsD;AACtD,KAAK,SAAS,CAAC,CAAC,SAAS,MAAM,IAAI;KAChC,CAAC,IAAI,MAAM,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,SAAS,KAAK,GACnD,KAAK,GACL,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;CAClC,CAAC;AAEF,wBAAgB,OAAO,CAAC,CAAC,SAAS,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAQ9D;AA2BD,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,YAAY,GAAG,SAAS,CA0B1E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,CAAC,EAAE,YAAY,GAAG,MAAM,CAMhE;AAuDD,wBAAgB,sBAAsB,CACpC,KAAK,CAAC,EAAE,YAAY,EACpB,WAAW,CAAC,EAAE,mBAAmB,GAChC,eAAe,CAEjB;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,oBAAoB,EAC7B,WAAW,CAAC,EAAE,mBAAmB,GAChC,gBAAgB,CAOlB;AAED,wBAAgB,2BAA2B,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,EAC5E,OAAO,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,GAC7D,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAmBvD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAEpE;AAWD,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAe7E"}
+{"version":3,"file":"owner.d.ts","sourceRoot":"","sources":["../../src/tasks/owner.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AAQvE,OAAO,KAAK,EACV,oBAAoB,EACpB,gBAAgB,EACjB,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EAEL,KAAK,mBAAmB,EACzB,MAAM,oBAAoB,CAAC;AAM5B,UAAU,YAAY;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACjD,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,CAAC,YAAY,EAAE,oBAAoB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,CAAC,YAAY,EAAE,oBAAoB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,WAAW,CAAC,EAAE,mBAAmB,CAAC;CACnC;AAED,MAAM,MAAM,eAAe,GAAG,oBAAoB,CAAC;AAEnD,UAAU,YAAY;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;sDAEsD;AACtD,KAAK,SAAS,CAAC,CAAC,SAAS,MAAM,IAAI;KAChC,CAAC,IAAI,MAAM,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,SAAS,KAAK,GACnD,KAAK,GACL,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;CAClC,CAAC;AAEF,wBAAgB,OAAO,CAAC,CAAC,SAAS,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAQ9D;AA2BD,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,YAAY,GAAG,SAAS,CA0B1E;AAED,wBAAgB,0BAA0B,CACxC,QAAQ,CAAC,EAAE,YAAY,GACtB,MAAM,GAAG,SAAS,CAUpB;AAED,wBAAgB,mBAAmB,CAAC,KAAK,CAAC,EAAE,YAAY,GAAG,MAAM,CAWhE;AAuDD,wBAAgB,sBAAsB,CACpC,KAAK,CAAC,EAAE,YAAY,EACpB,WAAW,CAAC,EAAE,mBAAmB,GAChC,eAAe,CAEjB;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,oBAAoB,EAC7B,WAAW,CAAC,EAAE,mBAAmB,GAChC,gBAAgB,CAOlB;AAED,wBAAgB,2BAA2B,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,EAC5E,OAAO,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,GAC7D,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAmBvD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,YAAY,CAEpE;AAWD,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAe7E"}

package/dist/tasks/owner.js

@@ -1,6 +1,6 @@
 import { hash, randomUUID } from 'node:crypto';
 import { z } from 'zod';
-import { getRequestId, runWithRequestContext } from '../lib/core.js';
+import { getRequestId, resolveMcpSessionOwnerKey, runWithRequestContext, } from '../lib/core.js';
 import { isObject } from '../lib/utils.js';
 import { sanitizeToolCallMeta, } from './call-contract.js';
 export function compact(obj) {
@@ -54,13 +54,21 @@ export function parseHandlerExtra(extra) {
     }
     return parsed;
 }
+export function buildAuthenticatedOwnerKey(authInfo) {
+    const authClientId = typeof authInfo?.clientId === 'string' ? authInfo.clientId : '';
+    const authToken = typeof authInfo?.token === 'string' ? authInfo.token : '';
+    if (authClientId || authToken) {
+        return `auth:${hash('sha256', `${authClientId}:${authToken}`, 'hex')}`;
+    }
+    return undefined;
+}
 export function resolveTaskOwnerKey(extra) {
-    if (extra?.sessionId)
-        return `session:${extra.sessionId}`;
-    if (extra?.authInfo?.clientId)
-        return `client:${extra.authInfo.clientId}`;
-    if (extra?.authInfo?.token)
-        return `token:${hash('sha256', extra.authInfo.token, 'hex')}`;
+    const authenticatedOwnerKey = buildAuthenticatedOwnerKey(extra?.authInfo);
+    if (authenticatedOwnerKey)
+        return authenticatedOwnerKey;
+    if (extra?.sessionId) {
+        return (resolveMcpSessionOwnerKey(extra.sessionId) ?? `session:${extra.sessionId}`);
+    }
     return 'default';
 }
 function resolveRequestIdFromExtra(extra) {

package/dist/tasks/registry.d.ts

@@ -1,18 +1,20 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { ServerResult } from '@modelcontextprotocol/sdk/types.js';
 import type { ToolHandlerExtra } from '../lib/mcp-interop.js';
-export type TaskCapableToolSupport = 'optional' | 'forbidden';
+export type TaskCapableToolSupport = 'required' | 'optional' | 'forbidden';
 export interface TaskCapableToolDescriptor<TArgs = unknown> {
     name: string;
     parseArguments: (args: unknown) => TArgs;
     execute: (args: TArgs, extra?: ToolHandlerExtra) => Promise<ServerResult>;
     getCompletionStatusMessage?: (result: ServerResult) => string | undefined;
     taskSupport?: TaskCapableToolSupport;
+    immediateResponse?: string;
 }
-export declare function registerTaskCapableTool<TArgs>(descriptor: TaskCapableToolDescriptor<TArgs>): void;
-export declare function unregisterTaskCapableTool(name: string): void;
-export declare function getTaskCapableTool(name: string): TaskCapableToolDescriptor | undefined;
-export declare function getTaskCapableToolSupport(name: string): TaskCapableToolSupport | undefined;
-export declare function hasTaskCapableTool(name: string): boolean;
-export declare function hasRegisteredTaskCapableTools(): boolean;
-export declare function setTaskCapableToolSupport(name: string, support: TaskCapableToolSupport): void;
+export declare function registerTaskCapableTool<TArgs>(server: McpServer, descriptor: TaskCapableToolDescriptor<TArgs>): void;
+export declare function unregisterTaskCapableTool(server: McpServer, name: string): void;
+export declare function getTaskCapableTool(server: McpServer, name: string): TaskCapableToolDescriptor | undefined;
+export declare function getTaskCapableToolSupport(server: McpServer, name: string): TaskCapableToolSupport | undefined;
+export declare function hasTaskCapableTool(server: McpServer, name: string): boolean;
+export declare function hasRegisteredTaskCapableTools(server: McpServer): boolean;
+export declare function setTaskCapableToolSupport(server: McpServer, name: string, support: TaskCapableToolSupport): void;
 //# sourceMappingURL=registry.d.ts.map