@ai-sdk/mcp 1.0.0-beta.9 → 1.0.0

package/dist/index.js

@@ -30,8 +30,11 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
+  ElicitResultSchema: () => ElicitResultSchema,
+  ElicitationRequestSchema: () => ElicitationRequestSchema,
   UnauthorizedError: () => UnauthorizedError,
   auth: () => auth,
+  createMCPClient: () => createMCPClient,
   experimental_createMCPClient: () => createMCPClient
 });
 module.exports = __toCommonJS(src_exports);
@@ -77,6 +80,7 @@ var SUPPORTED_PROTOCOL_VERSIONS = [
   "2025-03-26",
   "2024-11-05"
 ];
+var ToolMetaSchema = import_v4.z.optional(import_v4.z.record(import_v4.z.string(), import_v4.z.unknown()));
 var ClientOrServerImplementationSchema = import_v4.z.looseObject({
   name: import_v4.z.string(),
   version: import_v4.z.string()
@@ -89,6 +93,9 @@ var RequestSchema = import_v4.z.object({
   method: import_v4.z.string(),
   params: import_v4.z.optional(BaseParamsSchema)
 });
+var ElicitationCapabilitySchema = import_v4.z.object({
+  applyDefaults: import_v4.z.optional(import_v4.z.boolean())
+}).loose();
 var ServerCapabilitiesSchema = import_v4.z.looseObject({
   experimental: import_v4.z.optional(import_v4.z.object({}).loose()),
   logging: import_v4.z.optional(import_v4.z.object({}).loose()),
@@ -107,8 +114,12 @@ var ServerCapabilitiesSchema = import_v4.z.looseObject({
     import_v4.z.looseObject({
       listChanged: import_v4.z.optional(import_v4.z.boolean())
     })
-  )
+  ),
+  elicitation: import_v4.z.optional(ElicitationCapabilitySchema)
 });
+var ClientCapabilitiesSchema = import_v4.z.object({
+  elicitation: import_v4.z.optional(ElicitationCapabilitySchema)
+}).loose();
 var InitializeResultSchema = ResultSchema.extend({
   protocolVersion: import_v4.z.string(),
   capabilities: ServerCapabilitiesSchema,
@@ -129,7 +140,8 @@ var ToolSchema = import_v4.z.object({
     import_v4.z.object({
       title: import_v4.z.optional(import_v4.z.string())
     }).loose()
-  )
+  ),
+  _meta: ToolMetaSchema
 }).loose();
 var ListToolsResultSchema = PaginatedResultSchema.extend({
   tools: import_v4.z.array(ToolSchema)
@@ -207,6 +219,48 @@ var ReadResourceResultSchema = ResultSchema.extend({
     import_v4.z.union([TextResourceContentsSchema, BlobResourceContentsSchema])
   )
 });
+var PromptArgumentSchema = import_v4.z.object({
+  name: import_v4.z.string(),
+  description: import_v4.z.optional(import_v4.z.string()),
+  required: import_v4.z.optional(import_v4.z.boolean())
+}).loose();
+var PromptSchema = import_v4.z.object({
+  name: import_v4.z.string(),
+  title: import_v4.z.optional(import_v4.z.string()),
+  description: import_v4.z.optional(import_v4.z.string()),
+  arguments: import_v4.z.optional(import_v4.z.array(PromptArgumentSchema))
+}).loose();
+var ListPromptsResultSchema = PaginatedResultSchema.extend({
+  prompts: import_v4.z.array(PromptSchema)
+});
+var PromptMessageSchema = import_v4.z.object({
+  role: import_v4.z.union([import_v4.z.literal("user"), import_v4.z.literal("assistant")]),
+  content: import_v4.z.union([
+    TextContentSchema,
+    ImageContentSchema,
+    EmbeddedResourceSchema
+  ])
+}).loose();
+var GetPromptResultSchema = ResultSchema.extend({
+  description: import_v4.z.optional(import_v4.z.string()),
+  messages: import_v4.z.array(PromptMessageSchema)
+});
+var ElicitationRequestParamsSchema = BaseParamsSchema.extend({
+  message: import_v4.z.string(),
+  requestedSchema: import_v4.z.unknown()
+});
+var ElicitationRequestSchema = RequestSchema.extend({
+  method: import_v4.z.literal("elicitation/create"),
+  params: ElicitationRequestParamsSchema
+});
+var ElicitResultSchema = ResultSchema.extend({
+  action: import_v4.z.union([
+    import_v4.z.literal("accept"),
+    import_v4.z.literal("decline"),
+    import_v4.z.literal("cancel")
+  ]),
+  content: import_v4.z.optional(import_v4.z.record(import_v4.z.string(), import_v4.z.unknown()))
+});
 
 // src/tool/json-rpc-message.ts
 var JSONRPC_VERSION = "2.0";
@@ -782,7 +836,8 @@ async function refreshAuthorization(authorizationServerUrl, {
     tokenUrl = new URL("/token", authorizationServerUrl);
   }
   const headers = new Headers({
-    "Content-Type": "application/x-www-form-urlencoded"
+    "Content-Type": "application/x-www-form-urlencoded",
+    Accept: "application/json"
   });
   const params = new URLSearchParams({
     grant_type: grantType,
@@ -1525,13 +1580,16 @@ var DefaultMCPClient = class {
   constructor({
     transport: transportConfig,
     name: name3 = "ai-sdk-mcp-client",
-    onUncaughtError
+    version = CLIENT_VERSION,
+    onUncaughtError,
+    capabilities
   }) {
     this.requestMessageId = 0;
     this.responseHandlers = /* @__PURE__ */ new Map();
     this.serverCapabilities = {};
     this.isClosed = true;
     this.onUncaughtError = onUncaughtError;
+    this.clientCapabilities = capabilities != null ? capabilities : {};
     if (isCustomMcpTransport(transportConfig)) {
       this.transport = transportConfig;
     } else {
@@ -1541,18 +1599,22 @@ var DefaultMCPClient = class {
     this.transport.onerror = (error) => this.onError(error);
     this.transport.onmessage = (message) => {
       if ("method" in message) {
-        this.onError(
-          new MCPClientError({
-            message: "Unsupported message type"
-          })
-        );
+        if ("id" in message) {
+          this.onRequestMessage(message);
+        } else {
+          this.onError(
+            new MCPClientError({
+              message: "Unsupported message type"
+            })
+          );
+        }
         return;
       }
       this.onResponse(message);
     };
     this.clientInfo = {
       name: name3,
-      version: CLIENT_VERSION
+      version
     };
   }
   async init() {
@@ -1564,7 +1626,7 @@ var DefaultMCPClient = class {
           method: "initialize",
           params: {
             protocolVersion: LATEST_PROTOCOL_VERSION,
-            capabilities: {},
+            capabilities: this.clientCapabilities,
             clientInfo: this.clientInfo
           }
         },
@@ -1617,6 +1679,14 @@ var DefaultMCPClient = class {
           });
         }
         break;
+      case "prompts/list":
+      case "prompts/get":
+        if (!this.serverCapabilities.prompts) {
+          throw new MCPClientError({
+            message: `Server does not support prompts`
+          });
+        }
+        break;
       default:
         throw new MCPClientError({
           message: `Unsupported method: ${method}`
@@ -1749,6 +1819,35 @@ var DefaultMCPClient = class {
       throw error;
     }
   }
+  async listPromptsInternal({
+    params,
+    options
+  } = {}) {
+    try {
+      return this.request({
+        request: { method: "prompts/list", params },
+        resultSchema: ListPromptsResultSchema,
+        options
+      });
+    } catch (error) {
+      throw error;
+    }
+  }
+  async getPromptInternal({
+    name: name3,
+    args,
+    options
+  }) {
+    try {
+      return this.request({
+        request: { method: "prompts/get", params: { name: name3, arguments: args } },
+        resultSchema: GetPromptResultSchema,
+        options
+      });
+    } catch (error) {
+      throw error;
+    }
+  }
   async notification(notification) {
     const jsonrpcNotification = {
       ...notification,
@@ -1771,7 +1870,8 @@ var DefaultMCPClient = class {
         name: name3,
         description,
         inputSchema,
-        annotations
+        annotations,
+        _meta
       } of listToolsResult.tools) {
         const title = annotations == null ? void 0 : annotations.title;
         if (schemas !== "automatic" && !(name3 in schemas)) {
@@ -1798,7 +1898,7 @@ var DefaultMCPClient = class {
           inputSchema: schemas[name3].inputSchema,
           execute
         });
-        tools[name3] = toolWithExecute;
+        tools[name3] = { ...toolWithExecute, _meta };
       }
       return tools;
     } catch (error) {
@@ -1822,6 +1922,90 @@ var DefaultMCPClient = class {
   } = {}) {
     return this.listResourceTemplatesInternal({ options });
   }
+  experimental_listPrompts({
+    params,
+    options
+  } = {}) {
+    return this.listPromptsInternal({ params, options });
+  }
+  experimental_getPrompt({
+    name: name3,
+    arguments: args,
+    options
+  }) {
+    return this.getPromptInternal({ name: name3, args, options });
+  }
+  onElicitationRequest(schema, handler) {
+    if (schema !== ElicitationRequestSchema) {
+      throw new MCPClientError({
+        message: "Unsupported request schema. Only ElicitationRequestSchema is supported."
+      });
+    }
+    this.elicitationRequestHandler = handler;
+  }
+  async onRequestMessage(request) {
+    try {
+      if (request.method !== "elicitation/create") {
+        await this.transport.send({
+          jsonrpc: "2.0",
+          id: request.id,
+          error: {
+            code: -32601,
+            message: `Unsupported request method: ${request.method}`
+          }
+        });
+        return;
+      }
+      if (!this.elicitationRequestHandler) {
+        await this.transport.send({
+          jsonrpc: "2.0",
+          id: request.id,
+          error: {
+            code: -32601,
+            message: "No elicitation handler registered on client"
+          }
+        });
+        return;
+      }
+      const parsedRequest = ElicitationRequestSchema.safeParse({
+        method: request.method,
+        params: request.params
+      });
+      if (!parsedRequest.success) {
+        await this.transport.send({
+          jsonrpc: "2.0",
+          id: request.id,
+          error: {
+            code: -32602,
+            message: `Invalid elicitation request: ${parsedRequest.error.message}`,
+            data: parsedRequest.error.issues
+          }
+        });
+        return;
+      }
+      try {
+        const result = await this.elicitationRequestHandler(parsedRequest.data);
+        const validatedResult = ElicitResultSchema.parse(result);
+        await this.transport.send({
+          jsonrpc: "2.0",
+          id: request.id,
+          result: validatedResult
+        });
+      } catch (error) {
+        await this.transport.send({
+          jsonrpc: "2.0",
+          id: request.id,
+          error: {
+            code: -32603,
+            message: error instanceof Error ? error.message : "Failed to handle elicitation request"
+          }
+        });
+        this.onError(error);
+      }
+    } catch (error) {
+      this.onError(error);
+    }
+  }
   onClose() {
     if (this.isClosed) return;
     this.isClosed = true;
@@ -1861,8 +2045,11 @@ var DefaultMCPClient = class {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ElicitResultSchema,
+  ElicitationRequestSchema,
   UnauthorizedError,
   auth,
+  createMCPClient,
   experimental_createMCPClient
 });
 //# sourceMappingURL=index.js.map