@ai-sdk/mcp 2.0.2 → 2.0.3

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @ai-sdk/mcp
 
+## 2.0.3
+
+### Patch Changes
+
+- 68a739a: feat(mcp): allow MCP client to use server completions
+- Updated dependencies [6a436e3]
+  - @ai-sdk/provider-utils@5.0.1
+
 ## 2.0.2
 
 ### Patch Changes

package/dist/index.d.ts

@@ -374,6 +374,7 @@ declare const InitializeResultSchema: z.ZodObject<{
     capabilities: z.ZodObject<{
         experimental: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
         logging: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
+        completions: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
         prompts: z.ZodOptional<z.ZodObject<{
             listChanged: z.ZodOptional<z.ZodBoolean>;
         }, z.core.$loose>>;
@@ -499,6 +500,33 @@ declare const ReadResourceResultSchema: z.ZodObject<{
     }, z.core.$loose>]>>;
 }, z.core.$loose>;
 type ReadResourceResult = z.infer<typeof ReadResourceResultSchema>;
+declare const CompleteRequestParamsSchema: z.ZodObject<{
+    _meta: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
+    ref: z.ZodUnion<readonly [z.ZodObject<{
+        type: z.ZodLiteral<"ref/prompt">;
+        name: z.ZodString;
+    }, z.core.$loose>, z.ZodObject<{
+        type: z.ZodLiteral<"ref/resource">;
+        uri: z.ZodString;
+    }, z.core.$loose>]>;
+    argument: z.ZodObject<{
+        name: z.ZodString;
+        value: z.ZodString;
+    }, z.core.$loose>;
+    context: z.ZodOptional<z.ZodObject<{
+        arguments: z.ZodRecord<z.ZodString, z.ZodString>;
+    }, z.core.$loose>>;
+}, z.core.$loose>;
+type CompleteRequestParams = z.infer<typeof CompleteRequestParamsSchema>;
+declare const CompleteResultSchema: z.ZodObject<{
+    _meta: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
+    completion: z.ZodObject<{
+        values: z.ZodArray<z.ZodString>;
+        total: z.ZodOptional<z.ZodNumber>;
+        hasMore: z.ZodOptional<z.ZodBoolean>;
+    }, z.core.$loose>;
+}, z.core.$loose>;
+type CompleteResult = z.infer<typeof CompleteResultSchema>;
 declare const ListPromptsResultSchema: z.ZodObject<{
     _meta: z.ZodOptional<z.ZodObject<{}, z.core.$loose>>;
     nextCursor: z.ZodOptional<z.ZodString>;
@@ -660,6 +688,9 @@ interface MCPClient {
         arguments?: Record<string, unknown>;
         options?: RequestOptions;
     }): Promise<GetPromptResult>;
+    complete(args: CompleteRequestParams & {
+        options?: RequestOptions;
+    }): Promise<CompleteResult>;
     onElicitationRequest(schema: typeof ElicitationRequestSchema, handler: (request: ElicitationRequest) => Promise<ElicitResult> | ElicitResult): void;
     close: () => Promise<void>;
 }
@@ -721,4 +752,4 @@ declare function readMCPAppResource({ client, uri, options, }: {
     options?: RequestOptions;
 }): Promise<MCPAppResource>;
 
-export { type CallToolResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type InitializeResult, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPAppResource, type MCPAppResourceCSP, type MCPAppResourceMeta, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, MCP_APP_MIME_TYPE, type McpProviderMetadata, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient, mcpAppClientCapabilities, readMCPAppResource, splitMCPAppTools };
+export { type CallToolResult, type CompleteRequestParams, type CompleteResult, type Configuration, type ElicitResult, ElicitResultSchema, type ElicitationRequest, ElicitationRequestSchema, type InitializeResult, type JSONRPCError, type JSONRPCMessage, type JSONRPCNotification, type JSONRPCRequest, type JSONRPCResponse, type ListToolsResult, type MCPAppResource, type MCPAppResourceCSP, type MCPAppResourceMeta, type MCPClient, type ClientCapabilities as MCPClientCapabilities, type MCPClientConfig, type MCPTransport, MCP_APP_MIME_TYPE, type McpProviderMetadata, type OAuthAuthorizationServerInformation, type OAuthClientInformation, type OAuthClientMetadata, type OAuthClientProvider, type OAuthTokens, UnauthorizedError, auth, createMCPClient, type MCPClient as experimental_MCPClient, type ClientCapabilities as experimental_MCPClientCapabilities, type MCPClientConfig as experimental_MCPClientConfig, createMCPClient as experimental_createMCPClient, mcpAppClientCapabilities, readMCPAppResource, splitMCPAppTools };

package/dist/index.js

@@ -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()),
@@ -2164,6 +2193,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) {
@@ -2344,6 +2380,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,
@@ -2503,6 +2549,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({