@mastra/rag 1.2.7 → 1.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.js CHANGED
@@ -4141,7 +4141,7 @@ var TitleExtractor = class extends BaseExtractor {
4141
4141
  name: "title-extractor",
4142
4142
  instructions: "You are a title extractor. You are given a list of nodes and you need to extract the title from the nodes."
4143
4143
  });
4144
- const result = await miniAgent.generateVNext(
4144
+ const result = await miniAgent.generate(
4145
4145
  [{ role: "user", content: this.combineTemplate.format({ context: combinedTitles }) }],
4146
4146
  { format: "mastra" }
4147
4147
  );
@@ -4152,7 +4152,7 @@ var TitleExtractor = class extends BaseExtractor {
4152
4152
  name: "title-extractor",
4153
4153
  instructions: "You are a title extractor. You are given a list of nodes and you need to extract the title from the nodes."
4154
4154
  });
4155
- const result = await miniAgent.generate([
4155
+ const result = await miniAgent.generateLegacy([
4156
4156
  { role: "user", content: this.combineTemplate.format({ context: combinedTitles }) }
4157
4157
  ]);
4158
4158
  title = result.text;
@@ -4173,13 +4173,13 @@ var TitleExtractor = class extends BaseExtractor {
4173
4173
  const titleJobs = nodes.map(async (node) => {
4174
4174
  let completion;
4175
4175
  if (this.llm.specificationVersion === "v2") {
4176
- const result = await miniAgent.generateVNext(
4176
+ const result = await miniAgent.generate(
4177
4177
  [{ role: "user", content: this.nodeTemplate.format({ context: node.getContent() }) }],
4178
4178
  { format: "mastra" }
4179
4179
  );
4180
4180
  completion = result.text;
4181
4181
  } else {
4182
- const result = await miniAgent.generate([
4182
+ const result = await miniAgent.generateLegacy([
4183
4183
  { role: "user", content: this.nodeTemplate.format({ context: node.getContent() }) }
4184
4184
  ]);
4185
4185
  completion = result.text;
@@ -4239,10 +4239,10 @@ var SummaryExtractor = class extends BaseExtractor {
4239
4239
  });
4240
4240
  let summary = "";
4241
4241
  if (this.llm.specificationVersion === "v2") {
4242
- const result = await miniAgent.generateVNext([{ role: "user", content: prompt }], { format: "mastra" });
4242
+ const result = await miniAgent.generate([{ role: "user", content: prompt }], { format: "mastra" });
4243
4243
  summary = result.text;
4244
4244
  } else {
4245
- const result = await miniAgent.generate([{ role: "user", content: prompt }]);
4245
+ const result = await miniAgent.generateLegacy([{ role: "user", content: prompt }]);
4246
4246
  summary = result.text;
4247
4247
  }
4248
4248
  if (!summary) {
@@ -4324,10 +4324,10 @@ var QuestionsAnsweredExtractor = class extends BaseExtractor {
4324
4324
  });
4325
4325
  let questionsText = "";
4326
4326
  if (this.llm.specificationVersion === "v2") {
4327
- const result2 = await miniAgent.generateVNext([{ role: "user", content: prompt }], { format: "mastra" });
4327
+ const result2 = await miniAgent.generate([{ role: "user", content: prompt }], { format: "mastra" });
4328
4328
  questionsText = result2.text;
4329
4329
  } else {
4330
- const result2 = await miniAgent.generate([{ role: "user", content: prompt }]);
4330
+ const result2 = await miniAgent.generateLegacy([{ role: "user", content: prompt }]);
4331
4331
  questionsText = result2.text;
4332
4332
  }
4333
4333
  if (!questionsText) {
@@ -4395,7 +4395,7 @@ var KeywordExtractor = class extends BaseExtractor {
4395
4395
  instructions: "You are a keyword extractor. You are given a node and you need to extract the keywords from the node."
4396
4396
  });
4397
4397
  if (this.llm.specificationVersion === "v2") {
4398
- const result = await miniAgent.generateVNext(
4398
+ const result = await miniAgent.generate(
4399
4399
  [
4400
4400
  {
4401
4401
  role: "user",
@@ -4409,7 +4409,7 @@ var KeywordExtractor = class extends BaseExtractor {
4409
4409
  );
4410
4410
  keywords = result.text;
4411
4411
  } else {
4412
- const result = await miniAgent.generate([
4412
+ const result = await miniAgent.generateLegacy([
4413
4413
  {
4414
4414
  role: "user",
4415
4415
  content: this.promptTemplate.format({ context: node.getContent(), maxKeywords: this.keywords.toString() })
@@ -6551,9 +6551,9 @@ Always return just the number, no explanation.`,
6551
6551
  const model = await this.agent.getModel();
6552
6552
  let response;
6553
6553
  if (model.specificationVersion === "v2") {
6554
- response = await this.agent.generateVNext(prompt);
6555
- } else {
6556
6554
  response = await this.agent.generate(prompt);
6555
+ } else {
6556
+ response = await this.agent.generateLegacy(prompt);
6557
6557
  }
6558
6558
  return parseFloat(response.text);
6559
6559
  }