@ai-sdk/mcp 0.0.5 → 0.0.7

package/dist/index.mjs

@@ -172,6 +172,32 @@ var ReadResourceResultSchema = ResultSchema.extend({
     z.union([TextResourceContentsSchema, BlobResourceContentsSchema])
   )
 });
+var PromptArgumentSchema = z.object({
+  name: z.string(),
+  description: z.optional(z.string()),
+  required: z.optional(z.boolean())
+}).loose();
+var PromptSchema = z.object({
+  name: z.string(),
+  title: z.optional(z.string()),
+  description: z.optional(z.string()),
+  arguments: z.optional(z.array(PromptArgumentSchema))
+}).loose();
+var ListPromptsResultSchema = PaginatedResultSchema.extend({
+  prompts: z.array(PromptSchema)
+});
+var PromptMessageSchema = z.object({
+  role: z.union([z.literal("user"), z.literal("assistant")]),
+  content: z.union([
+    TextContentSchema,
+    ImageContentSchema,
+    EmbeddedResourceSchema
+  ])
+}).loose();
+var GetPromptResultSchema = ResultSchema.extend({
+  description: z.optional(z.string()),
+  messages: z.array(PromptMessageSchema)
+});
 
 // src/tool/json-rpc-message.ts
 var JSONRPC_VERSION = "2.0";
@@ -1586,6 +1612,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}`
@@ -1718,6 +1752,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,
@@ -1783,6 +1846,19 @@ var DefaultMCPClient = class {
   } = {}) {
     return this.listResourceTemplatesInternal({ options });
   }
+  listPrompts({
+    params,
+    options
+  } = {}) {
+    return this.listPromptsInternal({ params, options });
+  }
+  getPrompt({
+    name: name3,
+    arguments: args,
+    options
+  }) {
+    return this.getPromptInternal({ name: name3, args, options });
+  }
   onClose() {
     if (this.isClosed) return;
     this.isClosed = true;