@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.
Files changed (151) hide show
  1. package/README.md +1 -2
  2. package/dist/action/index.d.ts +3 -3
  3. package/dist/agent/index.d.ts +4 -4
  4. package/dist/agent/index.js +4 -4
  5. package/dist/base.d.ts +1 -1
  6. package/dist/base.js +2 -2
  7. package/dist/bundler/index.d.ts +12 -4
  8. package/dist/bundler/index.js +3 -3
  9. package/dist/{chunk-D6SEHWUL.js → chunk-4LJFWC2Q.js} +1 -1
  10. package/dist/chunk-5DYJZVB7.js +66 -0
  11. package/dist/chunk-AD6IIOCI.js +11 -0
  12. package/dist/{chunk-5NQ3MEZM.js → chunk-BOS3IA23.js} +0 -1
  13. package/dist/chunk-CQU72KBX.js +54 -0
  14. package/dist/{chunk-SKEBFXFK.js → chunk-DINPURQT.js} +5 -10
  15. package/dist/{chunk-FFZKACHZ.js → chunk-E3TBVBIO.js} +12 -84
  16. package/dist/{chunk-ADU6ZLCT.js → chunk-ECUVL2G5.js} +1 -1
  17. package/dist/{chunk-WLG2YHOZ.js → chunk-G4MCO7XF.js} +1 -1
  18. package/dist/{chunk-IO3KMCXV.js → chunk-ICMEXHKD.js} +0 -1
  19. package/dist/{chunk-HILJW64N.js → chunk-IE37CBXB.js} +20 -9
  20. package/dist/chunk-IEFQZ3IL.js +78 -0
  21. package/dist/{chunk-GTJIVA2D.js → chunk-IJ55HGH4.js} +1 -1
  22. package/dist/{chunk-BCNHUSA6.js → chunk-JJ57BXQR.js} +1 -1
  23. package/dist/{chunk-RIATJJC3.js → chunk-MCB4M5W4.js} +3 -3
  24. package/dist/chunk-NNNAWW2H.js +66 -0
  25. package/dist/{chunk-T5ZUXYWC.js → chunk-NYBGZL6Z.js} +1 -1
  26. package/dist/{chunk-654VMNVR.js → chunk-QX2CHXA7.js} +12 -5
  27. package/dist/chunk-RAR7V3E3.js +66 -0
  28. package/dist/{chunk-Y7YCQUOD.js → chunk-RI3ECMVF.js} +2 -3
  29. package/dist/{chunk-SMVXLIV7.js → chunk-U7WDR4WO.js} +1 -1
  30. package/dist/{chunk-J556NL2S.js → chunk-UCOTMN3J.js} +2 -2
  31. package/dist/chunk-VPSYTVWP.js +66 -0
  32. package/dist/{chunk-KHPTZCPA.js → chunk-WBPDZBUT.js} +5 -33
  33. package/dist/{chunk-Z2KRB7Z4.js → chunk-XDXRDRTQ.js} +1 -1
  34. package/dist/chunk-ZA7MIXNW.js +66 -0
  35. package/dist/deployer/index.d.ts +7 -4
  36. package/dist/deployer/index.js +4 -4
  37. package/dist/embeddings/index.js +1 -1
  38. package/dist/embeddings/model/providers/bedrock.d.ts +42 -0
  39. package/dist/embeddings/model/providers/bedrock.js +3 -0
  40. package/dist/embeddings/model/providers/bedrock.test.d.ts +2 -0
  41. package/dist/embeddings/model/providers/bedrock.test.js +37 -0
  42. package/dist/embeddings/model/providers/cohere.d.ts +34 -0
  43. package/dist/embeddings/model/providers/cohere.js +3 -0
  44. package/dist/embeddings/model/providers/cohere.test.d.ts +2 -0
  45. package/dist/embeddings/model/providers/cohere.test.js +37 -0
  46. package/dist/embeddings/model/providers/embedder.d.ts +28 -0
  47. package/dist/embeddings/model/providers/embedder.js +2 -0
  48. package/dist/embeddings/model/providers/google.d.ts +34 -0
  49. package/dist/embeddings/model/providers/google.js +3 -0
  50. package/dist/embeddings/model/providers/google.test.d.ts +2 -0
  51. package/dist/embeddings/model/providers/google.test.js +37 -0
  52. package/dist/embeddings/model/providers/mistral.d.ts +34 -0
  53. package/dist/embeddings/model/providers/mistral.js +3 -0
  54. package/dist/embeddings/model/providers/mistral.test.d.ts +2 -0
  55. package/dist/embeddings/model/providers/mistral.test.js +37 -0
  56. package/dist/embeddings/model/providers/mock.d.ts +20 -0
  57. package/dist/embeddings/model/providers/mock.js +3 -0
  58. package/dist/embeddings/model/providers/mock.test.d.ts +2 -0
  59. package/dist/embeddings/model/providers/mock.test.js +29 -0
  60. package/dist/embeddings/model/providers/openai.d.ts +34 -0
  61. package/dist/embeddings/model/providers/openai.js +3 -0
  62. package/dist/embeddings/model/providers/openai.test.d.ts +2 -0
  63. package/dist/embeddings/model/providers/openai.test.js +37 -0
  64. package/dist/embeddings/model/providers/voyage.d.ts +34 -0
  65. package/dist/embeddings/model/providers/voyage.js +3 -0
  66. package/dist/embeddings/model/providers/voyage.test.d.ts +2 -0
  67. package/dist/embeddings/model/providers/voyage.test.js +37 -0
  68. package/dist/eval/index.d.ts +4 -4
  69. package/dist/filter/index.d.ts +1 -1
  70. package/dist/filter/index.js +1 -1
  71. package/dist/{index-D-mqcYwN.d.ts → index-B48181D5.d.ts} +6 -13
  72. package/dist/{index-Cktbxvxx.d.ts → index-CBZ2mk2H.d.ts} +0 -1
  73. package/dist/index.d.ts +7 -8
  74. package/dist/index.js +17 -18
  75. package/dist/integration/index.d.ts +5 -5
  76. package/dist/llm/index.d.ts +4 -4
  77. package/dist/llm/index.js +3 -3
  78. package/dist/{model-CD0nIpJn.d.ts → llm/model/index.d.ts} +17 -3
  79. package/dist/llm/model/index.js +5 -0
  80. package/dist/llm/model/providers/anthropic-vertex.d.ts +5 -5
  81. package/dist/llm/model/providers/anthropic-vertex.js +3 -3
  82. package/dist/llm/model/providers/anthropic.d.ts +5 -5
  83. package/dist/llm/model/providers/anthropic.js +3 -3
  84. package/dist/llm/model/providers/azure.d.ts +5 -5
  85. package/dist/llm/model/providers/azure.js +3 -3
  86. package/dist/llm/model/providers/baseten.d.ts +5 -5
  87. package/dist/llm/model/providers/baseten.js +4 -4
  88. package/dist/llm/model/providers/bedrock.d.ts +5 -5
  89. package/dist/llm/model/providers/bedrock.js +3 -3
  90. package/dist/llm/model/providers/cerebras.d.ts +5 -5
  91. package/dist/llm/model/providers/cerebras.js +3 -3
  92. package/dist/llm/model/providers/cohere.d.ts +5 -5
  93. package/dist/llm/model/providers/cohere.js +3 -3
  94. package/dist/llm/model/providers/deepinfra.d.ts +5 -5
  95. package/dist/llm/model/providers/deepinfra.js +3 -3
  96. package/dist/llm/model/providers/deepseek.d.ts +5 -5
  97. package/dist/llm/model/providers/deepseek.js +3 -3
  98. package/dist/llm/model/providers/fireworks.d.ts +5 -5
  99. package/dist/llm/model/providers/fireworks.js +3 -3
  100. package/dist/llm/model/providers/google-vertex.d.ts +5 -5
  101. package/dist/llm/model/providers/google-vertex.js +3 -3
  102. package/dist/llm/model/providers/google.d.ts +5 -5
  103. package/dist/llm/model/providers/google.js +3 -3
  104. package/dist/llm/model/providers/grok.d.ts +5 -5
  105. package/dist/llm/model/providers/grok.js +3 -3
  106. package/dist/llm/model/providers/groq.d.ts +5 -5
  107. package/dist/llm/model/providers/groq.js +3 -3
  108. package/dist/llm/model/providers/lmstudio.d.ts +5 -5
  109. package/dist/llm/model/providers/lmstudio.js +4 -4
  110. package/dist/llm/model/providers/mistral.d.ts +5 -5
  111. package/dist/llm/model/providers/mistral.js +3 -3
  112. package/dist/llm/model/providers/mock.d.ts +5 -5
  113. package/dist/llm/model/providers/mock.js +3 -3
  114. package/dist/llm/model/providers/ollama.d.ts +5 -5
  115. package/dist/llm/model/providers/ollama.js +3 -3
  116. package/dist/llm/model/providers/openai-compat.d.ts +5 -5
  117. package/dist/llm/model/providers/openai-compat.js +4 -4
  118. package/dist/llm/model/providers/openai.d.ts +5 -5
  119. package/dist/llm/model/providers/openai.js +4 -4
  120. package/dist/llm/model/providers/openai.test.js +4 -4
  121. package/dist/llm/model/providers/perplexity.d.ts +5 -5
  122. package/dist/llm/model/providers/perplexity.js +3 -3
  123. package/dist/llm/model/providers/portkey.d.ts +5 -5
  124. package/dist/llm/model/providers/portkey.js +3 -3
  125. package/dist/llm/model/providers/togetherai.d.ts +5 -5
  126. package/dist/llm/model/providers/togetherai.js +3 -3
  127. package/dist/logger/index.d.ts +1 -1
  128. package/dist/logger/index.js +1 -1
  129. package/dist/mastra/index.d.ts +5 -7
  130. package/dist/mastra/index.js +4 -4
  131. package/dist/memory/index.d.ts +4 -4
  132. package/dist/memory/index.js +3 -3
  133. package/dist/relevance/index.d.ts +4 -4
  134. package/dist/relevance/index.js +5 -5
  135. package/dist/storage/index.d.ts +5 -5
  136. package/dist/storage/index.js +3 -3
  137. package/dist/tools/index.d.ts +5 -5
  138. package/dist/tts/index.d.ts +1 -1
  139. package/dist/tts/index.js +3 -3
  140. package/dist/vector/index.d.ts +1 -1
  141. package/dist/vector/index.js +4 -4
  142. package/dist/vector/libsql/index.d.ts +31 -0
  143. package/dist/vector/libsql/index.js +662 -0
  144. package/dist/{workflow-4u4hE9mG.d.ts → workflow-CSwxE-4q.d.ts} +1 -19
  145. package/dist/workflows/index.d.ts +6 -6
  146. package/dist/workflows/index.js +3 -3
  147. package/package.json +14 -8
  148. package/dist/chunk-NIYVERVP.js +0 -199
  149. package/dist/engine/index.d.ts +0 -59
  150. package/dist/engine/index.js +0 -4
  151. package/dist/engine-EwEG-4Iv.d.ts +0 -127
@@ -1,4 +1,4 @@
1
- import { MastraBundler } from './chunk-RIATJJC3.js';
1
+ import { MastraBundler } from './chunk-MCB4M5W4.js';
2
2
  import { __name } from './chunk-AJJZUHB4.js';
3
3
 
4
4
  // src/deployer/index.ts
@@ -1,6 +1,6 @@
1
- import { MastraBase } from './chunk-WLG2YHOZ.js';
1
+ import { MastraBase } from './chunk-G4MCO7XF.js';
2
2
  import { __name } from './chunk-AJJZUHB4.js';
3
- import dotenv from 'dotenv';
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 = dotenv.parse(content);
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,4 +1,4 @@
1
- import { MastraLLM } from './chunk-654VMNVR.js';
1
+ import { MastraLLM } from './chunk-QX2CHXA7.js';
2
2
  import { __name } from './chunk-AJJZUHB4.js';
3
3
  import { createOpenAI } from '@ai-sdk/openai';
4
4
 
@@ -1,6 +1,6 @@
1
1
  import { delay } from './chunk-KNPBNSJ7.js';
2
- import { MastraBase } from './chunk-WLG2YHOZ.js';
3
- import { RegisteredLogger } from './chunk-IO3KMCXV.js';
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-WLG2YHOZ.js';
4
- import { RegisteredLogger, LogLevel } from './chunk-IO3KMCXV.js';
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,4 +1,4 @@
1
- import { Agent } from './chunk-SKEBFXFK.js';
1
+ import { Agent } from './chunk-DINPURQT.js';
2
2
  import { __name, __publicField } from './chunk-AJJZUHB4.js';
3
3
  import { CohereClient } from 'cohere-ai';
4
4
 
@@ -1,5 +1,5 @@
1
- import { embed, embedMany } from './chunk-5NQ3MEZM.js';
2
- import { MastraBase } from './chunk-WLG2YHOZ.js';
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-Y7YCQUOD.js';
2
+ import { LLM } from './chunk-RI3ECMVF.js';
3
3
  import { InstrumentClass } from './chunk-6ZVFVYLE.js';
4
- import { LogLevel, createLogger, noopLogger } from './chunk-IO3KMCXV.js';
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
- const agent = this.agents?.[key];
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
  }
@@ -1,4 +1,4 @@
1
- import { MastraBase } from './chunk-WLG2YHOZ.js';
1
+ import { MastraBase } from './chunk-G4MCO7XF.js';
2
2
  import { __name, __publicField } from './chunk-AJJZUHB4.js';
3
3
  import { createClient } from '@libsql/client';
4
4
 
@@ -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 };
@@ -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-Cktbxvxx.js';
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
- declare abstract class MastraDeployer extends MastraBundler {
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 };
@@ -1,5 +1,5 @@
1
- export { MastraDeployer } from '../chunk-BCNHUSA6.js';
2
- import '../chunk-RIATJJC3.js';
3
- import '../chunk-WLG2YHOZ.js';
4
- import '../chunk-IO3KMCXV.js';
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';
@@ -1,2 +1,2 @@
1
- export { embed, embedMany } from '../chunk-5NQ3MEZM.js';
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,3 @@
1
+ export { BedrockEmbedder, embed, embedMany } from '../../../chunk-IEFQZ3IL.js';
2
+ import '../../../chunk-AD6IIOCI.js';
3
+ import '../../../chunk-AJJZUHB4.js';
@@ -0,0 +1,2 @@
1
+
2
+ export { }
@@ -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 };
@@ -0,0 +1,3 @@
1
+ export { CohereEmbedder, embed, embedMany } from '../../../chunk-VPSYTVWP.js';
2
+ import '../../../chunk-AD6IIOCI.js';
3
+ import '../../../chunk-AJJZUHB4.js';
@@ -0,0 +1,2 @@
1
+
2
+ export { }