@taskforcehq/taskforce 0.3.327 → 0.3.328

package/dist/mcp/documentAssetRegistrar.js

@@ -85,6 +85,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
                         properties: descriptor.inputSchema.properties,
                         ...(descriptor.inputSchema.required ? { required: descriptor.inputSchema.required } : {}),
                     },
+                    ...(descriptor.annotations ? { annotations: descriptor.annotations } : {}),
                 };
             },
             handler: bindDomainTool(executeTool, name),
@@ -116,6 +117,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
     }));
     register('link_document', (context) => ({
         description: context.describeTool('Link an existing canonical document to a task, initiative, or workstream without copying it. The operation is idempotent. Use search_documents first when you do not know the exact D-123 reference or asset id.'),
+        annotations: { idempotentHint: true },
         inputSchema: {
             properties: {
                 documentId: stringProperty('Canonical document reference such as D-123, document id, or asset id.'),
@@ -139,6 +141,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
     }));
     register('rehome_document', (context) => ({
         description: context.describeTool('Atomically move an existing canonical document reference between a task, initiative, or workstream without copying or deleting the document. Use this instead of separate link_document and unlink_document calls when correcting planning scope.'),
+        annotations: { idempotentHint: true },
         inputSchema: {
             properties: {
                 documentId: stringProperty('Canonical document reference such as D-123, document id, or asset id.'),
@@ -159,18 +162,18 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
     }));
     register('save_workstream_attachment', (context) => ({
         description: context.describeTool(context.runtimeMode === 'cloud'
-            ? 'Save generated content and attach it to a workstream. Provide inline content. Set overwrite=true to replace an existing canonical attachment with the same filename.'
-            : 'Save a file and attach it to a workstream. Provide inline content or an approved local sourcePath. Set overwrite=true to replace an existing canonical attachment with the same filename.'),
+            ? 'Save inline content as a workstream-owned attachment. Use overwrite=true to replace the same filename.'
+            : 'Save inline content or an approved local file as a workstream-owned attachment. Use overwrite=true to replace the same filename.'),
         inputSchema: {
             properties: {
                 id: stringProperty('Workstream ID or workstream reference such as WS-123.'),
                 filename: stringProperty('Destination filename for the attachment.'),
-                content: stringProperty('Inline file content; use this for AI-authored markdown/text attachments.'),
+                content: stringProperty('Inline text content.'),
                 ...(context.runtimeMode === 'local' ? {
                     sourcePath: stringProperty("Existing local file. Allowed roots: project, registered Git worktrees, Taskforce-managed sources, approved agent workspaces, or the bound profile's ~/.taskforce/qa/<username>/artifacts."),
                 } : {}),
                 caption: stringProperty('Optional attachment caption.'),
-                overwrite: booleanProperty('When true, overwrite an existing canonical workstream attachment with the same filename instead of creating a new one.'),
+                overwrite: booleanProperty('Replace the existing attachment with the same filename.'),
             },
             required: ['id', 'filename'],
         },
@@ -199,18 +202,18 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
     }));
     register('save_initiative_attachment', (context) => ({
         description: context.describeTool(context.runtimeMode === 'cloud'
-            ? 'Save generated content and attach it to an initiative. Provide inline content. Set overwrite=true to replace an existing canonical attachment with the same filename.'
-            : 'Save a file and attach it to an initiative. Provide inline content or an approved local sourcePath. Set overwrite=true to replace an existing canonical attachment with the same filename.'),
+            ? 'Save inline content as an initiative-owned attachment. Use overwrite=true to replace the same filename.'
+            : 'Save inline content or an approved local file as an initiative-owned attachment. Use overwrite=true to replace the same filename.'),
         inputSchema: {
             properties: {
                 id: stringProperty('Initiative ID or initiative reference such as IN-123.'),
                 filename: stringProperty('Destination filename for the attachment.'),
-                content: stringProperty('Inline file content; use this for AI-authored markdown/text attachments.'),
+                content: stringProperty('Inline text content.'),
                 ...(context.runtimeMode === 'local' ? {
                     sourcePath: stringProperty("Existing local file. Allowed roots: project, registered Git worktrees, Taskforce-managed sources, approved agent workspaces, or the bound profile's ~/.taskforce/qa/<username>/artifacts."),
                 } : {}),
                 caption: stringProperty('Optional attachment caption.'),
-                overwrite: booleanProperty('When true, overwrite an existing canonical initiative attachment with the same filename instead of creating a new one.'),
+                overwrite: booleanProperty('Replace the existing attachment with the same filename.'),
             },
             required: ['id', 'filename'],
         },
@@ -263,6 +266,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
                 format: stringProperty('Optional read mode when including content.', ['metadata', 'text', 'content', 'full', 'download']),
                 maxChars: numberProperty('Optional max inline character count for readable documents. Longer content is truncated with size metadata and a download hint.'),
             },
+            required: ['documentId'],
         },
     }));
     register('update_document', (context) => ({
@@ -466,13 +470,14 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
         },
     }));
     register('add_document_review_comment', (context) => ({
-        description: context.describeTool('Add one follow-up comment to an existing review session. Prefer review_document for multi-comment first-pass reviews.'),
+        description: context.describeTool('Add one follow-up comment to an existing review session. Prefer review_document for a complete review. Pass idempotencyKey only for exact retries after an ambiguous timeout.'),
         inputSchema: {
             properties: {
                 sessionId: stringProperty('Document review session id.'),
                 body: stringProperty('Comment text.'),
                 anchor: documentReviewAnchorProperty(),
                 parentCommentId: stringProperty('Optional root comment id. When supplied, adds a reply to that one-level thread; replies cannot target other replies.'),
+                idempotencyKey: { type: 'string', minLength: 1, maxLength: 128, description: 'Optional opaque retry key scoped to the active workspace, actor, and tool.' },
             },
             required: ['sessionId', 'body'],
         },
@@ -540,6 +545,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
                 }),
                 expectedSizeBytes: numberProperty('Optional exact byte count for integrity verification. The save is rejected before attachment when the persisted bytes would differ.'),
                 expectedSha256: stringProperty('Optional lowercase or uppercase SHA-256 hex digest for integrity verification. The save is rejected before attachment when the persisted bytes would differ.'),
+                displayName: stringProperty('Optional attachment display name. Defaults to filename and is independent from caption.'),
                 caption: stringProperty('Optional attachment caption.'),
                 overwrite: booleanProperty('When true, overwrite an existing canonical task attachment with the same filename instead of creating a new one.'),
             },
@@ -548,6 +554,7 @@ export function registerDocumentAssetTools(registerTool, executeTool) {
     }));
     register('set_task_plan', (context) => ({
         description: context.describeTool('Create or replace the canonical Markdown implementation plan for a task. Identical retries are no-ops. Optionally replace the task description with a concise summary after the plan is verified; a description failure preserves the valid plan and returns partial-success recovery guidance.'),
+        annotations: { idempotentHint: true },
         inputSchema: {
             properties: {
                 taskId: stringProperty('Task ID or task reference such as T-123.'),

package/dist/mcp/localServiceProxy.js

@@ -3,6 +3,7 @@ import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
 import { JSONRPCMessageSchema, } from '@modelcontextprotocol/sdk/types.js';
 import { normalizeMcpConnectionId, parseMcpClientId } from './clientRegistry.js';
 import { normalizeMcpToolProfile } from './toolProfiles.js';
+import { localServiceCloudTargetsEqual, parseLocalServiceCloudTarget, } from '../config/localServiceCloudTarget.js';
 import { discoverOrStartLocalService, LOCAL_SERVICE_API_BASE_PATH, LOCAL_SERVICE_PROTOCOL_VERSION, } from '../service/localServiceClientLifecycle.js';
 const REQUEST_TIMEOUT_MS = 120_000;
 const MAX_IN_FLIGHT = 32;
@@ -24,6 +25,7 @@ async function readJson(response, label) {
     }
 }
 function validateReadiness(readiness, descriptor) {
+    const readinessCloudTarget = parseLocalServiceCloudTarget(readiness.cloudTarget);
     if (!readiness.ready)
         throw new Error('Taskforce shared local service is not accepting new work.');
     if (readiness.protocolVersion !== LOCAL_SERVICE_PROTOCOL_VERSION
@@ -31,7 +33,8 @@ function validateReadiness(readiness, descriptor) {
         || readiness.generation !== descriptor.generation
         || readiness.pid !== descriptor.pid
         || readiness.endpoint !== descriptor.endpoint
-        || readiness.dataIdentity !== descriptor.ownershipKey) {
+        || readiness.dataIdentity !== descriptor.ownershipKey
+        || !localServiceCloudTargetsEqual(readinessCloudTarget, descriptor.cloudTarget)) {
         throw new Error('Taskforce shared local service readiness does not match its descriptor.');
     }
 }

package/dist/mcp/runtime.d.ts

@@ -16,6 +16,18 @@ import { createDurableTaskCommentMutationRouter } from '../sync/v3/durableTaskCo
 import { createDurableTaskChecklistMutationRouter } from '../sync/v3/durableTaskChecklistMutationRouter.js';
 import { createDurableTaskAttachmentLinksMutationRouter } from '../sync/v3/durableTaskAttachmentLinksMutationRouter.js';
 import type { LocalDataOwnerPermit } from '../server/localDataOwnerPermit.js';
+export type McpToolCallTiming = {
+    requestId: string | null;
+    toolName: string;
+    isWrite: boolean;
+    outcome: 'success' | 'failure';
+    totalMs: number;
+    preparationMs: number;
+    validationMs: number;
+    handlerMs: number;
+    responseBytes: number;
+    subphases: Record<string, number>;
+};
 export interface TaskforceMcpServerOptions {
     projectRoot?: string;
     tenantId?: string;
@@ -34,6 +46,7 @@ export interface TaskforceMcpServerOptions {
         mutationKind: string;
         payloadFingerprint: string;
     }) => void;
+    onToolCallTiming?: (timing: McpToolCallTiming) => void;
     operationIdentityScope?: {
         generation: string;
         sessionId: string;
@@ -45,10 +58,12 @@ export interface TaskforceMcpServerOptions {
     durableTaxonomyMutationRouter?: ReturnType<typeof createDurableTaxonomyMutationRouter>;
     durableInitiativeMutationRouterFactory?: (actorRef: string, options?: {
         operationId?: string;
+        commentId?: string;
         deferPostCommitNotification?: (notify: () => void) => void;
     }) => ReturnType<typeof createDurableInitiativeMutationRouter>;
     durableWorkstreamMutationRouterFactory?: (actorRef: string, options?: {
         operationId?: string;
+        commentId?: string;
         deferPostCommitNotification?: (notify: () => void) => void;
     }) => ReturnType<typeof createDurableWorkstreamMutationRouter>;
     durableTaskRelationshipMutationRouter?: ReturnType<typeof createDurableTaskRelationshipMutationRouter>;
@@ -62,6 +77,7 @@ export interface TaskforceMcpServerOptions {
         userId?: string | null;
         tokenId?: string | null;
         aiProfileId?: string | null;
+        aiProfileIdentityAsserted?: boolean;
         aiProfileToken?: AiProfileBootstrapConfig['aiProfileToken'];
         aiProfileName?: AiProfileBootstrapConfig['aiProfileName'];
         aiProfileIcon?: AiProfileBootstrapConfig['aiProfileIcon'];