@taiga-ai/mcp-server 0.14.0-rc.426 → 0.16.0-dev.433

package/src/tools/wait.ts

@@ -1,4 +1,4 @@
-import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { McpServer } from '@modelcontextprotocol/server';
 import { BUILD_RUN_TERMINAL_STATUSES, resolveAutoBuild } from '@taiga/types';
 import { z } from 'zod';
 import { TaigaApiError, type TaigaClient } from '../client.js';
@@ -49,7 +49,7 @@ type Verdict = { reached: true } | { reached: false; stoppedBecause: string } |
 
 interface Progress {
   progressToken: string | number | undefined;
-  sendNotification: (notification: {
+  notify: (notification: {
     method: 'notifications/progress';
     params: { progressToken: string | number; progress: number; total: number; message: string };
   }) => Promise<void>;
@@ -128,7 +128,7 @@ async function waitFor(
     if (attempt > 0 && progress.progressToken !== undefined) {
       // Best effort: a progress line the client cannot take does not end the wait.
       await progress
-        .sendNotification({
+        .notify({
           method: 'notifications/progress',
           params: {
             progressToken: progress.progressToken,
@@ -355,14 +355,14 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
           'call again to keep waiting. Conditions are states, not events: an initiative whose latest run is already ' +
           'settled answers run_settled at once; item is null when not even the first read answered in time. ' +
           'Planning takes minutes and a build up to an hour.',
-        inputSchema: {
+        inputSchema: z.object({
           itemId: z.string().uuid().describe('Backlog item ID -- get from taiga_list_backlog_items'),
           until: z.enum(ITEM_CONDITIONS).describe('What to wait for'),
           timeoutSeconds,
-        },
+        }),
         annotations: READ_ONLY,
       },
-      (params, extra) =>
+      (params, ctx) =>
         runTool(async () => {
           const gate: GateDefault = { product: undefined };
           const read = async (signal: AbortSignal) => {
@@ -392,9 +392,9 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
             (params.timeoutSeconds ?? DEFAULT_TIMEOUT_SECONDS) * 1000,
             options,
             {
-              progressToken: extra._meta?.progressToken,
-              sendNotification: extra.sendNotification,
-              signal: extra.signal,
+              progressToken: ctx.mcpReq._meta?.progressToken,
+              notify: ctx.mcpReq.notify,
+              signal: ctx.mcpReq.signal,
             }
           );
           const { now, ...outcome } = result;
@@ -416,14 +416,14 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
           'changed waits on, since a person resuming the run is a change), or at the timeout with reached: false: ' +
           'call again to keep waiting. Conditions are states: a run already settled answers settled at once; run ' +
           'is null when not even the first read answered in time.',
-        inputSchema: {
+        inputSchema: z.object({
           buildId: z.string().uuid().describe('Build run ID -- get from taiga_list_builds'),
           until: z.enum(BUILD_CONDITIONS).describe('What to wait for'),
           timeoutSeconds,
-        },
+        }),
         annotations: READ_ONLY,
       },
-      (params, extra) =>
+      (params, ctx) =>
         runTool(async () => {
           const result = await waitFor(
             async (signal) => (await client.getBuildRun(params.buildId, signal)).data as Row,
@@ -431,9 +431,9 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
             (params.timeoutSeconds ?? DEFAULT_TIMEOUT_SECONDS) * 1000,
             options,
             {
-              progressToken: extra._meta?.progressToken,
-              sendNotification: extra.sendNotification,
-              signal: extra.signal,
+              progressToken: ctx.mcpReq._meta?.progressToken,
+              notify: ctx.mcpReq.notify,
+              signal: ctx.mcpReq.signal,
             }
           );
           const { now, ...outcome } = result;
@@ -456,13 +456,13 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
           'a refresh. Returns the job as taiga_get_job reads it, with what came of an initiative request, as soon ' +
           'as it settles (reached: true), or at the timeout with reached: false: call again to keep waiting. Most ' +
           'jobs take a few minutes.',
-        inputSchema: {
+        inputSchema: z.object({
           jobId: z.string().uuid().describe('Agent job ID -- a request tool returns it'),
           timeoutSeconds,
-        },
+        }),
         annotations: READ_ONLY,
       },
-      (params, extra) =>
+      (params, ctx) =>
         runTool(async () => {
           const result = await waitFor(
             async (signal) => (await client.getAgentJob(params.jobId, signal)).data as Row,
@@ -470,9 +470,9 @@ export function registerWaitTools(server: McpServer, client: TaigaClient, option
             (params.timeoutSeconds ?? DEFAULT_TIMEOUT_SECONDS) * 1000,
             options,
             {
-              progressToken: extra._meta?.progressToken,
-              sendNotification: extra.sendNotification,
-              signal: extra.signal,
+              progressToken: ctx.mcpReq._meta?.progressToken,
+              notify: ctx.mcpReq.notify,
+              signal: ctx.mcpReq.signal,
             }
           );
           const { now, ...outcome } = result;