@mastra/rag 1.2.2 → 1.2.3-alpha.0

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @mastra/rag@1.2.1 build /home/runner/work/mastra/mastra/packages/rag
+> @mastra/rag@1.2.3-alpha.0 build /home/runner/work/mastra/mastra/packages/rag
 > tsup --silent --config tsup.config.ts
 

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @mastra/rag
 
+## 1.2.3-alpha.0
+
+### Patch Changes
+
+- [#7124](https://github.com/mastra-ai/mastra/pull/7124) [`29de0e1`](https://github.com/mastra-ai/mastra/commit/29de0e1b0a7173317ae7d1ab0c0993167c659f2b) Thanks [@abhiaiyer91](https://github.com/abhiaiyer91)! - MastraEmbeddingModel and ts hack
+
+- [#7120](https://github.com/mastra-ai/mastra/pull/7120) [`b415a4e`](https://github.com/mastra-ai/mastra/commit/b415a4e05b2546ccb9b5ba31a2bdf23deb1c7fa8) Thanks [@NikAiyer](https://github.com/NikAiyer)! - added providerOptions to createVectorQueryTool and createGraphRAGTool
+
+- [#7125](https://github.com/mastra-ai/mastra/pull/7125) [`6ac697e`](https://github.com/mastra-ai/mastra/commit/6ac697edcc2435482c247cba615277ec4765dcc4) Thanks [@abhiaiyer91](https://github.com/abhiaiyer91)! - improveEmbeddingModelStuff
+
+- Updated dependencies [[`637f323`](https://github.com/mastra-ai/mastra/commit/637f32371d79a8f78c52c0d53411af0915fcec67), [`29de0e1`](https://github.com/mastra-ai/mastra/commit/29de0e1b0a7173317ae7d1ab0c0993167c659f2b), [`6ac697e`](https://github.com/mastra-ai/mastra/commit/6ac697edcc2435482c247cba615277ec4765dcc4)]:
+  - @mastra/core@0.15.3-alpha.1
+
 ## 1.2.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -9,7 +9,7 @@ var big_js = require('big.js');
 var relevance = require('@mastra/core/relevance');
 var ZeroEntropy = require('zeroentropy');
 var tools = require('@mastra/core/tools');
-var ai = require('ai');
+var vector = require('@mastra/core/vector');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
@@ -6914,13 +6914,25 @@ var vectorQuerySearch = async ({
   topK,
   includeVectors = false,
   maxRetries = 2,
-  databaseConfig = {}
+  databaseConfig = {},
+  providerOptions
 }) => {
-  const { embedding } = await ai.embed({
-    value: queryText,
-    model,
-    maxRetries
-  });
+  let embeddingResult;
+  if (model.specificationVersion === "v2") {
+    embeddingResult = await vector.embedV2({
+      model,
+      value: queryText,
+      maxRetries,
+      ...providerOptions && { providerOptions }
+    });
+  } else {
+    embeddingResult = await vector.embedV1({
+      value: queryText,
+      model,
+      maxRetries
+    });
+  }
+  const embedding = embeddingResult.embedding;
   const queryParams = {
     indexName,
     queryVector: embedding,
@@ -7101,6 +7113,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) {
@@ -7143,7 +7156,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 });
@@ -7215,6 +7229,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;
@@ -7268,7 +7283,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 });