latitudesh-typescript-sdk 0.8.30 → 0.9.0

package/src/lib/config.ts

@@ -74,8 +74,8 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "2023-06-01",
-  sdkVersion: "0.8.30",
-  genVersion: "2.943.0",
+  sdkVersion: "0.9.0",
+  genVersion: "2.950.2",
   userAgent:
-    "speakeasy-sdk/typescript 0.8.30 2.943.0 2023-06-01 latitudesh-typescript-sdk",
+    "speakeasy-sdk/typescript 0.9.0 2.950.2 2023-06-01 latitudesh-typescript-sdk",
 } as const;

package/src/mcp-server/cli/start/impl.ts

@@ -2,8 +2,8 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { SSEServerTransport } from "@modelcontextprotocol/server-legacy/sse";
+import { serveStdio } from "@modelcontextprotocol/server/stdio";
 import express from "express";
 import { SDKOptions } from "../../../lib/config.js";
 import { LocalContext } from "../../cli.js";
@@ -44,8 +44,7 @@ export async function main(this: LocalContext, flags: StartCommandFlags) {
 
 async function startStdio(flags: StartCommandFlags) {
   const logger = createConsoleLogger(flags["log-level"]);
-  const transport = new StdioServerTransport();
-  const server = createMCPServer({
+  const createServer = createMCPServerFactory({
     logger,
     allowedTools: flags.tool,
     bearer: flags.bearer ?? "",
@@ -54,7 +53,7 @@ async function startStdio(flags: StartCommandFlags) {
     serverIdx: flags["server-index"],
     latitudeApiKey: flags["latitude-api-key"],
   });
-  await server.connect(transport);
+  const server = serveStdio(createServer);
 
   const abort = async () => {
     await server.close();
@@ -64,6 +63,12 @@ async function startStdio(flags: StartCommandFlags) {
   process.on("SIGINT", abort);
 }
 
+function createMCPServerFactory(
+  options: Parameters<typeof createMCPServer>[0],
+) {
+  return () => createMCPServer(options);
+}
+
 async function startSSE(flags: StartCommandFlags) {
   const logger = createConsoleLogger(flags["log-level"]);
   const app = express();

package/src/mcp-server/mcp-server.ts

@@ -19,7 +19,7 @@ const routes = buildRouteMap({
 export const app = buildApplication(routes, {
   name: "mcp",
   versionInfo: {
-    currentVersion: "0.8.30",
+    currentVersion: "0.9.0",
   },
 });
 

package/src/mcp-server/prompts.ts

@@ -2,13 +2,11 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import {
   GetPromptResult,
-  ServerNotification,
-  ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
+  isCompletable,
+  McpServer,
+} from "@modelcontextprotocol/server";
 import {
   objectOutputType,
   ZodOptional,
@@ -19,8 +17,8 @@ import {
 import { LatitudeshCore } from "../core.js";
 import { ConsoleLogger } from "./console-logger.js";
 import { MCPScope } from "./scopes.js";
+import { mcpSchema, MCPServerContext, mcpServerContext } from "./shared.js";
 
-// '@modelcontextprotocol/sdk' currently does not export this type
 export type PromptArgsRawShape = {
   [k: string]:
     | ZodType<string, ZodTypeDef, string>
@@ -37,7 +35,7 @@ export type PromptDefinition<
     prompt: (
       client: LatitudeshCore,
       args: objectOutputType<Args, ZodTypeAny>,
-      extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+      extra: MCPServerContext,
     ) => GetPromptResult | Promise<GetPromptResult>;
   }
   : {
@@ -47,7 +45,7 @@ export type PromptDefinition<
     args?: undefined;
     prompt: (
       client: LatitudeshCore,
-      extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+      extra: MCPServerContext,
     ) => GetPromptResult | Promise<GetPromptResult>;
   };
 
@@ -90,30 +88,27 @@ export function createRegisterPrompt(
     }
 
     if (prompt.args) {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      } else {
-        server.prompt(
-          prompt.name,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      }
+      const shape = Object.fromEntries(
+        Object.entries(prompt.args).map(([name, arg]) => {
+          let field = arg;
+          while (field instanceof ZodOptional && !isCompletable(field)) {
+            field = field.unwrap();
+          }
+          return [name, field];
+        }),
+      );
+      server.registerPrompt(prompt.name, {
+        ...(prompt.description === undefined
+          ? {}
+          : { description: prompt.description }),
+        argsSchema: { ...mcpSchema(prompt.args), shape },
+      }, async (args, ctx) => prompt.prompt(sdk, args, mcpServerContext(ctx)));
     } else {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          async (ctx) => prompt.prompt(sdk, ctx),
-        );
-      } else {
-        server.prompt(prompt.name, async (ctx) => prompt.prompt(sdk, ctx));
-      }
+      server.registerPrompt(prompt.name, {
+        ...(prompt.description === undefined
+          ? {}
+          : { description: prompt.description }),
+      }, async (ctx) => prompt.prompt(sdk, mcpServerContext(ctx)));
     }
 
     logger.debug("Registered prompt", { name: prompt.name });

package/src/mcp-server/resources.ts

@@ -2,27 +2,29 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
+import type { ResourceTemplate as LegacyResourceTemplate } from "@modelcontextprotocol/sdk/server/mcp.js";
 import {
   McpServer,
+  ReadResourceResult,
   ResourceMetadata,
   ResourceTemplate,
-} from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import { Variables } from "@modelcontextprotocol/sdk/shared/uriTemplate.js";
-import {
-  ReadResourceResult,
-  ServerNotification,
-  ServerRequest,
-} from "@modelcontextprotocol/sdk/types.js";
+  Variables,
+} from "@modelcontextprotocol/server";
 import { LatitudeshCore } from "../core.js";
 import { ConsoleLogger } from "./console-logger.js";
 import { MCPScope } from "./scopes.js";
-import { isAsyncIterable, isBinaryData, valueToBase64 } from "./shared.js";
+import {
+  isAsyncIterable,
+  isBinaryData,
+  MCPServerContext,
+  mcpServerContext,
+  valueToBase64,
+} from "./shared.js";
 
 export type ReadResourceCallback = (
   client: LatitudeshCore,
   uri: URL,
-  extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+  extra: MCPServerContext,
 ) => ReadResourceResult | Promise<ReadResourceResult>;
 
 export type ResourceDefinition = {
@@ -38,7 +40,7 @@ export type ReadResourceTemplateCallback = (
   client: LatitudeshCore,
   uri: URL,
   vars: Variables,
-  extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+  extra: MCPServerContext,
 ) => ReadResourceResult | Promise<ReadResourceResult>;
 
 export type ResourceTemplateDefinition = {
@@ -46,7 +48,7 @@ export type ResourceTemplateDefinition = {
   description: string;
   metadata?: ResourceMetadata;
   scopes?: MCPScope[];
-  resource: ResourceTemplate;
+  resource: ResourceTemplate | LegacyResourceTemplate;
   read: ReadResourceTemplateCallback;
 };
 
@@ -129,11 +131,11 @@ export function createRegisterResource(
       description: resource.description,
     };
 
-    server.resource(
+    server.registerResource(
       resource.name,
       resource.resource,
       metadata,
-      async (uri, ctx) => resource.read(sdk, uri, ctx),
+      async (uri, ctx) => resource.read(sdk, uri, mcpServerContext(ctx)),
     );
 
     logger.debug("Registered resource", { name: resource.name });
@@ -164,13 +166,32 @@ export function createRegisterResourceTemplate(
       description: resource.description,
     };
 
-    server.resource(
+    server.registerResource(
       resource.name,
-      resource.resource,
+      normalizeResourceTemplate(resource.resource),
       metadata,
-      async (uri, vars, ctx) => resource.read(sdk, uri, vars, ctx),
+      async (uri, vars, ctx) =>
+        resource.read(sdk, uri, vars, mcpServerContext(ctx)),
     );
 
     logger.debug("Registered resource template", { name: resource.name });
   };
 }
+
+function normalizeResourceTemplate(
+  resource: ResourceTemplate | LegacyResourceTemplate,
+): ResourceTemplate {
+  if (resource instanceof ResourceTemplate) return resource;
+  const complete: NonNullable<
+    ConstructorParameters<typeof ResourceTemplate>[1]["complete"]
+  > = {};
+  for (const variable of resource.uriTemplate.variableNames) {
+    const callback = resource.completeCallback(variable);
+    if (callback) complete[variable] = callback;
+  }
+  const list = resource.listCallback;
+  return new ResourceTemplate(resource.uriTemplate.toString(), {
+    list: list ? (ctx) => list(mcpServerContext(ctx)) : undefined,
+    complete,
+  });
+}

package/src/mcp-server/server.ts

@@ -2,7 +2,7 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { McpServer } from "@modelcontextprotocol/server";
 import { LatitudeshCore } from "../core.js";
 import { SDKOptions } from "../lib/config.js";
 import type { ConsoleLogger } from "./console-logger.js";
@@ -41,8 +41,11 @@ import { tool$elasticIpsListElasticIpBgpSessions } from "./tools/elasticIpsListE
 import { tool$elasticIpsListElasticIps } from "./tools/elasticIpsListElasticIps.js";
 import { tool$elasticIpsUpdateElasticIp } from "./tools/elasticIpsUpdateElasticIp.js";
 import { tool$eventsList } from "./tools/eventsList.js";
+import { tool$filesystemStorageAttachFilesystem } from "./tools/filesystemStorageAttachFilesystem.js";
 import { tool$filesystemStorageCreateFilesystem } from "./tools/filesystemStorageCreateFilesystem.js";
 import { tool$filesystemStorageDeleteFilesystem } from "./tools/filesystemStorageDeleteFilesystem.js";
+import { tool$filesystemStorageDetachFilesystem } from "./tools/filesystemStorageDetachFilesystem.js";
+import { tool$filesystemStorageGetFilesystemAttachment } from "./tools/filesystemStorageGetFilesystemAttachment.js";
 import { tool$filesystemStorageListFilesystems } from "./tools/filesystemStorageListFilesystems.js";
 import { tool$filesystemStorageUpdateFilesystem } from "./tools/filesystemStorageUpdateFilesystem.js";
 import { tool$firewallsAssignmentsCreate } from "./tools/firewallsAssignmentsCreate.js";
@@ -70,9 +73,11 @@ import { tool$lksDeleteLKSNodePool } from "./tools/lksDeleteLKSNodePool.js";
 import { tool$lksGetLKSCluster } from "./tools/lksGetLKSCluster.js";
 import { tool$lksGetLKSClusterKubeconfig } from "./tools/lksGetLKSClusterKubeconfig.js";
 import { tool$lksGetLKSNodePool } from "./tools/lksGetLKSNodePool.js";
+import { tool$lksListLKSAllocatableServers } from "./tools/lksListLKSAllocatableServers.js";
 import { tool$lksListLKSAvailableVersions } from "./tools/lksListLKSAvailableVersions.js";
 import { tool$lksListLKSClusters } from "./tools/lksListLKSClusters.js";
 import { tool$lksListLKSNodePools } from "./tools/lksListLKSNodePools.js";
+import { tool$lksListLKSNodes } from "./tools/lksListLKSNodes.js";
 import { tool$lksListLKSSites } from "./tools/lksListLKSSites.js";
 import { tool$lksUpdateLKSCluster } from "./tools/lksUpdateLKSCluster.js";
 import { tool$lksUpdateLKSNodePool } from "./tools/lksUpdateLKSNodePool.js";
@@ -126,6 +131,7 @@ import { tool$publicNetworksCreatePublicNetwork } from "./tools/publicNetworksCr
 import { tool$publicNetworksDestroyPublicNetwork } from "./tools/publicNetworksDestroyPublicNetwork.js";
 import { tool$publicNetworksGetPublicNetwork } from "./tools/publicNetworksGetPublicNetwork.js";
 import { tool$publicNetworksGetPublicNetworks } from "./tools/publicNetworksGetPublicNetworks.js";
+import { tool$publicNetworksUpdatePublicNetworkIp } from "./tools/publicNetworksUpdatePublicNetworkIp.js";
 import { tool$regionsFetch } from "./tools/regionsFetch.js";
 import { tool$regionsGet } from "./tools/regionsGet.js";
 import { tool$rolesGet } from "./tools/rolesGet.js";
@@ -221,7 +227,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "Latitudesh",
-    version: "0.8.30",
+    version: "0.9.0",
   });
 
   const client = new LatitudeshCore({
@@ -306,8 +312,10 @@ export function createMCPServer(deps: {
   tool(tool$lksGetLKSNodePool);
   tool(tool$lksDeleteLKSNodePool);
   tool(tool$lksUpdateLKSNodePool);
+  tool(tool$lksListLKSNodes);
   tool(tool$lksListLKSAvailableVersions);
   tool(tool$lksListLKSSites);
+  tool(tool$lksListLKSAllocatableServers);
   tool(tool$plansList);
   tool(tool$plansGet);
   tool(tool$plansGetBandwidth);
@@ -319,6 +327,7 @@ export function createMCPServer(deps: {
   tool(tool$publicNetworksCreatePublicNetwork);
   tool(tool$publicNetworksGetPublicNetwork);
   tool(tool$publicNetworksDestroyPublicNetwork);
+  tool(tool$publicNetworksUpdatePublicNetworkIp);
   tool(tool$projectsList);
   tool(tool$projectsCreate);
   tool(tool$projectsUpdate);
@@ -384,6 +393,9 @@ export function createMCPServer(deps: {
   tool(tool$filesystemStorageListFilesystems);
   tool(tool$filesystemStorageDeleteFilesystem);
   tool(tool$filesystemStorageUpdateFilesystem);
+  tool(tool$filesystemStorageGetFilesystemAttachment);
+  tool(tool$filesystemStorageAttachFilesystem);
+  tool(tool$filesystemStorageDetachFilesystem);
   tool(tool$blockStorageListVolumes);
   tool(tool$blockStorageCreateVolume);
   tool(tool$blockStorageRetrieveVolume);

package/src/mcp-server/shared.ts

@@ -2,8 +2,175 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
+import type { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
+import type {
+  ServerNotification,
+  ServerRequest,
+} from "@modelcontextprotocol/sdk/types.js";
+import { ErrorCode, McpError } from "@modelcontextprotocol/sdk/types.js";
+import { StandardSchemaWithJSON } from "@modelcontextprotocol/server";
+import type { ServerContext } from "@modelcontextprotocol/server";
+import { zodToJsonSchema } from "zod-to-json-schema";
 import * as z from "zod/v3";
 import { bytesToBase64 } from "../lib/base64.js";
+
+export type MCPServerContext =
+  & ServerContext
+  & RequestHandlerExtra<ServerRequest, ServerNotification>;
+
+export function mcpServerContext(ctx: ServerContext): MCPServerContext {
+  return {
+    ...ctx,
+    signal: ctx.mcpReq.signal,
+    requestId: ctx.mcpReq.id,
+    ...(ctx.mcpReq._meta ? { _meta: ctx.mcpReq._meta } : {}),
+    ...(ctx.http?.authInfo ? { authInfo: ctx.http.authInfo } : {}),
+    ...(ctx.http?.req
+      ? {
+        requestInfo: {
+          headers: Object.fromEntries(ctx.http.req.headers),
+          url: new URL(ctx.http.req.url),
+        },
+      }
+      : {}),
+    ...(ctx.http?.closeSSE ? { closeSSEStream: ctx.http.closeSSE } : {}),
+    ...(ctx.http?.closeStandaloneSSE
+      ? { closeStandaloneSSEStream: ctx.http.closeStandaloneSSE }
+      : {}),
+    sendNotification: async (notification) => {
+      if (!ctx.mcpReq.signal.aborted) await ctx.mcpReq.notify(notification);
+    },
+    sendRequest: async (request, resultSchema, options) => {
+      if (ctx.mcpReq.signal.aborted) {
+        throw new McpError(ErrorCode.ConnectionClosed, "Request was cancelled");
+      }
+      return (ctx.mcpReq.send as MCPServerContext["sendRequest"])(
+        request,
+        resultSchema,
+        options,
+      );
+    },
+  };
+}
+
+export function mcpSchema<A extends z.ZodRawShape>(
+  shape: A,
+): StandardSchemaWithJSON<z.input<z.ZodObject<A>>, z.output<z.ZodObject<A>>> {
+  const schema = z.object(shape);
+  const jsonSchema = (
+    direction: "input" | "output",
+    target: string,
+  ): Record<string, unknown> => {
+    if (target !== "draft-07" && target !== "draft-2020-12") {
+      throw new Error(`Unsupported JSON Schema target: ${target}`);
+    }
+    const result = JSON.parse(JSON.stringify(
+      zodToJsonSchema(schema, {
+        target: target === "draft-07" ? "jsonSchema7" : "jsonSchema2019-09",
+        pipeStrategy: direction,
+        effectStrategy: direction === "output" ? "any" : "input",
+        strictUnions: direction === "input",
+      }),
+      (_key, value) => typeof value === "bigint" ? value.toString() : value,
+    ));
+    return target === "draft-2020-12" ? jsonSchema2020(result) : result;
+  };
+
+  return {
+    "~standard": {
+      version: 1,
+      vendor: "zod",
+      jsonSchema: {
+        input: ({ target }) => jsonSchema("input", target),
+        output: ({ target }) => jsonSchema("output", target),
+      },
+      validate: async (value) => {
+        const result = await schema.safeParseAsync(value);
+        return result.success
+          ? { value: result.data }
+          : { issues: result.error.issues };
+      },
+    },
+  };
+}
+
+function jsonSchema2020(
+  schema: Record<string, unknown>,
+): Record<string, unknown> {
+  const paths = new Map<string, string>();
+  const references: Record<string, unknown>[] = [];
+  const visit = (value: unknown, from: string, to: string): void => {
+    if (!value || typeof value !== "object" || Array.isArray(value)) return;
+    const node = value as Record<string, unknown>;
+    paths.set(from, to);
+    if (typeof node["$ref"] === "string") references.push(node);
+    const entries = Object.entries(node);
+    const tuple = Array.isArray(node["items"]);
+    if (tuple) {
+      const items = node["items"] as unknown[];
+      if (items.length) node["prefixItems"] = items;
+      delete node["items"];
+      if (node["additionalItems"] !== undefined) {
+        node["items"] = node["additionalItems"];
+      }
+      delete node["additionalItems"];
+    }
+    for (const [key, child] of entries) {
+      const nextKey = tuple && key === "items"
+        ? "prefixItems"
+        : tuple && key === "additionalItems"
+        ? "items"
+        : key;
+      if (
+        ["properties", "patternProperties", "definitions", "$defs"].includes(
+          key,
+        )
+      ) {
+        for (
+          const [name, item] of Object.entries(child as Record<string, unknown>)
+        ) {
+          const escapedName = name.replace(/~/g, "~0").replace(/\//g, "~1");
+          visit(
+            item,
+            `${from}/${key}/${name}`,
+            `${to}/${nextKey}/${escapedName}`,
+          );
+        }
+      } else if (
+        ["allOf", "anyOf", "oneOf", "prefixItems"].includes(key)
+        || (tuple && key === "items")
+      ) {
+        (child as unknown[]).forEach((item, index) =>
+          visit(item, `${from}/${key}/${index}`, `${to}/${nextKey}/${index}`)
+        );
+      } else if (
+        [
+          "items",
+          "additionalItems",
+          "additionalProperties",
+          "propertyNames",
+          "contains",
+          "not",
+          "if",
+          "then",
+          "else",
+          "unevaluatedProperties",
+          "unevaluatedItems",
+        ].includes(key)
+      ) {
+        visit(child, `${from}/${key}`, `${to}/${nextKey}`);
+      }
+    }
+  };
+  visit(schema, "#", "#");
+  for (const reference of references) {
+    const ref = reference["$ref"] as string;
+    reference["$ref"] = paths.get(ref) ?? ref;
+  }
+  schema["$schema"] = "https://json-schema.org/draft/2020-12/schema";
+  return schema;
+}
+
 type BinaryData =
   | Uint8Array
   | ArrayBuffer

package/src/mcp-server/tools/apiKeysCreate.ts

@@ -21,7 +21,7 @@ Create a new API Key that is tied to the current user account. The created API k
     const [result, apiCall] = await apiKeysCreate(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/apiKeysDelete.ts

@@ -21,7 +21,7 @@ Delete an existing API Key. Once deleted, the API Key can no longer be used to a
     const [result, apiCall] = await apiKeysDelete(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/apiKeysList.ts

@@ -14,7 +14,7 @@ Returns a list of all API keys.
   tool: async (client, ctx) => {
     const [result, apiCall] = await apiKeysList(
       client,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/apiKeysUpdate.ts

@@ -22,7 +22,7 @@ Use PATCH to update settings without rotating the token.
     const [result, apiCall] = await apiKeysUpdate(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/apiKeysUpdateApiKey.ts

@@ -22,7 +22,7 @@ Use PUT to rotate the token.
     const [result, apiCall] = await apiKeysUpdateApiKey(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/baselinesPreviewCreateBaseline.ts

@@ -27,7 +27,7 @@ selected platforms — so a baseline that saves here can be dispatched verbatim.
       const [result, apiCall] = await baselinesPreviewCreateBaseline(
         client,
         args.request,
-        { fetchOptions: { signal: ctx.signal } },
+        { fetchOptions: { signal: ctx.mcpReq.signal } },
       ).$inspect();
 
       if (!result.ok) {

package/src/mcp-server/tools/baselinesPreviewDestroyBaseline.ts

@@ -24,7 +24,7 @@ Delete a baseline.
       const [result, apiCall] = await baselinesPreviewDestroyBaseline(
         client,
         args.request,
-        { fetchOptions: { signal: ctx.signal } },
+        { fetchOptions: { signal: ctx.mcpReq.signal } },
       ).$inspect();
 
       if (!result.ok) {

package/src/mcp-server/tools/baselinesPreviewGetBaseline.ts

@@ -23,7 +23,7 @@ Retrieve a single baseline.
     const [result, apiCall] = await baselinesPreviewGetBaseline(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/baselinesPreviewGetBaselines.ts

@@ -18,7 +18,7 @@ servers to be delivered with — plan, SSH keys, user data, disk layout and BIOS
   tool: async (client, ctx) => {
     const [result, apiCall] = await baselinesPreviewGetBaselines(
       client,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/billingListUsage.ts

@@ -21,7 +21,7 @@ Returns the billing usage of a project
     const [result, apiCall] = await billingListUsage(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/blockStorageCreateVolume.ts

@@ -20,7 +20,7 @@ Allows you to add persistent storage to a project. These volumes can be used to
     const [result, apiCall] = await blockStorageCreateVolume(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/blockStorageDeleteVolume.ts

@@ -20,7 +20,7 @@ Allows you to remove a volume from a project.`,
     const [result, apiCall] = await blockStorageDeleteVolume(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/blockStorageListVolumes.ts

@@ -20,7 +20,7 @@ Lists all the volumes from a team.`,
     const [result, apiCall] = await blockStorageListVolumes(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {

package/src/mcp-server/tools/blockStorageMapVolume.ts

@@ -20,7 +20,7 @@ Maps a high performance volume to a server over NVMe-TCP.`,
     const [result, apiCall] = await blockStorageMapVolume(
       client,
       args.request,
-      { fetchOptions: { signal: ctx.signal } },
+      { fetchOptions: { signal: ctx.mcpReq.signal } },
     ).$inspect();
 
     if (!result.ok) {