@mastra/mcp 0.13.2 → 0.13.3-alpha.0

package/dist/index.js

@@ -1714,13 +1714,13 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
     const capturedResourceOptions = this.resourceOptions;
     if (!capturedResourceOptions) return;
     if (capturedResourceOptions.listResources) {
-      serverInstance.setRequestHandler(ListResourcesRequestSchema, async () => {
+      serverInstance.setRequestHandler(ListResourcesRequestSchema, async (_request, extra) => {
         this.logger.debug("Handling ListResources request");
         if (this.definedResources) {
           return { resources: this.definedResources };
         } else {
           try {
-            const resources = await capturedResourceOptions.listResources();
+            const resources = await capturedResourceOptions.listResources({ extra });
             this.definedResources = resources;
             this.logger.debug(`Fetched and cached ${this.definedResources.length} resources.`);
             return { resources: this.definedResources };
@@ -1732,12 +1732,12 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
       });
     }
     if (capturedResourceOptions.getResourceContent) {
-      serverInstance.setRequestHandler(ReadResourceRequestSchema, async (request) => {
+      serverInstance.setRequestHandler(ReadResourceRequestSchema, async (request, extra) => {
         const startTime = Date.now();
         const uri = request.params.uri;
         this.logger.debug(`Handling ReadResource request for URI: ${uri}`);
         if (!this.definedResources) {
-          const resources = await this.resourceOptions?.listResources?.();
+          const resources = await this.resourceOptions?.listResources?.({ extra });
           if (!resources) throw new Error("Failed to load resources");
           this.definedResources = resources;
         }
@@ -1747,7 +1747,7 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
           throw new Error(`Resource not found: ${uri}`);
         }
         try {
-          const resourcesOrResourceContent = await capturedResourceOptions.getResourceContent({ uri });
+          const resourcesOrResourceContent = await capturedResourceOptions.getResourceContent({ uri, extra });
           const resourcesContent = Array.isArray(resourcesOrResourceContent) ? resourcesOrResourceContent : [resourcesOrResourceContent];
           const contents = resourcesContent.map((resourceContent) => {
             const contentItem = {
@@ -1775,13 +1775,13 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
       });
     }
     if (capturedResourceOptions.resourceTemplates) {
-      serverInstance.setRequestHandler(ListResourceTemplatesRequestSchema, async () => {
+      serverInstance.setRequestHandler(ListResourceTemplatesRequestSchema, async (_request, extra) => {
         this.logger.debug("Handling ListResourceTemplates request");
         if (this.definedResourceTemplates) {
           return { resourceTemplates: this.definedResourceTemplates };
         } else {
           try {
-            const templates = await capturedResourceOptions.resourceTemplates();
+            const templates = await capturedResourceOptions.resourceTemplates({ extra });
             this.definedResourceTemplates = templates;
             this.logger.debug(`Fetched and cached ${this.definedResourceTemplates.length} resource templates.`);
             return { resourceTemplates: this.definedResourceTemplates };
@@ -1812,7 +1812,7 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
     const capturedPromptOptions = this.promptOptions;
     if (!capturedPromptOptions) return;
     if (capturedPromptOptions.listPrompts) {
-      serverInstance.setRequestHandler(ListPromptsRequestSchema, async () => {
+      serverInstance.setRequestHandler(ListPromptsRequestSchema, async (_request, extra) => {
         this.logger.debug("Handling ListPrompts request");
         if (this.definedPrompts) {
           return {
@@ -1820,7 +1820,7 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
           };
         } else {
           try {
-            const prompts = await capturedPromptOptions.listPrompts();
+            const prompts = await capturedPromptOptions.listPrompts({ extra });
             for (const prompt of prompts) {
               PromptSchema.parse(prompt);
             }
@@ -1841,11 +1841,11 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
     if (capturedPromptOptions.getPromptMessages) {
       serverInstance.setRequestHandler(
         GetPromptRequestSchema,
-        async (request) => {
+        async (request, extra) => {
           const startTime = Date.now();
           const { name, version, arguments: args } = request.params;
           if (!this.definedPrompts) {
-            const prompts = await this.promptOptions?.listPrompts?.();
+            const prompts = await this.promptOptions?.listPrompts?.({ extra });
             if (!prompts) throw new Error("Failed to load prompts");
             this.definedPrompts = prompts;
           }
@@ -1866,7 +1866,7 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
           try {
             let messages = [];
             if (capturedPromptOptions.getPromptMessages) {
-              messages = await capturedPromptOptions.getPromptMessages({ name, version, args });
+              messages = await capturedPromptOptions.getPromptMessages({ name, version, args, extra });
             }
             const duration = Date.now() - startTime;
             this.logger.info(