feat: enhance tool imports and add UI types

- Updated imports in various tool files to include InferUITool for type inference.
- Added UI type exports for multiple tools including:
  - AlphaVantageCryptoUITool
  - ArxivUITool
  - BrowserToolUITool
  - CsvToJsonUITool
  - DataValidatorJSONUITool
  - EditorAgentUITool
  - EvaluateResultUITool
  - ExecaUITool
  - ExtractLearningsUITool
  - FinancialChartTools
  - FinnhubTools
  - PgSqlTool
  - PnpmBuildUITool
  - PolygonStockQuotesUITool
  - WeatherUITool
  - WebScraperUITool
  - WriteNoteUITool

Author: ssdeanx

app/chat/providers/chat-context.tsx

@@ -99,9 +99,10 @@ export function ChatProvider({
   const [pendingConfirmations, setPendingConfirmations] = useState<PendingConfirmation[]>([])
 
   const {
+    regenerate,
+    stop,
     messages,
     sendMessage: aiSendMessage,
-    stop,
     status: aiStatus,
     error: aiError,
   } = useChat({
@@ -111,9 +112,14 @@ export function ChatProvider({
       prepareSendMessagesRequest({ messages }) {
         return {
           id: selectedAgent,
-          messages: messages,
+          messages,
+          memory: {
+            thread: "user-1",
+            resource: "user-1",
+          },
           requestMetadata: {
-            agentId: selectedAgent
+            agentId: selectedAgent,
+            resourceId: selectedAgent
           },
           body: {
             messages,

lib/utils.ts

@@ -2,5 +2,5 @@ import { type ClassValue, clsx } from "clsx"
 import { twMerge } from "tailwind-merge"
 
 export function cn(...inputs: ClassValue[]) {
-    return twMerge(clsx(inputs))
-}
+  return twMerge(clsx(inputs))
+}

src/mastra/tools/alpha-vantage.tool.ts

@@ -1,5 +1,5 @@
 import { AISpanType, InternalSpans } from "@mastra/core/ai-tracing";
-import { createTool } from "@mastra/core/tools";
+import { InferUITool, createTool } from "@mastra/core/tools";
 import { z } from "zod";
 
 /**
@@ -548,3 +548,7 @@ export const alphaVantageTool = createTool({
     }
   }
 });
+
+export type AlphaVantageCryptoUITool = InferUITool<typeof alphaVantageCryptoTool>;
+export type AlphaVantageStockUITool = InferUITool<typeof alphaVantageStockTool>;
+export type AlphaVantageUITool = InferUITool<typeof alphaVantageTool>;

src/mastra/tools/arxiv.tool.ts

@@ -1,5 +1,5 @@
 import { AISpanType, InternalSpans } from "@mastra/core/ai-tracing";
-import { createTool } from "@mastra/core/tools";
+import { InferUITool, createTool } from "@mastra/core/tools";
 import { z } from "zod";
 
 // In-memory counter to track tool calls per request
@@ -274,6 +274,8 @@ export const arxivTool = createTool({
   }
 });
 
+export type ArxivUITool = InferUITool<typeof arxivTool>;
+
 /**
  * ArXiv PDF Parser Tool
  *
@@ -442,6 +444,8 @@ export const arxivPdfParserTool = createTool({
   }
 });
 
+export type ArxivPdfParserUITool = InferUITool<typeof arxivPdfParserTool>;
+
 /**
  * ArXiv Paper Downloader Tool
  *
@@ -580,6 +584,8 @@ export const arxivPaperDownloaderTool = createTool({
   }
 });
 
+export type ArxivPaperDownloaderUITool = InferUITool<typeof arxivPaperDownloaderTool>;
+
 /**
  * Simple XML parser for ArXiv API response
  * In production, consider using a proper XML parsing library

src/mastra/tools/browser-tool.ts

@@ -1,4 +1,7 @@
-import { createTool } from '@mastra/core/tools';
+/*
+ * Not working
+*/
+import { InferUITool, createTool } from "@mastra/core/tools";
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing';
 import { MDocument } from '@mastra/rag';
 import { chromium, Browser } from 'playwright-core';
@@ -508,3 +511,12 @@ export const monitorPageTool = createTool({
     }
   },
 });
+
+export type BrowserToolUITool = InferUITool<typeof browserTool>;
+export type ScreenshotToolUITool = InferUITool<typeof screenshotTool>;
+export type PdfGeneratorToolUITool = InferUITool<typeof pdfGeneratorTool>;
+export type ClickAndExtractToolUITool = InferUITool<typeof clickAndExtractTool>;
+export type FillFormToolUITool = InferUITool<typeof fillFormTool>;
+export type GoogleSearchUITool = InferUITool<typeof googleSearch>;
+export type ExtractTablesToolUITool = InferUITool<typeof extractTablesTool>;
+export type MonitorPageToolUITool = InferUITool<typeof monitorPageTool>;

src/mastra/tools/calendar-tool.ts

@@ -1,4 +1,4 @@
-import { createTool } from '@mastra/core/tools';
+import { InferUITools, createTool } from "@mastra/core/tools";
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing';
 import { execSync } from 'child_process';
 import { z } from 'zod';

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

@@ -1,5 +1,5 @@
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing'
-import { createTool } from '@mastra/core/tools'
+import { InferUITool, createTool } from "@mastra/core/tools";
 import { z } from 'zod'
 import { log } from '../config/logger'
 
@@ -208,3 +208,7 @@ export const copywriterTool = createTool({
         }
     },
 })
+
+// InferUITools expects a ToolSet (an object mapping ids to tools), not a single Tool.
+// Provide an object with the tool id as the key to satisfy the constraint.
+export type CopywriterUITool = InferUITool<typeof copywriterTool>;

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

@@ -1,5 +1,5 @@
 import { AISpanType, InternalSpans } from "@mastra/core/ai-tracing";
-import { createTool } from "@mastra/core/tools";
+import { InferUITool, createTool } from "@mastra/core/tools";
 import { parse } from "csv-parse/sync";
 import * as fs from "node:fs/promises";
 import { z } from "zod";
@@ -82,3 +82,5 @@ export const csvToJsonTool = createTool({
     }
   },
 });
+
+export type CsvToJsonUITool = InferUITool<typeof csvToJsonTool>;

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

@@ -14,7 +14,7 @@
 // approvalDate: 9/22
 
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing'
-import { createTool } from '@mastra/core/tools'
+import { InferUITool, createTool } from "@mastra/core/tools";
 import * as fs from 'fs/promises'
 import * as path from 'path'
 import { pipeline } from 'stream/promises'
@@ -91,6 +91,8 @@ export const readDataFileTool = createTool({
     },
 })
 
+export type ReadDataFileUITool = InferUITool<typeof readDataFileTool>;
+
 export const writeDataFileTool = createTool({
     id: 'write-data-file',
     description:
@@ -150,6 +152,8 @@ export const writeDataFileTool = createTool({
     },
 })
 
+export type WriteDataFileUITool = InferUITool<typeof writeDataFileTool>;
+
 export const deleteDataFileTool = createTool({
     id: 'delete-data-file',
     description: 'Deletes a file within the data directory.',
@@ -195,6 +199,8 @@ export const deleteDataFileTool = createTool({
     },
 })
 
+export type DeleteDataFileUITool = InferUITool<typeof deleteDataFileTool>;
+
 export const listDataDirTool = createTool({
     id: 'list-data-directory',
     description:
@@ -241,6 +247,9 @@ export const listDataDirTool = createTool({
         }
     },
 })
+
+export type ListDataDirUITool = InferUITool<typeof listDataDirTool>;
+
 export const copyDataFileTool = createTool({
     id: 'copy-data-file',
     description: 'Copies a file within the data directory to a new location.',
@@ -300,6 +309,8 @@ export const copyDataFileTool = createTool({
     },
 })
 
+export type CopyDataFileUITool = InferUITool<typeof copyDataFileTool>;
+
 export const moveDataFileTool = createTool({
     id: 'move-data-file',
     description: 'Moves or renames a file within the data directory.',
@@ -359,6 +370,8 @@ export const moveDataFileTool = createTool({
     },
 })
 
+export type MoveDataFileUITool = InferUITool<typeof moveDataFileTool>;
+
 export const searchDataFilesTool = createTool({
     id: 'search-data-files',
     description:
@@ -463,6 +476,8 @@ export const searchDataFilesTool = createTool({
     },
 })
 
+export type SearchDataFilesUITool = InferUITool<typeof searchDataFilesTool>;
+
 export const getDataFileInfoTool = createTool({
     id: 'get-data-file-info',
     description:
@@ -525,6 +540,7 @@ export const getDataFileInfoTool = createTool({
     },
 })
 
+export type GetDataFileInfoUITool = InferUITool<typeof getDataFileInfoTool>;
 export const createDataDirTool = createTool({
     id: 'create-data-directory',
     description: 'Creates a new directory within the data directory.',
@@ -568,6 +584,8 @@ export const createDataDirTool = createTool({
     },
 })
 
+export type CreateDataDirUITool = InferUITool<typeof createDataDirTool>;
+
 export const removeDataDirTool = createTool({
     id: 'remove-data-directory',
     description: 'Removes an empty directory within the data directory.',
@@ -617,6 +635,8 @@ export const removeDataDirTool = createTool({
     },
 })
 
+export type RemoveDataDirUITool = InferUITool<typeof removeDataDirTool>;
+
 export const archiveDataTool = createTool({
     id: 'archive-data',
     description:
@@ -682,6 +702,8 @@ export const archiveDataTool = createTool({
     },
 })
 
+export type ArchiveDataUITool = InferUITool<typeof archiveDataTool>;
+
 export const backupDataTool = createTool({
     id: 'backup-data',
     description:
@@ -749,3 +771,5 @@ export const backupDataTool = createTool({
         }
     },
 })
+
+export type BackupDataUITool = InferUITool<typeof backupDataTool>;

src/mastra/tools/data-processing-tools.ts

@@ -1,5 +1,5 @@
 import { AISpanType, InternalSpans } from '@mastra/core/ai-tracing'
-import { createTool } from '@mastra/core/tools'
+import { InferUITool, createTool } from "@mastra/core/tools";
 import * as fs from 'fs/promises'
 import * as path from 'path'
 import { DOMParser } from 'xmldom'
@@ -1232,3 +1232,14 @@ function jsonToXML(data: JSONLike, rootElement = 'root'): string {
 
     return objectToXML(data, rootElement)
 }
+
+
+export type ReadCSVDataToolUITool = InferUITool<typeof readCSVDataTool>;
+export type  ImageToCSVUITool = InferUITool<typeof imageToCSVTool>;
+export type DataProcessingUITool = InferUITool<typeof convertDataFormatTool>;
+export type ExcalidrawValidationUITool = InferUITool<typeof validateExcalidrawTool>;
+export type SVGProcessingUITool = InferUITool<typeof processSVGTool>;
+export type XMLProcessingUITool = InferUITool<typeof processXMLTool>;
+export type DataValidationUITool = InferUITool<typeof validateDataTool>;
+
+