@ai-sdk/mcp 0.0.8 → 0.0.9

package/dist/index.mjs

@@ -59,6 +59,9 @@ var RequestSchema = z.object({
   method: z.string(),
   params: z.optional(BaseParamsSchema)
 });
+var ElicitationCapabilitySchema = z.object({
+  applyDefaults: z.optional(z.boolean())
+}).loose();
 var ServerCapabilitiesSchema = z.looseObject({
   experimental: z.optional(z.object({}).loose()),
   logging: z.optional(z.object({}).loose()),
@@ -77,8 +80,12 @@ var ServerCapabilitiesSchema = z.looseObject({
     z.looseObject({
       listChanged: z.optional(z.boolean())
     })
-  )
+  ),
+  elicitation: z.optional(ElicitationCapabilitySchema)
 });
+var ClientCapabilitiesSchema = z.object({
+  elicitation: z.optional(ElicitationCapabilitySchema)
+}).loose();
 var InitializeResultSchema = ResultSchema.extend({
   protocolVersion: z.string(),
   capabilities: ServerCapabilitiesSchema,
@@ -198,6 +205,22 @@ var GetPromptResultSchema = ResultSchema.extend({
   description: z.optional(z.string()),
   messages: z.array(PromptMessageSchema)
 });
+var ElicitationRequestParamsSchema = BaseParamsSchema.extend({
+  message: z.string(),
+  requestedSchema: z.unknown()
+});
+var ElicitationRequestSchema = RequestSchema.extend({
+  method: z.literal("elicitation/create"),
+  params: ElicitationRequestParamsSchema
+});
+var ElicitResultSchema = ResultSchema.extend({
+  action: z.union([
+    z.literal("accept"),
+    z.literal("decline"),
+    z.literal("cancel")
+  ]),
+  content: z.optional(z.record(z.string(), z.unknown()))
+});
 
 // src/tool/json-rpc-message.ts
 var JSONRPC_VERSION = "2.0";
@@ -1520,13 +1543,15 @@ var DefaultMCPClient = class {
   constructor({
     transport: transportConfig,
     name: name3 = "ai-sdk-mcp-client",
-    onUncaughtError
+    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 {
@@ -1536,11 +1561,15 @@ 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);
@@ -1559,7 +1588,7 @@ var DefaultMCPClient = class {
           method: "initialize",
           params: {
             protocolVersion: LATEST_PROTOCOL_VERSION,
-            capabilities: {},
+            capabilities: this.clientCapabilities,
             clientInfo: this.clientInfo
           }
         },
@@ -1859,6 +1888,77 @@ var DefaultMCPClient = class {
   }) {
     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;
@@ -1897,6 +1997,8 @@ var DefaultMCPClient = class {
   }
 };
 export {
+  ElicitResultSchema,
+  ElicitationRequestSchema,
   UnauthorizedError,
   auth,
   createMCPClient as experimental_createMCPClient