@mastra/rag 1.2.2 → 1.2.3-alpha.0

package/dist/index.js

@@ -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 { embed } from 'ai';
+import { embedV2, embedV1 } from '@mastra/core/vector';
 
 var __create = Object.create;
 var __defProp = Object.defineProperty;
@@ -6908,13 +6908,25 @@ var vectorQuerySearch = async ({
   topK,
   includeVectors = false,
   maxRetries = 2,
-  databaseConfig = {}
+  databaseConfig = {},
+  providerOptions
 }) => {
-  const { embedding } = await embed({
-    value: queryText,
-    model,
-    maxRetries
-  });
+  let embeddingResult;
+  if (model.specificationVersion === "v2") {
+    embeddingResult = await embedV2({
+      model,
+      value: queryText,
+      maxRetries,
+      ...providerOptions && { providerOptions }
+    });
+  } else {
+    embeddingResult = await embedV1({
+      value: queryText,
+      model,
+      maxRetries
+    });
+  }
+  const embedding = embeddingResult.embedding;
   const queryParams = {
     indexName,
     queryVector: embedding,
@@ -7095,6 +7107,7 @@ var createGraphRAGTool = (options) => {
       const topK = runtimeContext.get("topK") ?? context.topK ?? 10;
       const filter = runtimeContext.get("filter") ?? context.filter;
       const queryText = context.queryText;
+      const providerOptions = runtimeContext.get("providerOptions") ?? options.providerOptions;
       const enableFilter = !!runtimeContext.get("filter") || (options.enableFilter ?? false);
       const logger = mastra?.getLogger();
       if (!logger) {
@@ -7137,7 +7150,8 @@ var createGraphRAGTool = (options) => {
           model,
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue,
-          includeVectors: true
+          includeVectors: true,
+          providerOptions
         });
         if (logger) {
           logger.debug("vectorQuerySearch returned results", { count: results.length });
@@ -7209,6 +7223,7 @@ var createVectorQueryTool = (options) => {
       const reranker = runtimeContext.get("reranker") ?? options.reranker;
       const databaseConfig = runtimeContext.get("databaseConfig") ?? options.databaseConfig;
       const model = runtimeContext.get("model") ?? options.model;
+      const providerOptions = runtimeContext.get("providerOptions") ?? options.providerOptions;
       if (!indexName) throw new Error(`indexName is required, got: ${indexName}`);
       if (!vectorStoreName) throw new Error(`vectorStoreName is required, got: ${vectorStoreName}`);
       const topK = runtimeContext.get("topK") ?? context.topK ?? 10;
@@ -7262,7 +7277,8 @@ var createVectorQueryTool = (options) => {
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue,
           includeVectors,
-          databaseConfig
+          databaseConfig,
+          providerOptions
         });
         if (logger) {
           logger.debug("vectorQuerySearch returned results", { count: results.length });