@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/CHANGELOG.md +22 -0
- package/dist/index.cjs +12 -12
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +12 -12
- package/dist/index.js.map +1 -1
- package/package.json +5 -5
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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
|
}
|