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

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;
@@ -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,7 +6974,14 @@ var vectorQuerySearch = async ({
   providerOptions
 }) => {
   let embeddingResult;
-  if (model.specificationVersion === "v2") {
+  if (model.specificationVersion === "v3") {
+    embeddingResult = await embedV3({
+      model,
+      value: queryText,
+      maxRetries,
+      ...providerOptions && { providerOptions }
+    });
+  } else if (model.specificationVersion === "v2") {
     embeddingResult = await embedV2({
       model,
       value: queryText,