@mastra/mcp 0.10.2-alpha.1 → 0.10.3-alpha.0

package/dist/index.js

@@ -6,13 +6,13 @@ import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
 import { StdioClientTransport, getDefaultEnvironment } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { DEFAULT_REQUEST_TIMEOUT_MSEC } from '@modelcontextprotocol/sdk/shared/protocol.js';
-import { ListToolsRequestSchema, CallToolRequestSchema, ListResourcesRequestSchema, ReadResourceRequestSchema, ListResourceTemplatesRequestSchema, SubscribeRequestSchema, UnsubscribeRequestSchema, ListResourcesResultSchema, ReadResourceResultSchema, ListResourceTemplatesResultSchema, ResourceUpdatedNotificationSchema, ResourceListChangedNotificationSchema, CallToolResultSchema, JSONRPCMessageSchema, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+import { ListToolsRequestSchema, CallToolRequestSchema, ListResourcesRequestSchema, ReadResourceRequestSchema, ListResourceTemplatesRequestSchema, SubscribeRequestSchema, UnsubscribeRequestSchema, ListPromptsRequestSchema, PromptSchema, GetPromptRequestSchema, ListResourcesResultSchema, ReadResourceResultSchema, ListResourceTemplatesResultSchema, ListPromptsResultSchema, GetPromptResultSchema, PromptListChangedNotificationSchema, ResourceUpdatedNotificationSchema, ResourceListChangedNotificationSchema, CallToolResultSchema, JSONRPCMessageSchema, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
 import { asyncExitHook, gracefulExit } from 'exit-hook';
 import { z } from 'zod';
 import { convertJsonSchemaToZod } from 'zod-from-json-schema';
 import equal from 'fast-deep-equal';
 import { v5 } from 'uuid';
-import { randomUUID } from 'node:crypto';
+import { randomUUID } from 'crypto';
 import { createTool, makeCoreTool } from '@mastra/core';
 import { Agent } from '@mastra/core/agent';
 import { MCPServerBase } from '@mastra/core/mcp';
@@ -24,6 +24,58 @@ import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/
 import { streamSSE } from 'hono/streaming';
 
 // src/client/client.ts
+var PromptClientActions = class {
+  client;
+  logger;
+  constructor({ client, logger }) {
+    this.client = client;
+    this.logger = logger;
+  }
+  /**
+   * Get all prompts from the connected MCP server.
+   * @returns A list of prompts with their versions.
+   */
+  async list() {
+    try {
+      const response = await this.client.listPrompts();
+      if (response && response.prompts && Array.isArray(response.prompts)) {
+        return response.prompts.map((prompt) => ({ ...prompt, version: prompt.version || "" }));
+      } else {
+        this.logger.warn(`Prompts response from server ${this.client.name} did not have expected structure.`, {
+          response
+        });
+        return [];
+      }
+    } catch (e) {
+      if (e.code === ErrorCode.MethodNotFound) {
+        return [];
+      }
+      this.logger.error(`Error getting prompts from server ${this.client.name}`, {
+        error: e instanceof Error ? e.message : String(e)
+      });
+      throw new Error(
+        `Failed to fetch prompts from server ${this.client.name}: ${e instanceof Error ? e.stack || e.message : String(e)}`
+      );
+    }
+  }
+  /**
+   * Get a specific prompt.
+   * @param name The name of the prompt to get.
+   * @param args Optional arguments for the prompt.
+   * @param version Optional version of the prompt to get.
+   * @returns The prompt content.
+   */
+  async get({ name, args, version }) {
+    return this.client.getPrompt({ name, args, version });
+  }
+  /**
+   * Set a notification handler for when the list of available prompts changes.
+   * @param handler The callback function to handle the notification.
+   */
+  async onListChanged(handler) {
+    this.client.setPromptListChangedNotificationHandler(handler);
+  }
+};
 var ResourceClientActions = class {
   client;
   logger;
@@ -53,7 +105,6 @@ var ResourceClientActions = class {
       this.logger.error(`Error getting resources from server ${this.client.name}`, {
         error: e instanceof Error ? e.message : String(e)
       });
-      console.log("errorheere", e);
       throw new Error(
         `Failed to fetch resources from server ${this.client.name}: ${e instanceof Error ? e.stack || e.message : String(e)}`
       );
@@ -76,7 +127,6 @@ var ResourceClientActions = class {
         return [];
       }
     } catch (e) {
-      console.log({ errorcooode: e.code });
       if (e.code === ErrorCode.MethodNotFound) {
         return [];
       }
@@ -155,6 +205,7 @@ var InternalMastraMCPClient = class extends MastraBase {
   transport;
   currentOperationContext = null;
   resources;
+  prompts;
   constructor({
     name,
     version = "1.0.0",
@@ -179,6 +230,7 @@ var InternalMastraMCPClient = class extends MastraBase {
     );
     this.setupLogging();
     this.resources = new ResourceClientActions({ client: this, logger: this.logger });
+    this.prompts = new PromptClientActions({ client: this, logger: this.logger });
   }
   /**
    * Log a message at the specified level
@@ -362,6 +414,39 @@ var InternalMastraMCPClient = class extends MastraBase {
       timeout: this.timeout
     });
   }
+  /**
+   * Fetch the list of available prompts from the MCP server.
+   */
+  async listPrompts() {
+    this.log("debug", `Requesting prompts from MCP server`);
+    return await this.client.request({ method: "prompts/list" }, ListPromptsResultSchema, {
+      timeout: this.timeout
+    });
+  }
+  /**
+   * Get a prompt and its dynamic messages from the server.
+   * @param name The prompt name
+   * @param args Arguments for the prompt
+   * @param version (optional) The prompt version to retrieve
+   */
+  async getPrompt({ name, args, version }) {
+    this.log("debug", `Requesting prompt from MCP server: ${name}`);
+    return await this.client.request(
+      { method: "prompts/get", params: { name, arguments: args, version } },
+      GetPromptResultSchema,
+      { timeout: this.timeout }
+    );
+  }
+  /**
+   * Register a handler to be called when the prompt list changes on the server.
+   * Use this to refresh cached prompt lists in the client/UI if needed.
+   */
+  setPromptListChangedNotificationHandler(handler) {
+    this.log("debug", "Setting prompt list changed notification handler");
+    this.client.setNotificationHandler(PromptListChangedNotificationSchema, () => {
+      handler();
+    });
+  }
   setResourceUpdatedNotificationHandler(handler) {
     this.log("debug", "Setting resource updated notification handler");
     this.client.setNotificationHandler(ResourceUpdatedNotificationSchema, (notification) => {
@@ -551,6 +636,31 @@ To fix this you have three different options:
       }
     };
   }
+  get prompts() {
+    this.addToInstanceCache();
+    return {
+      list: async () => {
+        const allPrompts = {};
+        for (const serverName of Object.keys(this.serverConfigs)) {
+          try {
+            const internalClient = await this.getConnectedClientForServer(serverName);
+            allPrompts[serverName] = await internalClient.prompts.list();
+          } catch (error) {
+            this.logger.error(`Failed to list prompts from server ${serverName}`, { error });
+          }
+        }
+        return allPrompts;
+      },
+      get: async ({ serverName, name, args, version }) => {
+        const internalClient = await this.getConnectedClientForServer(serverName);
+        return internalClient.prompts.get({ name, args, version });
+      },
+      onListChanged: async (serverName, handler) => {
+        const internalClient = await this.getConnectedClientForServer(serverName);
+        return internalClient.prompts.onListChanged(handler);
+      }
+    };
+  }
   addToInstanceCache() {
     if (!mcpClientInstances.has(this.id)) {
       mcpClientInstances.set(this.id, this);
@@ -759,6 +869,34 @@ var SSETransport = class {
   }
 };
 
+// src/server/promptActions.ts
+var ServerPromptActions = class {
+  getLogger;
+  getSdkServer;
+  clearDefinedPrompts;
+  constructor(dependencies) {
+    this.getLogger = dependencies.getLogger;
+    this.getSdkServer = dependencies.getSdkServer;
+    this.clearDefinedPrompts = dependencies.clearDefinedPrompts;
+  }
+  /**
+   * Notifies the server that the overall list of available prompts has changed.
+   * This will clear the internal cache of defined prompts and send a list_changed notification to clients.
+   */
+  async notifyListChanged() {
+    this.getLogger().info("Prompt list change externally notified. Clearing definedPrompts and sending notification.");
+    this.clearDefinedPrompts();
+    try {
+      await this.getSdkServer().sendPromptListChanged();
+    } catch (error) {
+      this.getLogger().error("Failed to send prompt list changed notification:", {
+        error: error instanceof Error ? error.message : String(error)
+      });
+      throw error;
+    }
+  }
+};
+
 // src/server/resourceActions.ts
 var ServerResourceActions = class {
   getSubscriptions;
@@ -823,11 +961,16 @@ var MCPServer = class extends MCPServerBase {
   listResourceTemplatesHandlerIsRegistered = false;
   subscribeResourceHandlerIsRegistered = false;
   unsubscribeResourceHandlerIsRegistered = false;
+  listPromptsHandlerIsRegistered = false;
+  getPromptHandlerIsRegistered = false;
   definedResources;
   definedResourceTemplates;
   resourceOptions;
+  definedPrompts;
+  promptOptions;
   subscriptions = /* @__PURE__ */ new Set();
   resources;
+  prompts;
   /**
    * Get the current stdio transport.
    */
@@ -852,6 +995,12 @@ var MCPServer = class extends MCPServerBase {
   getStreamableHTTPTransport() {
     return this.streamableHTTPTransport;
   }
+  /**
+   * Get the current server instance.
+   */
+  getServer() {
+    return this.server;
+  }
   /**
    * Construct a new MCPServer instance.
    * @param opts - Configuration options for the server, including registry metadata.
@@ -859,6 +1008,7 @@ var MCPServer = class extends MCPServerBase {
   constructor(opts) {
     super(opts);
     this.resourceOptions = opts.resources;
+    this.promptOptions = opts.prompts;
     const capabilities = {
       tools: {},
       logging: { enabled: true }
@@ -866,6 +1016,9 @@ var MCPServer = class extends MCPServerBase {
     if (opts.resources) {
       capabilities.resources = { subscribe: true, listChanged: true };
     }
+    if (opts.prompts) {
+      capabilities.prompts = { listChanged: true };
+    }
     this.server = new Server({ name: this.name, version: this.version }, { capabilities });
     this.logger.info(
       `Initialized MCPServer '${this.name}' v${this.version} (ID: ${this.id}) with tools: ${Object.keys(this.convertedTools).join(", ")} and resources. Capabilities: ${JSON.stringify(capabilities)}`
@@ -882,6 +1035,12 @@ var MCPServer = class extends MCPServerBase {
         this.registerListResourceTemplatesHandler();
       }
     }
+    if (opts.prompts) {
+      this.registerListPromptsHandler();
+      this.registerGetPromptHandler({
+        getPromptMessagesCallback: opts.prompts.getPromptMessages
+      });
+    }
     this.resources = new ServerResourceActions({
       getSubscriptions: () => this.subscriptions,
       getLogger: () => this.logger,
@@ -893,6 +1052,13 @@ var MCPServer = class extends MCPServerBase {
         this.definedResourceTemplates = void 0;
       }
     });
+    this.prompts = new ServerPromptActions({
+      getLogger: () => this.logger,
+      getSdkServer: () => this.server,
+      clearDefinedPrompts: () => {
+        this.definedPrompts = void 0;
+      }
+    });
   }
   convertAgentsToTools(agentsConfig, definedConvertedTools) {
     const agentTools = {};
@@ -1135,7 +1301,7 @@ var MCPServer = class extends MCPServerBase {
           content: [
             {
               type: "text",
-              text: JSON.stringify(result)
+              text: typeof result === "string" ? result : JSON.stringify(result)
             }
           ],
           isError: false
@@ -1319,6 +1485,95 @@ var MCPServer = class extends MCPServerBase {
       return {};
     });
   }
+  /**
+   * Register the ListPrompts handler.
+   */
+  registerListPromptsHandler() {
+    if (this.listPromptsHandlerIsRegistered) {
+      return;
+    }
+    this.listPromptsHandlerIsRegistered = true;
+    const capturedPromptOptions = this.promptOptions;
+    if (!capturedPromptOptions?.listPrompts) {
+      this.logger.warn("ListPrompts capability not supported by server configuration.");
+      return;
+    }
+    this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
+      this.logger.debug("Handling ListPrompts request");
+      if (this.definedPrompts) {
+        return {
+          prompts: this.definedPrompts?.map((p) => ({ ...p, version: p.version ?? void 0 }))
+        };
+      } else {
+        try {
+          const prompts = await capturedPromptOptions.listPrompts();
+          for (const prompt of prompts) {
+            PromptSchema.parse(prompt);
+          }
+          this.definedPrompts = prompts;
+          this.logger.debug(`Fetched and cached ${this.definedPrompts.length} prompts.`);
+          return {
+            prompts: this.definedPrompts?.map((p) => ({ ...p, version: p.version ?? void 0 }))
+          };
+        } catch (error) {
+          this.logger.error("Error fetching prompts via listPrompts():", {
+            error: error instanceof Error ? error.message : String(error)
+          });
+          throw error;
+        }
+      }
+    });
+  }
+  /**
+   * Register the GetPrompt handler.
+   */
+  registerGetPromptHandler({
+    getPromptMessagesCallback
+  }) {
+    if (this.getPromptHandlerIsRegistered) return;
+    this.getPromptHandlerIsRegistered = true;
+    this.server.setRequestHandler(
+      GetPromptRequestSchema,
+      async (request) => {
+        const startTime = Date.now();
+        const { name, version, arguments: args } = request.params;
+        if (!this.definedPrompts) {
+          const prompts = await this.promptOptions?.listPrompts?.();
+          if (!prompts) throw new Error("Failed to load prompts");
+          this.definedPrompts = prompts;
+        }
+        let prompt;
+        if (version) {
+          prompt = this.definedPrompts?.find((p) => p.name === name && p.version === version);
+        } else {
+          prompt = this.definedPrompts?.find((p) => p.name === name);
+        }
+        if (!prompt) throw new Error(`Prompt "${name}"${version ? ` (version ${version})` : ""} not found`);
+        if (prompt.arguments) {
+          for (const arg of prompt.arguments) {
+            if (arg.required && (args?.[arg.name] === void 0 || args?.[arg.name] === null)) {
+              throw new Error(`Missing required argument: ${arg.name}`);
+            }
+          }
+        }
+        try {
+          let messages = [];
+          if (getPromptMessagesCallback) {
+            messages = await getPromptMessagesCallback({ name, version, args });
+          }
+          const duration = Date.now() - startTime;
+          this.logger.info(
+            `Prompt '${name}'${version ? ` (version ${version})` : ""} retrieved successfully in ${duration}ms.`
+          );
+          return { prompt, messages };
+        } catch (error) {
+          const duration = Date.now() - startTime;
+          this.logger.error(`Failed to get content for prompt '${name}' in ${duration}ms`, { error });
+          throw error;
+        }
+      }
+    );
+  }
   /**
    * Start the MCP server using stdio transport (for Windsurf integration).
    */

package/integration-tests/node_modules/.bin/vitest

@@ -10,9 +10,9 @@ case `uname` in
 esac
 
 if [ -z "$NODE_PATH" ]; then
-  export NODE_PATH="/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@1.6.1_@edge-runtime+vm@3.2.0_@types+node@20.17.32_jsdom@26.0.0_bufferutil@4.0.9__b7b40f6481a5fb1015de024b8d25deb4/node_modules/vitest/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@1.6.1_@edge-runtime+vm@3.2.0_@types+node@20.17.32_jsdom@26.0.0_bufferutil@4.0.9__b7b40f6481a5fb1015de024b8d25deb4/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/node_modules"
+  export NODE_PATH="/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@3.2.2_@edge-runtime+vm@3.2.0_@types+debug@4.1.12_@types+node@20.19.0_@vitest+ui@_f2f8c5e73aab614bca86ded968276706/node_modules/vitest/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@3.2.2_@edge-runtime+vm@3.2.0_@types+debug@4.1.12_@types+node@20.19.0_@vitest+ui@_f2f8c5e73aab614bca86ded968276706/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/node_modules"
 else
-  export NODE_PATH="/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@1.6.1_@edge-runtime+vm@3.2.0_@types+node@20.17.32_jsdom@26.0.0_bufferutil@4.0.9__b7b40f6481a5fb1015de024b8d25deb4/node_modules/vitest/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@1.6.1_@edge-runtime+vm@3.2.0_@types+node@20.17.32_jsdom@26.0.0_bufferutil@4.0.9__b7b40f6481a5fb1015de024b8d25deb4/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/node_modules:$NODE_PATH"
+  export NODE_PATH="/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@3.2.2_@edge-runtime+vm@3.2.0_@types+debug@4.1.12_@types+node@20.19.0_@vitest+ui@_f2f8c5e73aab614bca86ded968276706/node_modules/vitest/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/vitest@3.2.2_@edge-runtime+vm@3.2.0_@types+debug@4.1.12_@types+node@20.19.0_@vitest+ui@_f2f8c5e73aab614bca86ded968276706/node_modules:/home/runner/work/mastra/mastra/node_modules/.pnpm/node_modules:$NODE_PATH"
 fi
 if [ -x "$basedir/node" ]; then
   exec "$basedir/node"  "$basedir/../vitest/vitest.mjs" "$@"

package/integration-tests/package.json

@@ -12,14 +12,14 @@
     "@mastra/client-js": "workspace:*",
     "@mastra/mcp": "workspace:*",
     "dotenv": "^16.5.0",
-    "zod": "^3.24.3"
+    "zod": "^3.25.56"
   },
   "devDependencies": {
     "@testing-library/react": "^16.2.0",
-    "@types/node": "^20.17.27",
+    "@types/node": "^20.17.57",
     "mastra": "workspace:*",
     "typescript": "^5.8.2",
-    "vitest": "^1.6.1",
+    "vitest": "^3.2.2",
     "@mastra/core": "workspace:*"
   },
   "peerDependencies": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mcp",
-  "version": "0.10.2-alpha.1",
+  "version": "0.10.3-alpha.0",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -32,7 +32,7 @@
   },
   "peerDependencies": {
     "zod": "^3.0.0",
-    "@mastra/core": "^0.10.1-alpha.0"
+    "@mastra/core": "^0.10.2-alpha.0"
   },
   "devDependencies": {
     "@ai-sdk/anthropic": "^1.1.15",
@@ -40,18 +40,18 @@
     "ai": "4.3.16",
     "@hono/node-server": "^1.13.8",
     "@mendable/firecrawl-js": "^1.24.0",
-    "@microsoft/api-extractor": "^7.52.5",
-    "@types/node": "^20.17.27",
-    "eslint": "^9.23.0",
+    "@microsoft/api-extractor": "^7.52.8",
+    "@types/node": "^20.17.57",
+    "eslint": "^9.28.0",
     "hono-mcp-server-sse-transport": "0.0.6",
-    "tsup": "^8.4.0",
+    "tsup": "^8.5.0",
     "tsx": "^4.19.3",
     "typescript": "^5.8.2",
-    "vitest": "^3.1.2",
-    "zod": "^3.24.3",
+    "vitest": "^3.2.2",
+    "zod": "^3.25.56",
     "zod-to-json-schema": "^3.24.5",
-    "@internal/lint": "0.0.7",
-    "@mastra/core": "0.10.2-alpha.2"
+    "@internal/lint": "0.0.10",
+    "@mastra/core": "0.10.4-alpha.1"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/__fixtures__/weather.ts

@@ -1,10 +1,10 @@
 import type { IncomingMessage, ServerResponse } from 'http';
 import { createServer } from 'http';
 import { createTool } from '@mastra/core';
-import type { Resource, ResourceTemplate } from '@modelcontextprotocol/sdk/types.js';
+import type { Prompt, PromptMessage, Resource, ResourceTemplate } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
 import { MCPServer } from '../server/server';
-import type { MCPServerResources, MCPServerResourceContent } from '../server/server';
+import type { MCPServerResources, MCPServerResourceContent, MCPServerPrompts } from '../server/types';
 
 const getWeather = async (location: string) => {
   // Return mock data for testing
@@ -119,6 +119,36 @@ const weatherResourceContents: Record<string, MCPServerResourceContent> = {
   },
 };
 
+const weatherPrompts: Prompt[] = [
+  {
+    name: 'current',
+    version: 'v1',
+    description: 'Get current weather for a location',
+    mimeType: 'application/json',
+    content: JSON.stringify({
+      location: 'Current weather for San Francisco',
+    }),
+  },
+  {
+    name: 'forecast',
+    version: 'v1',
+    description: 'Get weather forecast for a location',
+    mimeType: 'application/json',
+    content: JSON.stringify({
+      location: 'Forecast for San Francisco',
+    }),
+  },
+  {
+    name: 'historical',
+    version: 'v1',
+    description: 'Get historical weather data for a location',
+    mimeType: 'application/json',
+    content: JSON.stringify({
+      location: 'Historical weather for San Francisco',
+    }),
+  },
+];
+
 const mcpServerResources: MCPServerResources = {
   listResources: async () => weatherResourceDefinitions,
   getResourceContent: async ({ uri }: { uri: string }) => {
@@ -130,6 +160,25 @@ const mcpServerResources: MCPServerResources = {
   resourceTemplates: async () => weatherResourceTemplatesDefinitions,
 };
 
+const mcpServerPrompts: MCPServerPrompts = {
+  listPrompts: async () => weatherPrompts,
+  getPromptMessages: async ({ name }: { name: string }): Promise<PromptMessage[]> => {
+    const prompt = weatherPrompts.find(p => p.name === name);
+    if (!prompt) {
+      throw new Error(`Mock prompt not found for ${name}`);
+    }
+    return [
+      {
+        role: 'user',
+        content: {
+          type: 'text',
+          text: prompt.content as string,
+        },
+      },
+    ];
+  },
+};
+
 const mcpServer = new MCPServer({
   name: serverId,
   version: '1.0.0',
@@ -137,6 +186,7 @@ const mcpServer = new MCPServer({
     getWeather: weatherToolDefinition,
   },
   resources: mcpServerResources,
+  prompts: mcpServerPrompts,
 });
 
 const httpServer = createServer(async (req: IncomingMessage, res: ServerResponse) => {
@@ -193,12 +243,22 @@ const notificationInterval = setInterval(async () => {
     }
   }
 }, NOTIFICATION_INTERVAL_MS);
+
+const promptNotificationInterval = setInterval(async () => {
+  const listChangePrefix = `[${serverId}] IntervalListChange`;
+  try {
+    await mcpServer.prompts.notifyListChanged();
+  } catch (e: any) {
+    console.error(`${listChangePrefix} - Error sending promptListChanged via MCPServer: ${e.message}`);
+  }
+}, NOTIFICATION_INTERVAL_MS);
 // --- End Interval-based Notifications ---
 
 // Handle graceful shutdown
 process.on('SIGINT', async () => {
   console.log('Shutting down weather server...');
   clearInterval(notificationInterval); // Clear the interval
+  clearInterval(promptNotificationInterval); // Clear the interval
   await mcpServer.close();
   httpServer.close(() => {
     console.log('Weather server shut down complete');

package/src/client/client.test.ts

@@ -19,6 +19,7 @@ async function setupTestServer(withSessionManagement: boolean) {
         logging: {},
         tools: {},
         resources: {},
+        prompts: {},
       },
     },
   );
@@ -47,6 +48,27 @@ async function setupTestServer(withSessionManagement: boolean) {
     };
   });
 
+  mcpServer.prompt(
+    'greet',
+    'A simple greeting prompt',
+    () => {
+      return {
+        prompt: {
+          name: 'greet',
+          version: 'v1',
+          description: 'A simple greeting prompt',
+          mimeType: 'application/json',
+        },
+        messages: [
+          {
+            role: 'assistant',
+            content: { type: 'text', text: `Hello, World!` }
+          }
+        ]
+      };
+    },
+  );
+
   const serverTransport = new StreamableHTTPServerTransport({
     sessionIdGenerator: withSessionManagement ? () => randomUUID() : undefined,
   });
@@ -121,6 +143,30 @@ describe('MastraMCPClient with Streamable HTTP', () => {
       expect(readResult.contents.length).toBe(1);
       expect(readResult.contents[0].text).toBe('Hello, world!');
     });
+
+    it('should list prompts', async () => {
+      const {prompts} = await client.listPrompts();
+      expect(prompts).toBeInstanceOf(Array);
+      expect(prompts).toHaveLength(1);
+      expect(prompts[0]).toHaveProperty('name');
+      expect(prompts[0]).toHaveProperty('description');
+      expect(prompts[0].description).toBe('A simple greeting prompt');
+    });
+
+    it('should get a specific prompt', async () => {
+      const result = await client.getPrompt({name: 'greet'});
+      const {prompt, messages} = result;
+      expect(prompt).toBeDefined();
+      expect(prompt).toMatchObject({
+        name: 'greet',
+        version: 'v1',
+        description: expect.any(String),
+        mimeType: 'application/json',
+      });
+      expect(messages).toBeDefined();
+      const messageItem = messages[0];
+      expect(messageItem.content.text).toBe('Hello, World!');
+    });
   });
 
   describe('Stateful Mode', () => {