@mastra/core 0.2.0-alpha.89 → 0.2.0-alpha.92
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/README.md +1 -2
- package/dist/action/index.d.ts +3 -3
- package/dist/agent/index.d.ts +4 -4
- package/dist/agent/index.js +4 -4
- package/dist/base.d.ts +1 -1
- package/dist/base.js +2 -2
- package/dist/bundler/index.d.ts +12 -4
- package/dist/bundler/index.js +3 -3
- package/dist/{chunk-D6SEHWUL.js → chunk-4LJFWC2Q.js} +1 -1
- package/dist/chunk-5DYJZVB7.js +66 -0
- package/dist/chunk-AD6IIOCI.js +11 -0
- package/dist/{chunk-5NQ3MEZM.js → chunk-BOS3IA23.js} +0 -1
- package/dist/chunk-CQU72KBX.js +54 -0
- package/dist/{chunk-SKEBFXFK.js → chunk-DINPURQT.js} +5 -10
- package/dist/{chunk-FFZKACHZ.js → chunk-E3TBVBIO.js} +12 -84
- package/dist/{chunk-ADU6ZLCT.js → chunk-ECUVL2G5.js} +1 -1
- package/dist/{chunk-WLG2YHOZ.js → chunk-G4MCO7XF.js} +1 -1
- package/dist/{chunk-IO3KMCXV.js → chunk-ICMEXHKD.js} +0 -1
- package/dist/{chunk-HILJW64N.js → chunk-IE37CBXB.js} +20 -9
- package/dist/chunk-IEFQZ3IL.js +78 -0
- package/dist/{chunk-GTJIVA2D.js → chunk-IJ55HGH4.js} +1 -1
- package/dist/{chunk-BCNHUSA6.js → chunk-JJ57BXQR.js} +1 -1
- package/dist/{chunk-RIATJJC3.js → chunk-MCB4M5W4.js} +3 -3
- package/dist/chunk-NNNAWW2H.js +66 -0
- package/dist/{chunk-T5ZUXYWC.js → chunk-NYBGZL6Z.js} +1 -1
- package/dist/{chunk-654VMNVR.js → chunk-QX2CHXA7.js} +12 -5
- package/dist/chunk-RAR7V3E3.js +66 -0
- package/dist/{chunk-Y7YCQUOD.js → chunk-RI3ECMVF.js} +2 -3
- package/dist/{chunk-SMVXLIV7.js → chunk-U7WDR4WO.js} +1 -1
- package/dist/{chunk-J556NL2S.js → chunk-UCOTMN3J.js} +2 -2
- package/dist/chunk-VPSYTVWP.js +66 -0
- package/dist/{chunk-KHPTZCPA.js → chunk-WBPDZBUT.js} +5 -33
- package/dist/{chunk-Z2KRB7Z4.js → chunk-XDXRDRTQ.js} +1 -1
- package/dist/chunk-ZA7MIXNW.js +66 -0
- package/dist/deployer/index.d.ts +7 -4
- package/dist/deployer/index.js +4 -4
- package/dist/embeddings/index.js +1 -1
- package/dist/embeddings/model/providers/bedrock.d.ts +42 -0
- package/dist/embeddings/model/providers/bedrock.js +3 -0
- package/dist/embeddings/model/providers/bedrock.test.d.ts +2 -0
- package/dist/embeddings/model/providers/bedrock.test.js +37 -0
- package/dist/embeddings/model/providers/cohere.d.ts +34 -0
- package/dist/embeddings/model/providers/cohere.js +3 -0
- package/dist/embeddings/model/providers/cohere.test.d.ts +2 -0
- package/dist/embeddings/model/providers/cohere.test.js +37 -0
- package/dist/embeddings/model/providers/embedder.d.ts +28 -0
- package/dist/embeddings/model/providers/embedder.js +2 -0
- package/dist/embeddings/model/providers/google.d.ts +34 -0
- package/dist/embeddings/model/providers/google.js +3 -0
- package/dist/embeddings/model/providers/google.test.d.ts +2 -0
- package/dist/embeddings/model/providers/google.test.js +37 -0
- package/dist/embeddings/model/providers/mistral.d.ts +34 -0
- package/dist/embeddings/model/providers/mistral.js +3 -0
- package/dist/embeddings/model/providers/mistral.test.d.ts +2 -0
- package/dist/embeddings/model/providers/mistral.test.js +37 -0
- package/dist/embeddings/model/providers/mock.d.ts +20 -0
- package/dist/embeddings/model/providers/mock.js +3 -0
- package/dist/embeddings/model/providers/mock.test.d.ts +2 -0
- package/dist/embeddings/model/providers/mock.test.js +29 -0
- package/dist/embeddings/model/providers/openai.d.ts +34 -0
- package/dist/embeddings/model/providers/openai.js +3 -0
- package/dist/embeddings/model/providers/openai.test.d.ts +2 -0
- package/dist/embeddings/model/providers/openai.test.js +37 -0
- package/dist/embeddings/model/providers/voyage.d.ts +34 -0
- package/dist/embeddings/model/providers/voyage.js +3 -0
- package/dist/embeddings/model/providers/voyage.test.d.ts +2 -0
- package/dist/embeddings/model/providers/voyage.test.js +37 -0
- package/dist/eval/index.d.ts +4 -4
- package/dist/filter/index.d.ts +1 -1
- package/dist/filter/index.js +1 -1
- package/dist/{index-D-mqcYwN.d.ts → index-B48181D5.d.ts} +6 -13
- package/dist/{index-Cktbxvxx.d.ts → index-CBZ2mk2H.d.ts} +0 -1
- package/dist/index.d.ts +7 -8
- package/dist/index.js +17 -18
- package/dist/integration/index.d.ts +5 -5
- package/dist/llm/index.d.ts +4 -4
- package/dist/llm/index.js +3 -3
- package/dist/{model-CD0nIpJn.d.ts → llm/model/index.d.ts} +17 -3
- package/dist/llm/model/index.js +5 -0
- package/dist/llm/model/providers/anthropic-vertex.d.ts +5 -5
- package/dist/llm/model/providers/anthropic-vertex.js +3 -3
- package/dist/llm/model/providers/anthropic.d.ts +5 -5
- package/dist/llm/model/providers/anthropic.js +3 -3
- package/dist/llm/model/providers/azure.d.ts +5 -5
- package/dist/llm/model/providers/azure.js +3 -3
- package/dist/llm/model/providers/baseten.d.ts +5 -5
- package/dist/llm/model/providers/baseten.js +4 -4
- package/dist/llm/model/providers/bedrock.d.ts +5 -5
- package/dist/llm/model/providers/bedrock.js +3 -3
- package/dist/llm/model/providers/cerebras.d.ts +5 -5
- package/dist/llm/model/providers/cerebras.js +3 -3
- package/dist/llm/model/providers/cohere.d.ts +5 -5
- package/dist/llm/model/providers/cohere.js +3 -3
- package/dist/llm/model/providers/deepinfra.d.ts +5 -5
- package/dist/llm/model/providers/deepinfra.js +3 -3
- package/dist/llm/model/providers/deepseek.d.ts +5 -5
- package/dist/llm/model/providers/deepseek.js +3 -3
- package/dist/llm/model/providers/fireworks.d.ts +5 -5
- package/dist/llm/model/providers/fireworks.js +3 -3
- package/dist/llm/model/providers/google-vertex.d.ts +5 -5
- package/dist/llm/model/providers/google-vertex.js +3 -3
- package/dist/llm/model/providers/google.d.ts +5 -5
- package/dist/llm/model/providers/google.js +3 -3
- package/dist/llm/model/providers/grok.d.ts +5 -5
- package/dist/llm/model/providers/grok.js +3 -3
- package/dist/llm/model/providers/groq.d.ts +5 -5
- package/dist/llm/model/providers/groq.js +3 -3
- package/dist/llm/model/providers/lmstudio.d.ts +5 -5
- package/dist/llm/model/providers/lmstudio.js +4 -4
- package/dist/llm/model/providers/mistral.d.ts +5 -5
- package/dist/llm/model/providers/mistral.js +3 -3
- package/dist/llm/model/providers/mock.d.ts +5 -5
- package/dist/llm/model/providers/mock.js +3 -3
- package/dist/llm/model/providers/ollama.d.ts +5 -5
- package/dist/llm/model/providers/ollama.js +3 -3
- package/dist/llm/model/providers/openai-compat.d.ts +5 -5
- package/dist/llm/model/providers/openai-compat.js +4 -4
- package/dist/llm/model/providers/openai.d.ts +5 -5
- package/dist/llm/model/providers/openai.js +4 -4
- package/dist/llm/model/providers/openai.test.js +4 -4
- package/dist/llm/model/providers/perplexity.d.ts +5 -5
- package/dist/llm/model/providers/perplexity.js +3 -3
- package/dist/llm/model/providers/portkey.d.ts +5 -5
- package/dist/llm/model/providers/portkey.js +3 -3
- package/dist/llm/model/providers/togetherai.d.ts +5 -5
- package/dist/llm/model/providers/togetherai.js +3 -3
- package/dist/logger/index.d.ts +1 -1
- package/dist/logger/index.js +1 -1
- package/dist/mastra/index.d.ts +5 -7
- package/dist/mastra/index.js +4 -4
- package/dist/memory/index.d.ts +4 -4
- package/dist/memory/index.js +3 -3
- package/dist/relevance/index.d.ts +4 -4
- package/dist/relevance/index.js +5 -5
- package/dist/storage/index.d.ts +5 -5
- package/dist/storage/index.js +3 -3
- package/dist/tools/index.d.ts +5 -5
- package/dist/tts/index.d.ts +1 -1
- package/dist/tts/index.js +3 -3
- package/dist/vector/index.d.ts +1 -1
- package/dist/vector/index.js +4 -4
- package/dist/vector/libsql/index.d.ts +31 -0
- package/dist/vector/libsql/index.js +662 -0
- package/dist/{workflow-4u4hE9mG.d.ts → workflow-CSwxE-4q.d.ts} +1 -19
- package/dist/workflows/index.d.ts +6 -6
- package/dist/workflows/index.js +3 -3
- package/package.json +14 -8
- package/dist/chunk-NIYVERVP.js +0 -199
- package/dist/engine/index.d.ts +0 -59
- package/dist/engine/index.js +0 -4
- package/dist/engine-EwEG-4Iv.d.ts +0 -127
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { MastraBase } from './chunk-
|
|
1
|
+
import { MastraBase } from './chunk-G4MCO7XF.js';
|
|
2
2
|
import { __name } from './chunk-AJJZUHB4.js';
|
|
3
|
-
import
|
|
3
|
+
import { parse } from 'dotenv';
|
|
4
4
|
import { readFile } from 'fs/promises';
|
|
5
5
|
|
|
6
6
|
var _MastraBundler = class _MastraBundler extends MastraBase {
|
|
@@ -14,7 +14,7 @@ var _MastraBundler = class _MastraBundler extends MastraBase {
|
|
|
14
14
|
const envVars = /* @__PURE__ */ new Map();
|
|
15
15
|
for (const file of await this.getEnvFiles()) {
|
|
16
16
|
const content = await readFile(file, "utf-8");
|
|
17
|
-
const config =
|
|
17
|
+
const config = parse(content);
|
|
18
18
|
Object.entries(config).forEach(([key, value]) => {
|
|
19
19
|
envVars.set(key, value);
|
|
20
20
|
});
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { MastraEmbedder } from './chunk-AD6IIOCI.js';
|
|
2
|
+
import { __name, __publicField } from './chunk-AJJZUHB4.js';
|
|
3
|
+
import { createGoogleGenerativeAI } from '@ai-sdk/google';
|
|
4
|
+
import { embed as embed$1, embedMany as embedMany$1 } from 'ai';
|
|
5
|
+
|
|
6
|
+
async function embed(value, { apiKey = process.env.GOOGLE_API_KEY || "", model = "textembedding-gecko", baseURL, maxRetries = 3 }) {
|
|
7
|
+
const google = createGoogleGenerativeAI({
|
|
8
|
+
baseURL,
|
|
9
|
+
apiKey
|
|
10
|
+
});
|
|
11
|
+
const eModel = google.textEmbeddingModel(model);
|
|
12
|
+
return await embed$1({
|
|
13
|
+
model: eModel,
|
|
14
|
+
value,
|
|
15
|
+
maxRetries
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
__name(embed, "embed");
|
|
19
|
+
async function embedMany(values, { apiKey = process.env.GOOGLE_API_KEY || "", model = "textembedding-gecko", baseURL, maxRetries = 3 }) {
|
|
20
|
+
const google = createGoogleGenerativeAI({
|
|
21
|
+
baseURL,
|
|
22
|
+
apiKey
|
|
23
|
+
});
|
|
24
|
+
const eModel = google.textEmbeddingModel(model);
|
|
25
|
+
return await embedMany$1({
|
|
26
|
+
model: eModel,
|
|
27
|
+
values,
|
|
28
|
+
maxRetries
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
__name(embedMany, "embedMany");
|
|
32
|
+
var _GoogleEmbedder = class _GoogleEmbedder extends MastraEmbedder {
|
|
33
|
+
constructor({ apiKey = process.env.GOOGLE_API_KEY || "", model = "textembedding-gecko", baseURL }) {
|
|
34
|
+
super();
|
|
35
|
+
__publicField(this, "apiKey");
|
|
36
|
+
__publicField(this, "model");
|
|
37
|
+
__publicField(this, "baseURL");
|
|
38
|
+
this.apiKey = apiKey;
|
|
39
|
+
this.model = model;
|
|
40
|
+
this.baseURL = baseURL;
|
|
41
|
+
}
|
|
42
|
+
async embed(value, { maxRetries } = {
|
|
43
|
+
maxRetries: 3
|
|
44
|
+
}) {
|
|
45
|
+
return embed(value, {
|
|
46
|
+
apiKey: this.apiKey,
|
|
47
|
+
model: this.model,
|
|
48
|
+
baseURL: this.baseURL,
|
|
49
|
+
maxRetries
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
async embedMany(values, { maxRetries } = {
|
|
53
|
+
maxRetries: 3
|
|
54
|
+
}) {
|
|
55
|
+
return embedMany(values, {
|
|
56
|
+
apiKey: this.apiKey,
|
|
57
|
+
model: this.model,
|
|
58
|
+
baseURL: this.baseURL,
|
|
59
|
+
maxRetries
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
};
|
|
63
|
+
__name(_GoogleEmbedder, "GoogleEmbedder");
|
|
64
|
+
var GoogleEmbedder = _GoogleEmbedder;
|
|
65
|
+
|
|
66
|
+
export { GoogleEmbedder, embed, embedMany };
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { delay } from './chunk-KNPBNSJ7.js';
|
|
2
|
-
import { MastraBase } from './chunk-
|
|
3
|
-
import { RegisteredLogger } from './chunk-
|
|
2
|
+
import { MastraBase } from './chunk-G4MCO7XF.js';
|
|
3
|
+
import { RegisteredLogger } from './chunk-ICMEXHKD.js';
|
|
4
4
|
import { __name, __privateAdd, __privateSet, __privateGet } from './chunk-AJJZUHB4.js';
|
|
5
5
|
import { generateText, jsonSchema, generateObject, streamText, streamObject } from 'ai';
|
|
6
6
|
import { z } from 'zod';
|
|
@@ -86,8 +86,6 @@ var _MastraLLMBase = class _MastraLLMBase extends MastraBase {
|
|
|
86
86
|
_mastra = new WeakMap();
|
|
87
87
|
__name(_MastraLLMBase, "MastraLLMBase");
|
|
88
88
|
var MastraLLMBase = _MastraLLMBase;
|
|
89
|
-
|
|
90
|
-
// src/llm/model/model.ts
|
|
91
89
|
var _model, _mastra2;
|
|
92
90
|
var _MastraLLM = class _MastraLLM extends MastraLLMBase {
|
|
93
91
|
constructor({ model, mastra }) {
|
|
@@ -104,6 +102,15 @@ var _MastraLLM = class _MastraLLM extends MastraLLMBase {
|
|
|
104
102
|
}
|
|
105
103
|
}
|
|
106
104
|
}
|
|
105
|
+
__registerPrimitives(p) {
|
|
106
|
+
if (p.telemetry) {
|
|
107
|
+
this.__setTelemetry(p.telemetry);
|
|
108
|
+
}
|
|
109
|
+
if (p.logger) {
|
|
110
|
+
this.__setLogger(p.logger);
|
|
111
|
+
}
|
|
112
|
+
__privateSet(this, _mastra2, p);
|
|
113
|
+
}
|
|
107
114
|
convertTools(tools) {
|
|
108
115
|
this.logger.debug("Starting tool conversion for LLM");
|
|
109
116
|
const converted = Object.entries(tools || {}).reduce((memo, value) => {
|
|
@@ -398,4 +405,4 @@ _mastra2 = new WeakMap();
|
|
|
398
405
|
__name(_MastraLLM, "MastraLLM");
|
|
399
406
|
var MastraLLM = _MastraLLM;
|
|
400
407
|
|
|
401
|
-
export { MastraLLM };
|
|
408
|
+
export { MastraLLM, MastraLLMBase };
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { MastraEmbedder } from './chunk-AD6IIOCI.js';
|
|
2
|
+
import { __name, __publicField } from './chunk-AJJZUHB4.js';
|
|
3
|
+
import { embed as embed$1, embedMany as embedMany$1 } from 'ai';
|
|
4
|
+
import { createVoyage } from 'voyage-ai-provider';
|
|
5
|
+
|
|
6
|
+
async function embed(value, { apiKey = process.env.VOYAGE_API_KEY || "", model = "voyage-01", baseURL, maxRetries = 3 }) {
|
|
7
|
+
const voyage = createVoyage({
|
|
8
|
+
baseURL,
|
|
9
|
+
apiKey
|
|
10
|
+
});
|
|
11
|
+
const eModel = voyage.textEmbeddingModel(model);
|
|
12
|
+
return await embed$1({
|
|
13
|
+
model: eModel,
|
|
14
|
+
value,
|
|
15
|
+
maxRetries
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
__name(embed, "embed");
|
|
19
|
+
async function embedMany(values, { apiKey = process.env.VOYAGE_API_KEY || "", model = "voyage-01", baseURL, maxRetries = 3 }) {
|
|
20
|
+
const voyage = createVoyage({
|
|
21
|
+
baseURL,
|
|
22
|
+
apiKey
|
|
23
|
+
});
|
|
24
|
+
const eModel = voyage.textEmbeddingModel(model);
|
|
25
|
+
return await embedMany$1({
|
|
26
|
+
model: eModel,
|
|
27
|
+
values,
|
|
28
|
+
maxRetries
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
__name(embedMany, "embedMany");
|
|
32
|
+
var _VoyageEmbedder = class _VoyageEmbedder extends MastraEmbedder {
|
|
33
|
+
constructor({ apiKey = process.env.VOYAGE_API_KEY || "", model = "voyage-01", baseURL }) {
|
|
34
|
+
super();
|
|
35
|
+
__publicField(this, "apiKey");
|
|
36
|
+
__publicField(this, "model");
|
|
37
|
+
__publicField(this, "baseURL");
|
|
38
|
+
this.apiKey = apiKey;
|
|
39
|
+
this.model = model;
|
|
40
|
+
this.baseURL = baseURL;
|
|
41
|
+
}
|
|
42
|
+
async embed(value, { maxRetries } = {
|
|
43
|
+
maxRetries: 3
|
|
44
|
+
}) {
|
|
45
|
+
return embed(value, {
|
|
46
|
+
apiKey: this.apiKey,
|
|
47
|
+
model: this.model,
|
|
48
|
+
baseURL: this.baseURL,
|
|
49
|
+
maxRetries
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
async embedMany(values, { maxRetries } = {
|
|
53
|
+
maxRetries: 3
|
|
54
|
+
}) {
|
|
55
|
+
return embedMany(values, {
|
|
56
|
+
apiKey: this.apiKey,
|
|
57
|
+
model: this.model,
|
|
58
|
+
baseURL: this.baseURL,
|
|
59
|
+
maxRetries
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
};
|
|
63
|
+
__name(_VoyageEmbedder, "VoyageEmbedder");
|
|
64
|
+
var VoyageEmbedder = _VoyageEmbedder;
|
|
65
|
+
|
|
66
|
+
export { VoyageEmbedder, embed, embedMany };
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { InstrumentClass } from './chunk-6ZVFVYLE.js';
|
|
2
2
|
import { delay } from './chunk-KNPBNSJ7.js';
|
|
3
|
-
import { MastraBase } from './chunk-
|
|
4
|
-
import { RegisteredLogger, LogLevel } from './chunk-
|
|
3
|
+
import { MastraBase } from './chunk-G4MCO7XF.js';
|
|
4
|
+
import { RegisteredLogger, LogLevel } from './chunk-ICMEXHKD.js';
|
|
5
5
|
import { __name, __privateAdd, __privateSet, __privateGet } from './chunk-AJJZUHB4.js';
|
|
6
6
|
import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
|
|
7
7
|
import { createAnthropic } from '@ai-sdk/anthropic';
|
|
@@ -14,7 +14,6 @@ import { createXai } from '@ai-sdk/xai';
|
|
|
14
14
|
import { tool, generateText, jsonSchema, generateObject, streamText, streamObject } from 'ai';
|
|
15
15
|
import { createAnthropicVertex } from 'anthropic-vertex-ai';
|
|
16
16
|
import { z } from 'zod';
|
|
17
|
-
import 'dotenv/config';
|
|
18
17
|
|
|
19
18
|
function _ts_decorate(decorators, target, key, desc) {
|
|
20
19
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
1
|
+
import { MastraBase } from './chunk-G4MCO7XF.js';
|
|
2
|
+
import { embed, embedMany } from './chunk-BOS3IA23.js';
|
|
3
3
|
import { __name } from './chunk-AJJZUHB4.js';
|
|
4
4
|
|
|
5
5
|
// src/vector/index.ts
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { MastraEmbedder } from './chunk-AD6IIOCI.js';
|
|
2
|
+
import { __name, __publicField } from './chunk-AJJZUHB4.js';
|
|
3
|
+
import { createCohere } from '@ai-sdk/cohere';
|
|
4
|
+
import { embed as embed$1, embedMany as embedMany$1 } from 'ai';
|
|
5
|
+
|
|
6
|
+
async function embed(value, { apiKey = process.env.COHERE_API_KEY || "", model = "embed-english-v3.0", baseURL, maxRetries = 3 }) {
|
|
7
|
+
const cohere = createCohere({
|
|
8
|
+
baseURL,
|
|
9
|
+
apiKey
|
|
10
|
+
});
|
|
11
|
+
const eModel = cohere.textEmbeddingModel(model);
|
|
12
|
+
return await embed$1({
|
|
13
|
+
model: eModel,
|
|
14
|
+
value,
|
|
15
|
+
maxRetries
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
__name(embed, "embed");
|
|
19
|
+
async function embedMany(values, { apiKey = process.env.COHERE_API_KEY || "", model = "embed-english-v3.0", baseURL, maxRetries = 3 }) {
|
|
20
|
+
const cohere = createCohere({
|
|
21
|
+
baseURL,
|
|
22
|
+
apiKey
|
|
23
|
+
});
|
|
24
|
+
const eModel = cohere.textEmbeddingModel(model);
|
|
25
|
+
return await embedMany$1({
|
|
26
|
+
model: eModel,
|
|
27
|
+
values,
|
|
28
|
+
maxRetries
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
__name(embedMany, "embedMany");
|
|
32
|
+
var _CohereEmbedder = class _CohereEmbedder extends MastraEmbedder {
|
|
33
|
+
constructor({ apiKey = process.env.COHERE_API_KEY || "", model = "embed-english-v3.0", baseURL }) {
|
|
34
|
+
super();
|
|
35
|
+
__publicField(this, "apiKey");
|
|
36
|
+
__publicField(this, "model");
|
|
37
|
+
__publicField(this, "baseURL");
|
|
38
|
+
this.apiKey = apiKey;
|
|
39
|
+
this.model = model;
|
|
40
|
+
this.baseURL = baseURL;
|
|
41
|
+
}
|
|
42
|
+
async embed(value, { maxRetries } = {
|
|
43
|
+
maxRetries: 3
|
|
44
|
+
}) {
|
|
45
|
+
return embed(value, {
|
|
46
|
+
apiKey: this.apiKey,
|
|
47
|
+
model: this.model,
|
|
48
|
+
baseURL: this.baseURL,
|
|
49
|
+
maxRetries
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
async embedMany(values, { maxRetries } = {
|
|
53
|
+
maxRetries: 3
|
|
54
|
+
}) {
|
|
55
|
+
return embedMany(values, {
|
|
56
|
+
apiKey: this.apiKey,
|
|
57
|
+
model: this.model,
|
|
58
|
+
baseURL: this.baseURL,
|
|
59
|
+
maxRetries
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
};
|
|
63
|
+
__name(_CohereEmbedder, "CohereEmbedder");
|
|
64
|
+
var CohereEmbedder = _CohereEmbedder;
|
|
65
|
+
|
|
66
|
+
export { CohereEmbedder, embed, embedMany };
|
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
import { Telemetry } from './chunk-SDKEPBBH.js';
|
|
2
|
-
import { LLM } from './chunk-
|
|
2
|
+
import { LLM } from './chunk-RI3ECMVF.js';
|
|
3
3
|
import { InstrumentClass } from './chunk-6ZVFVYLE.js';
|
|
4
|
-
import { LogLevel, createLogger, noopLogger } from './chunk-
|
|
4
|
+
import { LogLevel, createLogger, noopLogger } from './chunk-ICMEXHKD.js';
|
|
5
5
|
import { __name, __publicField } from './chunk-AJJZUHB4.js';
|
|
6
|
-
import 'dotenv/config';
|
|
7
6
|
|
|
7
|
+
// src/mastra/index.ts
|
|
8
8
|
function _ts_decorate(decorators, target, key, desc) {
|
|
9
9
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
10
10
|
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
@@ -25,7 +25,6 @@ var _Mastra = class _Mastra {
|
|
|
25
25
|
__publicField(this, "telemetry");
|
|
26
26
|
__publicField(this, "tts");
|
|
27
27
|
__publicField(this, "deployer");
|
|
28
|
-
__publicField(this, "engine");
|
|
29
28
|
__publicField(this, "storage");
|
|
30
29
|
__publicField(this, "memory");
|
|
31
30
|
let logger;
|
|
@@ -42,6 +41,7 @@ var _Mastra = class _Mastra {
|
|
|
42
41
|
});
|
|
43
42
|
}
|
|
44
43
|
}
|
|
44
|
+
this.logger = logger;
|
|
45
45
|
if (config?.telemetry) {
|
|
46
46
|
this.telemetry = Telemetry.init(config.telemetry);
|
|
47
47
|
}
|
|
@@ -57,19 +57,6 @@ var _Mastra = class _Mastra {
|
|
|
57
57
|
this.deployer.__setTelemetry(this.telemetry);
|
|
58
58
|
}
|
|
59
59
|
}
|
|
60
|
-
if (config?.engine) {
|
|
61
|
-
if (this.telemetry) {
|
|
62
|
-
this.engine = this.telemetry.traceClass(config.engine, {
|
|
63
|
-
excludeMethods: [
|
|
64
|
-
"__setTelemetry",
|
|
65
|
-
"__getTelemetry"
|
|
66
|
-
]
|
|
67
|
-
});
|
|
68
|
-
this.engine.__setTelemetry(this.telemetry);
|
|
69
|
-
} else {
|
|
70
|
-
this.engine = config.engine;
|
|
71
|
-
}
|
|
72
|
-
}
|
|
73
60
|
if (config?.storage) {
|
|
74
61
|
if (this.telemetry) {
|
|
75
62
|
this.storage = this.telemetry.traceClass(config.storage, {
|
|
@@ -100,9 +87,6 @@ var _Mastra = class _Mastra {
|
|
|
100
87
|
});
|
|
101
88
|
this.vectors = vectors;
|
|
102
89
|
}
|
|
103
|
-
if (config?.engine) {
|
|
104
|
-
this.engine = config.engine;
|
|
105
|
-
}
|
|
106
90
|
if (config?.vectors) {
|
|
107
91
|
this.vectors = config.vectors;
|
|
108
92
|
}
|
|
@@ -143,7 +127,6 @@ var _Mastra = class _Mastra {
|
|
|
143
127
|
agent.__registerPrimitives({
|
|
144
128
|
logger: this.getLogger(),
|
|
145
129
|
telemetry: this.telemetry,
|
|
146
|
-
engine: this.engine,
|
|
147
130
|
storage: this.storage,
|
|
148
131
|
memory: this.memory,
|
|
149
132
|
agents,
|
|
@@ -160,7 +143,6 @@ var _Mastra = class _Mastra {
|
|
|
160
143
|
workflow.__registerPrimitives({
|
|
161
144
|
logger: this.getLogger(),
|
|
162
145
|
telemetry: this.telemetry,
|
|
163
|
-
engine: this.engine,
|
|
164
146
|
storage: this.storage,
|
|
165
147
|
memory: this.memory,
|
|
166
148
|
agents: this.agents,
|
|
@@ -171,7 +153,6 @@ var _Mastra = class _Mastra {
|
|
|
171
153
|
this.workflows[key] = workflow;
|
|
172
154
|
});
|
|
173
155
|
}
|
|
174
|
-
this.logger = logger;
|
|
175
156
|
this.setLogger({
|
|
176
157
|
logger
|
|
177
158
|
});
|
|
@@ -243,13 +224,7 @@ var _Mastra = class _Mastra {
|
|
|
243
224
|
this.logger = logger;
|
|
244
225
|
if (this.agents) {
|
|
245
226
|
Object.keys(this.agents).forEach((key) => {
|
|
246
|
-
|
|
247
|
-
if (agent) {
|
|
248
|
-
agent.__setLogger(this.logger);
|
|
249
|
-
if (agent.hasOwnMemory()) {
|
|
250
|
-
agent.getMemory()?.__setLogger(this.logger);
|
|
251
|
-
}
|
|
252
|
-
}
|
|
227
|
+
this.agents?.[key]?.__setLogger(this.logger);
|
|
253
228
|
});
|
|
254
229
|
}
|
|
255
230
|
if (this.workflows) {
|
|
@@ -268,9 +243,6 @@ var _Mastra = class _Mastra {
|
|
|
268
243
|
this.tts?.[key]?.__setLogger(this.logger);
|
|
269
244
|
});
|
|
270
245
|
}
|
|
271
|
-
if (this.engine) {
|
|
272
|
-
this.engine.__setLogger(this.logger);
|
|
273
|
-
}
|
|
274
246
|
if (this.storage) {
|
|
275
247
|
this.storage.__setLogger(this.logger);
|
|
276
248
|
}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { MastraEmbedder } from './chunk-AD6IIOCI.js';
|
|
2
|
+
import { __name, __publicField } from './chunk-AJJZUHB4.js';
|
|
3
|
+
import { createOpenAI } from '@ai-sdk/openai';
|
|
4
|
+
import { embed as embed$1, embedMany as embedMany$1 } from 'ai';
|
|
5
|
+
|
|
6
|
+
async function embed(value, { apiKey = process.env.OPENAI_API_KEY || "", model = "text-embedding-3-small", baseURL, maxRetries = 3 }) {
|
|
7
|
+
const openai = createOpenAI({
|
|
8
|
+
baseURL,
|
|
9
|
+
apiKey
|
|
10
|
+
});
|
|
11
|
+
const eModel = openai.textEmbeddingModel(model);
|
|
12
|
+
return await embed$1({
|
|
13
|
+
model: eModel,
|
|
14
|
+
value,
|
|
15
|
+
maxRetries
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
__name(embed, "embed");
|
|
19
|
+
async function embedMany(value, { apiKey = process.env.OPENAI_API_KEY || "", model = "text-embedding-3-small", baseURL, maxRetries = 3 }) {
|
|
20
|
+
const openai = createOpenAI({
|
|
21
|
+
baseURL,
|
|
22
|
+
apiKey
|
|
23
|
+
});
|
|
24
|
+
const eModel = openai.textEmbeddingModel(model);
|
|
25
|
+
return await embedMany$1({
|
|
26
|
+
model: eModel,
|
|
27
|
+
values: value,
|
|
28
|
+
maxRetries
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
__name(embedMany, "embedMany");
|
|
32
|
+
var _OpenAIEmbedder = class _OpenAIEmbedder extends MastraEmbedder {
|
|
33
|
+
constructor({ apiKey = process.env.OPENAI_API_KEY || "", model = "text-embedding-3-small", baseURL }) {
|
|
34
|
+
super();
|
|
35
|
+
__publicField(this, "apiKey");
|
|
36
|
+
__publicField(this, "model");
|
|
37
|
+
__publicField(this, "baseURL");
|
|
38
|
+
this.apiKey = apiKey;
|
|
39
|
+
this.model = model;
|
|
40
|
+
this.baseURL = baseURL;
|
|
41
|
+
}
|
|
42
|
+
async embed(value, { maxRetries } = {
|
|
43
|
+
maxRetries: 3
|
|
44
|
+
}) {
|
|
45
|
+
return embed(value, {
|
|
46
|
+
apiKey: this.apiKey,
|
|
47
|
+
model: this.model,
|
|
48
|
+
baseURL: this.baseURL,
|
|
49
|
+
maxRetries
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
async embedMany(values, { maxRetries } = {
|
|
53
|
+
maxRetries: 3
|
|
54
|
+
}) {
|
|
55
|
+
return embedMany(values, {
|
|
56
|
+
apiKey: this.apiKey,
|
|
57
|
+
model: this.model,
|
|
58
|
+
baseURL: this.baseURL,
|
|
59
|
+
maxRetries
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
};
|
|
63
|
+
__name(_OpenAIEmbedder, "OpenAIEmbedder");
|
|
64
|
+
var OpenAIEmbedder = _OpenAIEmbedder;
|
|
65
|
+
|
|
66
|
+
export { OpenAIEmbedder, embed, embedMany };
|
package/dist/deployer/index.d.ts
CHANGED
|
@@ -1,18 +1,21 @@
|
|
|
1
|
-
import { MastraBundler } from '../bundler/index.js';
|
|
1
|
+
import { IBundler, MastraBundler } from '../bundler/index.js';
|
|
2
2
|
import '../base.js';
|
|
3
3
|
import '@opentelemetry/api';
|
|
4
|
-
import '../index-
|
|
4
|
+
import '../index-CBZ2mk2H.js';
|
|
5
5
|
import 'pino';
|
|
6
6
|
import 'stream';
|
|
7
7
|
import '../telemetry-oCUM52DG.js';
|
|
8
8
|
import '@opentelemetry/sdk-node';
|
|
9
9
|
import '@opentelemetry/sdk-trace-base';
|
|
10
10
|
|
|
11
|
-
|
|
11
|
+
interface IDeployer extends IBundler {
|
|
12
|
+
deploy(outputDirectory: string): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
declare abstract class MastraDeployer extends MastraBundler implements IDeployer {
|
|
12
15
|
constructor({ name }: {
|
|
13
16
|
name: string;
|
|
14
17
|
});
|
|
15
18
|
abstract deploy(outputDirectory: string): Promise<void>;
|
|
16
19
|
}
|
|
17
20
|
|
|
18
|
-
export { MastraDeployer };
|
|
21
|
+
export { type IDeployer, MastraDeployer };
|
package/dist/deployer/index.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
export { MastraDeployer } from '../chunk-
|
|
2
|
-
import '../chunk-
|
|
3
|
-
import '../chunk-
|
|
4
|
-
import '../chunk-
|
|
1
|
+
export { MastraDeployer } from '../chunk-JJ57BXQR.js';
|
|
2
|
+
import '../chunk-MCB4M5W4.js';
|
|
3
|
+
import '../chunk-G4MCO7XF.js';
|
|
4
|
+
import '../chunk-ICMEXHKD.js';
|
|
5
5
|
import '../chunk-AJJZUHB4.js';
|
package/dist/embeddings/index.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export { embed, embedMany } from '../chunk-
|
|
1
|
+
export { embed, embedMany } from '../chunk-BOS3IA23.js';
|
|
2
2
|
import '../chunk-AJJZUHB4.js';
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { EmbedResult, EmbedManyResult } from 'ai';
|
|
2
|
+
import { MastraEmbedder } from './embedder.js';
|
|
3
|
+
|
|
4
|
+
type BedrockEmbeddingModelNames = 'amazon.titan-embed-text-v1' | 'amazon.titan-embed-image-v1' | 'cohere.embed-english-v3' | 'cohere.embed-multilingual-v3' | (string & {});
|
|
5
|
+
declare function embed(value: string, { region, accessKeyId, secretAccessKey, sessionToken, model, maxRetries, }: {
|
|
6
|
+
maxRetries?: number;
|
|
7
|
+
region?: string;
|
|
8
|
+
accessKeyId?: string;
|
|
9
|
+
secretAccessKey?: string;
|
|
10
|
+
sessionToken?: string;
|
|
11
|
+
model: BedrockEmbeddingModelNames;
|
|
12
|
+
}): Promise<EmbedResult<string>>;
|
|
13
|
+
declare function embedMany(values: string[], { region, accessKeyId, secretAccessKey, sessionToken, model, maxRetries, }: {
|
|
14
|
+
maxRetries?: number;
|
|
15
|
+
region?: string;
|
|
16
|
+
accessKeyId?: string;
|
|
17
|
+
secretAccessKey?: string;
|
|
18
|
+
sessionToken?: string;
|
|
19
|
+
model: BedrockEmbeddingModelNames;
|
|
20
|
+
}): Promise<EmbedManyResult<string>>;
|
|
21
|
+
declare class BedrockEmbedder extends MastraEmbedder {
|
|
22
|
+
region: string;
|
|
23
|
+
accessKeyId: string;
|
|
24
|
+
secretAccessKey: string;
|
|
25
|
+
sessionToken: string;
|
|
26
|
+
model: BedrockEmbeddingModelNames;
|
|
27
|
+
constructor({ region, accessKeyId, secretAccessKey, sessionToken, model, }?: {
|
|
28
|
+
region?: string;
|
|
29
|
+
accessKeyId?: string;
|
|
30
|
+
secretAccessKey?: string;
|
|
31
|
+
sessionToken?: string;
|
|
32
|
+
model?: BedrockEmbeddingModelNames;
|
|
33
|
+
});
|
|
34
|
+
embed(value: string, { maxRetries }?: {
|
|
35
|
+
maxRetries?: number;
|
|
36
|
+
}): Promise<EmbedResult<string>>;
|
|
37
|
+
embedMany(values: string[], { maxRetries }?: {
|
|
38
|
+
maxRetries?: number;
|
|
39
|
+
}): Promise<EmbedManyResult<string>>;
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
export { BedrockEmbedder, type BedrockEmbeddingModelNames, embed, embedMany };
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { BedrockEmbedder } from '../../../chunk-IEFQZ3IL.js';
|
|
2
|
+
import '../../../chunk-AD6IIOCI.js';
|
|
3
|
+
import '../../../chunk-AJJZUHB4.js';
|
|
4
|
+
import dotenv from 'dotenv';
|
|
5
|
+
import { describe, beforeEach, it, expect } from 'vitest';
|
|
6
|
+
|
|
7
|
+
dotenv.config();
|
|
8
|
+
describe("Bedrock Embeddings", () => {
|
|
9
|
+
let embedder;
|
|
10
|
+
beforeEach(() => {
|
|
11
|
+
embedder = new BedrockEmbedder({
|
|
12
|
+
model: "amazon.titan-embed-text-v1"
|
|
13
|
+
});
|
|
14
|
+
});
|
|
15
|
+
it.skip("should create an embedding for a single string value", async () => {
|
|
16
|
+
const value = "This is a test string";
|
|
17
|
+
const maxRetries = 3;
|
|
18
|
+
const embedding = await embedder.embed(value, {
|
|
19
|
+
maxRetries
|
|
20
|
+
});
|
|
21
|
+
console.log(embedding);
|
|
22
|
+
expect(embedding).toBeDefined();
|
|
23
|
+
});
|
|
24
|
+
it.skip("should create embeddings for an array of string values", async () => {
|
|
25
|
+
const values = [
|
|
26
|
+
"String 1",
|
|
27
|
+
"String 2",
|
|
28
|
+
"String 3"
|
|
29
|
+
];
|
|
30
|
+
const maxRetries = 3;
|
|
31
|
+
const embeddings = await embedder.embedMany(values, {
|
|
32
|
+
maxRetries
|
|
33
|
+
});
|
|
34
|
+
console.log(embeddings);
|
|
35
|
+
expect(embeddings).toBeDefined();
|
|
36
|
+
});
|
|
37
|
+
});
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import { EmbedResult, EmbedManyResult } from 'ai';
|
|
2
|
+
import { MastraEmbedder } from './embedder.js';
|
|
3
|
+
|
|
4
|
+
type CohereEmbeddingModelNames = 'embed-english-v3.0' | 'embed-english-light-v3.0' | 'embed-multilingual-v3.0' | 'embed-multilingual-light-v3.0' | (string & {});
|
|
5
|
+
declare function embed(value: string, { apiKey, model, baseURL, maxRetries, }: {
|
|
6
|
+
maxRetries?: number;
|
|
7
|
+
apiKey?: string;
|
|
8
|
+
model: CohereEmbeddingModelNames;
|
|
9
|
+
baseURL?: string;
|
|
10
|
+
}): Promise<EmbedResult<string>>;
|
|
11
|
+
declare function embedMany(values: string[], { apiKey, model, baseURL, maxRetries, }: {
|
|
12
|
+
maxRetries?: number;
|
|
13
|
+
apiKey?: string;
|
|
14
|
+
model: CohereEmbeddingModelNames;
|
|
15
|
+
baseURL?: string;
|
|
16
|
+
}): Promise<EmbedManyResult<string>>;
|
|
17
|
+
declare class CohereEmbedder extends MastraEmbedder {
|
|
18
|
+
apiKey: string;
|
|
19
|
+
model: CohereEmbeddingModelNames;
|
|
20
|
+
baseURL: string | undefined;
|
|
21
|
+
constructor({ apiKey, model, baseURL, }: {
|
|
22
|
+
apiKey?: string;
|
|
23
|
+
model: CohereEmbeddingModelNames;
|
|
24
|
+
baseURL?: string;
|
|
25
|
+
});
|
|
26
|
+
embed(value: string, { maxRetries }?: {
|
|
27
|
+
maxRetries?: number;
|
|
28
|
+
}): Promise<EmbedResult<string>>;
|
|
29
|
+
embedMany(values: string[], { maxRetries }?: {
|
|
30
|
+
maxRetries?: number;
|
|
31
|
+
}): Promise<EmbedManyResult<string>>;
|
|
32
|
+
}
|
|
33
|
+
|
|
34
|
+
export { CohereEmbedder, type CohereEmbeddingModelNames, embed, embedMany };
|