@ai-sdk/mcp 1.0.56 → 1.0.57

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @ai-sdk/mcp
 
+## 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

@@ -534,6 +534,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;
     /**

package/dist/index.d.ts

@@ -534,6 +534,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;
     /**

package/dist/index.js

@@ -1914,6 +1914,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 +1999,7 @@ var DefaultMCPClient = class {
     clientName = name3 != null ? name3 : "ai-sdk-mcp-client",
     version = CLIENT_VERSION,
     onUncaughtError,
+    maxRetries,
     capabilities
   }) {
     this.requestMessageId = 0;
@@ -1958,6 +2008,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 +2197,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;