@ai-sdk/mcp 1.0.54 → 1.0.55

package/dist/index.mjs

@@ -78,6 +78,7 @@ var ElicitationCapabilitySchema = z.object({
 var ServerCapabilitiesSchema = z.looseObject({
   experimental: z.optional(z.object({}).loose()),
   logging: z.optional(z.object({}).loose()),
+  completions: z.optional(z.object({}).loose()),
   prompts: z.optional(
     z.looseObject({
       listChanged: z.optional(z.boolean())
@@ -222,6 +223,34 @@ var ReadResourceResultSchema = ResultSchema.extend({
     z.union([TextResourceContentsSchema, BlobResourceContentsSchema])
   )
 });
+var PromptReferenceSchema = z.object({
+  type: z.literal("ref/prompt"),
+  name: z.string()
+}).loose();
+var ResourceReferenceSchema = z.object({
+  type: z.literal("ref/resource"),
+  uri: z.string()
+}).loose();
+var CompletionArgumentSchema = z.object({
+  name: z.string(),
+  value: z.string()
+}).loose();
+var CompleteRequestParamsSchema = BaseParamsSchema.extend({
+  ref: z.union([PromptReferenceSchema, ResourceReferenceSchema]),
+  argument: CompletionArgumentSchema,
+  context: z.optional(
+    z.object({
+      arguments: z.record(z.string(), z.string())
+    }).loose()
+  )
+});
+var CompleteResultSchema = ResultSchema.extend({
+  completion: z.object({
+    values: z.array(z.string()).max(100),
+    total: z.optional(z.number().int()),
+    hasMore: z.optional(z.boolean())
+  }).loose()
+});
 var PromptArgumentSchema = z.object({
   name: z.string(),
   description: z.optional(z.string()),
@@ -1989,6 +2018,13 @@ var DefaultMCPClient = class {
     switch (method) {
       case "initialize":
         break;
+      case "completion/complete":
+        if (!this.serverCapabilities.completions) {
+          throw new MCPClientError({
+            message: `Server does not support completions`
+          });
+        }
+        break;
       case "tools/list":
       case "tools/call":
         if (!this.serverCapabilities.tools) {
@@ -2171,6 +2207,16 @@ var DefaultMCPClient = class {
       throw error;
     }
   }
+  async completeInternal({
+    options,
+    ...params
+  }) {
+    return this.request({
+      request: { method: "completion/complete", params },
+      resultSchema: CompleteResultSchema,
+      options
+    });
+  }
   async notification(notification) {
     const jsonrpcNotification = {
       ...notification,
@@ -2318,6 +2364,9 @@ var DefaultMCPClient = class {
   }) {
     return this.getPromptInternal({ name: name3, args, options });
   }
+  complete(args) {
+    return this.completeInternal(args);
+  }
   onElicitationRequest(schema, handler) {
     if (schema !== ElicitationRequestSchema) {
       throw new MCPClientError({