@mastra/core 0.1.27-alpha.31 → 0.1.27-alpha.33

package/dist/core.cjs.development.js

@@ -15,7 +15,6 @@ var openai = require('@ai-sdk/openai');
 var xai = require('@ai-sdk/xai');
 var ai = require('ai');
 var anthropicVertexAi = require('anthropic-vertex-ai');
-var voyageAiProvider = require('voyage-ai-provider');
 var redis = require('@upstash/redis');
 var fs = require('fs');
 var path = require('path');
@@ -29,6 +28,7 @@ var semanticConventions = require('@opentelemetry/semantic-conventions');
 var radash = require('radash');
 var sift = require('sift');
 var xstate = require('xstate');
+var voyageAiProvider = require('voyage-ai-provider');
 
 function _arrayLikeToArray(r, a) {
   (null == a || a > r.length) && (a = r.length);
@@ -1293,129 +1293,16 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
     }
     return modelDef;
   };
-  _proto.createEmbedding = /*#__PURE__*/function () {
-    var _createEmbedding = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref4) {
-      var model, value, maxRetries, embeddingModel, openai$1, cohere$1, amazon, google$1, mistral$1, voyage;
-      return _regeneratorRuntime().wrap(function _callee$(_context) {
-        while (1) switch (_context.prev = _context.next) {
-          case 0:
-            model = _ref4.model, value = _ref4.value, maxRetries = _ref4.maxRetries;
-            if (!(model.provider === 'OPEN_AI')) {
-              _context.next = 6;
-              break;
-            }
-            openai$1 = openai.createOpenAI({
-              apiKey: process.env.OPENAI_API_KEY
-            });
-            embeddingModel = openai$1.embedding(model.name);
-            _context.next = 32;
-            break;
-          case 6:
-            if (!(model.provider === 'COHERE')) {
-              _context.next = 11;
-              break;
-            }
-            cohere$1 = cohere.createCohere({
-              apiKey: process.env.COHERE_API_KEY
-            });
-            embeddingModel = cohere$1.embedding(model.name);
-            _context.next = 32;
-            break;
-          case 11:
-            if (!(model.provider === 'AMAZON')) {
-              _context.next = 16;
-              break;
-            }
-            amazon = amazonBedrock.createAmazonBedrock({
-              region: process.env.AWS_REGION || '',
-              accessKeyId: process.env.AWS_ACCESS_KEY_ID || '',
-              secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY || '',
-              sessionToken: process.env.AWS_SESSION_TOKEN || ''
-            });
-            embeddingModel = amazon.embedding(model.name);
-            _context.next = 32;
-            break;
-          case 16:
-            if (!(model.provider === 'GOOGLE')) {
-              _context.next = 21;
-              break;
-            }
-            google$1 = google.createGoogleGenerativeAI({
-              baseURL: 'https://generativelanguage.googleapis.com/v1beta',
-              apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY || ''
-            });
-            embeddingModel = google$1.textEmbeddingModel(model.name);
-            _context.next = 32;
-            break;
-          case 21:
-            if (!(model.provider === 'MISTRAL')) {
-              _context.next = 26;
-              break;
-            }
-            mistral$1 = mistral.createMistral({
-              baseURL: 'https://api.mistral.ai/v1',
-              apiKey: process.env.MISTRAL_API_KEY || ''
-            });
-            embeddingModel = mistral$1.textEmbeddingModel(model.name);
-            _context.next = 32;
-            break;
-          case 26:
-            if (!(model.provider === 'VOYAGE')) {
-              _context.next = 31;
-              break;
-            }
-            voyage = voyageAiProvider.createVoyage({
-              baseURL: 'https://api.voyageai.com/v1',
-              apiKey: process.env.VOYAGE_API_KEY || ''
-            });
-            embeddingModel = voyage.textEmbeddingModel(model.name);
-            _context.next = 32;
-            break;
-          case 31:
-            throw new Error("Invalid embedding model");
-          case 32:
-            if (!(value instanceof Array)) {
-              _context.next = 36;
-              break;
-            }
-            _context.next = 35;
-            return ai.embedMany({
-              model: embeddingModel,
-              values: value,
-              maxRetries: maxRetries
-            });
-          case 35:
-            return _context.abrupt("return", _context.sent);
-          case 36:
-            _context.next = 38;
-            return ai.embed({
-              model: embeddingModel,
-              value: value,
-              maxRetries: maxRetries
-            });
-          case 38:
-            return _context.abrupt("return", _context.sent);
-          case 39:
-          case "end":
-            return _context.stop();
-        }
-      }, _callee);
-    }));
-    function createEmbedding(_x) {
-      return _createEmbedding.apply(this, arguments);
-    }
-    return createEmbedding;
-  }();
   _proto.getParams = /*#__PURE__*/function () {
-    var _getParams = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref5) {
+    var _getParams = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref4) {
       var tools, resultTool, model, toolsConverted, answerTool, modelDef;
-      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
-        while (1) switch (_context2.prev = _context2.next) {
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
           case 0:
-            tools = _ref5.tools, resultTool = _ref5.resultTool, model = _ref5.model;
-            toolsConverted = Object.entries(tools).reduce(function (memo, _ref6) {
-              var key = _ref6[0],
-                val = _ref6[1];
+            tools = _ref4.tools, resultTool = _ref4.resultTool, model = _ref4.model;
+            toolsConverted = Object.entries(tools).reduce(function (memo, _ref5) {
+              var key = _ref5[0],
+                val = _ref5[1];
               memo[key] = ai.tool(val);
               return memo;
             }, {});
@@ -1426,29 +1313,29 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               };
             }
             if (!('type' in model)) {
-              _context2.next = 8;
+              _context.next = 8;
               break;
             }
             modelDef = this.createModelDef({
               model: model
             });
-            _context2.next = 15;
+            _context.next = 15;
             break;
           case 8:
             if (!(model.model instanceof Function)) {
-              _context2.next = 14;
+              _context.next = 14;
               break;
             }
-            _context2.next = 11;
+            _context.next = 11;
             return model.model();
           case 11:
-            modelDef = _context2.sent;
-            _context2.next = 15;
+            modelDef = _context.sent;
+            _context.next = 15;
             break;
           case 14:
             modelDef = model.model;
           case 15:
-            return _context2.abrupt("return", {
+            return _context.abrupt("return", {
               toolsConverted: toolsConverted,
               modelDef: modelDef,
               answerTool: answerTool,
@@ -1456,11 +1343,11 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             });
           case 16:
           case "end":
-            return _context2.stop();
+            return _context.stop();
         }
-      }, _callee2, this);
+      }, _callee, this);
     }));
-    function getParams(_x2) {
+    function getParams(_x) {
       return _getParams.apply(this, arguments);
     }
     return getParams;
@@ -1504,9 +1391,9 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
   };
   _proto.createOutputSchema = function createOutputSchema(output) {
     var _this2 = this;
-    var schema = Object.entries(output).reduce(function (memo, _ref7) {
-      var k = _ref7[0],
-        v = _ref7[1];
+    var schema = Object.entries(output).reduce(function (memo, _ref6) {
+      var k = _ref6[0],
+        v = _ref6[1];
       if (_this2.isBaseOutputType(v.type)) {
         memo[k] = _this2.baseOutputTypeSchema(v.type);
       }
@@ -1531,12 +1418,12 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
     return zod.z.object(schema);
   };
   _proto.generate = /*#__PURE__*/function () {
-    var _generate = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(messages, _temp) {
-      var _ref8, schema, stream, _ref8$maxSteps, maxSteps, onFinish, onStepFinish, enabledTools, convertedTools, runId, msgs;
-      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
-        while (1) switch (_context3.prev = _context3.next) {
+    var _generate = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(messages, _temp) {
+      var _ref7, schema, stream, _ref7$maxSteps, maxSteps, onFinish, onStepFinish, enabledTools, convertedTools, runId, msgs;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
           case 0:
-            _ref8 = _temp === void 0 ? {} : _temp, schema = _ref8.schema, stream = _ref8.stream, _ref8$maxSteps = _ref8.maxSteps, maxSteps = _ref8$maxSteps === void 0 ? 5 : _ref8$maxSteps, onFinish = _ref8.onFinish, onStepFinish = _ref8.onStepFinish, enabledTools = _ref8.enabledTools, convertedTools = _ref8.convertedTools, runId = _ref8.runId;
+            _ref7 = _temp === void 0 ? {} : _temp, schema = _ref7.schema, stream = _ref7.stream, _ref7$maxSteps = _ref7.maxSteps, maxSteps = _ref7$maxSteps === void 0 ? 5 : _ref7$maxSteps, onFinish = _ref7.onFinish, onStepFinish = _ref7.onStepFinish, enabledTools = _ref7.enabledTools, convertedTools = _ref7.convertedTools, runId = _ref7.runId;
             if (Array.isArray(messages)) {
               msgs = messages == null ? void 0 : messages.map(function (m) {
                 if (typeof m === 'string') {
@@ -1554,10 +1441,10 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               }];
             }
             if (!(stream && schema)) {
-              _context3.next = 6;
+              _context2.next = 6;
               break;
             }
-            _context3.next = 5;
+            _context2.next = 5;
             return this.__streamObject({
               messages: msgs,
               structuredOutput: schema,
@@ -1569,13 +1456,13 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               runId: runId
             });
           case 5:
-            return _context3.abrupt("return", _context3.sent);
+            return _context2.abrupt("return", _context2.sent);
           case 6:
             if (!stream) {
-              _context3.next = 10;
+              _context2.next = 10;
               break;
             }
-            _context3.next = 9;
+            _context2.next = 9;
             return this.__stream({
               messages: msgs,
               onStepFinish: onStepFinish,
@@ -1586,13 +1473,13 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               runId: runId
             });
           case 9:
-            return _context3.abrupt("return", _context3.sent);
+            return _context2.abrupt("return", _context2.sent);
           case 10:
             if (!schema) {
-              _context3.next = 14;
+              _context2.next = 14;
               break;
             }
-            _context3.next = 13;
+            _context2.next = 13;
             return this.__textObject({
               messages: msgs,
               structuredOutput: schema,
@@ -1603,9 +1490,9 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               runId: runId
             });
           case 13:
-            return _context3.abrupt("return", _context3.sent);
+            return _context2.abrupt("return", _context2.sent);
           case 14:
-            _context3.next = 16;
+            _context2.next = 16;
             return this.__text({
               messages: msgs,
               onStepFinish: onStepFinish,
@@ -1615,26 +1502,26 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               runId: runId
             });
           case 16:
-            return _context3.abrupt("return", _context3.sent);
+            return _context2.abrupt("return", _context2.sent);
           case 17:
           case "end":
-            return _context3.stop();
+            return _context2.stop();
         }
-      }, _callee3, this);
+      }, _callee2, this);
     }));
-    function generate(_x3, _x4) {
+    function generate(_x2, _x3) {
       return _generate.apply(this, arguments);
     }
     return generate;
   }();
   _proto.__text = /*#__PURE__*/function () {
-    var _text = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref9) {
+    var _text = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref8) {
       var _this3 = this;
-      var messages, _onStepFinish, _ref9$maxSteps, maxSteps, enabledTools, runId, convertedTools, model, modelToPass, params, argsForExecute;
-      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
-        while (1) switch (_context5.prev = _context5.next) {
+      var messages, _onStepFinish, _ref8$maxSteps, maxSteps, enabledTools, runId, convertedTools, model, modelToPass, params, argsForExecute;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
           case 0:
-            messages = _ref9.messages, _onStepFinish = _ref9.onStepFinish, _ref9$maxSteps = _ref9.maxSteps, maxSteps = _ref9$maxSteps === void 0 ? 5 : _ref9$maxSteps, enabledTools = _ref9.enabledTools, runId = _ref9.runId, convertedTools = _ref9.convertedTools;
+            messages = _ref8.messages, _onStepFinish = _ref8.onStepFinish, _ref8$maxSteps = _ref8.maxSteps, maxSteps = _ref8$maxSteps === void 0 ? 5 : _ref8$maxSteps, enabledTools = _ref8.enabledTools, runId = _ref8.runId, convertedTools = _ref8.convertedTools;
             model = _classPrivateFieldLooseBase(this, _model)[_model];
             _classPrivateFieldLooseBase(this, _log$2)[_log$2](LogLevel.DEBUG, "Generating text with " + messages.length + " messages", runId);
             if ('name' in model) {
@@ -1649,71 +1536,71 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               modelToPass = model;
             }
-            _context5.next = 6;
+            _context4.next = 6;
             return this.getParams({
               tools: convertedTools || this.convertTools(enabledTools || {}),
               model: modelToPass
             });
           case 6:
-            params = _context5.sent;
+            params = _context4.sent;
             argsForExecute = {
               model: params.modelDef,
               tools: _extends({}, params.toolsConverted, params.answerTool),
               toolChoice: params.toolChoice,
               maxSteps: maxSteps,
               onStepFinish: function () {
-                var _onStepFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(props) {
+                var _onStepFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(props) {
                   var _props$response, _props$response2;
-                  return _regeneratorRuntime().wrap(function _callee4$(_context4) {
-                    while (1) switch (_context4.prev = _context4.next) {
+                  return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+                    while (1) switch (_context3.prev = _context3.next) {
                       case 0:
                         _onStepFinish == null || _onStepFinish(JSON.stringify(props, null, 2));
                         if (!(props != null && (_props$response = props.response) != null && (_props$response = _props$response.headers) != null && _props$response['x-ratelimit-remaining-tokens'] && parseInt(props == null || (_props$response2 = props.response) == null || (_props$response2 = _props$response2.headers) == null ? void 0 : _props$response2['x-ratelimit-remaining-tokens'], 10) < 2000)) {
-                          _context4.next = 5;
+                          _context3.next = 5;
                           break;
                         }
                         _classPrivateFieldLooseBase(_this3, _logger$2)[_logger$2].warn('Rate limit approaching, waiting 10 seconds');
-                        _context4.next = 5;
+                        _context3.next = 5;
                         return delay(10 * 1000);
                       case 5:
                       case "end":
-                        return _context4.stop();
+                        return _context3.stop();
                     }
-                  }, _callee4);
+                  }, _callee3);
                 }));
-                function onStepFinish(_x6) {
+                function onStepFinish(_x5) {
                   return _onStepFinish2.apply(this, arguments);
                 }
                 return onStepFinish;
               }()
             };
-            _context5.next = 10;
+            _context4.next = 10;
             return ai.generateText(_extends({
               messages: messages
             }, argsForExecute, {
               experimental_telemetry: this.experimental_telemetry
             }));
           case 10:
-            return _context5.abrupt("return", _context5.sent);
+            return _context4.abrupt("return", _context4.sent);
           case 11:
           case "end":
-            return _context5.stop();
+            return _context4.stop();
         }
-      }, _callee5, this);
+      }, _callee4, this);
     }));
-    function __text(_x5) {
+    function __text(_x4) {
       return _text.apply(this, arguments);
     }
     return __text;
   }();
   _proto.__textObject = /*#__PURE__*/function () {
-    var _textObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(_ref10) {
+    var _textObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref9) {
       var _this4 = this;
-      var messages, _onStepFinish3, _ref10$maxSteps, maxSteps, enabledTools, convertedTools, structuredOutput, runId, model, modelToPass, params, argsForExecute, schema, output;
-      return _regeneratorRuntime().wrap(function _callee7$(_context7) {
-        while (1) switch (_context7.prev = _context7.next) {
+      var messages, _onStepFinish3, _ref9$maxSteps, maxSteps, enabledTools, convertedTools, structuredOutput, runId, model, modelToPass, params, argsForExecute, schema, output;
+      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
           case 0:
-            messages = _ref10.messages, _onStepFinish3 = _ref10.onStepFinish, _ref10$maxSteps = _ref10.maxSteps, maxSteps = _ref10$maxSteps === void 0 ? 5 : _ref10$maxSteps, enabledTools = _ref10.enabledTools, convertedTools = _ref10.convertedTools, structuredOutput = _ref10.structuredOutput, runId = _ref10.runId;
+            messages = _ref9.messages, _onStepFinish3 = _ref9.onStepFinish, _ref9$maxSteps = _ref9.maxSteps, maxSteps = _ref9$maxSteps === void 0 ? 5 : _ref9$maxSteps, enabledTools = _ref9.enabledTools, convertedTools = _ref9.convertedTools, structuredOutput = _ref9.structuredOutput, runId = _ref9.runId;
             model = _classPrivateFieldLooseBase(this, _model)[_model];
             _classPrivateFieldLooseBase(this, _log$2)[_log$2](LogLevel.DEBUG, "Generating text with " + messages.length + " messages", runId);
             if ('name' in model) {
@@ -1728,39 +1615,39 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               modelToPass = model;
             }
-            _context7.next = 6;
+            _context6.next = 6;
             return this.getParams({
               tools: convertedTools || this.convertTools(enabledTools || {}),
               model: modelToPass
             });
           case 6:
-            params = _context7.sent;
+            params = _context6.sent;
             argsForExecute = {
               model: params.modelDef,
               tools: _extends({}, params.toolsConverted, params.answerTool),
               toolChoice: params.toolChoice,
               maxSteps: maxSteps,
               onStepFinish: function () {
-                var _onStepFinish4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(props) {
+                var _onStepFinish4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(props) {
                   var _props$response3, _props$response4;
-                  return _regeneratorRuntime().wrap(function _callee6$(_context6) {
-                    while (1) switch (_context6.prev = _context6.next) {
+                  return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+                    while (1) switch (_context5.prev = _context5.next) {
                       case 0:
                         _onStepFinish3 == null || _onStepFinish3(JSON.stringify(props, null, 2));
                         if (!(props != null && (_props$response3 = props.response) != null && (_props$response3 = _props$response3.headers) != null && _props$response3['x-ratelimit-remaining-tokens'] && parseInt(props == null || (_props$response4 = props.response) == null || (_props$response4 = _props$response4.headers) == null ? void 0 : _props$response4['x-ratelimit-remaining-tokens'], 10) < 2000)) {
-                          _context6.next = 5;
+                          _context5.next = 5;
                           break;
                         }
                         _classPrivateFieldLooseBase(_this4, _logger$2)[_logger$2].warn('Rate limit approaching, waiting 10 seconds', runId);
-                        _context6.next = 5;
+                        _context5.next = 5;
                         return delay(10 * 1000);
                       case 5:
                       case "end":
-                        return _context6.stop();
+                        return _context5.stop();
                     }
-                  }, _callee6);
+                  }, _callee5);
                 }));
-                function onStepFinish(_x8) {
+                function onStepFinish(_x7) {
                   return _onStepFinish4.apply(this, arguments);
                 }
                 return onStepFinish;
@@ -1776,7 +1663,7 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               schema = this.createOutputSchema(structuredOutput);
             }
-            _context7.next = 12;
+            _context6.next = 12;
             return ai.generateObject(_extends({
               messages: messages
             }, argsForExecute, {
@@ -1785,26 +1672,26 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               experimental_telemetry: this.experimental_telemetry
             }));
           case 12:
-            return _context7.abrupt("return", _context7.sent);
+            return _context6.abrupt("return", _context6.sent);
           case 13:
           case "end":
-            return _context7.stop();
+            return _context6.stop();
         }
-      }, _callee7, this);
+      }, _callee6, this);
     }));
-    function __textObject(_x7) {
+    function __textObject(_x6) {
       return _textObject.apply(this, arguments);
     }
     return __textObject;
   }();
   _proto.__stream = /*#__PURE__*/function () {
-    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_ref11) {
+    var _stream = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref10) {
       var _this5 = this;
-      var messages, _onStepFinish5, _onFinish, _ref11$maxSteps, maxSteps, enabledTools, runId, convertedTools, model, modelToPass, params, argsForExecute;
-      return _regeneratorRuntime().wrap(function _callee10$(_context10) {
-        while (1) switch (_context10.prev = _context10.next) {
+      var messages, _onStepFinish5, _onFinish, _ref10$maxSteps, maxSteps, enabledTools, runId, convertedTools, model, modelToPass, params, argsForExecute;
+      return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
           case 0:
-            messages = _ref11.messages, _onStepFinish5 = _ref11.onStepFinish, _onFinish = _ref11.onFinish, _ref11$maxSteps = _ref11.maxSteps, maxSteps = _ref11$maxSteps === void 0 ? 5 : _ref11$maxSteps, enabledTools = _ref11.enabledTools, runId = _ref11.runId, convertedTools = _ref11.convertedTools;
+            messages = _ref10.messages, _onStepFinish5 = _ref10.onStepFinish, _onFinish = _ref10.onFinish, _ref10$maxSteps = _ref10.maxSteps, maxSteps = _ref10$maxSteps === void 0 ? 5 : _ref10$maxSteps, enabledTools = _ref10.enabledTools, runId = _ref10.runId, convertedTools = _ref10.convertedTools;
             model = _classPrivateFieldLooseBase(this, _model)[_model];
             _classPrivateFieldLooseBase(this, _log$2)[_log$2](LogLevel.DEBUG, "Streaming text with " + messages.length + " messages", runId);
             if ('name' in model) {
@@ -1819,88 +1706,88 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               modelToPass = model;
             }
-            _context10.next = 6;
+            _context9.next = 6;
             return this.getParams({
               tools: convertedTools || this.convertTools(enabledTools),
               model: modelToPass
             });
           case 6:
-            params = _context10.sent;
+            params = _context9.sent;
             argsForExecute = {
               model: params.modelDef,
               tools: _extends({}, params.toolsConverted, params.answerTool),
               toolChoice: params.toolChoice,
               maxSteps: maxSteps,
               onStepFinish: function () {
-                var _onStepFinish6 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(props) {
+                var _onStepFinish6 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(props) {
                   var _props$response5, _props$response6;
-                  return _regeneratorRuntime().wrap(function _callee8$(_context8) {
-                    while (1) switch (_context8.prev = _context8.next) {
+                  return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+                    while (1) switch (_context7.prev = _context7.next) {
                       case 0:
                         _onStepFinish5 == null || _onStepFinish5(JSON.stringify(props, null, 2));
                         if (!(props != null && (_props$response5 = props.response) != null && (_props$response5 = _props$response5.headers) != null && _props$response5['x-ratelimit-remaining-tokens'] && parseInt(props == null || (_props$response6 = props.response) == null || (_props$response6 = _props$response6.headers) == null ? void 0 : _props$response6['x-ratelimit-remaining-tokens'], 10) < 2000)) {
-                          _context8.next = 5;
+                          _context7.next = 5;
                           break;
                         }
                         _classPrivateFieldLooseBase(_this5, _logger$2)[_logger$2].warn('Rate limit approaching, waiting 10 seconds', runId);
-                        _context8.next = 5;
+                        _context7.next = 5;
                         return delay(10 * 1000);
                       case 5:
                       case "end":
-                        return _context8.stop();
+                        return _context7.stop();
                     }
-                  }, _callee8);
+                  }, _callee7);
                 }));
-                function onStepFinish(_x10) {
+                function onStepFinish(_x9) {
                   return _onStepFinish6.apply(this, arguments);
                 }
                 return onStepFinish;
               }(),
               onFinish: function () {
-                var _onFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(props) {
-                  return _regeneratorRuntime().wrap(function _callee9$(_context9) {
-                    while (1) switch (_context9.prev = _context9.next) {
+                var _onFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(props) {
+                  return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+                    while (1) switch (_context8.prev = _context8.next) {
                       case 0:
                         _onFinish == null || _onFinish(JSON.stringify(props, null, 2));
                       case 1:
                       case "end":
-                        return _context9.stop();
+                        return _context8.stop();
                     }
-                  }, _callee9);
+                  }, _callee8);
                 }));
-                function onFinish(_x11) {
+                function onFinish(_x10) {
                   return _onFinish2.apply(this, arguments);
                 }
                 return onFinish;
               }()
             };
-            _context10.next = 10;
+            _context9.next = 10;
             return ai.streamText(_extends({
               messages: messages
             }, argsForExecute, {
               experimental_telemetry: this.experimental_telemetry
             }));
           case 10:
-            return _context10.abrupt("return", _context10.sent);
+            return _context9.abrupt("return", _context9.sent);
           case 11:
           case "end":
-            return _context10.stop();
+            return _context9.stop();
         }
-      }, _callee10, this);
+      }, _callee9, this);
     }));
-    function __stream(_x9) {
+    function __stream(_x8) {
       return _stream.apply(this, arguments);
     }
     return __stream;
   }();
   _proto.__streamObject = /*#__PURE__*/function () {
-    var _streamObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(_ref12) {
+    var _streamObject = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(_ref11) {
       var _this6 = this;
-      var messages, _onStepFinish7, _onFinish3, _ref12$maxSteps, maxSteps, enabledTools, convertedTools, structuredOutput, runId, model, modelToPass, params, argsForExecute, schema, output;
-      return _regeneratorRuntime().wrap(function _callee13$(_context13) {
-        while (1) switch (_context13.prev = _context13.next) {
+      var messages, _onStepFinish7, _onFinish3, _ref11$maxSteps, maxSteps, enabledTools, convertedTools, structuredOutput, runId, model, modelToPass, params, argsForExecute, schema, output;
+      return _regeneratorRuntime().wrap(function _callee12$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
           case 0:
-            messages = _ref12.messages, _onStepFinish7 = _ref12.onStepFinish, _onFinish3 = _ref12.onFinish, _ref12$maxSteps = _ref12.maxSteps, maxSteps = _ref12$maxSteps === void 0 ? 5 : _ref12$maxSteps, enabledTools = _ref12.enabledTools, convertedTools = _ref12.convertedTools, structuredOutput = _ref12.structuredOutput, runId = _ref12.runId;
+            messages = _ref11.messages, _onStepFinish7 = _ref11.onStepFinish, _onFinish3 = _ref11.onFinish, _ref11$maxSteps = _ref11.maxSteps, maxSteps = _ref11$maxSteps === void 0 ? 5 : _ref11$maxSteps, enabledTools = _ref11.enabledTools, convertedTools = _ref11.convertedTools, structuredOutput = _ref11.structuredOutput, runId = _ref11.runId;
             model = _classPrivateFieldLooseBase(this, _model)[_model];
             _classPrivateFieldLooseBase(this, _log$2)[_log$2](LogLevel.DEBUG, "Streaming text with " + messages.length + " messages", runId);
             if ('name' in model) {
@@ -1915,56 +1802,56 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               modelToPass = model;
             }
-            _context13.next = 6;
+            _context12.next = 6;
             return this.getParams({
               tools: convertedTools || this.convertTools(enabledTools),
               model: modelToPass
             });
           case 6:
-            params = _context13.sent;
+            params = _context12.sent;
             argsForExecute = {
               model: params.modelDef,
               tools: _extends({}, params.toolsConverted, params.answerTool),
               toolChoice: params.toolChoice,
               maxSteps: maxSteps,
               onStepFinish: function () {
-                var _onStepFinish8 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(props) {
+                var _onStepFinish8 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(props) {
                   var _props$response7, _props$response8;
-                  return _regeneratorRuntime().wrap(function _callee11$(_context11) {
-                    while (1) switch (_context11.prev = _context11.next) {
+                  return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+                    while (1) switch (_context10.prev = _context10.next) {
                       case 0:
                         _onStepFinish7 == null || _onStepFinish7(JSON.stringify(props, null, 2));
                         if (!(props != null && (_props$response7 = props.response) != null && (_props$response7 = _props$response7.headers) != null && _props$response7['x-ratelimit-remaining-tokens'] && parseInt(props == null || (_props$response8 = props.response) == null || (_props$response8 = _props$response8.headers) == null ? void 0 : _props$response8['x-ratelimit-remaining-tokens'], 10) < 2000)) {
-                          _context11.next = 5;
+                          _context10.next = 5;
                           break;
                         }
                         _classPrivateFieldLooseBase(_this6, _logger$2)[_logger$2].warn('Rate limit approaching, waiting 10 seconds', runId);
-                        _context11.next = 5;
+                        _context10.next = 5;
                         return delay(10 * 1000);
                       case 5:
                       case "end":
-                        return _context11.stop();
+                        return _context10.stop();
                     }
-                  }, _callee11);
+                  }, _callee10);
                 }));
-                function onStepFinish(_x13) {
+                function onStepFinish(_x12) {
                   return _onStepFinish8.apply(this, arguments);
                 }
                 return onStepFinish;
               }(),
               onFinish: function () {
-                var _onFinish4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(props) {
-                  return _regeneratorRuntime().wrap(function _callee12$(_context12) {
-                    while (1) switch (_context12.prev = _context12.next) {
+                var _onFinish4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(props) {
+                  return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+                    while (1) switch (_context11.prev = _context11.next) {
                       case 0:
                         _onFinish3 == null || _onFinish3(JSON.stringify(props, null, 2));
                       case 1:
                       case "end":
-                        return _context12.stop();
+                        return _context11.stop();
                     }
-                  }, _callee12);
+                  }, _callee11);
                 }));
-                function onFinish(_x14) {
+                function onFinish(_x13) {
                   return _onFinish4.apply(this, arguments);
                 }
                 return onFinish;
@@ -1980,7 +1867,7 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
             } else {
               schema = this.createOutputSchema(structuredOutput);
             }
-            _context13.next = 12;
+            _context12.next = 12;
             return ai.streamObject(_extends({
               messages: messages
             }, argsForExecute, {
@@ -1989,14 +1876,14 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               experimental_telemetry: this.experimental_telemetry
             }));
           case 12:
-            return _context13.abrupt("return", _context13.sent);
+            return _context12.abrupt("return", _context12.sent);
           case 13:
           case "end":
-            return _context13.stop();
+            return _context12.stop();
         }
-      }, _callee13, this);
+      }, _callee12, this);
     }));
-    function __streamObject(_x12) {
+    function __streamObject(_x11) {
       return _streamObject.apply(this, arguments);
     }
     return __streamObject;
@@ -3957,6 +3844,18 @@ exports.Mastra = (_workflows = /*#__PURE__*/_classPrivateFieldLooseKey("workflow
     }
     return agent;
   };
+  _proto.getAgents = function getAgents() {
+    return Array.from(this.agents.entries()).map(function (_ref3) {
+      var name = _ref3[0],
+        agent = _ref3[1];
+      return {
+        name: name,
+        instructions: agent.instructions,
+        modelProvider: agent.model.provider,
+        modelName: agent.model.name
+      };
+    });
+  };
   _proto.getWorkflow = function getWorkflow(name) {
     var workflow = _classPrivateFieldLooseBase(this, _workflows)[_workflows].get(name);
     if (!workflow) {
@@ -3980,7 +3879,7 @@ exports.Mastra = (_workflows = /*#__PURE__*/_classPrivateFieldLooseKey("workflow
       throw new Error("Tool with name " + String(name) + " not found");
     }
     var hydratedExecutor = /*#__PURE__*/function () {
-      var _ref3 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(params, runId) {
+      var _ref4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(params, runId) {
         return _regeneratorRuntime().wrap(function _callee2$(_context2) {
           while (1) switch (_context2.prev = _context2.next) {
             case 0:
@@ -4006,7 +3905,7 @@ exports.Mastra = (_workflows = /*#__PURE__*/_classPrivateFieldLooseKey("workflow
         }, _callee2);
       }));
       return function hydratedExecutor(_x4, _x5) {
-        return _ref3.apply(this, arguments);
+        return _ref4.apply(this, arguments);
       };
     }();
     return _extends({}, tool, {
@@ -4019,9 +3918,9 @@ exports.Mastra = (_workflows = /*#__PURE__*/_classPrivateFieldLooseKey("workflow
     });
   };
   _proto.availableIntegrations = function availableIntegrations() {
-    return Array.from(this.integrations.entries()).map(function (_ref4) {
-      var name = _ref4[0],
-        integration = _ref4[1];
+    return Array.from(this.integrations.entries()).map(function (_ref5) {
+      var name = _ref5[0],
+        integration = _ref5[1];
       return {
         name: name,
         integration: integration
@@ -4031,8 +3930,8 @@ exports.Mastra = (_workflows = /*#__PURE__*/_classPrivateFieldLooseKey("workflow
   _proto.getTools = function getTools() {
     return this.tools;
   };
-  _proto.setLogger = function setLogger(_ref5) {
-    var logger = _ref5.logger;
+  _proto.setLogger = function setLogger(_ref6) {
+    var logger = _ref6.logger;
     this.logger = logger;
   };
   _proto.getLogger = function getLogger() {
@@ -5482,6 +5381,120 @@ var MastraMemory = /*#__PURE__*/function () {
   return MastraMemory;
 }();
 
+function embed(_x) {
+  return _embed.apply(this, arguments);
+}
+function _embed() {
+  _embed = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref) {
+    var model, value, maxRetries, embeddingModel, openai$1, cohere$1, amazon, google$1, mistral$1, voyage;
+    return _regeneratorRuntime().wrap(function _callee$(_context) {
+      while (1) switch (_context.prev = _context.next) {
+        case 0:
+          model = _ref.model, value = _ref.value, maxRetries = _ref.maxRetries;
+          if (!(model.provider === 'OPEN_AI')) {
+            _context.next = 6;
+            break;
+          }
+          openai$1 = openai.createOpenAI({
+            apiKey: model.apiKey || process.env.OPENAI_API_KEY
+          });
+          embeddingModel = openai$1.embedding(model.name);
+          _context.next = 32;
+          break;
+        case 6:
+          if (!(model.provider === 'COHERE')) {
+            _context.next = 11;
+            break;
+          }
+          cohere$1 = cohere.createCohere({
+            apiKey: model.apiKey || process.env.COHERE_API_KEY
+          });
+          embeddingModel = cohere$1.embedding(model.name);
+          _context.next = 32;
+          break;
+        case 11:
+          if (!(model.provider === 'AMAZON')) {
+            _context.next = 16;
+            break;
+          }
+          amazon = amazonBedrock.createAmazonBedrock({
+            region: process.env.AWS_REGION || '',
+            accessKeyId: process.env.AWS_ACCESS_KEY_ID || '',
+            secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY || '',
+            sessionToken: process.env.AWS_SESSION_TOKEN || ''
+          });
+          embeddingModel = amazon.embedding(model.name);
+          _context.next = 32;
+          break;
+        case 16:
+          if (!(model.provider === 'GOOGLE')) {
+            _context.next = 21;
+            break;
+          }
+          google$1 = google.createGoogleGenerativeAI({
+            baseURL: 'https://generativelanguage.googleapis.com/v1beta',
+            apiKey: model.apiKey || process.env.GOOGLE_GENERATIVE_AI_API_KEY || ''
+          });
+          embeddingModel = google$1.textEmbeddingModel(model.name);
+          _context.next = 32;
+          break;
+        case 21:
+          if (!(model.provider === 'MISTRAL')) {
+            _context.next = 26;
+            break;
+          }
+          mistral$1 = mistral.createMistral({
+            baseURL: 'https://api.mistral.ai/v1',
+            apiKey: model.apiKey || process.env.MISTRAL_API_KEY || ''
+          });
+          embeddingModel = mistral$1.textEmbeddingModel(model.name);
+          _context.next = 32;
+          break;
+        case 26:
+          if (!(model.provider === 'VOYAGE')) {
+            _context.next = 31;
+            break;
+          }
+          voyage = voyageAiProvider.createVoyage({
+            baseURL: 'https://api.voyageai.com/v1',
+            apiKey: model.apiKey || process.env.VOYAGE_API_KEY || ''
+          });
+          embeddingModel = voyage.textEmbeddingModel(model.name);
+          _context.next = 32;
+          break;
+        case 31:
+          throw new Error("Invalid embedding model");
+        case 32:
+          if (!(value instanceof Array)) {
+            _context.next = 36;
+            break;
+          }
+          _context.next = 35;
+          return ai.embedMany({
+            model: embeddingModel,
+            values: value,
+            maxRetries: maxRetries
+          });
+        case 35:
+          return _context.abrupt("return", _context.sent);
+        case 36:
+          _context.next = 38;
+          return ai.embed({
+            model: embeddingModel,
+            value: value,
+            maxRetries: maxRetries
+          });
+        case 38:
+          return _context.abrupt("return", _context.sent);
+        case 39:
+        case "end":
+          return _context.stop();
+      }
+    }, _callee);
+  }));
+  return _embed.apply(this, arguments);
+}
+
 exports.BaseLogger = BaseLogger;
 exports.ConsoleLogger = ConsoleLogger;
 exports.FileLogger = FileLogger;
@@ -5502,6 +5515,7 @@ exports.createLogger = createLogger;
 exports.createMultiLogger = createMultiLogger;
 exports.createSync = createSync;
 exports.createTool = createTool;
+exports.embed = embed;
 exports.hasActiveTelemetry = hasActiveTelemetry;
 exports.withSpan = withSpan;
 //# sourceMappingURL=core.cjs.development.js.map