@mastra/core 0.1.27-alpha.27 → 0.1.27-alpha.28

package/dist/core.cjs.development.js

@@ -15,6 +15,7 @@ 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');
@@ -1294,7 +1295,7 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
   };
   _proto.createEmbedding = /*#__PURE__*/function () {
     var _createEmbedding = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref4) {
-      var model, value, maxRetries, embeddingModel, openai$1, cohere$1;
+      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:
@@ -1307,7 +1308,7 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               apiKey: process.env.OPENAI_API_KEY
             });
             embeddingModel = openai$1.embedding(model.name);
-            _context.next = 12;
+            _context.next = 32;
             break;
           case 6:
             if (!(model.provider === 'COHERE')) {
@@ -1318,33 +1319,83 @@ var LLM = (_model = /*#__PURE__*/_classPrivateFieldLooseKey("model"), _tools$1 =
               apiKey: process.env.COHERE_API_KEY
             });
             embeddingModel = cohere$1.embedding(model.name);
-            _context.next = 12;
+            _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 12:
+          case 32:
             if (!(value instanceof Array)) {
-              _context.next = 16;
+              _context.next = 36;
               break;
             }
-            _context.next = 15;
+            _context.next = 35;
             return ai.embedMany({
               model: embeddingModel,
               values: value,
               maxRetries: maxRetries
             });
-          case 15:
+          case 35:
             return _context.abrupt("return", _context.sent);
-          case 16:
-            _context.next = 18;
+          case 36:
+            _context.next = 38;
             return ai.embed({
               model: embeddingModel,
               value: value,
               maxRetries: maxRetries
             });
-          case 18:
+          case 38:
             return _context.abrupt("return", _context.sent);
-          case 19:
+          case 39:
           case "end":
             return _context.stop();
         }