@mastra/rag 2.0.0-beta.3 → 2.0.0-beta.5

package/dist/index.js

@@ -1,4 +1,4 @@
-import { Agent } from '@mastra/core/agent';
+import { Agent, isSupportedLanguageModel } from '@mastra/core/agent';
 import { randomUUID, createHash } from 'crypto';
 import { z } from 'zod';
 import { parse } from 'node-html-better-parser';
@@ -7,7 +7,7 @@ import { Big } from 'big.js';
 import { createSimilarityPrompt } from '@mastra/core/relevance';
 import ZeroEntropy from 'zeroentropy';
 import { createTool } from '@mastra/core/tools';
-import { embedV2, embedV1 } from '@mastra/core/vector';
+import { embedV3, embedV2, embedV1 } from '@mastra/core/vector';
 
 var __create = Object.create;
 var __defProp = Object.defineProperty;
@@ -15,7 +15,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);
 };
@@ -896,7 +896,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 };
 }
@@ -4135,7 +4135,7 @@ var TitleExtractor = class extends BaseExtractor {
       const titleCandidates = await this.getTitlesCandidates(nodes);
       const combinedTitles = titleCandidates.join(", ");
       let title = "";
-      if (this.llm.specificationVersion === "v2") {
+      if (isSupportedLanguageModel(this.llm)) {
         const miniAgent = new Agent({
           id: "title-extractor",
           model: this.llm,
@@ -4174,7 +4174,7 @@ var TitleExtractor = class extends BaseExtractor {
     });
     const titleJobs = nodes.map(async (node) => {
       let completion;
-      if (this.llm.specificationVersion === "v2") {
+      if (isSupportedLanguageModel(this.llm)) {
         const result = await miniAgent.generate([
           { role: "user", content: this.nodeTemplate.format({ context: node.getContent() }) }
         ]);
@@ -4240,7 +4240,7 @@ var SummaryExtractor = class extends BaseExtractor {
       instructions: "You are a summary extractor. You are given a node and you need to extract the summary from the node."
     });
     let summary = "";
-    if (this.llm.specificationVersion === "v2") {
+    if (isSupportedLanguageModel(this.llm)) {
       const result = await miniAgent.generate([{ role: "user", content: prompt }]);
       summary = result.text;
     } else {
@@ -4326,7 +4326,7 @@ var QuestionsAnsweredExtractor = class extends BaseExtractor {
       instructions: "You are a question extractor. You are given a node and you need to extract the questions from the node."
     });
     let questionsText = "";
-    if (this.llm.specificationVersion === "v2") {
+    if (isSupportedLanguageModel(this.llm)) {
       const result2 = await miniAgent.generate([{ role: "user", content: prompt }]);
       questionsText = result2.text;
     } else {
@@ -4398,7 +4398,7 @@ var KeywordExtractor = class extends BaseExtractor {
         name: "keyword-extractor",
         instructions: "You are a keyword extractor. You are given a node and you need to extract the keywords from the node."
       });
-      if (this.llm.specificationVersion === "v2") {
+      if (isSupportedLanguageModel(this.llm)) {
         const result = await miniAgent.generate([
           {
             role: "user",
@@ -6588,7 +6588,7 @@ Always return just the number, no explanation.`,
     const prompt = createSimilarityPrompt(query, text);
     const model = await this.agent.getModel();
     let response;
-    if (model.specificationVersion === "v2") {
+    if (isSupportedLanguageModel(model)) {
       response = await this.agent.generate(prompt);
     } else {
       response = await this.agent.generateLegacy(prompt);
@@ -6974,11 +6974,20 @@ var vectorQuerySearch = async ({
   providerOptions
 }) => {
   let embeddingResult;
-  if (model.specificationVersion === "v2") {
+  if (model.specificationVersion === "v3") {
+    embeddingResult = await embedV3({
+      model,
+      value: queryText,
+      maxRetries,
+      // Type assertion needed: providerOptions type is a union, but embedV3 expects specific version
+      ...providerOptions && { providerOptions }
+    });
+  } else if (model.specificationVersion === "v2") {
     embeddingResult = await embedV2({
       model,
       value: queryText,
       maxRetries,
+      // Type assertion needed: providerOptions type is a union, but embedV2 expects specific version
       ...providerOptions && { providerOptions }
     });
   } else {
@@ -7102,6 +7111,106 @@ var filterSchema = z.object({
   filter: 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) => {
@@ -7144,7 +7253,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,
@@ -7161,7 +7271,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;
@@ -7173,36 +7283,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 });
         }
@@ -7255,7 +7348,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
@@ -7295,41 +7388,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 });
         }