@mastra/rag 0.1.13-alpha.0 → 0.1.13-alpha.2

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/rag@0.1.13-alpha.0 build /home/runner/work/mastra/mastra/packages/rag
+> @mastra/rag@0.1.13-alpha.2 build /home/runner/work/mastra/mastra/packages/rag
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 27075ms
+TSC ⚡️ Build success in 27290ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/packages/rag/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/packages/rag/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 38725ms
+DTS ⚡️ Build success in 36864ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 92.25 KB
-ESM ⚡️ Build success in 1988ms
-CJS dist/index.cjs 92.97 KB
-CJS ⚡️ Build success in 1989ms
+ESM dist/index.js 92.31 KB
+ESM ⚡️ Build success in 1806ms
+CJS dist/index.cjs 93.03 KB
+CJS ⚡️ Build success in 1806ms

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @mastra/rag
 
+## 0.1.13-alpha.2
+
+### Patch Changes
+
+- ac30427: Updated inputschema for vectorquery tool
+
+## 0.1.13-alpha.1
+
+### Patch Changes
+
+- Updated dependencies [6794797]
+  - @mastra/core@0.6.4-alpha.1
+
 ## 0.1.13-alpha.0
 
 ### Patch Changes

package/dist/index.cjs

@@ -1878,11 +1878,11 @@ var createGraphRAGTool = ({
   let isInitialized = false;
   const baseSchema = {
     queryText: zod.z.string().describe(queryTextDescription),
-    topK: zod.z.any().describe(topKDescription)
+    topK: zod.z.coerce.number().describe(topKDescription)
   };
   const inputSchema = enableFilter ? zod.z.object({
     ...baseSchema,
-    filter: zod.z.string().describe(filterDescription)
+    filter: zod.z.coerce.string().describe(filterDescription)
   }).passthrough() : zod.z.object(baseSchema).passthrough();
   return tools.createTool({
     id: toolId,
@@ -1893,7 +1893,8 @@ var createGraphRAGTool = ({
     description: toolDescription,
     execute: async ({ context: { queryText, topK, filter }, mastra }) => {
       const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
       if (vectorStore) {
         let queryFilter = {};
         if (enableFilter) {
@@ -1901,15 +1902,15 @@ var createGraphRAGTool = ({
             try {
               return typeof filter === "string" ? JSON.parse(filter) : filter;
             } catch (error) {
-              if (mastra.logger) {
-                mastra.logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
+              if (logger) {
+                logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,
@@ -1961,11 +1962,11 @@ var createVectorQueryTool = ({
   const toolDescription = description || defaultVectorQueryDescription();
   const baseSchema = {
     queryText: zod.z.string().describe(queryTextDescription),
-    topK: zod.z.any().describe(topKDescription)
+    topK: zod.z.coerce.number().describe(topKDescription)
   };
   const inputSchema = enableFilter ? zod.z.object({
     ...baseSchema,
-    filter: zod.z.string().describe(filterDescription)
+    filter: zod.z.coerce.string().describe(filterDescription)
   }).passthrough() : zod.z.object(baseSchema).passthrough();
   return tools.createTool({
     id: toolId,
@@ -1976,7 +1977,8 @@ var createVectorQueryTool = ({
     description: toolDescription,
     execute: async ({ context: { queryText, topK, filter }, mastra }) => {
       const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
       if (vectorStore) {
         let queryFilter = {};
         if (enableFilter && filter) {
@@ -1984,15 +1986,15 @@ var createVectorQueryTool = ({
             try {
               return typeof filter === "string" ? JSON.parse(filter) : filter;
             } catch (error) {
-              if (mastra.logger) {
-                mastra.logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
+              if (logger) {
+                logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
         }
         const { results } = await vectorQuerySearch({
           indexName,

package/dist/index.js

@@ -1876,11 +1876,11 @@ var createGraphRAGTool = ({
   let isInitialized = false;
   const baseSchema = {
     queryText: z.string().describe(queryTextDescription),
-    topK: z.any().describe(topKDescription)
+    topK: z.coerce.number().describe(topKDescription)
   };
   const inputSchema = enableFilter ? z.object({
     ...baseSchema,
-    filter: z.string().describe(filterDescription)
+    filter: z.coerce.string().describe(filterDescription)
   }).passthrough() : z.object(baseSchema).passthrough();
   return createTool({
     id: toolId,
@@ -1891,7 +1891,8 @@ var createGraphRAGTool = ({
     description: toolDescription,
     execute: async ({ context: { queryText, topK, filter }, mastra }) => {
       const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
       if (vectorStore) {
         let queryFilter = {};
         if (enableFilter) {
@@ -1899,15 +1900,15 @@ var createGraphRAGTool = ({
             try {
               return typeof filter === "string" ? JSON.parse(filter) : filter;
             } catch (error) {
-              if (mastra.logger) {
-                mastra.logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
+              if (logger) {
+                logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,
@@ -1959,11 +1960,11 @@ var createVectorQueryTool = ({
   const toolDescription = description || defaultVectorQueryDescription();
   const baseSchema = {
     queryText: z.string().describe(queryTextDescription),
-    topK: z.any().describe(topKDescription)
+    topK: z.coerce.number().describe(topKDescription)
   };
   const inputSchema = enableFilter ? z.object({
     ...baseSchema,
-    filter: z.string().describe(filterDescription)
+    filter: z.coerce.string().describe(filterDescription)
   }).passthrough() : z.object(baseSchema).passthrough();
   return createTool({
     id: toolId,
@@ -1974,7 +1975,8 @@ var createVectorQueryTool = ({
     description: toolDescription,
     execute: async ({ context: { queryText, topK, filter }, mastra }) => {
       const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
       if (vectorStore) {
         let queryFilter = {};
         if (enableFilter && filter) {
@@ -1982,15 +1984,15 @@ var createVectorQueryTool = ({
             try {
               return typeof filter === "string" ? JSON.parse(filter) : filter;
             } catch (error) {
-              if (mastra.logger) {
-                mastra.logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
+              if (logger) {
+                logger.warn("Failed to parse filter as JSON, using empty filter", { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
         }
         const { results } = await vectorQuerySearch({
           indexName,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/rag",
-  "version": "0.1.13-alpha.0",
+  "version": "0.1.13-alpha.2",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -29,7 +29,7 @@
     "node-html-better-parser": "^1.4.7",
     "pathe": "^2.0.3",
     "zod": "^3.24.2",
-    "@mastra/core": "^0.6.4-alpha.0"
+    "@mastra/core": "^0.6.4-alpha.1"
   },
   "peerDependencies": {
     "ai": "^4.0.0"

package/src/tools/graph-rag.ts

@@ -46,13 +46,13 @@ export const createGraphRAGTool = ({
 
   const baseSchema = {
     queryText: z.string().describe(queryTextDescription),
-    topK: z.any().describe(topKDescription),
+    topK: z.coerce.number().describe(topKDescription),
   };
   const inputSchema = enableFilter
     ? z
         .object({
           ...baseSchema,
-          filter: z.string().describe(filterDescription),
+          filter: z.coerce.string().describe(filterDescription),
         })
         .passthrough()
     : z.object(baseSchema).passthrough();
@@ -70,7 +70,8 @@ export const createGraphRAGTool = ({
           : typeof topK === 'string' && !isNaN(Number(topK))
             ? Number(topK)
             : 10;
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
 
       if (vectorStore) {
         let queryFilter = {};
@@ -80,15 +81,15 @@ export const createGraphRAGTool = ({
               return typeof filter === 'string' ? JSON.parse(filter) : filter;
             } catch (error) {
               // Log the error and use empty object
-              if (mastra.logger) {
-                mastra.logger.warn('Failed to parse filter as JSON, using empty filter', { filter, error });
+              if (logger) {
+                logger.warn('Failed to parse filter as JSON, using empty filter', { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,

package/src/tools/vector-query.test.ts

@@ -28,11 +28,21 @@ describe('createVectorQueryTool', () => {
         // Mock vector store methods
       },
     },
+    getVector: vi.fn(() => ({
+      testStore: {
+        // Mock vector store methods
+      },
+    })),
     logger: {
       debug: vi.fn(),
       warn: vi.fn(),
       info: vi.fn(),
     },
+    getLogger: vi.fn(() => ({
+      debug: vi.fn(),
+      warn: vi.fn(),
+      info: vi.fn(),
+    })),
   };
 
   beforeEach(() => {
@@ -78,7 +88,7 @@ describe('createVectorQueryTool', () => {
       // Get the Zod schema
       const schema = tool.__inputSchema;
 
-      // Test various filter inputs that should all work
+      // Test various filter inputs that should coerce to string
       const testCases = [
         // String inputs
         { filter: '{"field": "value"}' },
@@ -86,9 +96,6 @@ describe('createVectorQueryTool', () => {
         { filter: 'simple-string' },
         // Empty
         { filter: '' },
-      ];
-
-      const invalidTestCases = [
         { filter: { field: 'value' } },
         { filter: {} },
         { filter: 123 },
@@ -106,16 +113,6 @@ describe('createVectorQueryTool', () => {
         ).not.toThrow();
       });
 
-      invalidTestCases.forEach(({ filter }) => {
-        expect(() =>
-          schema.parse({
-            queryText: 'test query',
-            topK: 5,
-            filter,
-          }),
-        ).toThrow();
-      });
-
       // Verify that all parsed values are strings
       testCases.forEach(({ filter }) => {
         const result = schema.parse({

package/src/tools/vector-query.ts

@@ -34,13 +34,13 @@ export const createVectorQueryTool = ({
   // Create base schema with required fields
   const baseSchema = {
     queryText: z.string().describe(queryTextDescription),
-    topK: z.any().describe(topKDescription),
+    topK: z.coerce.number().describe(topKDescription),
   };
   const inputSchema = enableFilter
     ? z
         .object({
           ...baseSchema,
-          filter: z.string().describe(filterDescription),
+          filter: z.coerce.string().describe(filterDescription),
         })
         .passthrough()
     : z.object(baseSchema).passthrough();
@@ -59,7 +59,8 @@ export const createVectorQueryTool = ({
             ? Number(topK)
             : 10;
 
-      const vectorStore = mastra?.vectors?.[vectorStoreName];
+      const vectorStore = mastra?.getVector(vectorStoreName);
+      const logger = mastra?.getLogger();
 
       // Get relevant chunks from the vector database
       if (vectorStore) {
@@ -70,15 +71,15 @@ export const createVectorQueryTool = ({
               return typeof filter === 'string' ? JSON.parse(filter) : filter;
             } catch (error) {
               // Log the error and use empty object
-              if (mastra.logger) {
-                mastra.logger.warn('Failed to parse filter as JSON, using empty filter', { filter, error });
+              if (logger) {
+                logger.warn('Failed to parse filter as JSON, using empty filter', { filter, error });
               }
               return {};
             }
           })();
         }
-        if (mastra.logger) {
-          mastra.logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
+        if (logger) {
+          logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
         }
 
         const { results } = await vectorQuerySearch({