feat: Add TracingContext type to various tools for enhanced tracing support

- Updated copywriter-agent-tool.ts to include TracingContext in the execute function.
- Modified csv-to-json.tool.ts to utilize TracingContext in the execute function signature.
- Enhanced data-file-manager.ts by adding TracingContext type to multiple execute functions.
- Included TracingContext in the execute function of execa-tool.ts for better tracing.
- Added TracingContext type to pdf-data-conversion.tool.ts for improved tracing capabilities.
- Updated weather-tool.ts to incorporate TracingContext in the tool's context.
- Enhanced web-scraper-tool.ts by adding TracingContext to the execute function across multiple tools.
- Modified write-note.ts to include TracingContext in the execute function for better tracing.

Author: ssdeanx

src/mastra/tools/copywriter-agent-tool.ts

@@ -2,7 +2,7 @@ import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing'
 import { InferUITool, createTool } from "@mastra/core/tools";
 import { z } from 'zod'
 import { log } from '../config/logger'
-
+import type { TracingContext } from '@mastra/core/ai-tracing';
 // Define runtime context for this tool
 export interface CopywriterToolContext {
     userId?: string
@@ -70,7 +70,7 @@ export const copywriterTool = createTool({
             .optional()
             .describe('Approximate word count of the content'),
     }),
-    execute: async ({ context, mastra, writer, tracingContext }) => {
+    execute: async ({ context, mastra, writer, tracingContext } ) => {
         const {
             topic,
             contentType = 'blog',

src/mastra/tools/csv-to-json.tool.ts

@@ -3,6 +3,7 @@ import { InferUITool, createTool } from "@mastra/core/tools";
 import { parse } from "csv-parse/sync";
 import * as fs from "node:fs/promises";
 import { z } from "zod";
+import type { TracingContext } from '@mastra/core/ai-tracing';
 
 const csvToolContextSchema = z.object({
   maxRows: z.number().optional(),
@@ -33,7 +34,7 @@ export const csvToJsonTool = createTool({
     data: z.array(z.any()).describe("Parsed JSON data"),
     error: z.string().optional(),
   }),
-  execute: async ({ context, writer, runtimeContext, tracingContext }) => {
+  execute: async ({ context, writer, runtimeContext, tracingContext }: { context: { csvData?: string; filePath?: string; options?: { delimiter?: string; columns?: boolean; trim?: boolean; skip_empty_lines?: boolean } }, writer?: any, runtimeContext?: any, tracingContext?: TracingContext }) => {
     await writer?.write({ type: 'progress', data: { message: '📊 Starting CSV to JSON conversion' } });
     const rootSpan = tracingContext?.currentSpan?.createChildSpan({
       type: AISpanType.TOOL_CALL,
@@ -61,11 +62,18 @@ export const csvToJsonTool = createTool({
         throw new Error("Either csvData or filePath must be provided");
       }
 
+      const options = context.options ?? {
+        delimiter: ",",
+        columns: true,
+        trim: true,
+        skip_empty_lines: true,
+      };
+
       const records = parse(contentToParse, {
-        delimiter: context.options.delimiter,
-        columns: context.options.columns,
-        trim: context.options.trim,
-        skip_empty_lines: context.options.skip_empty_lines,
+        delimiter: options.delimiter,
+        columns: options.columns,
+        trim: options.trim,
+        skip_empty_lines: options.skip_empty_lines,
       });
 
       if (maxRows !== undefined && records.length > maxRows) {

src/mastra/tools/data-file-manager.ts

@@ -21,6 +21,7 @@ import { pipeline } from 'stream/promises'
 import * as zlib from 'zlib'
 import { z } from 'zod'
 import { log } from '../config/logger'
+import type { TracingContext } from '@mastra/core/ai-tracing';
 
 // Define runtime context for these tools
 export interface DataFileManagerContext {
@@ -58,7 +59,7 @@ export const readDataFileTool = createTool({
             ),
     }),
     outputSchema: z.string().describe('The content of the file as a string.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { fileName: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: '📖 Reading file: ' + context.fileName } });
         const readSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -108,7 +109,7 @@ export const writeDataFileTool = createTool({
     outputSchema: z
         .string()
         .describe('A confirmation string indicating success.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { fileName: string, content: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: '💾 Writing to file: ' + context.fileName } });
         const writeSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -167,7 +168,7 @@ export const deleteDataFileTool = createTool({
     outputSchema: z
         .string()
         .describe('A confirmation string indicating success.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { fileName: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: '🗑️ Deleting file: ' + context.fileName } });
         const deleteSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -216,7 +217,7 @@ export const listDataDirTool = createTool({
     outputSchema: z
         .array(z.string())
         .describe('An array of file and directory names.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { dirPath?: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: '📂 Listing directory: ' + (context.dirPath ?? 'docs/data') } });
         const listSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -268,7 +269,7 @@ export const copyDataFileTool = createTool({
     outputSchema: z
         .string()
         .describe('A confirmation string indicating success.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { sourceFile: string, destFile: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: `📋 Copying file: ${context.sourceFile} to ${context.destFile}` } });
         const copySpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -329,7 +330,7 @@ export const moveDataFileTool = createTool({
     outputSchema: z
         .string()
         .describe('A confirmation string indicating success.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { sourceFile: string, destFile: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: `🚚 Moving file: ${context.sourceFile} to ${context.destFile}` } });
         const moveSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -396,7 +397,7 @@ export const searchDataFilesTool = createTool({
     outputSchema: z
         .array(z.string())
         .describe('An array of matching file paths.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { pattern: string, searchContent?: boolean, dirPath?: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: `🔍 Searching for pattern: "${context.pattern}"` } });
         const searchSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
@@ -498,7 +499,7 @@ export const getDataFileInfoTool = createTool({
             isDirectory: z.boolean(),
         })
         .describe('File metadata information.'),
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { fileName: string }, writer?: any, tracingContext?: TracingContext }) => {
         await writer?.write({ type: 'progress', data: { message: 'ℹ️ Getting info for file: ' + context.fileName } });
         const infoSpan = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,

src/mastra/tools/execa-tool.ts

@@ -5,7 +5,7 @@ import { execa, ExecaError } from 'execa';
 import { Transform } from 'stream';
 import { z } from 'zod';
 import { log } from '../config/logger';
-
+import type { TracingContext } from '@mastra/core/ai-tracing';
 // Create transform stream that applies chalk
 const colorTransform = new Transform({
   transform(chunk, _encoding, callback) {
@@ -27,7 +27,7 @@ export const execaTool = createTool({
         message: z.string(),
     }),
 
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { command: string; args: string[] }, writer?: any, tracingContext?: TracingContext }) => {
         const span = tracingContext?.currentSpan?.createChildSpan({
             type: AISpanType.TOOL_CALL,
             name: 'execa-tool',

src/mastra/tools/pdf-data-conversion.tool.ts

@@ -1,3 +1,4 @@
+import { context } from '@opentelemetry/api';
 // Kilocode: Tool Approval
 // owner: team-data
 // justification: PDF parsing and markdown conversion for RAG indexing
@@ -25,6 +26,7 @@ import {
 	logError,
 	logToolExecution,
 } from '../config/logger'
+import type { TracingContext } from '@mastra/core/ai-tracing';
 
 // Lazy-loaded pdf-parse to avoid demo parsing errors
 let pdfParse: unknown = null
@@ -498,7 +500,7 @@ Perfect for RAG indexing, documentation conversion, and content processing.
   `,
 	inputSchema: PdfToMarkdownInputSchema,
 	outputSchema: PdfToMarkdownOutputSchema,
-	execute: async ({ context, tracingContext, writer }) => {
+	execute: async ({ context, tracingContext, writer } ) => {
 		const startTime = Date.now()
 		logToolExecution('pdf-to-markdown', { input: context })
 

src/mastra/tools/weather-tool.ts

@@ -3,6 +3,7 @@ import { z } from 'zod'
 import { log } from '../config/logger'
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing'
 import { RuntimeContext } from '@mastra/core/runtime-context'
+import type { TracingContext } from '@mastra/core/ai-tracing';
 
 // Define the Zod schema for the runtime context
 const weatherToolContextSchema = z.object({

src/mastra/tools/web-scraper-tool.ts

@@ -24,7 +24,7 @@ import * as fs from 'node:fs/promises'
 import * as path from 'node:path'
 import { JSDOM } from 'jsdom'
 import { log } from '../config/logger'
-
+import type { TracingContext } from '@mastra/core/ai-tracing';
 // Enhanced HTML processing with JSDOM
 const DANGEROUS_TAGS = new Set([
     'script',
@@ -408,7 +408,7 @@ export const webScraperTool = createTool({
         'Extracts structured data from web pages using JSDOM and Cheerio with enhanced security and error handling.',
     inputSchema: webScraperInputSchema,
     outputSchema: webScraperOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { url: string; selector?: string; extractAttributes?: string[]; saveMarkdown?: boolean; markdownFileName?: string }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: `🌐 Starting web scrape for ${context.url}` } });
         toolCallCounters.set('web-scraper', (toolCallCounters.get('web-scraper') ?? 0) + 1)
         const scrapeSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -741,7 +741,7 @@ export const batchWebScraperTool = createTool({
         'Scrape multiple web pages concurrently with enhanced JSDOM processing and rate limiting.',
     inputSchema: batchWebScraperInputSchema,
     outputSchema: batchWebScraperOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { urls: string[]; selector?: string; maxConcurrent?: number; saveResults?: boolean; baseFileName?: string }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: `🌐 Batch scraping ${context.urls.length} URLs` } });
         toolCallCounters.set('batch-web-scraper', (toolCallCounters.get('batch-web-scraper') ?? 0) + 1)
         const batchSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -1000,7 +1000,7 @@ export const siteMapExtractorTool = createTool({
         'Extract a comprehensive site map by crawling internal links with enhanced JSDOM processing and rate limiting.',
     inputSchema: siteMapExtractorInputSchema,
     outputSchema: siteMapExtractorOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { url: string; maxDepth?: number; maxPages?: number; includeExternal?: boolean; saveMap?: boolean }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: `🗺️ Starting site map extraction for ${context.url}` } });
         toolCallCounters.set('site-map-extractor', (toolCallCounters.get('site-map-extractor') ?? 0) + 1)
         const mapSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -1259,7 +1259,7 @@ export const linkExtractorTool = createTool({
         'Extract and analyze all links from a web page with enhanced JSDOM processing and filtering.',
     inputSchema: linkExtractorInputSchema,
     outputSchema: linkExtractorOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { url: string; linkTypes?: ('internal' | 'external' | 'all')[]; includeAnchors?: boolean; filterPatterns?: string[] }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: `🔗 Extracting links from ${context.url}` } });
         toolCallCounters.set('link-extractor', (toolCallCounters.get('link-extractor') ?? 0) + 1)
         const linkSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -1489,7 +1489,7 @@ export const htmlToMarkdownTool = createTool({
         'Convert HTML content to well-formatted markdown with enhanced JSDOM parsing and security.',
     inputSchema: htmlToMarkdownInputSchema,
     outputSchema: htmlToMarkdownOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { html: string; saveToFile?: boolean; fileName?: string }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: '🔄 Converting HTML to markdown...' } });
         toolCallCounters.set('html-to-markdown', (toolCallCounters.get('html-to-markdown') ?? 0) + 1)
         const convertSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -1621,7 +1621,7 @@ export const listScrapedContentTool = createTool({
         'List all scraped content files stored in the data directory with enhanced security.',
     inputSchema: listScrapedContentInputSchema,
     outputSchema: listScrapedContentOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { pattern?: string; includeMetadata?: boolean }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: '📂 Listing scraped content files...' } });
         toolCallCounters.set('list-scraped-content', (toolCallCounters.get('list-scraped-content') ?? 0) + 1)
         const listSpan = tracingContext?.currentSpan?.createChildSpan({
@@ -1812,7 +1812,7 @@ export const contentCleanerTool = createTool({
         'Clean HTML content by removing unwanted elements with enhanced JSDOM processing and security.',
     inputSchema: contentCleanerInputSchema,
     outputSchema: contentCleanerOutputSchema,
-    execute: async ({ context, writer, tracingContext }) => {
+    execute: async ({ context, writer, tracingContext }: { context: { html: string; removeScripts?: boolean; removeStyles?: boolean; removeComments?: boolean; preserveStructure?: boolean }, writer?: any, tracingContext?: TracingContext }) => {
       await writer?.write({ type: 'progress', data: { message: '🧹 Starting content cleaning...' } });
         toolCallCounters.set('content-cleaner', (toolCallCounters.get('content-cleaner') ?? 0) + 1)
         const cleanSpan = tracingContext?.currentSpan?.createChildSpan({

src/mastra/tools/write-note.ts

@@ -3,7 +3,7 @@ import { AISpanType } from "@mastra/core/ai-tracing";
 import { z } from "zod";
 import path from "node:path";
 import fs from "fs/promises";
-
+import type { TracingContext } from '@mastra/core/ai-tracing';
 
 const NOTES_DIR = path.join(process.cwd(), "notes");
 
@@ -21,7 +21,7 @@ export const writeNoteTool = createTool({
       .describe("The markdown content of the note."),
   }),
   outputSchema: z.string().nonempty(),
-  execute: async ({ context, tracingContext }) => {
+  execute: async ({ context, tracingContext }: { context: { title: string; content: string }, tracingContext?: TracingContext }) => {
     const startTime = Date.now();
     const span = tracingContext?.currentSpan?.createChildSpan({
       type: AISpanType.TOOL_CALL,