@growthbook/mcp 1.3.1 → 1.4.3

package/server/tools/{experiments.js → experiments/experiments.js}

@@ -1,6 +1,7 @@
 import { z } from "zod";
-import { generateLinkToGrowthBook, getDocsMetadata, handleResNotOk, SUPPORTED_FILE_EXTENSIONS, paginationSchema, } from "../utils.js";
-import { getDefaults } from "./defaults.js";
+import { generateLinkToGrowthBook, getDocsMetadata, handleResNotOk, SUPPORTED_FILE_EXTENSIONS, paginationSchema, fetchWithRateLimit, } from "../../utils.js";
+import { getDefaults } from "../defaults.js";
+import { handleSummaryMode } from "./experiment-summary.js";
 export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin, user, }) {
     /**
      * Tool: get_experiments
@@ -11,13 +12,29 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             .describe("The ID of the project to filter experiments by")
             .optional(),
         mode: z
-            .enum(["default", "analyze"])
-            .default("default")
-            .describe("The mode to use to fetch experiments. Default mode returns summary info about experiments. Analyze mode will also fetch experiment results, allowing for better analysis, interpretation, and reporting."),
+            .enum(["metadata", "summary", "full"])
+            .default("metadata")
+            .describe("The mode to use to fetch experiments. Metadata mode returns experiment config without results. Summary mode fetches results and returns pruned key stats for quick analysis. Full mode fetches and returns complete results data. WARNING: Full mode may return large payloads."),
         ...paginationSchema,
     }, {
         readOnlyHint: true,
-    }, async ({ limit, offset, mostRecent, project, mode }) => {
+    }, async ({ limit, offset, mostRecent, project, mode }, extra) => {
+        const progressToken = extra._meta?.progressToken;
+        const totalSteps = mode === "summary" ? 5 : mode === "full" ? 3 : 2;
+        const reportProgress = async (progress, message) => {
+            if (progressToken) {
+                await server.server.notification({
+                    method: "notifications/progress",
+                    params: {
+                        progressToken,
+                        progress,
+                        total: totalSteps,
+                        ...(message && { message }),
+                    },
+                });
+            }
+        };
+        await reportProgress(1, "Fetching experiments...");
         try {
             // Default behavior
             if (!mostRecent || offset > 0) {
@@ -28,7 +45,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
                 if (project) {
                     defaultQueryParams.append("projectId", project);
                 }
-                const defaultRes = await fetch(`${baseApiUrl}/api/v1/experiments?${defaultQueryParams.toString()}`, {
+                const defaultRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments?${defaultQueryParams.toString()}`, {
                     headers: {
                         Authorization: `Bearer ${apiKey}`,
                         "Content-Type": "application/json",
@@ -37,10 +54,15 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
                 await handleResNotOk(defaultRes);
                 const data = await defaultRes.json();
                 const experiments = data.experiments;
-                if (mode === "analyze") {
+                if (mode === "full" || mode === "summary") {
+                    await reportProgress(2, "Fetching experiment results...");
                     for (const [index, experiment] of experiments.entries()) {
+                        if (experiment.status === "draft") {
+                            experiments[index].result = undefined;
+                            continue;
+                        }
                         try {
-                            const resultsRes = await fetch(`${baseApiUrl}/api/v1/experiments/${experiment.id}/results`, {
+                            const resultsRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments/${experiment.id}/results`, {
                                 headers: {
                                     Authorization: `Bearer ${apiKey}`,
                                 },
@@ -54,12 +76,32 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
                         }
                     }
                 }
+                if (mode === "summary") {
+                    const summaryExperiments = await handleSummaryMode(experiments, baseApiUrl, apiKey, reportProgress);
+                    const summaryExperimentsWithPagination = {
+                        summary: summaryExperiments,
+                        limit: data.limit,
+                        offset: data.offset,
+                        total: data.total,
+                        hasMore: data.hasMore,
+                        nextOffset: data.nextOffset,
+                    };
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify(summaryExperimentsWithPagination),
+                            },
+                        ],
+                    };
+                }
+                await reportProgress(2, "Processing results...");
                 return {
-                    content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                    content: [{ type: "text", text: JSON.stringify(data) }],
                 };
             }
             // Most recent behavior
-            const countRes = await fetch(`${baseApiUrl}/api/v1/experiments?limit=1`, {
+            const countRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments?limit=1`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                 },
@@ -75,7 +117,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             if (project) {
                 mostRecentQueryParams.append("projectId", project);
             }
-            const mostRecentRes = await fetch(`${baseApiUrl}/api/v1/experiments?${mostRecentQueryParams.toString()}`, {
+            const mostRecentRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments?${mostRecentQueryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                 },
@@ -85,10 +127,11 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             if (mostRecentData.experiments &&
                 Array.isArray(mostRecentData.experiments)) {
                 mostRecentData.experiments = mostRecentData.experiments.reverse();
-                if (mode === "analyze") {
+                if (mode === "full" || mode === "summary") {
+                    await reportProgress(2, "Fetching experiment results...");
                     for (const [index, experiment,] of mostRecentData.experiments.entries()) {
                         try {
-                            const resultsRes = await fetch(`${baseApiUrl}/api/v1/experiments/${experiment.id}/results`, {
+                            const resultsRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments/${experiment.id}/results`, {
                                 headers: {
                                     Authorization: `Bearer ${apiKey}`,
                                 },
@@ -103,10 +146,30 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
                     }
                 }
             }
+            if (mode === "summary") {
+                const experiments = Array.isArray(mostRecentData.experiments)
+                    ? mostRecentData.experiments
+                    : [];
+                const summaryExperiments = await handleSummaryMode(experiments, baseApiUrl, apiKey, reportProgress);
+                const summaryExperimentsWithPagination = {
+                    summary: summaryExperiments,
+                    limit: mostRecentData.limit,
+                    offset: mostRecentData.offset,
+                    total: mostRecentData.total,
+                    hasMore: mostRecentData.hasMore,
+                    nextOffset: mostRecentData.nextOffset,
+                };
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify(summaryExperimentsWithPagination),
+                        },
+                    ],
+                };
+            }
             return {
-                content: [
-                    { type: "text", text: JSON.stringify(mostRecentData, null, 2) },
-                ],
+                content: [{ type: "text", text: JSON.stringify(mostRecentData) }],
             };
         }
         catch (error) {
@@ -119,14 +182,14 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
     server.tool("get_experiment", "Gets a single experiment from GrowthBook", {
         experimentId: z.string().describe("The ID of the experiment to get"),
         mode: z
-            .enum(["default", "analyze"])
-            .default("default")
-            .describe("The mode to use to fetch the experiment. Default mode returns summary info about the experiment. Analyze mode will also fetch experiment results, allowing for better analysis, interpretation, and reporting."),
+            .enum(["metadata", "full"])
+            .default("metadata")
+            .describe("The mode to use to fetch the experiment. Metadata mode returns summary info about the experiment. Full mode fetches results and returns complete results data."),
     }, {
         readOnlyHint: true,
     }, async ({ experimentId, mode }) => {
         try {
-            const res = await fetch(`${baseApiUrl}/api/v1/experiments/${experimentId}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments/${experimentId}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -134,10 +197,13 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             });
             await handleResNotOk(res);
             const data = await res.json();
-            // If analyze mode, fetch results
-            if (mode === "analyze") {
+            // If analyze or summary mode, fetch results
+            if (mode === "full") {
+                if (data.status === "draft") {
+                    data.result = null;
+                }
                 try {
-                    const resultsRes = await fetch(`${baseApiUrl}/api/v1/experiments/${experimentId}/results`, {
+                    const resultsRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments/${experimentId}/results`, {
                         headers: {
                             Authorization: `Bearer ${apiKey}`,
                         },
@@ -152,7 +218,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             }
             const linkToGrowthBook = generateLinkToGrowthBook(appOrigin, "experiment", experimentId);
             const text = `
-    ${JSON.stringify(data, null, 2)}
+    ${JSON.stringify(data)}
     
     [View the experiment in GrowthBook](${linkToGrowthBook})
     `;
@@ -173,7 +239,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
         try {
             const queryParams = new URLSearchParams();
             queryParams.append("limit", "100");
-            const res = await fetch(`${baseApiUrl}/api/v1/attributes?${queryParams.toString()}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/attributes?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -182,7 +248,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             await handleResNotOk(res);
             const data = await res.json();
             return {
-                content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                content: [{ type: "text", text: JSON.stringify(data) }],
             };
         }
         catch (error) {
@@ -258,7 +324,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
             ...(project && { project }),
         };
         try {
-            const experimentRes = await fetch(`${baseApiUrl}/api/v1/experiments`, {
+            const experimentRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/experiments`, {
                 method: "POST",
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
@@ -298,7 +364,7 @@ export function registerExperimentTools({ server, baseApiUrl, apiKey, appOrigin,
                     }, {}),
                 },
             };
-            const flagRes = await fetch(`${baseApiUrl}/api/v1/features`, {
+            const flagRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features`, {
                 method: "POST",
                 headers: {
                     Authorization: `Bearer ${apiKey}`,

package/server/tools/features.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { getDocsMetadata, handleResNotOk, generateLinkToGrowthBook, paginationSchema, featureFlagSchema, } from "../utils.js";
+import { getDocsMetadata, handleResNotOk, generateLinkToGrowthBook, paginationSchema, featureFlagSchema, fetchWithRateLimit, } from "../utils.js";
 import { exec } from "child_process";
 import { getDefaults } from "./defaults.js";
 export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, user, }) {
@@ -24,7 +24,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             environments = defaults.environments;
         }
         else {
-            const envRes = await fetch(`${baseApiUrl}/api/v1/features/environments`, {
+            const envRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features/environments`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -51,7 +51,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             ...(project && { project }),
         };
         try {
-            const res = await fetch(`${baseApiUrl}/api/v1/features`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features`, {
                 method: "POST",
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
@@ -63,7 +63,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             const data = await res.json();
             const { docs, language, stub } = getDocsMetadata(fileExtension);
             const linkToGrowthBook = generateLinkToGrowthBook(appOrigin, "features", id);
-            const text = `This is the API response: ${JSON.stringify(data, null, 2)}
+            const text = `This is the API response: ${JSON.stringify(data)}
 
         Additionally, here is a template of what to show to the user:
 
@@ -124,7 +124,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
                     return acc;
                 }, {}),
             };
-            const res = await fetch(`${baseApiUrl}/api/v1/features/${featureId}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features/${featureId}`, {
                 method: "POST",
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
@@ -136,7 +136,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             const data = await res.json();
             const linkToGrowthBook = generateLinkToGrowthBook(appOrigin, "features", featureId);
             const { docs, language, stub } = getDocsMetadata(fileExtension);
-            const text = `This is the API response: ${JSON.stringify(data, null, 2)}
+            const text = `This is the API response: ${JSON.stringify(data)}
       
         Additionally, here is a template of what to show to the user:
 
@@ -176,7 +176,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             if (project) {
                 queryParams.append("projectId", project);
             }
-            const res = await fetch(`${baseApiUrl}/api/v1/features?${queryParams.toString()}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -185,7 +185,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             await handleResNotOk(res);
             const data = await res.json();
             return {
-                content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                content: [{ type: "text", text: JSON.stringify(data) }],
             };
         }
         catch (error) {
@@ -201,7 +201,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
         readOnlyHint: true,
     }, async ({ id }) => {
         try {
-            const res = await fetch(`${baseApiUrl}/api/v1/features/${id}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features/${id}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -211,7 +211,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
             const data = await res.json();
             const linkToGrowthBook = generateLinkToGrowthBook(appOrigin, "features", id);
             const text = `
-        ${JSON.stringify(data.feature, null, 2)}
+        ${JSON.stringify(data)}
 
         Share information about the feature flag with the user. In particular, give details about the enabled environments,
         rules for each environment, and the default value. If the feature flag is archived or doesnt exist, inform the user and
@@ -241,7 +241,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
                 limit: limit?.toString(),
                 offset: offset?.toString(),
             });
-            const res = await fetch(`${baseApiUrl}/api/v1/features?${queryParams.toString()}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/features?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -264,7 +264,7 @@ export function registerFeatureTools({ server, baseApiUrl, apiKey, appOrigin, us
                 });
             });
             const text = `
-        ${JSON.stringify(filteredSafeRollouts, null, 2)}
+        ${JSON.stringify(filteredSafeRollouts)}
 
         Share information about the rolled-back or released safe rollout rules with the user. Safe Rollout rules are stored under
         environmentSettings, keyed by environment and are within the rules array with a type of "safe-rollout". Ask the user if they

package/server/tools/metrics.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { generateLinkToGrowthBook, handleResNotOk, paginationSchema, } from "../utils.js";
+import { generateLinkToGrowthBook, handleResNotOk, paginationSchema, fetchWithRateLimit, } from "../utils.js";
 export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, }) {
     /**
      * Tool: get_metrics
@@ -21,7 +21,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
             if (project) {
                 queryParams.append("projectId", project);
             }
-            const metricsRes = await fetch(`${baseApiUrl}/api/v1/metrics?${queryParams.toString()}`, {
+            const metricsRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/metrics?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -29,7 +29,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
             });
             await handleResNotOk(metricsRes);
             const metricsData = await metricsRes.json();
-            const factMetricRes = await fetch(`${baseApiUrl}/api/v1/fact-metrics?${queryParams.toString()}`, {
+            const factMetricRes = await fetchWithRateLimit(`${baseApiUrl}/api/v1/fact-metrics?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -42,9 +42,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
                 factMetrics: factMetricData,
             };
             return {
-                content: [
-                    { type: "text", text: JSON.stringify(metricData, null, 2) },
-                ],
+                content: [{ type: "text", text: JSON.stringify(metricData) }],
             };
         }
         catch (error) {
@@ -62,7 +60,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
         try {
             let res;
             if (metricId.startsWith("fact__")) {
-                res = await fetch(`${baseApiUrl}/api/v1/fact-metrics/${metricId}`, {
+                res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/fact-metrics/${metricId}`, {
                     headers: {
                         Authorization: `Bearer ${apiKey}`,
                         "Content-Type": "application/json",
@@ -70,7 +68,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
                 });
             }
             else {
-                res = await fetch(`${baseApiUrl}/api/v1/metrics/${metricId}`, {
+                res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/metrics/${metricId}`, {
                     headers: {
                         Authorization: `Bearer ${apiKey}`,
                         "Content-Type": "application/json",
@@ -84,7 +82,7 @@ export function registerMetricsTools({ server, baseApiUrl, apiKey, appOrigin, })
                 content: [
                     {
                         type: "text",
-                        text: JSON.stringify(data, null, 2) +
+                        text: JSON.stringify(data) +
                             `\n**Critical** Show the user the link to the metric in GrowthBook: [View the metric in GrowthBook](${linkToGrowthBook})
           `,
                     },

package/server/tools/projects.js

@@ -1,4 +1,4 @@
-import { handleResNotOk, paginationSchema, } from "../utils.js";
+import { handleResNotOk, paginationSchema, fetchWithRateLimit, } from "../utils.js";
 /**
  * Tool: get_projects
  */
@@ -13,7 +13,7 @@ export function registerProjectTools({ server, baseApiUrl, apiKey, }) {
             offset: offset.toString(),
         });
         try {
-            const res = await fetch(`${baseApiUrl}/api/v1/projects?${queryParams.toString()}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/projects?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -22,7 +22,7 @@ export function registerProjectTools({ server, baseApiUrl, apiKey, }) {
             await handleResNotOk(res);
             const data = await res.json();
             return {
-                content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                content: [{ type: "text", text: JSON.stringify(data) }],
             };
         }
         catch (error) {

package/server/tools/sdk-connections.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { handleResNotOk, paginationSchema, } from "../utils.js";
+import { handleResNotOk, paginationSchema, fetchWithRateLimit, } from "../utils.js";
 export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
     /**
      * Tool: get_sdk_connections
@@ -21,7 +21,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
             if (project) {
                 queryParams.append("projectId", project);
             }
-            const res = await fetch(`${baseApiUrl}/api/v1/sdk-connections?${queryParams.toString()}`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/sdk-connections?${queryParams.toString()}`, {
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
                     "Content-Type": "application/json",
@@ -30,7 +30,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
             await handleResNotOk(res);
             const data = await res.json();
             return {
-                content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                content: [{ type: "text", text: JSON.stringify(data) }],
             };
         }
         catch (error) {
@@ -84,7 +84,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
     }, async ({ name, language, environment, projects }) => {
         if (!environment) {
             try {
-                const res = await fetch(`${baseApiUrl}/api/v1/environments`, {
+                const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/environments`, {
                     headers: {
                         Authorization: `Bearer ${apiKey}`,
                         "Content-Type": "application/json",
@@ -92,7 +92,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
                 });
                 await handleResNotOk(res);
                 const data = await res.json();
-                const text = `${JSON.stringify(data, null, 2)}
+                const text = `${JSON.stringify(data)}
     
         Here is the list of environments. Ask the user to select one and use the key in the create_sdk_connection tool.
         `;
@@ -113,7 +113,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
             ...(projects && { projects }),
         };
         try {
-            const res = await fetch(`${baseApiUrl}/api/v1/sdk-connections`, {
+            const res = await fetchWithRateLimit(`${baseApiUrl}/api/v1/sdk-connections`, {
                 method: "POST",
                 headers: {
                     Authorization: `Bearer ${apiKey}`,
@@ -124,7 +124,7 @@ export function registerSdkConnectionTools({ server, baseApiUrl, apiKey, }) {
             await handleResNotOk(res);
             const data = await res.json();
             return {
-                content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
+                content: [{ type: "text", text: JSON.stringify(data) }],
             };
         }
         catch (error) {

package/server/tools/search.js

@@ -7,35 +7,54 @@ export function registerSearchTools({ server }) {
     server.tool("search_growthbook_docs", "Search the GrowthBook docs on how to use a feature", {
         query: z
             .string()
+            .min(1)
             .describe("The search query to look up in the GrowthBook docs."),
+        maxResults: z
+            .number()
+            .min(1)
+            .max(10)
+            .default(5)
+            .optional()
+            .describe("Maximum number of results to return (1-10, default: 5). More results may provide better context but increase response size."),
     }, {
         readOnlyHint: true,
-    }, async ({ query }) => {
-        const hits = await searchGrowthBookDocs(query);
+    }, async ({ query, maxResults = 5 }) => {
+        const results = await searchGrowthBookDocs(query, {
+            hitsPerPage: maxResults,
+        });
+        if (results.length === 0) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `No results found for "${query}". Try using different keywords or a more specific search term.`,
+                    },
+                ],
+            };
+        }
         return {
-            content: hits.slice(0, 5).map((hit) => {
-                // Algolia typically returns content in various fields
-                const content = hit.content ||
-                    hit.text ||
-                    hit._snippetResult?.content?.value ||
-                    hit._highlightResult?.content?.value;
-                const snippet = hit._snippetResult?.content?.value ||
-                    hit._highlightResult?.content?.value;
-                const title = hit.title || hit.hierarchy?.lvl0 || hit.hierarchy?.lvl1;
-                const url = hit.url || hit.anchor;
-                let text = "";
-                if (title) {
-                    text += `**${title}**\n`;
+            content: results.map((result) => {
+                let text = `**${result.title}**\n`;
+                // Add hierarchy/breadcrumb if available (helps with context)
+                if (result.hierarchy && result.hierarchy.length > 0) {
+                    text += `📍 ${result.hierarchy.join(" > ")}\n`;
                 }
-                if (url) {
-                    text += `URL: ${url}\n`;
+                // Add URL (important for reference)
+                if (result.url && result.url !== "#") {
+                    text += `🔗 ${result.url}\n`;
                 }
-                if (snippet || content) {
-                    text += `\n${snippet || content}`;
+                // Add snippet/content (the most important part - make it prominent)
+                if (result.snippet) {
+                    text += `\n${result.snippet}`;
+                }
+                else if (result.content) {
+                    // Fallback to full content if no snippet, but truncate intelligently
+                    const truncated = result.content.substring(0, 300);
+                    text += `\n${truncated}${result.content.length > 300 ? "..." : ""}`;
                 }
                 return {
                     type: "text",
-                    text: text || JSON.stringify(hit),
+                    text: text.trim(),
                 };
             }),
         };

package/server/types/types.js

@@ -0,0 +1 @@
+export {};