@mastra/rag 0.1.18-alpha.4 → 0.1.18-alpha.5

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/rag@0.1.18-alpha.4 build /home/runner/work/mastra/mastra/packages/rag
+> @mastra/rag@0.1.18-alpha.5 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 28135ms
+TSC ⚡️ Build success in 25972ms
 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 40051ms
+DTS ⚡️ Build success in 37741ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 93.37 KB
-CJS ⚡️ Build success in 2052ms
-ESM dist/index.js 92.65 KB
-ESM ⚡️ Build success in 2052ms
+CJS dist/index.cjs 96.18 KB
+CJS ⚡️ Build success in 2138ms
+ESM dist/index.js 95.46 KB
+ESM ⚡️ Build success in 2138ms

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @mastra/rag
 
+## 0.1.18-alpha.5
+
+### Patch Changes
+
+- 18d0758: [MASTRA-3073] added error logging and more debug logs to vector-query and graph-rag
+
 ## 0.1.18-alpha.4
 
 ### Patch Changes

package/dist/index.cjs

@@ -1904,10 +1904,24 @@ 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
-      if (vectorStore) {
+      if (!logger) {
+        console.warn(
+          "[GraphRAGTool] Logger not initialized: no debug or error logs will be recorded for this tool execution."
+        );
+      }
+      if (logger) {
+        logger.debug("[GraphRAGTool] execute called with:", { queryText, topK, filter });
+      }
+      try {
+        const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
+        const vectorStore = mastra?.getVector(vectorStoreName);
+        if (!vectorStore) {
+          if (logger) {
+            logger.error("Vector store not found", { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
         let queryFilter = {};
         if (enableFilter) {
           queryFilter = (() => {
@@ -1922,7 +1936,7 @@ var createGraphRAGTool = ({
           })();
         }
         if (logger) {
-          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+          logger.debug("Prepared vector query parameters:", { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,
@@ -1933,6 +1947,9 @@ var createGraphRAGTool = ({
           topK: topKValue,
           includeVectors: true
         });
+        if (logger) {
+          logger.debug("vectorQuerySearch returned results", { count: results.length });
+        }
         if (!isInitialized) {
           const chunks = results.map((result) => ({
             text: result?.metadata?.text,
@@ -1941,8 +1958,13 @@ var createGraphRAGTool = ({
           const embeddings = results.map((result) => ({
             vector: result.vector || []
           }));
+          if (logger) {
+            logger.debug("Initializing graph", { chunkCount: chunks.length, embeddingCount: embeddings.length });
+          }
           graphRag.createGraph(chunks, embeddings);
           isInitialized = true;
+        } else if (logger) {
+          logger.debug("Graph already initialized, skipping graph construction");
         }
         const rerankedResults = graphRag.query({
           query: queryEmbedding,
@@ -1950,14 +1972,26 @@ var createGraphRAGTool = ({
           randomWalkSteps: graphOptions.randomWalkSteps,
           restartProb: graphOptions.restartProb
         });
+        if (logger) {
+          logger.debug("GraphRAG query returned results", { count: rerankedResults.length });
+        }
         const relevantChunks = rerankedResults.map((result) => result.content);
+        if (logger) {
+          logger.debug("Returning relevant context chunks", { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks
         };
+      } catch (err) {
+        if (logger) {
+          logger.error("Unexpected error in VectorQueryTool execute", {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : void 0
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: []
-      };
     }
   });
 };
@@ -1988,10 +2022,24 @@ 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
-      if (vectorStore) {
+      if (!logger) {
+        console.warn(
+          "[VectorQueryTool] Logger not initialized: no debug or error logs will be recorded for this tool execution."
+        );
+      }
+      if (logger) {
+        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter });
+      }
+      try {
+        const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
+        const vectorStore = mastra?.getVector(vectorStoreName);
+        if (!vectorStore) {
+          if (logger) {
+            logger.error("Vector store not found", { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
         let queryFilter = {};
         if (enableFilter && filter) {
           queryFilter = (() => {
@@ -2006,7 +2054,7 @@ var createVectorQueryTool = ({
           })();
         }
         if (logger) {
-          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter });
         }
         const { results } = await vectorQuerySearch({
           indexName,
@@ -2016,22 +2064,43 @@ var createVectorQueryTool = ({
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue
         });
+        if (logger) {
+          logger.debug("vectorQuerySearch returned results", { count: results.length });
+        }
         if (reranker) {
+          if (logger) {
+            logger.debug("Reranking results", { rerankerModel: reranker.model, rerankerOptions: reranker.options });
+          }
           const rerankedResults = await rerank(results, queryText, reranker.model, {
             ...reranker.options,
             topK: reranker.options?.topK || topKValue
           });
+          if (logger) {
+            logger.debug("Reranking complete", { rerankedCount: rerankedResults.length });
+          }
           const relevantChunks2 = rerankedResults.map(({ result }) => result?.metadata);
+          if (logger) {
+            logger.debug("Returning reranked relevant context chunks", { count: relevantChunks2.length });
+          }
           return { relevantContext: relevantChunks2 };
         }
         const relevantChunks = results.map((result) => result?.metadata);
+        if (logger) {
+          logger.debug("Returning relevant context chunks", { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks
         };
+      } catch (err) {
+        if (logger) {
+          logger.error("Unexpected error in VectorQueryTool execute", {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : void 0
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: []
-      };
     }
   });
 };

package/dist/index.js

@@ -1902,10 +1902,24 @@ 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
-      if (vectorStore) {
+      if (!logger) {
+        console.warn(
+          "[GraphRAGTool] Logger not initialized: no debug or error logs will be recorded for this tool execution."
+        );
+      }
+      if (logger) {
+        logger.debug("[GraphRAGTool] execute called with:", { queryText, topK, filter });
+      }
+      try {
+        const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
+        const vectorStore = mastra?.getVector(vectorStoreName);
+        if (!vectorStore) {
+          if (logger) {
+            logger.error("Vector store not found", { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
         let queryFilter = {};
         if (enableFilter) {
           queryFilter = (() => {
@@ -1920,7 +1934,7 @@ var createGraphRAGTool = ({
           })();
         }
         if (logger) {
-          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+          logger.debug("Prepared vector query parameters:", { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,
@@ -1931,6 +1945,9 @@ var createGraphRAGTool = ({
           topK: topKValue,
           includeVectors: true
         });
+        if (logger) {
+          logger.debug("vectorQuerySearch returned results", { count: results.length });
+        }
         if (!isInitialized) {
           const chunks = results.map((result) => ({
             text: result?.metadata?.text,
@@ -1939,8 +1956,13 @@ var createGraphRAGTool = ({
           const embeddings = results.map((result) => ({
             vector: result.vector || []
           }));
+          if (logger) {
+            logger.debug("Initializing graph", { chunkCount: chunks.length, embeddingCount: embeddings.length });
+          }
           graphRag.createGraph(chunks, embeddings);
           isInitialized = true;
+        } else if (logger) {
+          logger.debug("Graph already initialized, skipping graph construction");
         }
         const rerankedResults = graphRag.query({
           query: queryEmbedding,
@@ -1948,14 +1970,26 @@ var createGraphRAGTool = ({
           randomWalkSteps: graphOptions.randomWalkSteps,
           restartProb: graphOptions.restartProb
         });
+        if (logger) {
+          logger.debug("GraphRAG query returned results", { count: rerankedResults.length });
+        }
         const relevantChunks = rerankedResults.map((result) => result.content);
+        if (logger) {
+          logger.debug("Returning relevant context chunks", { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks
         };
+      } catch (err) {
+        if (logger) {
+          logger.error("Unexpected error in VectorQueryTool execute", {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : void 0
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: []
-      };
     }
   });
 };
@@ -1986,10 +2020,24 @@ 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
-      if (vectorStore) {
+      if (!logger) {
+        console.warn(
+          "[VectorQueryTool] Logger not initialized: no debug or error logs will be recorded for this tool execution."
+        );
+      }
+      if (logger) {
+        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter });
+      }
+      try {
+        const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
+        const vectorStore = mastra?.getVector(vectorStoreName);
+        if (!vectorStore) {
+          if (logger) {
+            logger.error("Vector store not found", { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
         let queryFilter = {};
         if (enableFilter && filter) {
           queryFilter = (() => {
@@ -2004,7 +2052,7 @@ var createVectorQueryTool = ({
           })();
         }
         if (logger) {
-          logger.debug("Using this filter and topK:", { queryFilter, topK: topKValue });
+          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter });
         }
         const { results } = await vectorQuerySearch({
           indexName,
@@ -2014,22 +2062,43 @@ var createVectorQueryTool = ({
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue
         });
+        if (logger) {
+          logger.debug("vectorQuerySearch returned results", { count: results.length });
+        }
         if (reranker) {
+          if (logger) {
+            logger.debug("Reranking results", { rerankerModel: reranker.model, rerankerOptions: reranker.options });
+          }
           const rerankedResults = await rerank(results, queryText, reranker.model, {
             ...reranker.options,
             topK: reranker.options?.topK || topKValue
           });
+          if (logger) {
+            logger.debug("Reranking complete", { rerankedCount: rerankedResults.length });
+          }
           const relevantChunks2 = rerankedResults.map(({ result }) => result?.metadata);
+          if (logger) {
+            logger.debug("Returning reranked relevant context chunks", { count: relevantChunks2.length });
+          }
           return { relevantContext: relevantChunks2 };
         }
         const relevantChunks = results.map((result) => result?.metadata);
+        if (logger) {
+          logger.debug("Returning relevant context chunks", { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks
         };
+      } catch (err) {
+        if (logger) {
+          logger.error("Unexpected error in VectorQueryTool execute", {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : void 0
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: []
-      };
     }
   });
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/rag",
-  "version": "0.1.18-alpha.4",
+  "version": "0.1.18-alpha.5",
   "description": "",
   "type": "module",
   "main": "dist/index.js",

package/src/tools/graph-rag.ts

@@ -64,16 +64,31 @@ export const 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
+      if (!logger) {
+        console.warn(
+          '[GraphRAGTool] Logger not initialized: no debug or error logs will be recorded for this tool execution.',
+        );
+      }
+      if (logger) {
+        logger.debug('[GraphRAGTool] execute called with:', { queryText, topK, filter });
+      }
+      try {
+        const topKValue =
+          typeof topK === 'number' && !isNaN(topK)
+            ? topK
+            : typeof topK === 'string' && !isNaN(Number(topK))
+              ? Number(topK)
+              : 10;
+        const vectorStore = mastra?.getVector(vectorStoreName);
+
+        if (!vectorStore) {
+          if (logger) {
+            logger.error('Vector store not found', { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
 
-      if (vectorStore) {
         let queryFilter = {};
         if (enableFilter) {
           queryFilter = (() => {
@@ -89,7 +104,7 @@ export const createGraphRAGTool = ({
           })();
         }
         if (logger) {
-          logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
+          logger.debug('Prepared vector query parameters:', { queryFilter, topK: topKValue });
         }
         const { results, queryEmbedding } = await vectorQuerySearch({
           indexName,
@@ -100,6 +115,9 @@ export const createGraphRAGTool = ({
           topK: topKValue,
           includeVectors: true,
         });
+        if (logger) {
+          logger.debug('vectorQuerySearch returned results', { count: results.length });
+        }
 
         // Initialize graph if not done yet
         if (!isInitialized) {
@@ -112,8 +130,13 @@ export const createGraphRAGTool = ({
             vector: result.vector || [],
           }));
 
+          if (logger) {
+            logger.debug('Initializing graph', { chunkCount: chunks.length, embeddingCount: embeddings.length });
+          }
           graphRag.createGraph(chunks, embeddings);
           isInitialized = true;
+        } else if (logger) {
+          logger.debug('Graph already initialized, skipping graph construction');
         }
 
         // Get reranked results using GraphRAG
@@ -123,16 +146,27 @@ export const createGraphRAGTool = ({
           randomWalkSteps: graphOptions.randomWalkSteps,
           restartProb: graphOptions.restartProb,
         });
-
+        if (logger) {
+          logger.debug('GraphRAG query returned results', { count: rerankedResults.length });
+        }
         // Extract and combine relevant chunks
         const relevantChunks = rerankedResults.map(result => result.content);
+        if (logger) {
+          logger.debug('Returning relevant context chunks', { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks,
         };
+      } catch (err) {
+        if (logger) {
+          logger.error('Unexpected error in VectorQueryTool execute', {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : undefined,
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: [],
-      };
     },
   });
 };

package/src/tools/vector-query.ts

@@ -52,18 +52,32 @@ export const 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?.getVector(vectorStoreName);
       const logger = mastra?.getLogger();
+      if (!logger) {
+        console.warn(
+          '[VectorQueryTool] Logger not initialized: no debug or error logs will be recorded for this tool execution.',
+        );
+      }
+      if (logger) {
+        logger.debug('[VectorQueryTool] execute called with:', { queryText, topK, filter });
+      }
+      try {
+        const topKValue =
+          typeof topK === 'number' && !isNaN(topK)
+            ? topK
+            : typeof topK === 'string' && !isNaN(Number(topK))
+              ? Number(topK)
+              : 10;
 
-      // Get relevant chunks from the vector database
-      if (vectorStore) {
+        const vectorStore = mastra?.getVector(vectorStoreName);
+
+        if (!vectorStore) {
+          if (logger) {
+            logger.error('Vector store not found', { vectorStoreName });
+          }
+          return { relevantContext: [] };
+        }
+        // Get relevant chunks from the vector database
         let queryFilter = {};
         if (enableFilter && filter) {
           queryFilter = (() => {
@@ -79,7 +93,7 @@ export const createVectorQueryTool = ({
           })();
         }
         if (logger) {
-          logger.debug('Using this filter and topK:', { queryFilter, topK: topKValue });
+          logger.debug('Prepared vector query parameters', { queryText, topK: topKValue, queryFilter });
         }
 
         const { results } = await vectorQuerySearch({
@@ -90,23 +104,44 @@ export const createVectorQueryTool = ({
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : undefined,
           topK: topKValue,
         });
+        if (logger) {
+          logger.debug('vectorQuerySearch returned results', { count: results.length });
+        }
         if (reranker) {
+          if (logger) {
+            logger.debug('Reranking results', { rerankerModel: reranker.model, rerankerOptions: reranker.options });
+          }
           const rerankedResults = await rerank(results, queryText, reranker.model, {
             ...reranker.options,
             topK: reranker.options?.topK || topKValue,
           });
+          if (logger) {
+            logger.debug('Reranking complete', { rerankedCount: rerankedResults.length });
+          }
           const relevantChunks = rerankedResults.map(({ result }) => result?.metadata);
+          if (logger) {
+            logger.debug('Returning reranked relevant context chunks', { count: relevantChunks.length });
+          }
           return { relevantContext: relevantChunks };
         }
 
         const relevantChunks = results.map(result => result?.metadata);
+        if (logger) {
+          logger.debug('Returning relevant context chunks', { count: relevantChunks.length });
+        }
         return {
           relevantContext: relevantChunks,
         };
+      } catch (err) {
+        if (logger) {
+          logger.error('Unexpected error in VectorQueryTool execute', {
+            error: err,
+            errorMessage: err instanceof Error ? err.message : String(err),
+            errorStack: err instanceof Error ? err.stack : undefined,
+          });
+        }
+        return { relevantContext: [] };
       }
-      return {
-        relevantContext: [],
-      };
     },
   });
 };