@mastra/rag 2.0.0-beta.4 → 2.0.0-beta.6

package/dist/index.cjs

@@ -21,7 +21,7 @@ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __knownSymbol = (name14, symbol15) => (symbol15 = Symbol[name14]) ? symbol15 : Symbol.for("Symbol." + name14);
+var __knownSymbol = (name14, symbol15) => (symbol15 = Symbol[name14]) ? symbol15 : /* @__PURE__ */ Symbol.for("Symbol." + name14);
 var __typeError = (msg) => {
   throw TypeError(msg);
 };
@@ -902,7 +902,7 @@ function loadApiKey({
   }
   return apiKey;
 }
-var validatorSymbol = Symbol.for("vercel.ai.validator");
+var validatorSymbol = /* @__PURE__ */ Symbol.for("vercel.ai.validator");
 function validator(validate) {
   return { [validatorSymbol]: true, validate };
 }
@@ -6985,6 +6985,7 @@ var vectorQuerySearch = async ({
       model,
       value: queryText,
       maxRetries,
+      // Type assertion needed: providerOptions type is a union, but embedV3 expects specific version
       ...providerOptions && { providerOptions }
     });
   } else if (model.specificationVersion === "v2") {
@@ -6992,6 +6993,7 @@ var vectorQuerySearch = async ({
       model,
       value: queryText,
       maxRetries,
+      // Type assertion needed: providerOptions type is a union, but embedV2 expects specific version
       ...providerOptions && { providerOptions }
     });
   } else {
@@ -7115,6 +7117,106 @@ var filterSchema = zod.z.object({
   filter: zod.z.coerce.string().describe(filterDescription)
 });
 
+// src/utils/tool-helpers.ts
+function isValidMastraVector(value) {
+  return value !== null && value !== void 0 && typeof value === "object";
+}
+function buildContextString(context) {
+  const parts = [];
+  if (context.vectorStoreName) {
+    parts.push(`vectorStoreName="${context.vectorStoreName}"`);
+  }
+  if (context.requestContext) {
+    const schemaId = context.requestContext.get?.("schemaId");
+    const tenantId = context.requestContext.get?.("tenantId");
+    if (schemaId) parts.push(`schemaId="${schemaId}"`);
+    if (tenantId) parts.push(`tenantId="${tenantId}"`);
+  }
+  if (context.mastra) {
+    parts.push("mastra=provided");
+  }
+  return parts.length > 0 ? ` (context: ${parts.join(", ")})` : "";
+}
+async function resolveVectorStore(options, context, logger) {
+  const { requestContext, mastra, vectorStoreName, fallbackOnInvalid = false } = context;
+  if ("vectorStore" in options && options.vectorStore !== void 0) {
+    const vectorStoreOption = options.vectorStore;
+    if (typeof vectorStoreOption === "function") {
+      const resolved = await vectorStoreOption({ requestContext, mastra });
+      if (!isValidMastraVector(resolved)) {
+        const contextStr = buildContextString(context);
+        const receivedType = resolved === null ? "null" : resolved === void 0 ? "undefined" : typeof resolved;
+        if (fallbackOnInvalid) {
+          if (mastra && vectorStoreName) {
+            return mastra.getVector(vectorStoreName);
+          }
+          return void 0;
+        }
+        throw new Error(
+          `VectorStoreResolver returned invalid value: expected MastraVector instance, got ${receivedType}${contextStr}`
+        );
+      }
+      return resolved;
+    }
+    if (!isValidMastraVector(vectorStoreOption)) {
+      const contextStr = buildContextString(context);
+      const receivedType = vectorStoreOption === null ? "null" : vectorStoreOption === void 0 ? "undefined" : typeof vectorStoreOption;
+      if (fallbackOnInvalid) {
+        if (mastra && vectorStoreName) {
+          return mastra.getVector(vectorStoreName);
+        }
+        return void 0;
+      }
+      throw new Error(`vectorStore option is not a valid MastraVector instance: got ${receivedType}${contextStr}`);
+    }
+    return vectorStoreOption;
+  }
+  if (mastra) {
+    return mastra.getVector(vectorStoreName);
+  }
+  return void 0;
+}
+function coerceTopK(topK, defaultValue = 10) {
+  if (typeof topK === "number") {
+    if (Number.isFinite(topK) && topK > 0) {
+      return topK;
+    }
+    return defaultValue;
+  }
+  if (typeof topK === "string") {
+    const parsed = Number(topK);
+    if (Number.isFinite(parsed) && parsed > 0) {
+      return parsed;
+    }
+    return defaultValue;
+  }
+  return defaultValue;
+}
+function parseFilterValue(filter, logger) {
+  if (!filter) {
+    return {};
+  }
+  if (typeof filter === "string") {
+    try {
+      return JSON.parse(filter);
+    } catch (error) {
+      if (logger) {
+        logger.error("Invalid filter", { filter, error });
+      }
+      throw new Error(`Invalid filter format: ${error instanceof Error ? error.message : String(error)}`);
+    }
+  }
+  if (typeof filter !== "object" || filter === null || Array.isArray(filter)) {
+    if (logger) {
+      logger.error("Invalid filter", { filter, error: "Filter must be a plain object" });
+    }
+    throw new Error(
+      `Invalid filter format: expected a plain object, got ${Array.isArray(filter) ? "array" : typeof filter}`
+    );
+  }
+  return filter;
+}
+
 // src/utils/convert-sources.ts
 var convertToSources = (results) => {
   return results.map((result) => {
@@ -7157,7 +7259,8 @@ var defaultGraphOptions = {
 // src/tools/graph-rag.ts
 var createGraphRAGTool = (options) => {
   const { model, id, description } = options;
-  const toolId = id || `GraphRAG ${options.vectorStoreName} ${options.indexName} Tool`;
+  const storeName = options["vectorStoreName"] ? options.vectorStoreName : "DirectVectorStore";
+  const toolId = id || `GraphRAG ${storeName} ${options.indexName} Tool`;
   const toolDescription = description || defaultGraphRagDescription();
   const graphOptions = {
     ...defaultGraphOptions,
@@ -7174,7 +7277,7 @@ var createGraphRAGTool = (options) => {
     execute: async (inputData, context) => {
       const { requestContext, mastra } = context || {};
       const indexName = requestContext?.get("indexName") ?? options.indexName;
-      const vectorStoreName = requestContext?.get("vectorStoreName") ?? options.vectorStoreName;
+      const vectorStoreName = "vectorStore" in options ? storeName : requestContext?.get("vectorStoreName") ?? storeName;
       if (!indexName) throw new Error(`indexName is required, got: ${indexName}`);
       if (!vectorStoreName) throw new Error(`vectorStoreName is required, got: ${vectorStoreName}`);
       const includeSources = requestContext?.get("includeSources") ?? options.includeSources ?? true;
@@ -7186,36 +7289,19 @@ var createGraphRAGTool = (options) => {
       const providerOptions = requestContext?.get("providerOptions") ?? options.providerOptions;
       const enableFilter = !!requestContext?.get("filter") || (options.enableFilter ?? false);
       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);
+        const topKValue = coerceTopK(topK);
+        const vectorStore = await resolveVectorStore(options, { requestContext, mastra, vectorStoreName });
         if (!vectorStore) {
           if (logger) {
-            logger.error("Vector store not found", { vectorStoreName });
+            logger.error(`Vector store '${vectorStoreName}' not found`);
           }
           return { relevantContext: [], sources: [] };
         }
-        let queryFilter = {};
-        if (enableFilter) {
-          queryFilter = (() => {
-            try {
-              return typeof filter === "string" ? JSON.parse(filter) : filter;
-            } catch (error) {
-              if (logger) {
-                logger.error("Invalid filter", { filter, error });
-              }
-              throw new Error(`Invalid filter format: ${error instanceof Error ? error.message : String(error)}`);
-            }
-          })();
-        }
+        const queryFilter = enableFilter && filter ? parseFilterValue(filter, logger) : {};
         if (logger) {
           logger.debug("Prepared vector query parameters:", { queryFilter, topK: topKValue });
         }
@@ -7268,7 +7354,7 @@ var createGraphRAGTool = (options) => {
         };
       } catch (err) {
         if (logger) {
-          logger.error("Unexpected error in VectorQueryTool execute", {
+          logger.error("Unexpected error in GraphRAGTool execute", {
             error: err,
             errorMessage: err instanceof Error ? err.message : String(err),
             errorStack: err instanceof Error ? err.stack : void 0
@@ -7308,41 +7394,19 @@ var createVectorQueryTool = (options) => {
       const queryText = inputData.queryText;
       const enableFilter = !!requestContext?.get("filter") || (options.enableFilter ?? false);
       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, databaseConfig });
       }
       try {
-        const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
-        let vectorStore = void 0;
-        if ("vectorStore" in options) {
-          vectorStore = options.vectorStore;
-        } else if (mastra) {
-          vectorStore = mastra.getVector(vectorStoreName);
-        }
+        const topKValue = coerceTopK(topK);
+        const vectorStore = await resolveVectorStore(options, { requestContext, mastra, vectorStoreName });
         if (!vectorStore) {
           if (logger) {
-            logger.error("Vector store not found", { vectorStoreName });
+            logger.error(`Vector store '${vectorStoreName}' not found`);
           }
           return { relevantContext: [], sources: [] };
         }
-        let queryFilter = {};
-        if (enableFilter && filter) {
-          queryFilter = (() => {
-            try {
-              return typeof filter === "string" ? JSON.parse(filter) : filter;
-            } catch (error) {
-              if (logger) {
-                logger.error("Invalid filter", { filter, error });
-              }
-              throw new Error(`Invalid filter format: ${error instanceof Error ? error.message : String(error)}`);
-            }
-          })();
-        }
+        const queryFilter = enableFilter && filter ? parseFilterValue(filter, logger) : {};
         if (logger) {
           logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter, databaseConfig });
         }