@ai-sdk/mcp 1.0.56 → 1.0.58

package/dist/index.mjs

@@ -1,50 +1,3 @@
-// src/tool/mcp-client.ts
-import {
-  asSchema,
-  dynamicTool,
-  jsonSchema,
-  safeParseJSON,
-  safeValidateTypes,
-  tool
-} from "@ai-sdk/provider-utils";
-
-// src/error/mcp-client-error.ts
-import { AISDKError } from "@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 = 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 AISDKError.hasMarker(error, marker);
-  }
-};
-
-// src/tool/mcp-sse-transport.ts
-import {
-  EventSourceParserStream,
-  withUserAgentSuffix,
-  getRuntimeEnvironmentUserAgent
-} from "@ai-sdk/provider-utils";
-
 // src/tool/json-rpc-message.ts
 import { parseJSON } from "@ai-sdk/provider-utils";
 import { z as z2 } from "zod/v4";
@@ -329,10 +282,61 @@ var JSONRPCMessageSchema = z2.union([
   JSONRPCResponseSchema,
   JSONRPCErrorSchema
 ]);
+function validateJSONRPCMessage(message) {
+  return JSONRPCMessageSchema.parse(message);
+}
 async function parseJSONRPCMessage(text) {
-  return JSONRPCMessageSchema.parse(await parseJSON({ text }));
+  return validateJSONRPCMessage(await parseJSON({ text }));
 }
 
+// src/tool/mcp-client.ts
+import {
+  asSchema,
+  dynamicTool,
+  jsonSchema,
+  retryWithExponentialBackoff,
+  safeParseJSON,
+  safeValidateTypes,
+  tool
+} from "@ai-sdk/provider-utils";
+
+// src/error/mcp-client-error.ts
+import { AISDKError } from "@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 = 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 AISDKError.hasMarker(error, marker);
+  }
+};
+
+// src/tool/mcp-sse-transport.ts
+import {
+  EventSourceParserStream,
+  withUserAgentSuffix,
+  getRuntimeEnvironmentUserAgent
+} from "@ai-sdk/provider-utils";
+
 // src/version.ts
 var VERSION = typeof __PACKAGE_VERSION__ !== "undefined" ? __PACKAGE_VERSION__ : "0.0.0-test";
 
@@ -1679,8 +1683,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")) {
@@ -1888,6 +1894,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
 }) {
@@ -1924,6 +1979,7 @@ var DefaultMCPClient = class {
     clientName = name3 != null ? name3 : "ai-sdk-mcp-client",
     version = CLIENT_VERSION,
     onUncaughtError,
+    maxRetries,
     capabilities
   }) {
     this.requestMessageId = 0;
@@ -1932,6 +1988,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;
@@ -2120,18 +2177,41 @@ var DefaultMCPClient = class {
       options
     });
   }
+  async callToolWithRetry({
+    options,
+    execute
+  }) {
+    if (this.maxRetries === 0) {
+      return execute();
+    }
+    return 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;
@@ -2489,6 +2569,7 @@ export {
   UnauthorizedError,
   auth,
   createMCPClient,
-  createMCPClient as experimental_createMCPClient
+  createMCPClient as experimental_createMCPClient,
+  validateJSONRPCMessage
 };
 //# sourceMappingURL=index.mjs.map