@ai-sdk/mcp 1.0.56 → 1.0.58

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @ai-sdk/mcp
 
+## 1.0.58
+
+### Patch Changes
+
+- f12f0d3: feat(mcp): expose a json-rpc message validator
+
+## 1.0.57
+
+### Patch Changes
+
+- ea1e95b: feat(mcp): add maxRetries option for failed mcp tool calls
+- Updated dependencies [ea1e95b]
+  - @ai-sdk/provider-utils@4.0.35
+
 ## 1.0.56
 
 ### Patch Changes

package/dist/index.d.mts

@@ -66,6 +66,7 @@ declare const JSONRPCMessageSchema: z.ZodUnion<readonly [z.ZodObject<{
     }, z.core.$strip>;
 }, z.core.$strict>]>;
 type JSONRPCMessage = z.infer<typeof JSONRPCMessageSchema>;
+declare function validateJSONRPCMessage(message: unknown): JSONRPCMessage;
 
 /**
  * OAuth 2.1 token response
@@ -534,6 +535,15 @@ interface MCPClientConfig {
     transport: MCPTransportConfig | MCPTransport;
     /** Optional callback for uncaught errors */
     onUncaughtError?: (error: unknown) => void;
+    /**
+     * Maximum number of retries for transient MCP tool call failures.
+     *
+     * Set to 0 to disable retries. Retries only apply to tools/call requests.
+     * JSON-RPC application errors, such as invalid params, are not retried.
+     *
+     * @default 0
+     */
+    maxRetries?: number;
     /** Optional client name, defaults to 'ai-sdk-mcp-client' */
     clientName?: string;
     /**
@@ -610,4 +620,4 @@ interface MCPClient {
     close: () => Promise<void>;
 }
 
-export { type CallToolResult, type CompleteRequestParams, type CompleteResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient };
+export { type CallToolResult, type CompleteRequestParams, type CompleteResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient, validateJSONRPCMessage };

package/dist/index.d.ts

@@ -66,6 +66,7 @@ declare const JSONRPCMessageSchema: z.ZodUnion<readonly [z.ZodObject<{
     }, z.core.$strip>;
 }, z.core.$strict>]>;
 type JSONRPCMessage = z.infer<typeof JSONRPCMessageSchema>;
+declare function validateJSONRPCMessage(message: unknown): JSONRPCMessage;
 
 /**
  * OAuth 2.1 token response
@@ -534,6 +535,15 @@ interface MCPClientConfig {
     transport: MCPTransportConfig | MCPTransport;
     /** Optional callback for uncaught errors */
     onUncaughtError?: (error: unknown) => void;
+    /**
+     * Maximum number of retries for transient MCP tool call failures.
+     *
+     * Set to 0 to disable retries. Retries only apply to tools/call requests.
+     * JSON-RPC application errors, such as invalid params, are not retried.
+     *
+     * @default 0
+     */
+    maxRetries?: number;
     /** Optional client name, defaults to 'ai-sdk-mcp-client' */
     clientName?: string;
     /**
@@ -610,4 +620,4 @@ interface MCPClient {
     close: () => Promise<void>;
 }
 
-export { type CallToolResult, type CompleteRequestParams, type CompleteResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient };
+export { type CallToolResult, type CompleteRequestParams, type CompleteResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient, validateJSONRPCMessage };

package/dist/index.js

@@ -35,46 +35,11 @@ __export(index_exports, {
   UnauthorizedError: () => UnauthorizedError,
   auth: () => auth,
   createMCPClient: () => createMCPClient,
-  experimental_createMCPClient: () => createMCPClient
+  experimental_createMCPClient: () => createMCPClient,
+  validateJSONRPCMessage: () => validateJSONRPCMessage
 });
 module.exports = __toCommonJS(index_exports);
 
-// src/tool/mcp-client.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
-
-// src/error/mcp-client-error.ts
-var import_provider = require("@ai-sdk/provider");
-var name = "AI_MCPClientError";
-var marker = `vercel.ai.error.${name}`;
-var symbol = Symbol.for(marker);
-var _a, _b;
-var MCPClientError = class extends (_b = import_provider.AISDKError, _a = symbol, _b) {
-  constructor({
-    name: name3 = "MCPClientError",
-    message,
-    cause,
-    data,
-    code,
-    statusCode,
-    url,
-    responseBody
-  }) {
-    super({ name: name3, message, cause });
-    this[_a] = true;
-    this.data = data;
-    this.code = code;
-    this.statusCode = statusCode;
-    this.url = url;
-    this.responseBody = responseBody;
-  }
-  static isInstance(error) {
-    return import_provider.AISDKError.hasMarker(error, marker);
-  }
-};
-
-// src/tool/mcp-sse-transport.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
-
 // src/tool/json-rpc-message.ts
 var import_provider_utils = require("@ai-sdk/provider-utils");
 var import_v42 = require("zod/v4");
@@ -359,10 +324,49 @@ var JSONRPCMessageSchema = import_v42.z.union([
   JSONRPCResponseSchema,
   JSONRPCErrorSchema
 ]);
+function validateJSONRPCMessage(message) {
+  return JSONRPCMessageSchema.parse(message);
+}
 async function parseJSONRPCMessage(text) {
-  return JSONRPCMessageSchema.parse(await (0, import_provider_utils.parseJSON)({ text }));
+  return validateJSONRPCMessage(await (0, import_provider_utils.parseJSON)({ text }));
 }
 
+// src/tool/mcp-client.ts
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
+
+// src/error/mcp-client-error.ts
+var import_provider = require("@ai-sdk/provider");
+var name = "AI_MCPClientError";
+var marker = `vercel.ai.error.${name}`;
+var symbol = Symbol.for(marker);
+var _a, _b;
+var MCPClientError = class extends (_b = import_provider.AISDKError, _a = symbol, _b) {
+  constructor({
+    name: name3 = "MCPClientError",
+    message,
+    cause,
+    data,
+    code,
+    statusCode,
+    url,
+    responseBody
+  }) {
+    super({ name: name3, message, cause });
+    this[_a] = true;
+    this.data = data;
+    this.code = code;
+    this.statusCode = statusCode;
+    this.url = url;
+    this.responseBody = responseBody;
+  }
+  static isInstance(error) {
+    return import_provider.AISDKError.hasMarker(error, marker);
+  }
+};
+
+// src/tool/mcp-sse-transport.ts
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+
 // src/version.ts
 var VERSION = typeof __PACKAGE_VERSION__ !== "undefined" ? __PACKAGE_VERSION__ : "0.0.0-test";
 
@@ -1705,8 +1709,10 @@ var HttpMCPTransport = class {
         const contentType = response.headers.get("content-type") || "";
         if (contentType.includes("application/json")) {
           const data = await response.json();
-          const messages = Array.isArray(data) ? data.map((m) => JSONRPCMessageSchema.parse(m)) : [JSONRPCMessageSchema.parse(data)];
-          for (const m of messages) (_d = this.onmessage) == null ? void 0 : _d.call(this, m);
+          const messages = Array.isArray(data) ? data.map((message2) => validateJSONRPCMessage(message2)) : [validateJSONRPCMessage(data)];
+          for (const jsonRpcMessage of messages) {
+            (_d = this.onmessage) == null ? void 0 : _d.call(this, jsonRpcMessage);
+          }
           return;
         }
         if (contentType.includes("text/event-stream")) {
@@ -1914,6 +1920,55 @@ function isCustomMcpTransport(transport) {
 
 // src/tool/mcp-client.ts
 var CLIENT_VERSION = "1.0.0";
+var DEFAULT_MAX_TOOL_CALL_RETRIES = 0;
+var DEFAULT_RETRY_ERROR_CODES = [
+  "ConnectionRefused",
+  "ConnectionClosed",
+  "FailedToOpenSocket",
+  "ECONNRESET",
+  "ECONNREFUSED",
+  "ETIMEDOUT",
+  "EPIPE"
+];
+function getErrorStatusCode(error) {
+  if (error != null && typeof error === "object" && "statusCode" in error && typeof error.statusCode === "number") {
+    return error.statusCode;
+  }
+  return void 0;
+}
+function getStringErrorCode(error) {
+  if (error != null && typeof error === "object" && "code" in error && typeof error.code === "string") {
+    return error.code;
+  }
+  return void 0;
+}
+function isRetryableMCPToolCallError(error) {
+  const statusCode = getErrorStatusCode(error);
+  if (statusCode != null) {
+    return statusCode === 408 || statusCode === 409 || statusCode === 429 || statusCode >= 500;
+  }
+  if (MCPClientError.isInstance(error) && error.code != null) {
+    return false;
+  }
+  const errorCode = getStringErrorCode(error);
+  return errorCode != null && DEFAULT_RETRY_ERROR_CODES.includes(errorCode);
+}
+function prepareMaxRetries(maxRetries) {
+  if (maxRetries == null) {
+    return DEFAULT_MAX_TOOL_CALL_RETRIES;
+  }
+  if (!Number.isInteger(maxRetries)) {
+    throw new MCPClientError({
+      message: "maxRetries must be an integer"
+    });
+  }
+  if (maxRetries < 0) {
+    throw new MCPClientError({
+      message: "maxRetries must be >= 0"
+    });
+  }
+  return maxRetries;
+}
 function mcpToModelOutput({
   output
 }) {
@@ -1950,6 +2005,7 @@ var DefaultMCPClient = class {
     clientName = name3 != null ? name3 : "ai-sdk-mcp-client",
     version = CLIENT_VERSION,
     onUncaughtError,
+    maxRetries,
     capabilities
   }) {
     this.requestMessageId = 0;
@@ -1958,6 +2014,7 @@ var DefaultMCPClient = class {
     this._serverInfo = { name: "", version: "" };
     this.isClosed = true;
     this.onUncaughtError = onUncaughtError;
+    this.maxRetries = prepareMaxRetries(maxRetries);
     this.clientCapabilities = capabilities != null ? capabilities : {};
     if (isCustomMcpTransport(transportConfig)) {
       this.transport = transportConfig;
@@ -2146,18 +2203,41 @@ var DefaultMCPClient = class {
       options
     });
   }
+  async callToolWithRetry({
+    options,
+    execute
+  }) {
+    if (this.maxRetries === 0) {
+      return execute();
+    }
+    return (0, import_provider_utils5.retryWithExponentialBackoff)({
+      maxRetries: this.maxRetries,
+      abortSignal: options == null ? void 0 : options.abortSignal,
+      shouldRetry: isRetryableMCPToolCallError,
+      createRetryError: ({ message, errors }) => new MCPClientError({
+        message,
+        cause: errors[errors.length - 1]
+      })
+    })(execute);
+  }
   async callTool({
     name: name3,
     args,
     options
   }) {
     try {
-      return this.request({
-        request: { method: "tools/call", params: { name: name3, arguments: args } },
-        resultSchema: CallToolResultSchema,
-        options: {
-          signal: options == null ? void 0 : options.abortSignal
-        }
+      return this.callToolWithRetry({
+        options,
+        execute: () => this.request({
+          request: {
+            method: "tools/call",
+            params: { name: name3, arguments: args }
+          },
+          resultSchema: CallToolResultSchema,
+          options: {
+            signal: options == null ? void 0 : options.abortSignal
+          }
+        })
       });
     } catch (error) {
       throw error;
@@ -2516,6 +2596,7 @@ var DefaultMCPClient = class {
   UnauthorizedError,
   auth,
   createMCPClient,
-  experimental_createMCPClient
+  experimental_createMCPClient,
+  validateJSONRPCMessage
 });
 //# sourceMappingURL=index.js.map