@mastra/mcp-docs-server 0.0.7 → 0.0.8

package/dist/stdio.js

@@ -1,77 +1,87 @@
 #!/usr/bin/env node
-import { fromPackageRoot, prepare } from './chunk-J7WZZETH.js';
+import { fromPackageRoot, prepare } from './chunk-QWYMT5LP.js';
 import fs2 from 'node:fs/promises';
-import { FastMCP } from 'tylerbarnes-fastmcp-fix';
-import { JSDOM } from 'jsdom';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { ListToolsRequestSchema, CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { JSDOM } from 'jsdom';
 import path2 from 'node:path';
 
 async function fetchBlogPosts() {
-  try {
-    const response = await fetch("https://mastra.ai/blog");
-    if (!response.ok) {
-      throw new Error("Failed to fetch blog posts");
-    }
-    const html = await response.text();
-    const dom = new JSDOM(html);
-    const document = dom.window.document;
-    const blogLinks = Array.from(document.querySelectorAll('a[href^="/blog/"]')).filter((link) => {
-      const href = link.getAttribute("href");
-      return href !== "/blog" && !href?.includes("authors");
-    }).map((link) => {
-      const h2 = link.querySelector("h2");
-      const title = h2?.textContent?.trim();
-      const href = link.getAttribute("href");
-      if (title && href) {
-        return `[${title}](${href})`;
-      }
-      return null;
-    }).filter(Boolean);
-    return "Mastra.ai Blog Posts:\n\n" + blogLinks.join("\n");
-  } catch (error) {
-    throw new Error("Failed to fetch blog posts " + JSON.stringify(error));
+  const response = await fetch("https://mastra.ai/blog");
+  if (!response.ok) {
+    throw new Error("Failed to fetch blog posts");
   }
+  const html = await response.text();
+  const dom = new JSDOM(html);
+  const document = dom.window.document;
+  const blogLinks = Array.from(document.querySelectorAll('a[href^="/blog/"]')).filter((link) => {
+    const href = link.getAttribute("href");
+    return href !== "/blog" && !href?.includes("authors");
+  }).map((link) => {
+    const h2 = link.querySelector("h2");
+    const title = h2?.textContent?.trim();
+    const href = link.getAttribute("href");
+    if (title && href) {
+      return `[${title}](${href})`;
+    }
+    return null;
+  }).filter(Boolean);
+  return "Mastra.ai Blog Posts:\n\n" + blogLinks.join("\n");
 }
 async function fetchBlogPost(url) {
-  try {
-    const response = await fetch(url);
-    if (!response.ok) {
-      throw new Error("Failed to fetch blog post");
-    }
-    const html = await response.text();
-    const dom = new JSDOM(html);
-    const document = dom.window.document;
-    const scripts = document.querySelectorAll("script");
-    scripts.forEach((script) => script.remove());
-    const content = document.body.textContent?.trim() || "";
-    if (!content) {
-      throw new Error("No content found in blog post");
-    }
-    return content;
-  } catch (error) {
-    throw new Error(`Failed to fetch blog post: ${error instanceof Error ? error.message : "Unknown error"}`);
+  const response = await fetch(url);
+  if (!response.ok) {
+    throw new Error("Failed to fetch blog post");
   }
+  const html = await response.text();
+  const dom = new JSDOM(html);
+  const document = dom.window.document;
+  const scripts = document.querySelectorAll("script");
+  scripts.forEach((script) => script.remove());
+  const content = document.body.textContent?.trim() || "";
+  if (!content) {
+    throw new Error("No content found in blog post");
+  }
+  return content;
 }
+var blogInputSchema = z.object({
+  url: z.string().describe(
+    "URL of a specific blog post to fetch. If the string /blog is passed as the url it returns a list of all blog posts."
+  )
+});
 var blogTool = {
   name: "mastraBlog",
   description: "Get Mastra.ai blog content. Without a URL, returns a list of all blog posts. With a URL, returns the specific blog post content in markdown format. The blog contains changelog posts as well as announcements and posts about Mastra features and AI news",
-  parameters: z.object({
-    url: z.string().describe(
-      "URL of a specific blog post to fetch. If the string /blog is passed as the url it returns a list of all blog posts."
-    )
-  }),
   execute: async (args) => {
     try {
+      let content;
       if (args.url !== `/blog`) {
-        return await fetchBlogPost(`https://mastra.ai${args.url}`);
+        content = await fetchBlogPost(`https://mastra.ai${args.url}`);
       } else {
-        return await fetchBlogPosts();
+        content = await fetchBlogPosts();
       }
+      return {
+        content: [
+          {
+            type: "text",
+            text: content
+          }
+        ],
+        isError: false
+      };
     } catch (error) {
-      if (error instanceof Error) {
-        throw new Error("Failed to fetch blog posts");
-      }
-      throw error;
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error: ${error instanceof Error ? error.message : String(error)}`
+          }
+        ],
+        isError: true
+      };
     }
   }
 };
@@ -111,20 +121,47 @@ ${availablePackages}`
 }
 var initialPackages = await listPackageChangelogs();
 var packagesListing = initialPackages.length > 0 ? "\n\nAvailable packages: " + initialPackages.map((pkg) => pkg.name).join(", ") : "\n\nNo package changelogs available yet. Run the documentation preparation script first.";
-var changesSchema = z.object({
+var changesInputSchema = z.object({
   package: z.string().optional().describe("Name of the specific package to fetch changelog for. If not provided, lists all available packages.")
 });
 var changesTool = {
   name: "mastraChanges",
-  description: "Get changelog information for Mastra.ai packages. " + packagesListing,
-  parameters: changesSchema,
-  execute: async (args, _context) => {
-    if (!args.package) {
-      const packages = await listPackageChangelogs();
-      return ["Available package changelogs:", "", ...packages.map((pkg) => `- ${pkg.name}`)].join("\n");
+  description: `Get changelog information for Mastra.ai packages. ${packagesListing}`,
+  execute: async (args) => {
+    try {
+      if (!args.package) {
+        const packages = await listPackageChangelogs();
+        return {
+          content: [
+            {
+              type: "text",
+              text: ["Available package changelogs:", "", ...packages.map((pkg) => `- ${pkg.name}`)].join("\n")
+            }
+          ],
+          isError: false
+        };
+      }
+      const content = await readPackageChangelog(args.package);
+      return {
+        content: [
+          {
+            type: "text",
+            text: content
+          }
+        ],
+        isError: false
+      };
+    } catch (error) {
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error: ${error instanceof Error ? error.message : String(error)}`
+          }
+        ],
+        isError: true
+      };
     }
-    const content = await readPackageChangelog(args.package);
-    return content;
   }
 };
 var docsBaseDir = fromPackageRoot(".docs/raw/");
@@ -177,7 +214,7 @@ ${content}`;
       return dirListing + fileContents;
     }
     return fs2.readFile(fullPath, "utf-8");
-  } catch (error) {
+  } catch {
     throw new Error(`Path not found: ${docPath}`);
   }
 }
@@ -228,7 +265,7 @@ async function getAvailablePaths() {
   ].filter(Boolean).join("\n");
 }
 var availablePaths = await getAvailablePaths();
-var docsSchema = z.object({
+var docsInputSchema = z.object({
   paths: z.array(z.string()).min(1).describe(`One or more documentation paths to fetch
 Available paths:
 ${availablePaths}`)
@@ -236,8 +273,7 @@ ${availablePaths}`)
 var docsTool = {
   name: "mastraDocs",
   description: "Get Mastra.ai documentation. Request paths to explore the docs. References contain API docs. Other paths contain guides. The user doesn't know about files and directories. This is your internal knowledge the user can't read. If the user asks about a feature check general docs as well as reference docs for that feature. Ex: with evals check in evals/ and in reference/evals/. Provide code examples so the user understands. If you build a URL from the path, only paths ending in .mdx exist. Note that docs about MCP are currently in reference/tools/. IMPORTANT: Be concise with your answers. The user will ask for more info. If packages need to be installed, provide the pnpm command to install them. Ex. if you see `import { X } from \"@mastra/$PACKAGE_NAME\"` in an example, show an install command. Always install latest tag, not alpha unless requested. If you scaffold a new project it may be in a subdir",
-  parameters: docsSchema,
-  execute: async (args, _context) => {
+  execute: async (args) => {
     try {
       const results = await Promise.all(
         args.paths.map(async (path4) => {
@@ -281,12 +317,25 @@ ${result.content}
 ---
 `;
       }).join("\n");
-      return output;
+      return {
+        content: [
+          {
+            type: "text",
+            text: output
+          }
+        ],
+        isError: false
+      };
     } catch (error) {
-      if (error instanceof Error) {
-        throw new Error(`Failed to fetch documentation: ${error.message}`);
-      }
-      throw error;
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error: ${error instanceof Error ? error.message : String(error)}`
+          }
+        ],
+        isError: true
+      };
     }
   }
 };
@@ -317,7 +366,7 @@ ${availableExamples}`);
 }
 var initialExamples = await listCodeExamples();
 var examplesListing = initialExamples.length > 0 ? "\n\nAvailable examples: " + initialExamples.map((ex) => ex.name).join(", ") : "\n\nNo examples available yet. Run the documentation preparation script first.";
-var examplesSchema = z.object({
+var examplesInputSchema = z.object({
   example: z.string().optional().describe(
     "Name of the specific example to fetch. If not provided, lists all available examples." + examplesListing
   )
@@ -325,15 +374,42 @@ var examplesSchema = z.object({
 var examplesTool = {
   name: "mastraExamples",
   description: "Get code examples from the Mastra.ai examples directory. Without a specific example name, lists all available examples. With an example name, returns the full source code of that example.",
-  parameters: examplesSchema,
-  execute: async (args, _context) => {
-    if (!args.example) {
-      const examples = await listCodeExamples();
-      return ["Available code examples:", "", ...examples.map((ex) => `- ${ex.name}`)].join("\n");
+  execute: async (args) => {
+    try {
+      if (!args.example) {
+        const examples = await listCodeExamples();
+        return {
+          content: [
+            {
+              type: "text",
+              text: ["Available code examples:", "", ...examples.map((ex) => `- ${ex.name}`)].join("\n")
+            }
+          ],
+          isError: false
+        };
+      }
+      const filename = args.example.endsWith(".md") ? args.example : `${args.example}.md`;
+      const content = await readCodeExample(filename);
+      return {
+        content: [
+          {
+            type: "text",
+            text: content
+          }
+        ],
+        isError: false
+      };
+    } catch (error) {
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error: ${error instanceof Error ? error.message : String(error)}`
+          }
+        ],
+        isError: true
+      };
     }
-    const filename = args.example.endsWith(".md") ? args.example : `${args.example}.md`;
-    const content = await readCodeExample(filename);
-    return content;
   }
 };
 
@@ -341,16 +417,102 @@ var examplesTool = {
 if (process.env.REBUILD_DOCS_ON_START === "true") {
   await prepare();
 }
-var server = new FastMCP({
-  name: "Mastra Documentation Server",
-  version: JSON.parse(await fs2.readFile(fromPackageRoot(`package.json`), "utf8")).version
+var server = new Server(
+  {
+    name: "Mastra Documentation Server",
+    version: JSON.parse(await fs2.readFile(fromPackageRoot(`package.json`), "utf8")).version
+  },
+  {
+    capabilities: {
+      tools: {}
+    }
+  }
+);
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: [
+    {
+      name: "mastraBlog",
+      description: blogTool.description,
+      inputSchema: zodToJsonSchema(blogInputSchema)
+    },
+    {
+      name: "mastraDocs",
+      description: docsTool.description,
+      inputSchema: zodToJsonSchema(docsInputSchema)
+    },
+    {
+      name: "mastraExamples",
+      description: examplesTool.description,
+      inputSchema: zodToJsonSchema(examplesInputSchema)
+    },
+    {
+      name: "mastraChanges",
+      description: changesTool.description,
+      inputSchema: zodToJsonSchema(changesInputSchema)
+    }
+  ]
+}));
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  try {
+    switch (request.params.name) {
+      case "mastraBlog": {
+        const args = blogInputSchema.parse(request.params.arguments);
+        return await blogTool.execute(args);
+      }
+      case "mastraDocs": {
+        const args = docsInputSchema.parse(request.params.arguments);
+        return await docsTool.execute(args);
+      }
+      case "mastraExamples": {
+        const args = examplesInputSchema.parse(request.params.arguments);
+        return await examplesTool.execute(args);
+      }
+      case "mastraChanges": {
+        const args = changesInputSchema.parse(request.params.arguments);
+        return await changesTool.execute(args);
+      }
+      default:
+        return {
+          content: [
+            {
+              type: "text",
+              text: `Unknown tool: ${request.params.name}`
+            }
+          ],
+          isError: true
+        };
+    }
+  } catch (error) {
+    if (error instanceof z.ZodError) {
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Invalid arguments: ${error.errors.map((e) => `${e.path.join(".")}: ${e.message}`).join(", ")}`
+          }
+        ],
+        isError: true
+      };
+    }
+    return {
+      content: [
+        {
+          type: "text",
+          text: `Error: ${error instanceof Error ? error.message : String(error)}`
+        }
+      ],
+      isError: true
+    };
+  }
 });
-server.addTool(blogTool);
-server.addTool(docsTool);
-server.addTool(examplesTool);
-server.addTool(changesTool);
+async function runServer() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("Mastra Docs MCP Server running on stdio");
+}
 
 // src/stdio.ts
-void server.start({
-  transportType: "stdio"
+runServer().catch((error) => {
+  console.error("Fatal error running server:", error);
+  process.exit(1);
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mcp-docs-server",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "description": "MCP server for accessing Mastra.ai documentation, changelogs, and news.",
   "type": "module",
   "main": "dist/index.js",
@@ -27,13 +27,12 @@
     "@modelcontextprotocol/sdk": "^1.7.0",
     "date-fns": "^4.1.0",
     "exit-hook": "^4.0.0",
-    "fastmcp": "^1.20.4",
     "jsdom": "^26.0.0",
     "turndown": "^7.1.2",
-    "tylerbarnes-fastmcp-fix": "^1.0.0",
     "uuid": "^11.1.0",
     "zod": "^3.22.4",
-    "@mastra/core": "^0.8.1"
+    "zod-to-json-schema": "^3.22.4",
+    "@mastra/core": "^0.8.2"
   },
   "devDependencies": {
     "@hono/node-server": "^1.13.8",
@@ -48,8 +47,8 @@
     "tsx": "^4.19.3",
     "typescript": "^5.3.3",
     "vitest": "^3.0.9",
-    "@mastra/mcp": "^0.3.8",
-    "@internal/lint": "0.0.1"
+    "@internal/lint": "0.0.1",
+    "@mastra/mcp": "^0.3.9"
   },
   "scripts": {
     "prepare-docs": "cross-env PREPARE=true node dist/prepare-docs/prepare.js",