@mastra/core 0.1.3 → 0.1.5

package/dist/core.esm.js

@@ -11,11 +11,17 @@ import qs from 'qs';
 import { serve } from 'inngest/next';
 import path from 'path';
 import * as fs from 'fs';
+import { readdirSync, readFileSync, existsSync, mkdirSync, writeFileSync } from 'fs';
 import first from 'lodash-es/first';
 import last from 'lodash-es/last';
 import { Inngest } from 'inngest';
 import { OAuth2Client } from '@badgateway/oauth2-client';
-import _, { isEmpty, omit, omitBy } from 'lodash-es';
+import _, { isEmpty, omit, pick, compact, omitBy } from 'lodash-es';
+import { openai, createOpenAI } from '@ai-sdk/openai';
+import { embed, tool, generateText, streamText } from 'ai';
+import { Pinecone } from '@pinecone-database/pinecone';
+import OpenAI from 'openai';
+import { createAnthropic } from '@ai-sdk/anthropic';
 
 var WorkflowStatusEnum = {
   DRAFT: 'DRAFT',
@@ -1841,7 +1847,7 @@ var makeCallback = function makeCallback(framework) {
 };
 
 var APP_ID = 'mastra';
-var client = /*#__PURE__*/new Inngest({
+var client$1 = /*#__PURE__*/new Inngest({
   id: APP_ID
 });
 
@@ -1948,7 +1954,7 @@ function _getBlueprints() {
 var createWorkflowHandler = function createWorkflowHandler(_ref2) {
   var blueprintDirPath = _ref2.blueprintDirPath,
     runBlueprint = _ref2.runBlueprint;
-  return client.createFunction({
+  return client$1.createFunction({
     id: 'workflow-executor',
     retries: 0
   }, {
@@ -2043,18 +2049,35 @@ var makeInngest = function makeInngest(framework) {
   // TODO: hook into framework to add framework functions to the inngest client
   var eventHandlers = framework.getGlobalEventHandlers();
   var globalEventHandlers = eventHandlers.map(function (eh) {
-    return client.createFunction({
+    return client$1.createFunction({
       id: eh.id
     }, {
       event: eh.event
-    }, eh.executor);
+    }, /*#__PURE__*/function () {
+      var _ref = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(props) {
+        return _regeneratorRuntime().wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              return _context.abrupt("return", eh.executor(_extends({}, props, {
+                mastra: framework
+              })));
+            case 1:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee);
+      }));
+      return function (_x) {
+        return _ref.apply(this, arguments);
+      };
+    }());
   });
   var systemWorkflowHandler = createWorkflowHandler({
     blueprintDirPath: (_framework$config$wor = framework.config.workflows) == null ? void 0 : _framework$config$wor.blueprintDirPath,
     runBlueprint: framework.runBlueprint
   });
   var handler = serve({
-    client: client,
+    client: client$1,
     functions: [].concat(globalEventHandlers, [systemWorkflowHandler])
   });
   // @ts-ignore
@@ -2413,19 +2436,6 @@ var Integration = /*#__PURE__*/function () {
         return _ref4.apply(this, arguments);
       };
     }();
-    this.getEventHandlers = function (_ref5) {
-      var makeWebhookUrl = _ref5.makeWebhookUrl;
-      return Object.keys(_this.events).map(function (eventKey) {
-        var _this$events$eventKey;
-        var eventHandler = (_this$events$eventKey = _this.events[eventKey]) == null ? void 0 : _this$events$eventKey.handler;
-        if (!eventHandler) return null;
-        return eventHandler({
-          integrationInstance: _this,
-          eventKey: eventKey,
-          makeWebhookUrl: makeWebhookUrl
-        });
-      }).filter(Boolean);
-    };
     if (!(config != null && config.name)) {
       throw new IntegrationError('Integration name must be defined');
     }
@@ -2470,12 +2480,12 @@ var Integration = /*#__PURE__*/function () {
     return getApiClient;
   }();
   _proto.getApi = /*#__PURE__*/function () {
-    var _getApi = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref6) {
+    var _getApi = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref5) {
       var connectionId;
       return _regeneratorRuntime().wrap(function _callee5$(_context5) {
         while (1) switch (_context5.prev = _context5.next) {
           case 0:
-            connectionId = _ref6.connectionId;
+            connectionId = _ref5.connectionId;
             _context5.next = 3;
             return this.getApiClient({
               connectionId: connectionId
@@ -2499,8 +2509,8 @@ var Integration = /*#__PURE__*/function () {
   _proto.getAuthenticator = function getAuthenticator() {
     throw new IntegrationError('Authenticator not implemented');
   };
-  _proto.attachDataLayer = function attachDataLayer(_ref7) {
-    var dataLayer = _ref7.dataLayer;
+  _proto.attachDataLayer = function attachDataLayer(_ref6) {
+    var dataLayer = _ref6.dataLayer;
     this.dataLayer = dataLayer;
   };
   _proto.registerApis = function registerApis() {
@@ -2557,14 +2567,14 @@ var Integration = /*#__PURE__*/function () {
     return processWebhookRequest;
   }();
   _proto.triggerEvent = /*#__PURE__*/function () {
-    var _triggerEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref8) {
+    var _triggerEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref7) {
       var key, data, user, event;
       return _regeneratorRuntime().wrap(function _callee8$(_context8) {
         while (1) switch (_context8.prev = _context8.next) {
           case 0:
-            key = _ref8.key, data = _ref8.data, user = _ref8.user;
+            key = _ref7.key, data = _ref7.data, user = _ref7.user;
             _context8.next = 3;
-            return client.send({
+            return client$1.send({
               name: key,
               data: data,
               user: user
@@ -2584,13 +2594,13 @@ var Integration = /*#__PURE__*/function () {
     return triggerEvent;
   }();
   _proto.test = /*#__PURE__*/function () {
-    var _test = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref9) {
+    var _test = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref8) {
       var _this$dataLayer;
       var connectionId, connection, _this$config$scopes, authenticator, bearer, desiredScopes, _bearer$scope, actualScopes, isMissingScopes;
       return _regeneratorRuntime().wrap(function _callee9$(_context9) {
         while (1) switch (_context9.prev = _context9.next) {
           case 0:
-            connectionId = _ref9.connectionId;
+            connectionId = _ref8.connectionId;
             _context9.next = 3;
             return (_this$dataLayer = this.dataLayer) == null ? void 0 : _this$dataLayer.getConnection({
               connectionId: connectionId,
@@ -2641,7 +2651,7 @@ var Integration = /*#__PURE__*/function () {
     return test;
   }();
   _proto.onConnectionCreated = /*#__PURE__*/function () {
-    var _onConnectionCreated = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_ref10) {
+    var _onConnectionCreated = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_ref9) {
       return _regeneratorRuntime().wrap(function _callee10$(_context10) {
         while (1) switch (_context10.prev = _context10.next) {
           case 0:
@@ -2657,7 +2667,7 @@ var Integration = /*#__PURE__*/function () {
     return onConnectionCreated;
   }();
   _proto.onDisconnect = /*#__PURE__*/function () {
-    var _onDisconnect = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(_ref11) {
+    var _onDisconnect = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(_ref10) {
       return _regeneratorRuntime().wrap(function _callee11$(_context11) {
         while (1) switch (_context11.prev = _context11.next) {
           case 0:
@@ -4047,9 +4057,9 @@ function _runActionsRecursively() {
                     data: data
                   });
                   executorResult = {};
-                  _ctx = mastraConnectionId ? {
+                  _ctx = mastraConnectionId ? _extends({}, ctx, {
                     connectionId: mastraConnectionId
-                  } : ctx;
+                  }) : ctx;
                   _context2.prev = 46;
                   _context2.next = 49;
                   return actionExecutor({
@@ -4216,266 +4226,1983 @@ function _blueprintRunner() {
   return _blueprintRunner.apply(this, arguments);
 }
 
-var Framework = /*#__PURE__*/function () {
-  function Framework(_ref) {
-    var _this = this;
-    var dataLayer = _ref.dataLayer,
-      config = _ref.config;
-    //global events grouped by Integration
-    this.globalEvents = new Map();
-    // global event handlers
-    this.globalEventHandlers = [];
-    // global apis grouped by Integration
-    this.globalApis = new Map();
-    this.integrations = new Map();
-    this.dataLayer = void 0;
-    this.config = void 0;
-    this.runBlueprint = /*#__PURE__*/function () {
-      var _ref3 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref2) {
-        var _this$availableIntegr;
-        var blueprint, _ref2$dataCtx, dataCtx, ctx, systemApis, systemEvents, availableIntegrations, availableIntegrationApis, availableIntegrationEvents, frameworkApis, frameworkEvents;
-        return _regeneratorRuntime().wrap(function _callee$(_context) {
-          while (1) switch (_context.prev = _context.next) {
-            case 0:
-              blueprint = _ref2.blueprint, _ref2$dataCtx = _ref2.dataCtx, dataCtx = _ref2$dataCtx === void 0 ? {} : _ref2$dataCtx, ctx = _ref2.ctx;
-              systemApis = _this.getSystemApis();
-              systemEvents = _this.getSystemEvents();
-              availableIntegrations = (_this$availableIntegr = _this.availableIntegrations()) == null ? void 0 : _this$availableIntegr.map(function (_ref4) {
-                var integration = _ref4.integration;
-                return integration;
-              });
-              availableIntegrationApis = availableIntegrations.reduce(function (acc, _ref5) {
-                var name = _ref5.name;
-                var apis = _this.getApisByIntegration(name);
-                return _extends({}, acc, apis);
-              }, {});
-              availableIntegrationEvents = availableIntegrations.reduce(function (acc, _ref6) {
-                var name = _ref6.name;
-                var events = _this.getEventsByIntegration(name);
-                return _extends({}, acc, events);
-              }, {});
-              frameworkApis = _extends({}, systemApis, availableIntegrationApis);
-              frameworkEvents = _extends({}, systemEvents, availableIntegrationEvents);
-              _context.next = 10;
-              return blueprintRunner({
-                dataCtx: dataCtx,
-                blueprint: blueprint,
-                frameworkApis: frameworkApis,
-                frameworkEvents: frameworkEvents,
-                ctx: ctx
-              });
-            case 10:
-            case "end":
-              return _context.stop();
-          }
-        }, _callee);
-      }));
-      return function (_x) {
-        return _ref3.apply(this, arguments);
-      };
-    }();
-    this.dataLayer = dataLayer;
-    this.config = config;
+function getAgentDir(_ref) {
+  var agentDir = _ref.agentDir;
+  var ARK_APP_DIR = process.env.ARK_APP_DIR || process.cwd();
+  return path.join(ARK_APP_DIR, agentDir);
+}
+function listAgentsJson(_ref2) {
+  var agentDir = _ref2.agentDir;
+  var agentDirPath = getAgentDir({
+    agentDir: agentDir
+  });
+  var agents = readdirSync(agentDirPath);
+  return agents;
+}
+function getAgentFile(agentFilePath) {
+  try {
+    var fileP = agentFilePath;
+    var fileB = readFileSync(fileP, 'utf8');
+    var json = JSON.parse(fileB);
+    return json;
+  } catch (e) {
+    console.error('No agent blueprint found:', agentFilePath);
+    return;
   }
-  Framework.init = function init(config) {
-    var _config$workflows$sys;
-    if (!config.db.uri) {
-      throw new Error('No database config/provider found');
-    }
-    var dataLayer = new DataLayer({
-      url: config.db.uri,
-      provider: config.db.provider
-    });
-    var framework = new Framework({
-      config: config,
-      dataLayer: dataLayer
-    });
-    // Register integrations
-    config.integrations.forEach(function (integration) {
-      framework.__registerIntgeration(integration);
-    });
-    // Register System apis
-    framework.__registerApis({
-      apis: (_config$workflows$sys = config.workflows.systemApis) == null ? void 0 : _config$workflows$sys.map(function (api) {
-        return _extends({}, api, {
-          integrationName: config.name
-        });
-      })
-    });
-    // Register System events
-    framework.__registerEvents({
-      events: config.workflows.systemEvents
-    });
-    return framework;
-  };
-  var _proto = Framework.prototype;
-  _proto.connectedIntegrations = /*#__PURE__*/function () {
-    var _connectedIntegrations = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref7) {
-      var _this2 = this;
-      var context, ints, connectionChecks;
-      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
-        while (1) switch (_context3.prev = _context3.next) {
+}
+function getAgentBlueprint(_ref3) {
+  var agentId = _ref3.agentId,
+    agentDir = _ref3.agentDir;
+  var agentDirPath = getAgentDir({
+    agentDir: agentDir
+  });
+  var agentBlueprintPath = path.join(agentDirPath, agentId + ".json");
+  return getAgentFile(agentBlueprintPath);
+}
+
+var VectorLayer = /*#__PURE__*/function () {
+  function VectorLayer() {
+    this.supportedProviders = ['PINECONE'];
+  }
+  var _proto = VectorLayer.prototype;
+  _proto.getPineconeIndexes = /*#__PURE__*/function () {
+    var _getPineconeIndexes = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
           case 0:
-            context = _ref7.context;
-            ints = this.availableIntegrations();
-            _context3.next = 4;
-            return Promise.all(ints.map(/*#__PURE__*/function () {
-              var _ref9 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref8) {
-                var integration, connection;
-                return _regeneratorRuntime().wrap(function _callee2$(_context2) {
-                  while (1) switch (_context2.prev = _context2.next) {
-                    case 0:
-                      integration = _ref8.integration;
-                      _context2.next = 3;
-                      return _this2.dataLayer.getConnection({
-                        connectionId: context.connectionId,
-                        name: integration.name
-                      });
-                    case 3:
-                      connection = _context2.sent;
-                      return _context2.abrupt("return", {
-                        integration: integration,
-                        connected: !!connection
-                      });
-                    case 5:
-                    case "end":
-                      return _context2.stop();
-                  }
-                }, _callee2);
-              }));
-              return function (_x3) {
-                return _ref9.apply(this, arguments);
-              };
-            }()));
-          case 4:
-            connectionChecks = _context3.sent;
-            return _context3.abrupt("return", connectionChecks.filter(function (_ref10) {
-              var connected = _ref10.connected;
-              return connected;
-            }).map(function (_ref11) {
-              var integration = _ref11.integration;
-              return integration;
-            }));
-          case 6:
+            return _context.abrupt("return", this.Pinecone.listIndexes());
+          case 1:
           case "end":
-            return _context3.stop();
+            return _context.stop();
         }
-      }, _callee3, this);
+      }, _callee, this);
     }));
-    function connectedIntegrations(_x2) {
-      return _connectedIntegrations.apply(this, arguments);
+    function getPineconeIndexes() {
+      return _getPineconeIndexes.apply(this, arguments);
     }
-    return connectedIntegrations;
+    return getPineconeIndexes;
   }();
-  _proto.__registerIntgeration = function __registerIntgeration(definition) {
-    var _this$globalEventHand;
-    var name = definition.name;
-    definition.attachDataLayer({
-      dataLayer: this.dataLayer
-    });
-    var router = this.createRouter();
-    definition.corePresets = {
-      redirectURI: router.makeRedirectURI()
-    };
-    this.integrations.set(name, definition);
-    definition.registerEvents();
-    this.__registerEvents({
-      events: definition.getEvents(),
-      integrationName: name
-    });
-    definition.registerApis();
-    definition._convertApiClientToSystemApis();
-    this.__registerApis({
-      apis: Object.values(definition.getApis()),
-      integrationName: name
-    });
-    (_this$globalEventHand = this.globalEventHandlers).push.apply(_this$globalEventHand, definition.getEventHandlers({
-      makeWebhookUrl: router.makeWebhookUrl
-    }));
-  };
-  _proto.__registerEvents = function __registerEvents(_ref12) {
-    var events = _ref12.events,
-      _ref12$integrationNam = _ref12.integrationName,
-      integrationName = _ref12$integrationNam === void 0 ? this.config.name : _ref12$integrationNam;
-    var integrationEvents = this.globalEvents.get(integrationName) || {};
-    this.globalEvents.set(integrationName, _extends({}, integrationEvents, events));
-  };
-  _proto.__registerApis = function __registerApis(_ref13) {
-    var apis = _ref13.apis,
-      _ref13$integrationNam = _ref13.integrationName,
-      integrationName = _ref13$integrationNam === void 0 ? this.config.name : _ref13$integrationNam;
-    var integrationApis = this.globalApis.get(integrationName) || {};
-    this.globalApis.set(integrationName, _extends({}, integrationApis, apis.reduce(function (acc, api) {
-      var _extends2;
-      return _extends({}, acc, (_extends2 = {}, _extends2[api.type] = api, _extends2));
-    }, {})));
-  };
-  _proto.availableIntegrations = function availableIntegrations() {
-    return Array.from(this.integrations.entries()).map(function (_ref14) {
-      var name = _ref14[0],
-        integration = _ref14[1];
-      return {
-        name: name,
-        integration: integration
-      };
-    });
-  };
-  _proto.getIntegration = function getIntegration(name) {
-    return this.integrations.get(name);
-  };
-  _proto.getGlobalEvents = function getGlobalEvents() {
-    return this.globalEvents;
-  };
-  _proto.getSystemEvents = function getSystemEvents() {
-    var events = this.globalEvents.get(this.config.name);
-    return events || {};
-  };
-  _proto.getEventsByIntegration = function getEventsByIntegration(name) {
-    return this.globalEvents.get(name);
-  };
-  _proto.getGlobalEventHandlers = function getGlobalEventHandlers() {
-    return this.globalEventHandlers;
-  };
-  _proto.getApis = function getApis() {
-    return this.globalApis;
-  };
-  _proto.getSystemApis = function getSystemApis() {
-    return this.globalApis.get(this.config.name);
-  };
-  _proto.getApisByIntegration = function getApisByIntegration(name, includeHidden) {
-    var integrationApis = this.globalApis.get(name);
-    if (includeHidden) {
-      return integrationApis;
-    }
-    return omitBy(integrationApis, function (value) {
-      return value.isHidden;
-    });
-  };
-  _proto.authenticatableIntegrations = function authenticatableIntegrations() {
-    return this.availableIntegrations().filter(function (_ref15) {
-      var integration = _ref15.integration;
-      try {
-        integration.getAuthenticator();
-        return true;
-      } catch (e) {
-        return false;
+  return _createClass(VectorLayer, [{
+    key: "Pinecone",
+    get: function get() {
+      if (!process.env.PINECONE_API_KEY) {
+        throw new Error('Pinecone API key not found');
       }
-    });
-  };
-  _proto.authenticator = function authenticator(name) {
-    var _int = this.getIntegration(name);
-    if (!_int) {
-      throw new Error("No integration exists for " + name);
+      return new Pinecone({
+        apiKey: process.env.PINECONE_API_KEY
+      });
+    }
+  }]);
+}();
+
+function getVectorProvider(provider) {
+  if (provider === 'PINECONE') {
+    var _VectorLayer = new VectorLayer(),
+      Pinecone = _VectorLayer.Pinecone;
+    return Pinecone;
+  }
+}
+var delay$1 = function delay(ms) {
+  return new Promise(function (resolve) {
+    return setTimeout(resolve, ms);
+  });
+};
+function executeGenericVectorSync(_x) {
+  return _executeGenericVectorSync.apply(this, arguments);
+}
+function _executeGenericVectorSync() {
+  _executeGenericVectorSync = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref) {
+    var _event$user, _event$data, _event$data2;
+    var event, mastra, connectionId, vector_provider, entities, systemName, vp, _iterator, _step, _yield$mastra$dataLay, _mastra$dataLayer, vectorE, integrationName, k_id, _mastra$dataLayer2, connection, _loop, _ret, _iterator2, _step2;
+    return _regeneratorRuntime().wrap(function _callee3$(_context4) {
+      while (1) switch (_context4.prev = _context4.next) {
+        case 0:
+          event = _ref.event, mastra = _ref.mastra;
+          connectionId = event == null || (_event$user = event.user) == null ? void 0 : _event$user.connectionId;
+          vector_provider = event == null || (_event$data = event.data) == null ? void 0 : _event$data.vector_provider;
+          entities = event == null || (_event$data2 = event.data) == null ? void 0 : _event$data2.entities;
+          systemName = mastra.config.name;
+          if (vector_provider) {
+            _context4.next = 8;
+            break;
+          }
+          console.error("No vector_provider defined for agent");
+          return _context4.abrupt("return");
+        case 8:
+          vp = getVectorProvider(vector_provider);
+          if (vp) {
+            _context4.next = 12;
+            break;
+          }
+          console.error('UNSUPPORTED VECTOR PROVIDER', vector_provider);
+          return _context4.abrupt("return");
+        case 12:
+          if (process.env.OPENAI_API_KEY) {
+            _context4.next = 15;
+            break;
+          }
+          console.error('NO OPENAI_API_KEY');
+          return _context4.abrupt("return");
+        case 15:
+          _iterator = _createForOfIteratorHelperLoose(entities);
+        case 16:
+          if ((_step = _iterator()).done) {
+            _context4.next = 50;
+            break;
+          }
+          vectorE = _step.value;
+          integrationName = vectorE.integration;
+          _context4.next = 21;
+          return (_mastra$dataLayer = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer.getConnection({
+            connectionId: connectionId,
+            name: integrationName
+          });
+        case 21:
+          _context4.t1 = _yield$mastra$dataLay = _context4.sent;
+          if (!(_context4.t1 == null)) {
+            _context4.next = 26;
+            break;
+          }
+          _context4.t2 = void 0;
+          _context4.next = 27;
+          break;
+        case 26:
+          _context4.t2 = _yield$mastra$dataLay.id;
+        case 27:
+          _context4.t0 = _context4.t2;
+          if (_context4.t0) {
+            _context4.next = 30;
+            break;
+          }
+          _context4.t0 = '';
+        case 30:
+          k_id = _context4.t0;
+          if (!(!k_id && integrationName === systemName)) {
+            _context4.next = 36;
+            break;
+          }
+          _context4.next = 34;
+          return (_mastra$dataLayer2 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer2.createConnection({
+            connection: {
+              connectionId: connectionId,
+              name: integrationName
+            },
+            credential: {
+              type: 'API_KEY',
+              value: connectionId,
+              scope: []
+            }
+          });
+        case 34:
+          connection = _context4.sent;
+          k_id = connection.id;
+        case 36:
+          if (k_id) {
+            _context4.next = 39;
+            break;
+          }
+          console.error('Error bootstrapping shit');
+          return _context4.abrupt("return");
+        case 39:
+          _loop = /*#__PURE__*/_regeneratorRuntime().mark(function _loop() {
+            var _mastra$dataLayer3, _entityRecords, _records;
+            var entity, entityRecords, records, _mastra$dataLayer4, _entityRecords2, _records2, _yield$mastra$trigger, _event, res, recordsMapped, entityTypeIndex, _recordsMapped$map, vectors;
+            return _regeneratorRuntime().wrap(function _loop$(_context3) {
+              while (1) switch (_context3.prev = _context3.next) {
+                case 0:
+                  entity = _step2.value;
+                  _context3.next = 3;
+                  return (_mastra$dataLayer3 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer3.getEntityRecordsByConnectionAndType({
+                    k_id: k_id,
+                    type: entity.name
+                  });
+                case 3:
+                  entityRecords = _context3.sent;
+                  records = (_entityRecords = entityRecords) == null ? void 0 : _entityRecords.records;
+                  if (!(!records || ((_records = records) == null ? void 0 : _records.length) === 0)) {
+                    _context3.next = 23;
+                    break;
+                  }
+                  // @TODO: SYNC THEM
+                  console.error('NO RECORDS');
+                  _context3.next = 9;
+                  return mastra.triggerEvent({
+                    key: entity.syncEvent,
+                    data: {},
+                    user: {
+                      connectionId: connectionId
+                    }
+                  });
+                case 9:
+                  _yield$mastra$trigger = _context3.sent;
+                  _event = _yield$mastra$trigger.event;
+                  console.log(_event, '####');
+                  _context3.next = 14;
+                  return _event.subscribe();
+                case 14:
+                  res = _context3.sent;
+                  console.log('Subscribe result', res);
+                  _context3.next = 18;
+                  return (_mastra$dataLayer4 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer4.getEntityRecordsByConnectionAndType({
+                    k_id: k_id,
+                    type: entity.name
+                  });
+                case 18:
+                  entityRecords = _context3.sent;
+                  records = (_entityRecords2 = entityRecords) == null ? void 0 : _entityRecords2.records;
+                  if (!(!records || ((_records2 = records) == null ? void 0 : _records2.length) === 0)) {
+                    _context3.next = 23;
+                    break;
+                  }
+                  console.error('NO RECORDS AFTER SYNC');
+                  return _context3.abrupt("return", {
+                    v: void 0
+                  });
+                case 23:
+                  recordsMapped = records.map(function (_ref7) {
+                    var data = _ref7.data,
+                      externalId = _ref7.externalId;
+                    return _extends({
+                      id: externalId
+                    }, pick(data, entity.fields));
+                  });
+                  console.log(recordsMapped, 'RECORDS');
+                  _context3.next = 27;
+                  return vp.createIndex({
+                    suppressConflicts: true,
+                    name: entity.name,
+                    dimension: 1536,
+                    metric: 'cosine',
+                    spec: {
+                      serverless: {
+                        cloud: 'aws',
+                        region: 'us-east-1'
+                      }
+                    }
+                  });
+                case 27:
+                  _context3.next = 29;
+                  return delay$1(5000);
+                case 29:
+                  entityTypeIndex = vp.index(entity.name);
+                  console.log(entityTypeIndex, 'INDEX');
+                  _context3.prev = 31;
+                  _context3.next = 34;
+                  return Promise.all((_recordsMapped$map = recordsMapped.map(/*#__PURE__*/function () {
+                    var _ref8 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(record) {
+                      var _yield$embed, embedding;
+                      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+                        while (1) switch (_context2.prev = _context2.next) {
+                          case 0:
+                            _context2.next = 2;
+                            return embed({
+                              model: openai.embedding('text-embedding-3-small'),
+                              value: JSON.stringify(record)
+                            });
+                          case 2:
+                            _yield$embed = _context2.sent;
+                            embedding = _yield$embed.embedding;
+                            return _context2.abrupt("return", {
+                              id: record.id,
+                              values: embedding,
+                              metadata: record
+                            });
+                          case 5:
+                          case "end":
+                            return _context2.stop();
+                        }
+                      }, _callee2);
+                    }));
+                    return function (_x5) {
+                      return _ref8.apply(this, arguments);
+                    };
+                  }())) != null ? _recordsMapped$map : []);
+                case 34:
+                  vectors = _context3.sent;
+                  console.log('UPSERTING', vectors);
+                  if (!(vectors.length > 0)) {
+                    _context3.next = 39;
+                    break;
+                  }
+                  _context3.next = 39;
+                  return entityTypeIndex.namespace(entity.name).upsert(vectors);
+                case 39:
+                  _context3.next = 44;
+                  break;
+                case 41:
+                  _context3.prev = 41;
+                  _context3.t0 = _context3["catch"](31);
+                  console.error(_context3.t0);
+                case 44:
+                  return _context3.abrupt("return", {
+                    v: void 0
+                  });
+                case 45:
+                case "end":
+                  return _context3.stop();
+              }
+            }, _loop, null, [[31, 41]]);
+          });
+          _iterator2 = _createForOfIteratorHelperLoose(vectorE.data);
+        case 41:
+          if ((_step2 = _iterator2()).done) {
+            _context4.next = 48;
+            break;
+          }
+          return _context4.delegateYield(_loop(), "t3", 43);
+        case 43:
+          _ret = _context4.t3;
+          if (!_ret) {
+            _context4.next = 46;
+            break;
+          }
+          return _context4.abrupt("return", _ret.v);
+        case 46:
+          _context4.next = 41;
+          break;
+        case 48:
+          _context4.next = 16;
+          break;
+        case 50:
+        case "end":
+          return _context4.stop();
+      }
+    }, _callee3);
+  }));
+  return _executeGenericVectorSync.apply(this, arguments);
+}
+function executeVectorSync(_x2) {
+  return _executeVectorSync.apply(this, arguments);
+}
+function _executeVectorSync() {
+  _executeVectorSync = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref2) {
+    var _event$user2, _agent$knowledge_sour3;
+    var event, mastra, agentDir, agent, connectionId, systemName, vp, _agent$knowledge_sour4, _iterator3, _step3, _yield$mastra$dataLay2, _mastra$dataLayer5, vectorE, integrationName, k_id, _mastra$dataLayer6, connection, _loop2, _ret2, _iterator4, _step4;
+    return _regeneratorRuntime().wrap(function _callee5$(_context7) {
+      while (1) switch (_context7.prev = _context7.next) {
+        case 0:
+          event = _ref2.event, mastra = _ref2.mastra;
+          agentDir = mastra.config.agents.agentDirPath;
+          agent = getAgentBlueprint({
+            agentDir: agentDir,
+            agentId: event.data.agentId
+          });
+          connectionId = event == null || (_event$user2 = event.user) == null ? void 0 : _event$user2.connectionId;
+          systemName = mastra.config.name;
+          if (agent.knowledge_sources.vector_provider) {
+            _context7.next = 8;
+            break;
+          }
+          console.error("No vector_provider defined for agent");
+          return _context7.abrupt("return");
+        case 8:
+          vp = getVectorProvider(agent == null || (_agent$knowledge_sour3 = agent.knowledge_sources) == null ? void 0 : _agent$knowledge_sour3.vector_provider);
+          if (vp) {
+            _context7.next = 12;
+            break;
+          }
+          console.error('UNSUPPORTED VECTOR PROVIDER', agent == null || (_agent$knowledge_sour4 = agent.knowledge_sources) == null ? void 0 : _agent$knowledge_sour4.vector_provider);
+          return _context7.abrupt("return");
+        case 12:
+          if (process.env.OPENAI_API_KEY) {
+            _context7.next = 15;
+            break;
+          }
+          console.error('NO OPENAI_API_KEY');
+          return _context7.abrupt("return");
+        case 15:
+          if (!(!agent || !(agent != null && agent.knowledge_sources))) {
+            _context7.next = 18;
+            break;
+          }
+          console.error('NO AGENT OR KNOWLEDGE SOURCES');
+          return _context7.abrupt("return");
+        case 18:
+          _iterator3 = _createForOfIteratorHelperLoose(agent.knowledge_sources.entities);
+        case 19:
+          if ((_step3 = _iterator3()).done) {
+            _context7.next = 53;
+            break;
+          }
+          vectorE = _step3.value;
+          integrationName = vectorE.integration;
+          _context7.next = 24;
+          return (_mastra$dataLayer5 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer5.getConnection({
+            connectionId: connectionId,
+            name: integrationName
+          });
+        case 24:
+          _context7.t1 = _yield$mastra$dataLay2 = _context7.sent;
+          if (!(_context7.t1 == null)) {
+            _context7.next = 29;
+            break;
+          }
+          _context7.t2 = void 0;
+          _context7.next = 30;
+          break;
+        case 29:
+          _context7.t2 = _yield$mastra$dataLay2.id;
+        case 30:
+          _context7.t0 = _context7.t2;
+          if (_context7.t0) {
+            _context7.next = 33;
+            break;
+          }
+          _context7.t0 = '';
+        case 33:
+          k_id = _context7.t0;
+          if (!(!k_id && integrationName === systemName)) {
+            _context7.next = 39;
+            break;
+          }
+          _context7.next = 37;
+          return (_mastra$dataLayer6 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer6.createConnection({
+            connection: {
+              connectionId: connectionId,
+              name: integrationName
+            },
+            credential: {
+              type: 'API_KEY',
+              value: connectionId,
+              scope: []
+            }
+          });
+        case 37:
+          connection = _context7.sent;
+          k_id = connection.id;
+        case 39:
+          if (k_id) {
+            _context7.next = 42;
+            break;
+          }
+          console.error('Error bootstrapping shit');
+          return _context7.abrupt("return");
+        case 42:
+          _loop2 = /*#__PURE__*/_regeneratorRuntime().mark(function _loop2() {
+            var _mastra$dataLayer7, _entityRecords3, _records3;
+            var entity, entityRecords, records, _mastra$dataLayer8, _entityRecords4, _records4, _yield$mastra$trigger2, _event2, res, recordsMapped, entityTypeIndex, _recordsMapped$map2, vectors;
+            return _regeneratorRuntime().wrap(function _loop2$(_context6) {
+              while (1) switch (_context6.prev = _context6.next) {
+                case 0:
+                  entity = _step4.value;
+                  _context6.next = 3;
+                  return (_mastra$dataLayer7 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer7.getEntityRecordsByConnectionAndType({
+                    k_id: k_id,
+                    type: entity.name
+                  });
+                case 3:
+                  entityRecords = _context6.sent;
+                  records = (_entityRecords3 = entityRecords) == null ? void 0 : _entityRecords3.records;
+                  if (!(!records || ((_records3 = records) == null ? void 0 : _records3.length) === 0)) {
+                    _context6.next = 23;
+                    break;
+                  }
+                  // @TODO: SYNC THEM
+                  console.error('NO RECORDS');
+                  _context6.next = 9;
+                  return mastra.triggerEvent({
+                    key: entity.syncEvent,
+                    data: {},
+                    user: {
+                      connectionId: connectionId
+                    }
+                  });
+                case 9:
+                  _yield$mastra$trigger2 = _context6.sent;
+                  _event2 = _yield$mastra$trigger2.event;
+                  console.log(_event2, '####');
+                  _context6.next = 14;
+                  return _event2.subscribe();
+                case 14:
+                  res = _context6.sent;
+                  console.log('Subscribe result', res);
+                  _context6.next = 18;
+                  return (_mastra$dataLayer8 = mastra.dataLayer) == null ? void 0 : _mastra$dataLayer8.getEntityRecordsByConnectionAndType({
+                    k_id: k_id,
+                    type: entity.name
+                  });
+                case 18:
+                  entityRecords = _context6.sent;
+                  records = (_entityRecords4 = entityRecords) == null ? void 0 : _entityRecords4.records;
+                  if (!(!records || ((_records4 = records) == null ? void 0 : _records4.length) === 0)) {
+                    _context6.next = 23;
+                    break;
+                  }
+                  console.error('NO RECORDS AFTER SYNC');
+                  return _context6.abrupt("return", {
+                    v: void 0
+                  });
+                case 23:
+                  recordsMapped = records.map(function (_ref9) {
+                    var data = _ref9.data,
+                      externalId = _ref9.externalId;
+                    return _extends({
+                      id: externalId
+                    }, pick(data, entity.fields));
+                  });
+                  console.log(recordsMapped, 'RECORDS');
+                  _context6.next = 27;
+                  return vp.createIndex({
+                    suppressConflicts: true,
+                    name: entity.name,
+                    dimension: 1536,
+                    metric: 'cosine',
+                    spec: {
+                      serverless: {
+                        cloud: 'aws',
+                        region: 'us-east-1'
+                      }
+                    }
+                  });
+                case 27:
+                  entityTypeIndex = vp.index(entity.name);
+                  console.log(entityTypeIndex, 'INDEX');
+                  _context6.prev = 29;
+                  _context6.next = 32;
+                  return Promise.all((_recordsMapped$map2 = recordsMapped.map(/*#__PURE__*/function () {
+                    var _ref10 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(record) {
+                      var _yield$embed2, embedding;
+                      return _regeneratorRuntime().wrap(function _callee4$(_context5) {
+                        while (1) switch (_context5.prev = _context5.next) {
+                          case 0:
+                            _context5.next = 2;
+                            return embed({
+                              model: openai.embedding('text-embedding-3-small'),
+                              value: JSON.stringify(record)
+                            });
+                          case 2:
+                            _yield$embed2 = _context5.sent;
+                            embedding = _yield$embed2.embedding;
+                            return _context5.abrupt("return", {
+                              id: record.id,
+                              values: embedding,
+                              metadata: record
+                            });
+                          case 5:
+                          case "end":
+                            return _context5.stop();
+                        }
+                      }, _callee4);
+                    }));
+                    return function (_x6) {
+                      return _ref10.apply(this, arguments);
+                    };
+                  }())) != null ? _recordsMapped$map2 : []);
+                case 32:
+                  vectors = _context6.sent;
+                  console.log('UPSERTING', vectors);
+                  if (!(vectors.length > 0)) {
+                    _context6.next = 37;
+                    break;
+                  }
+                  _context6.next = 37;
+                  return entityTypeIndex.namespace(entity.name).upsert(vectors);
+                case 37:
+                  _context6.next = 42;
+                  break;
+                case 39:
+                  _context6.prev = 39;
+                  _context6.t0 = _context6["catch"](29);
+                  console.error(_context6.t0);
+                case 42:
+                  return _context6.abrupt("return", {
+                    v: void 0
+                  });
+                case 43:
+                case "end":
+                  return _context6.stop();
+              }
+            }, _loop2, null, [[29, 39]]);
+          });
+          _iterator4 = _createForOfIteratorHelperLoose(vectorE.data);
+        case 44:
+          if ((_step4 = _iterator4()).done) {
+            _context7.next = 51;
+            break;
+          }
+          return _context7.delegateYield(_loop2(), "t3", 46);
+        case 46:
+          _ret2 = _context7.t3;
+          if (!_ret2) {
+            _context7.next = 49;
+            break;
+          }
+          return _context7.abrupt("return", _ret2.v);
+        case 49:
+          _context7.next = 44;
+          break;
+        case 51:
+          _context7.next = 19;
+          break;
+        case 53:
+        case "end":
+          return _context7.stop();
+      }
+    }, _callee5);
+  }));
+  return _executeVectorSync.apply(this, arguments);
+}
+function vectorQueryEngine(_x3) {
+  return _vectorQueryEngine.apply(this, arguments);
+}
+function _vectorQueryEngine() {
+  _vectorQueryEngine = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref3) {
+    var vector_provider, _ref3$topK, topK, indexName, content, entityType, vp, index, _yield$embed3, embedding, queryResponse;
+    return _regeneratorRuntime().wrap(function _callee6$(_context8) {
+      while (1) switch (_context8.prev = _context8.next) {
+        case 0:
+          vector_provider = _ref3.vector_provider, _ref3$topK = _ref3.topK, topK = _ref3$topK === void 0 ? 1 : _ref3$topK, indexName = _ref3.indexName, content = _ref3.content, entityType = _ref3.entityType;
+          vp = getVectorProvider(vector_provider);
+          if (vp) {
+            _context8.next = 5;
+            break;
+          }
+          console.error('UNSUPPORTED VECTOR PROVIDER', vector_provider);
+          return _context8.abrupt("return");
+        case 5:
+          index = vp.index(indexName);
+          _context8.next = 8;
+          return embed({
+            model: openai.embedding('text-embedding-3-small'),
+            value: content
+          });
+        case 8:
+          _yield$embed3 = _context8.sent;
+          embedding = _yield$embed3.embedding;
+          console.log(embedding, 'EMBEDDING');
+          _context8.next = 13;
+          return index.namespace(entityType).query({
+            vector: embedding,
+            topK: topK,
+            includeMetadata: true
+          });
+        case 13:
+          queryResponse = _context8.sent;
+          return _context8.abrupt("return", queryResponse);
+        case 15:
+        case "end":
+          return _context8.stop();
+      }
+    }, _callee6);
+  }));
+  return _vectorQueryEngine.apply(this, arguments);
+}
+function getVectorQueryApis(_ref4) {
+  var mastra = _ref4.mastra;
+  var agentDir = mastra.config.agents.agentDirPath;
+  var agents = listAgentsJson({
+    agentDir: agentDir
+  });
+  var agentData = agents.map(function (agentFile) {
+    var _agent$knowledge_sour;
+    var agentDirPath = getAgentDir({
+      agentDir: agentDir
+    });
+    var agent = getAgentFile(path.join(agentDirPath, agentFile));
+    if (!(agent != null && (_agent$knowledge_sour = agent.knowledge_sources) != null && _agent$knowledge_sour.vector_provider)) {
+      console.error("No vector_provider defined for agent");
+      return;
+    }
+    return agent;
+  }).filter(Boolean);
+  return agentData.flatMap(function (agent) {
+    var entities = agent == null ? void 0 : agent.knowledge_sources.entities;
+    return entities.flatMap(function (_ref5) {
+      var data = _ref5.data,
+        integration = _ref5.integration;
+      return data.map(function (entity) {
+        return {
+          integrationName: integration,
+          type: "get_" + entity.name + "_from_vector_" + entity.index,
+          label: "Provides " + entity.name + " information from Vector " + entity.index + " Store",
+          description: "Provides " + entity.name + " information from Vector " + entity.index + " Store",
+          schema: z.object({
+            content: z.string(),
+            topResult: z.number()
+          }),
+          executor: function () {
+            var _executor = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref6) {
+              var _agent$knowledge_sour2;
+              var data, res;
+              return _regeneratorRuntime().wrap(function _callee$(_context) {
+                while (1) switch (_context.prev = _context.next) {
+                  case 0:
+                    data = _ref6.data;
+                    _context.next = 3;
+                    return vectorQueryEngine({
+                      vector_provider: agent == null || (_agent$knowledge_sour2 = agent.knowledge_sources) == null ? void 0 : _agent$knowledge_sour2.vector_provider,
+                      indexName: entity.index,
+                      content: data.content,
+                      topK: data.topResult || 1,
+                      entityType: entity.name
+                    });
+                  case 3:
+                    res = _context.sent;
+                    console.log(JSON.stringify({
+                      res: res
+                    }, null, 2));
+                    // @TODO: make this a proper response
+                    return _context.abrupt("return", res);
+                  case 6:
+                  case "end":
+                    return _context.stop();
+                }
+              }, _callee);
+            }));
+            function executor(_x4) {
+              return _executor.apply(this, arguments);
+            }
+            return executor;
+          }()
+        };
+      });
+    });
+  }).filter(Boolean);
+}
+function agentVectorSyncEvent() {
+  return {
+    // @TODO: naming convention
+    id: 'agent-vector-sync',
+    event: 'AGENT_VECTOR_SYNC',
+    executor: executeVectorSync
+  };
+}
+function genericVectorySyncEvent() {
+  return {
+    // @TODO: naming convention
+    id: 'vector-sync',
+    event: 'VECTOR_SYNC',
+    executor: executeGenericVectorSync
+  };
+}
+
+function getAgentSystemApis(_ref) {
+  var mastra = _ref.mastra;
+  var agentDir = mastra.config.agents.agentDirPath;
+  var agents = listAgentsJson({
+    agentDir: agentDir
+  });
+  // Remove the .json extension from the agent names
+  var agentList = agents.map(function (agent) {
+    return agent.slice(0, -5);
+  });
+  return [{
+    integrationName: 'SYSTEM',
+    type: 'message_agent',
+    label: 'Send Message To Agent',
+    description: 'Sends a message to an Agent',
+    schema: z.object({
+      agentId: z["enum"](agentList),
+      message: z.string()
+    }),
+    outputSchema: z.object({
+      message: z.string()
+    }),
+    executor: function () {
+      var _executor = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref2) {
+        var data, executor, result, _run$content, thread, run;
+        return _regeneratorRuntime().wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              data = _ref2.data;
+              _context.next = 3;
+              return mastra.getAgent({
+                agentId: data.agentId,
+                connectionId: data.connectionId
+              });
+            case 3:
+              executor = _context.sent;
+              if (executor) {
+                _context.next = 6;
+                break;
+              }
+              throw new Error('Could not create agent executor');
+            case 6:
+              if (!(typeof executor === 'function')) {
+                _context.next = 13;
+                break;
+              }
+              _context.next = 9;
+              return executor({
+                prompt: data == null ? void 0 : data.message
+              });
+            case 9:
+              result = _context.sent;
+              return _context.abrupt("return", {
+                message: result == null ? void 0 : result.text
+              });
+            case 13:
+              _context.next = 15;
+              return executor.initializeThread([{
+                role: 'user',
+                content: data == null ? void 0 : data.message
+              }]);
+            case 15:
+              thread = _context.sent;
+              _context.next = 18;
+              return executor.watchRun({
+                threadId: thread.id
+              });
+            case 18:
+              run = _context.sent;
+              return _context.abrupt("return", {
+                message: run == null || (_run$content = run.content) == null || (_run$content = _run$content[0]) == null || (_run$content = _run$content.text) == null ? void 0 : _run$content.value
+              });
+            case 20:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee);
+      }));
+      function executor(_x) {
+        return _executor.apply(this, arguments);
+      }
+      return executor;
+    }()
+  }];
+}
+
+var loggerPath = /*#__PURE__*/path.join(/*#__PURE__*/process.cwd(), 'mastra-agent-logs');
+function createFileLogger(_ref) {
+  var destinationPath = _ref.destinationPath;
+  var fullPath = path.join(loggerPath, destinationPath);
+  return function (log) {
+    if (!existsSync(loggerPath)) {
+      mkdirSync(loggerPath, {
+        recursive: true
+      });
+      return writeFileSync(fullPath, JSON.stringify([_extends({}, log, {
+        createdAt: new Date()
+      })], null, 2));
+    }
+    var logs = JSON.parse(readFileSync(fullPath, 'utf-8'));
+    logs.push(_extends({}, log, {
+      createdAt: new Date()
+    }));
+    return writeFileSync(fullPath, JSON.stringify(logs, null, 2));
+  };
+}
+
+var client;
+if (process.env.OPENAI_API_KEY) {
+  client = /*#__PURE__*/new OpenAI({
+    apiKey: process.env.OPENAI_API_KEY
+  });
+}
+function getAssistantAgent(_x3) {
+  return _getAssistantAgent.apply(this, arguments);
+}
+function _getAssistantAgent() {
+  _getAssistantAgent = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(_ref3) {
+    var id, toolMap, logger, agent, handleRunStatus, handleRequiresAction, getRun;
+    return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+      while (1) switch (_context11.prev = _context11.next) {
+        case 0:
+          id = _ref3.id, toolMap = _ref3.toolMap;
+          logger = createFileLogger({
+            destinationPath: id + ".json"
+          });
+          _context11.next = 4;
+          return client.beta.assistants.retrieve(id);
+        case 4:
+          agent = _context11.sent;
+          handleRunStatus = /*#__PURE__*/function () {
+            var _ref5 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref4) {
+              var threadId, run, _messages$data, messages;
+              return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+                while (1) switch (_context3.prev = _context3.next) {
+                  case 0:
+                    threadId = _ref4.threadId, run = _ref4.run;
+                    if (!(run.status === 'completed')) {
+                      _context3.next = 8;
+                      break;
+                    }
+                    _context3.next = 4;
+                    return client.beta.threads.messages.list(threadId);
+                  case 4:
+                    messages = _context3.sent;
+                    return _context3.abrupt("return", (_messages$data = messages.data) == null ? void 0 : _messages$data[0]);
+                  case 8:
+                    if (!(run.status === 'requires_action')) {
+                      _context3.next = 14;
+                      break;
+                    }
+                    _context3.next = 11;
+                    return handleRequiresAction({
+                      run: run,
+                      threadId: threadId
+                    });
+                  case 11:
+                    return _context3.abrupt("return", _context3.sent);
+                  case 14:
+                    console.error('Run did not complete:', run);
+                    logger({
+                      statusCode: 400,
+                      message: JSON.stringify({
+                        message: "Run did not complete",
+                        event_type: 'run_error',
+                        metadata: {
+                          run: run
+                        }
+                      }, null, 2)
+                    });
+                  case 16:
+                  case "end":
+                    return _context3.stop();
+                }
+              }, _callee3);
+            }));
+            return function handleRunStatus(_x4) {
+              return _ref5.apply(this, arguments);
+            };
+          }();
+          handleRequiresAction = /*#__PURE__*/function () {
+            var _ref7 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref6) {
+              var threadId, run, toolOutputs;
+              return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+                while (1) switch (_context5.prev = _context5.next) {
+                  case 0:
+                    threadId = _ref6.threadId, run = _ref6.run;
+                    if (!(run.required_action && run.required_action.submit_tool_outputs && run.required_action.submit_tool_outputs.tool_calls)) {
+                      _context5.next = 23;
+                      break;
+                    }
+                    _context5.t0 = compact;
+                    _context5.next = 5;
+                    return Promise.all(run.required_action.submit_tool_outputs.tool_calls.map(/*#__PURE__*/function () {
+                      var _ref8 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(tool, index, tools) {
+                        var callInfo, toolMetadata, toolFn, args, output;
+                        return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+                          while (1) switch (_context4.prev = _context4.next) {
+                            case 0:
+                              callInfo = index + 1 + " of " + tools.length;
+                              console.log('Tool:', tool["function"].name, tool.id, Object.keys(toolMap));
+                              toolMetadata = {
+                                id: tool.id,
+                                fn: tool["function"].name,
+                                availableTools: Object.keys(toolMap)
+                              };
+                              logger({
+                                statusCode: 100,
+                                message: JSON.stringify({
+                                  message: "[local] Starting tool call " + callInfo + ": " + tool["function"].name,
+                                  metadata: {
+                                    tool: toolMetadata
+                                  }
+                                }, null, 2)
+                              });
+                              toolFn = toolMap == null ? void 0 : toolMap[tool["function"].name];
+                              if (toolFn) {
+                                _context4.next = 8;
+                                break;
+                              }
+                              logger({
+                                statusCode: 404,
+                                message: JSON.stringify({
+                                  message: "[local] No tool fn found: " + tool["function"].name,
+                                  metadata: {
+                                    tool: toolMetadata
+                                  }
+                                }, null, 2)
+                              });
+                              return _context4.abrupt("return");
+                            case 8:
+                              console.log('Executing tool:', tool["function"].name, tool.id, tool["function"].arguments);
+                              args = {};
+                              try {
+                                if (tool["function"].arguments) {
+                                  args = JSON.parse(tool["function"].arguments);
+                                  logger({
+                                    statusCode: 100,
+                                    message: JSON.stringify({
+                                      message: "[local] Passing args to tool call: " + tool["function"].name,
+                                      metadata: {
+                                        args: args,
+                                        tool: toolMetadata
+                                      }
+                                    }, null, 2)
+                                  });
+                                }
+                              } catch (e) {
+                                console.error(e);
+                              }
+                              _context4.next = 13;
+                              return toolFn(args);
+                            case 13:
+                              output = _context4.sent;
+                              logger({
+                                statusCode: 200,
+                                message: JSON.stringify({
+                                  message: "[local] Completed tool call " + callInfo + ": " + tool["function"].name,
+                                  metadata: {
+                                    output: output,
+                                    tool: toolMetadata
+                                  }
+                                }, null, 2)
+                              });
+                              return _context4.abrupt("return", {
+                                tool_call_id: tool.id,
+                                output: JSON.stringify(output)
+                              });
+                            case 16:
+                            case "end":
+                              return _context4.stop();
+                          }
+                        }, _callee4);
+                      }));
+                      return function (_x6, _x7, _x8) {
+                        return _ref8.apply(this, arguments);
+                      };
+                    }()));
+                  case 5:
+                    _context5.t1 = _context5.sent;
+                    toolOutputs = (0, _context5.t0)(_context5.t1);
+                    if (toolOutputs) {
+                      _context5.next = 11;
+                      break;
+                    }
+                    console.error('No tool outputs to submit.');
+                    logger({
+                      statusCode: 404,
+                      message: JSON.stringify({
+                        message: "No tool outputs submitted",
+                        metadata: {}
+                      }, null, 2)
+                    });
+                    return _context5.abrupt("return", handleRunStatus({
+                      threadId: threadId,
+                      run: run
+                    }));
+                  case 11:
+                    if (!(toolOutputs && (toolOutputs == null ? void 0 : toolOutputs.length) > 0)) {
+                      _context5.next = 20;
+                      break;
+                    }
+                    console.log(toolOutputs, '###### YOOOOOO');
+                    _context5.next = 15;
+                    return client.beta.threads.runs.submitToolOutputsAndPoll(threadId, run.id, {
+                      tool_outputs: toolOutputs
+                    });
+                  case 15:
+                    run = _context5.sent;
+                    logger({
+                      statusCode: 200,
+                      message: JSON.stringify({
+                        message: "Tool outputs submitted",
+                        metadata: {
+                          run: run
+                        }
+                      }, null, 2)
+                    });
+                    console.log('Tool outputs submitted successfully.');
+                    _context5.next = 22;
+                    break;
+                  case 20:
+                    logger({
+                      statusCode: 404,
+                      message: JSON.stringify({
+                        message: "No tool outputs to submit",
+                        metadata: {}
+                      }, null, 2)
+                    });
+                    console.log('No tool outputs to submit.');
+                  case 22:
+                    return _context5.abrupt("return", handleRunStatus({
+                      threadId: threadId,
+                      run: run
+                    }));
+                  case 23:
+                  case "end":
+                    return _context5.stop();
+                }
+              }, _callee5);
+            }));
+            return function handleRequiresAction(_x5) {
+              return _ref7.apply(this, arguments);
+            };
+          }();
+          getRun = /*#__PURE__*/function () {
+            var _ref10 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref9) {
+              var threadId, runId;
+              return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+                while (1) switch (_context6.prev = _context6.next) {
+                  case 0:
+                    threadId = _ref9.threadId, runId = _ref9.runId;
+                    _context6.next = 3;
+                    return client.beta.threads.runs.retrieve(threadId, runId);
+                  case 3:
+                    return _context6.abrupt("return", _context6.sent);
+                  case 4:
+                  case "end":
+                    return _context6.stop();
+                }
+              }, _callee6);
+            }));
+            return function getRun(_x9) {
+              return _ref10.apply(this, arguments);
+            };
+          }();
+          return _context11.abrupt("return", {
+            agent: agent,
+            initializeThread: function () {
+              var _initializeThread = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(messages) {
+                return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+                  while (1) switch (_context7.prev = _context7.next) {
+                    case 0:
+                      if (messages === void 0) {
+                        messages = [];
+                      }
+                      _context7.next = 3;
+                      return client.beta.threads.create({
+                        messages: messages
+                      });
+                    case 3:
+                      return _context7.abrupt("return", _context7.sent);
+                    case 4:
+                    case "end":
+                      return _context7.stop();
+                  }
+                }, _callee7);
+              }));
+              function initializeThread(_x10) {
+                return _initializeThread.apply(this, arguments);
+              }
+              return initializeThread;
+            }(),
+            listMessages: function () {
+              var _listMessages = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref11) {
+                var threadId;
+                return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+                  while (1) switch (_context8.prev = _context8.next) {
+                    case 0:
+                      threadId = _ref11.threadId;
+                      _context8.next = 3;
+                      return client.beta.threads.messages.list(threadId);
+                    case 3:
+                      return _context8.abrupt("return", _context8.sent);
+                    case 4:
+                    case "end":
+                      return _context8.stop();
+                  }
+                }, _callee8);
+              }));
+              function listMessages(_x11) {
+                return _listMessages.apply(this, arguments);
+              }
+              return listMessages;
+            }(),
+            getRun: getRun,
+            handleRunStatus: handleRunStatus,
+            handleRequiresAction: handleRequiresAction,
+            createUserMessage: function () {
+              var _createUserMessage = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref12) {
+                var threadId, content;
+                return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+                  while (1) switch (_context9.prev = _context9.next) {
+                    case 0:
+                      threadId = _ref12.threadId, content = _ref12.content;
+                      _context9.next = 3;
+                      return client.beta.threads.messages.create(threadId, {
+                        role: 'user',
+                        content: content
+                      });
+                    case 3:
+                      return _context9.abrupt("return", _context9.sent);
+                    case 4:
+                    case "end":
+                      return _context9.stop();
+                  }
+                }, _callee9);
+              }));
+              function createUserMessage(_x12) {
+                return _createUserMessage.apply(this, arguments);
+              }
+              return createUserMessage;
+            }(),
+            watchRun: function () {
+              var _watchRun = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_ref13) {
+                var runId, threadId, run;
+                return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+                  while (1) switch (_context10.prev = _context10.next) {
+                    case 0:
+                      runId = _ref13.runId, threadId = _ref13.threadId;
+                      if (!runId) {
+                        _context10.next = 7;
+                        break;
+                      }
+                      _context10.next = 4;
+                      return getRun({
+                        threadId: threadId,
+                        runId: runId
+                      });
+                    case 4:
+                      run = _context10.sent;
+                      _context10.next = 11;
+                      break;
+                    case 7:
+                      _context10.next = 9;
+                      return client.beta.threads.runs.createAndPoll(threadId, {
+                        assistant_id: id,
+                        tool_choice: 'required'
+                      });
+                    case 9:
+                      run = _context10.sent;
+                      logger({
+                        statusCode: 202,
+                        message: JSON.stringify({
+                          message: "Creating and polling run, tool choice required",
+                          metadata: {
+                            run: run,
+                            tool_choice: 'required',
+                            threadId: threadId,
+                            assistant_id: id
+                          }
+                        }, null, 2)
+                      });
+                    case 11:
+                      return _context10.abrupt("return", handleRunStatus({
+                        threadId: threadId,
+                        run: run
+                      }));
+                    case 12:
+                    case "end":
+                      return _context10.stop();
+                  }
+                }, _callee10);
+              }));
+              function watchRun(_x13) {
+                return _watchRun.apply(this, arguments);
+              }
+              return watchRun;
+            }()
+          });
+        case 9:
+        case "end":
+          return _context11.stop();
+      }
+    }, _callee11);
+  }));
+  return _getAssistantAgent.apply(this, arguments);
+}
+
+var delay = function delay(ms) {
+  return new Promise(function (resolve) {
+    return setTimeout(resolve, ms);
+  });
+};
+function getAgentParams(_ref) {
+  var tools = _ref.tools,
+    resultTool = _ref.resultTool,
+    model = _ref.model;
+  var toolsConverted = Object.entries(tools).reduce(function (memo, _ref2) {
+    var key = _ref2[0],
+      val = _ref2[1];
+    memo[key] = tool(val);
+    return memo;
+  }, {});
+  var answerTool = {};
+  if (resultTool) {
+    answerTool = {
+      answer: tool(resultTool)
+    };
+  }
+  var modelDef;
+  if (model.type === 'openai') {
+    var mName = model.name;
+    if (!mName) {
+      mName = "gpt-4o-2024-08-06";
+    }
+    var openai = createOpenAI({
+      apiKey: process.env.OPENAI_API_KEY
+    });
+    modelDef = openai(mName, {
+      structuredOutputs: true
+    });
+  } else if (model.type === 'anthropic') {
+    var _mName = model.name;
+    if (!_mName) {
+      _mName = "claude-3-5-sonnet-20240620";
+    }
+    var anthropic = createAnthropic({
+      apiKey: process.env.ANTHROPIC_API_KEY
+    });
+    modelDef = anthropic(_mName);
+  } else if (model.type === 'groq') {
+    var _process$env$GROQ_API;
+    modelDef = createOpenAICompatibleModel('https://api.groq.com/openai/v1', (_process$env$GROQ_API = process.env.GROQ_API_KEY) != null ? _process$env$GROQ_API : '', 'llama-3.2-90b-text-preview', model.name);
+  } else if (model.type === 'perplexity') {
+    var _process$env$PERPLEXI;
+    modelDef = createOpenAICompatibleModel('https://api.perplexity.ai/', (_process$env$PERPLEXI = process.env.PERPLEXITY_API_KEY) != null ? _process$env$PERPLEXI : '', 'llama-3.1-sonar-large-128k-chat', model.name);
+  } else if (model.type === 'fireworks') {
+    var _process$env$FIREWORK;
+    modelDef = createOpenAICompatibleModel('https://api.fireworks.ai/inference/v1', (_process$env$FIREWORK = process.env.FIREWORKS_API_KEY) != null ? _process$env$FIREWORK : '', 'llama-v3p1-70b-instruct', model.name);
+  }
+  return {
+    toolsConverted: toolsConverted,
+    modelDef: modelDef,
+    answerTool: answerTool,
+    toolChoice: (model == null ? void 0 : model.toolChoice) || 'required'
+  };
+}
+function createStreamAgent(_ref3) {
+  var agent_instructions = _ref3.agent_instructions,
+    _ref3$maxSteps = _ref3.maxSteps,
+    maxSteps = _ref3$maxSteps === void 0 ? 5 : _ref3$maxSteps,
+    tools = _ref3.tools,
+    resultTool = _ref3.resultTool,
+    _ref3$context = _ref3.context,
+    context = _ref3$context === void 0 ? [] : _ref3$context,
+    model = _ref3.model;
+  var params = getAgentParams({
+    tools: tools,
+    resultTool: resultTool,
+    model: model
+  });
+  return /*#__PURE__*/function () {
+    var _ref5 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref4) {
+      var prompt, argsForExecute;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            prompt = _ref4.prompt;
+            argsForExecute = {
+              model: params.modelDef,
+              tools: _extends({}, params.toolsConverted, params.answerTool),
+              toolChoice: (params == null ? void 0 : params.toolChoice) || 'required',
+              maxSteps: maxSteps,
+              // system: systemPrompt,
+              onStepFinish: function () {
+                var _onStepFinish = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(props) {
+                  var _props$response, _props$response2;
+                  return _regeneratorRuntime().wrap(function _callee$(_context) {
+                    while (1) switch (_context.prev = _context.next) {
+                      case 0:
+                        console.log(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)) {
+                          _context.next = 5;
+                          break;
+                        }
+                        console.log('Rate limit reached, waiting 10 seconds');
+                        _context.next = 5;
+                        return delay(10 * 1000);
+                      case 5:
+                      case "end":
+                        return _context.stop();
+                    }
+                  }, _callee);
+                }));
+                function onStepFinish(_x2) {
+                  return _onStepFinish.apply(this, arguments);
+                }
+                return onStepFinish;
+              }()
+            };
+            _context2.next = 4;
+            return streamText(_extends({
+              messages: [].concat(context, [{
+                role: 'user',
+                content: prompt
+              }, {
+                role: 'system',
+                content: agent_instructions
+              }])
+            }, argsForExecute));
+          case 4:
+            return _context2.abrupt("return", _context2.sent);
+          case 5:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2);
+    }));
+    return function (_x) {
+      return _ref5.apply(this, arguments);
+    };
+  }();
+}
+function createAgent(_ref6) {
+  var agent_instructions = _ref6.agent_instructions,
+    _ref6$maxSteps = _ref6.maxSteps,
+    maxSteps = _ref6$maxSteps === void 0 ? 5 : _ref6$maxSteps,
+    tools = _ref6.tools,
+    resultTool = _ref6.resultTool,
+    _ref6$context = _ref6.context,
+    context = _ref6$context === void 0 ? [] : _ref6$context,
+    model = _ref6.model;
+  var params = getAgentParams({
+    tools: tools,
+    resultTool: resultTool,
+    model: model
+  });
+  return /*#__PURE__*/function () {
+    var _ref8 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref7) {
+      var prompt, argsForExecute;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            prompt = _ref7.prompt;
+            argsForExecute = {
+              model: params.modelDef,
+              tools: _extends({}, params.toolsConverted, params.answerTool),
+              toolChoice: (params == null ? void 0 : params.toolChoice) || 'required',
+              maxSteps: maxSteps,
+              // system: systemPrompt,
+              onStepFinish: function () {
+                var _onStepFinish2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(props) {
+                  var _props$response3, _props$response4;
+                  return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+                    while (1) switch (_context3.prev = _context3.next) {
+                      case 0:
+                        console.log(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)) {
+                          _context3.next = 5;
+                          break;
+                        }
+                        console.log('Rate limit reached, waiting 10 seconds');
+                        _context3.next = 5;
+                        return delay(10 * 1000);
+                      case 5:
+                      case "end":
+                        return _context3.stop();
+                    }
+                  }, _callee3);
+                }));
+                function onStepFinish(_x4) {
+                  return _onStepFinish2.apply(this, arguments);
+                }
+                return onStepFinish;
+              }()
+            };
+            _context4.next = 4;
+            return generateText(_extends({
+              messages: [].concat(context, [{
+                role: 'user',
+                content: prompt
+              }, {
+                role: 'system',
+                content: agent_instructions
+              }])
+            }, argsForExecute));
+          case 4:
+            return _context4.abrupt("return", _context4.sent);
+          case 5:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4);
+    }));
+    return function (_x3) {
+      return _ref8.apply(this, arguments);
+    };
+  }();
+}
+function createOpenAICompatibleModel(baseURL, apiKey, defaultModelName, modelName) {
+  var client = createOpenAI({
+    baseURL: baseURL,
+    apiKey: apiKey
+  });
+  return client(modelName || defaultModelName);
+}
+
+function getAgent(_x) {
+  return _getAgent.apply(this, arguments);
+}
+function _getAgent() {
+  _getAgent = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref) {
+    var connectionId, agent, apis, tools, toolMap, assistant, _agent$model2, keyToModel, _tools, _toolMap, resultTool, schema, _agent$model;
+    return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+      while (1) switch (_context3.prev = _context3.next) {
+        case 0:
+          connectionId = _ref.connectionId, agent = _ref.agent, apis = _ref.apis;
+          if (!(agent.model.provider === 'OPEN_AI_ASSISTANT')) {
+            _context3.next = 10;
+            break;
+          }
+          tools = Object.keys(agent.tools);
+          toolMap = Object.entries(apis).reduce(function (memo, _ref2) {
+            var k = _ref2[0],
+              def = _ref2[1];
+            if (tools.includes(k)) {
+              memo[k] = /*#__PURE__*/function () {
+                var _ref3 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(props) {
+                  return _regeneratorRuntime().wrap(function _callee$(_context) {
+                    while (1) switch (_context.prev = _context.next) {
+                      case 0:
+                        return _context.abrupt("return", def.executor({
+                          data: props,
+                          ctx: {
+                            connectionId: connectionId
+                          }
+                        }));
+                      case 1:
+                      case "end":
+                        return _context.stop();
+                    }
+                  }, _callee);
+                }));
+                return function (_x2) {
+                  return _ref3.apply(this, arguments);
+                };
+              }();
+            }
+            return memo;
+          }, {});
+          _context3.next = 6;
+          return getAssistantAgent({
+            id: agent.id,
+            toolMap: toolMap
+          });
+        case 6:
+          assistant = _context3.sent;
+          return _context3.abrupt("return", assistant);
+        case 10:
+          if (!['OPEN_AI_VERCEL', 'ANTHROPIC_VERCEL', 'GROQ_VERCEL', 'PERPLEXITY_VERCEL', 'FIREWORKS_VERCEL'].includes(agent.model.provider)) {
+            _context3.next = 19;
+            break;
+          }
+          keyToModel = {
+            OPEN_AI_VERCEL: 'openai',
+            ANTHROPIC_VERCEL: 'anthropic',
+            GROQ_VERCEL: 'groq',
+            PERPLEXITY_VERCEL: 'perplexity',
+            FIREWORKS_VERCEL: 'fireworks'
+          };
+          _tools = Object.keys(agent.tools);
+          _toolMap = Object.entries(apis).reduce(function (memo, _ref4) {
+            var k = _ref4[0],
+              def = _ref4[1];
+            if (_tools.includes(k)) {
+              memo[k] = {
+                description: def.label,
+                parameters: def.schema,
+                execute: function () {
+                  var _execute = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(props) {
+                    return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+                      while (1) switch (_context2.prev = _context2.next) {
+                        case 0:
+                          return _context2.abrupt("return", def.executor({
+                            data: props,
+                            ctx: {
+                              connectionId: connectionId
+                            }
+                          }));
+                        case 1:
+                        case "end":
+                          return _context2.stop();
+                      }
+                    }, _callee2);
+                  }));
+                  function execute(_x3) {
+                    return _execute.apply(this, arguments);
+                  }
+                  return execute;
+                }()
+              };
+            }
+            return memo;
+          }, {});
+          resultTool = undefined;
+          if (agent.outputs.structured) {
+            schema = Object.entries(agent.outputs.structured.schema).reduce(function (memo, _ref5) {
+              var k = _ref5[0],
+                v = _ref5[1];
+              if (v.type === 'string') {
+                memo[k] = z.string();
+              }
+              if (v.type === 'array') {
+                var _v$items;
+                var itemType = v == null || (_v$items = v.items) == null ? void 0 : _v$items.type;
+                if (itemType === 'string') {
+                  memo[k] = z.array(z.string());
+                }
+              }
+              return memo;
+            }, {});
+            resultTool = {
+              description: agent.outputs.structured.description,
+              parameters: z.object(schema)
+            };
+          }
+          if (!(agent.model.generation_type === 'stream')) {
+            _context3.next = 18;
+            break;
+          }
+          return _context3.abrupt("return", createStreamAgent({
+            agent_instructions: agent.agent_instructions,
+            model: {
+              type: keyToModel[agent.model.provider],
+              name: agent.model.name,
+              toolChoice: ((_agent$model = agent.model) == null ? void 0 : _agent$model.toolChoice) || 'required'
+            },
+            tools: _toolMap,
+            resultTool: resultTool
+          }));
+        case 18:
+          return _context3.abrupt("return", createAgent({
+            agent_instructions: agent.agent_instructions,
+            model: {
+              type: keyToModel[agent.model.provider],
+              name: agent.model.name,
+              toolChoice: ((_agent$model2 = agent.model) == null ? void 0 : _agent$model2.toolChoice) || 'required'
+            },
+            tools: _toolMap,
+            resultTool: resultTool
+          }));
+        case 19:
+        case "end":
+          return _context3.stop();
+      }
+    }, _callee3);
+  }));
+  return _getAgent.apply(this, arguments);
+}
+
+var Mastra = /*#__PURE__*/function () {
+  function Mastra(_ref) {
+    var _this = this;
+    var dataLayer = _ref.dataLayer,
+      vectorLayer = _ref.vectorLayer,
+      config = _ref.config;
+    //global events grouped by Integration
+    this.globalEvents = new Map();
+    // global apis grouped by Integration
+    this.globalApis = new Map();
+    this.integrations = new Map();
+    this.dataLayer = void 0;
+    this.vectorLayer = void 0;
+    this.agentsConfig = {
+      agentDirPath: '',
+      vectorProvider: []
+    };
+    this.config = void 0;
+    this.runBlueprint = /*#__PURE__*/function () {
+      var _ref3 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref2) {
+        var _this$availableIntegr;
+        var blueprint, _ref2$dataCtx, dataCtx, ctx, systemApis, systemEvents, availableIntegrations, availableIntegrationApis, availableIntegrationEvents, frameworkApis, frameworkEvents;
+        return _regeneratorRuntime().wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              blueprint = _ref2.blueprint, _ref2$dataCtx = _ref2.dataCtx, dataCtx = _ref2$dataCtx === void 0 ? {} : _ref2$dataCtx, ctx = _ref2.ctx;
+              systemApis = _this.getSystemApis();
+              systemEvents = _this.getSystemEvents();
+              availableIntegrations = (_this$availableIntegr = _this.availableIntegrations()) == null ? void 0 : _this$availableIntegr.map(function (_ref4) {
+                var integration = _ref4.integration;
+                return integration;
+              });
+              availableIntegrationApis = availableIntegrations.reduce(function (acc, _ref5) {
+                var name = _ref5.name;
+                var apis = _this.getApisByIntegration(name);
+                return _extends({}, acc, apis);
+              }, {});
+              availableIntegrationEvents = availableIntegrations.reduce(function (acc, _ref6) {
+                var name = _ref6.name;
+                var events = _this.getEventsByIntegration(name);
+                return _extends({}, acc, events);
+              }, {});
+              frameworkApis = _extends({}, systemApis, availableIntegrationApis);
+              frameworkEvents = _extends({}, systemEvents, availableIntegrationEvents);
+              _context.next = 10;
+              return blueprintRunner({
+                dataCtx: dataCtx,
+                blueprint: blueprint,
+                frameworkApis: frameworkApis,
+                frameworkEvents: frameworkEvents,
+                ctx: ctx
+              });
+            case 10:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee);
+      }));
+      return function (_x) {
+        return _ref3.apply(this, arguments);
+      };
+    }();
+    this.dataLayer = dataLayer;
+    this.vectorLayer = vectorLayer;
+    this.config = config;
+  }
+  Mastra.init = function init(config) {
+    var _config$workflows$sys;
+    if (!config.db.uri) {
+      throw new Error('No database config/provider found');
+    }
+    var dataLayer = new DataLayer({
+      url: config.db.uri,
+      provider: config.db.provider
+    });
+    var vectorLayer = new VectorLayer();
+    var framework = new Mastra({
+      config: config,
+      dataLayer: dataLayer,
+      vectorLayer: vectorLayer
+    });
+    // Register integrations
+    config.integrations.forEach(function (integration) {
+      framework.__registerIntgeration(integration);
+    });
+    // Register system apis
+    framework.__registerApis({
+      apis: [].concat(getAgentSystemApis({
+        mastra: framework
+      }), (_config$workflows$sys = config.workflows.systemApis) == null ? void 0 : _config$workflows$sys.map(function (api) {
+        return _extends({}, api, {
+          integrationName: config.name
+        });
+      }), getVectorQueryApis({
+        mastra: framework
+      }))
+    });
+    // Register System events
+    framework.__registerEvents({
+      events: config.workflows.systemEvents
+    });
+    // Register vector sync event
+    framework.__registerEvents({
+      events: {
+        AGENT_VECTOR_SYNC: {
+          label: 'Sync vector data for an agent',
+          description: 'Sync vector data for an agent',
+          schema: z.object({
+            agentId: z.string()
+          }),
+          handler: agentVectorSyncEvent
+        },
+        VECTOR_SYNC: {
+          label: 'Sync vector data',
+          description: 'Sync vector data',
+          schema: z.object({
+            vector_provider: z.string(),
+            entities: z.array(z.object({
+              integration: z.string(),
+              data: z.array(z.object({
+                name: z.string(),
+                fields: z.array(z.string()),
+                syncEvent: z.string(),
+                index: z.string()
+              }))
+            }))
+          }),
+          handler: genericVectorySyncEvent
+        }
+      },
+      integrationName: config.name
+    });
+    // Register agent config
+    framework.agentsConfig = config.agents;
+    return framework;
+  };
+  var _proto = Mastra.prototype;
+  _proto.connectedIntegrations = /*#__PURE__*/function () {
+    var _connectedIntegrations = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref7) {
+      var _this2 = this;
+      var context, ints, connectionChecks;
+      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
+          case 0:
+            context = _ref7.context;
+            ints = this.availableIntegrations();
+            _context3.next = 4;
+            return Promise.all(ints.map(/*#__PURE__*/function () {
+              var _ref9 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref8) {
+                var integration, connection;
+                return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+                  while (1) switch (_context2.prev = _context2.next) {
+                    case 0:
+                      integration = _ref8.integration;
+                      _context2.next = 3;
+                      return _this2.dataLayer.getConnection({
+                        connectionId: context.connectionId,
+                        name: integration.name
+                      });
+                    case 3:
+                      connection = _context2.sent;
+                      return _context2.abrupt("return", {
+                        integration: integration,
+                        connected: !!connection
+                      });
+                    case 5:
+                    case "end":
+                      return _context2.stop();
+                  }
+                }, _callee2);
+              }));
+              return function (_x3) {
+                return _ref9.apply(this, arguments);
+              };
+            }()));
+          case 4:
+            connectionChecks = _context3.sent;
+            return _context3.abrupt("return", connectionChecks.filter(function (_ref10) {
+              var connected = _ref10.connected;
+              return connected;
+            }).map(function (_ref11) {
+              var integration = _ref11.integration;
+              return integration;
+            }));
+          case 6:
+          case "end":
+            return _context3.stop();
+        }
+      }, _callee3, this);
+    }));
+    function connectedIntegrations(_x2) {
+      return _connectedIntegrations.apply(this, arguments);
+    }
+    return connectedIntegrations;
+  }();
+  _proto.__registerIntgeration = function __registerIntgeration(definition) {
+    var name = definition.name;
+    definition.attachDataLayer({
+      dataLayer: this.dataLayer
+    });
+    var router = this.createRouter();
+    definition.corePresets = {
+      redirectURI: router.makeRedirectURI()
+    };
+    this.integrations.set(name, definition);
+    definition.registerEvents();
+    this.__registerEvents({
+      events: definition.getEvents(),
+      integrationName: name
+    });
+    definition.registerApis();
+    definition._convertApiClientToSystemApis();
+    this.__registerApis({
+      apis: Object.values(definition.getApis()),
+      integrationName: name
+    });
+  };
+  _proto.registerApi = function registerApi(name, api) {
+    var _extends2;
+    var integrationApis = this.globalApis.get(this.config.name) || {};
+    this.globalApis.set(this.config.name, _extends({}, integrationApis, (_extends2 = {}, _extends2[name] = _extends({}, api, {
+      integrationName: this.config.name
+    }), _extends2)));
+  };
+  _proto.registerEvent = function registerEvent(name, event) {
+    var _extends3;
+    var integrationEvents = this.globalEvents.get('SYSTEM') || {};
+    this.globalEvents.set('SYSTEM', _extends({}, integrationEvents, (_extends3 = {}, _extends3[name] = event, _extends3)));
+  };
+  _proto.__registerEvents = function __registerEvents(_ref12) {
+    var events = _ref12.events,
+      _ref12$integrationNam = _ref12.integrationName,
+      integrationName = _ref12$integrationNam === void 0 ? this.config.name : _ref12$integrationNam;
+    var integrationEvents = this.globalEvents.get(integrationName) || {};
+    this.globalEvents.set(integrationName, _extends({}, integrationEvents, events));
+  };
+  _proto.__registerApis = function __registerApis(_ref13) {
+    var apis = _ref13.apis,
+      _ref13$integrationNam = _ref13.integrationName,
+      integrationName = _ref13$integrationNam === void 0 ? this.config.name : _ref13$integrationNam;
+    var integrationApis = this.globalApis.get(integrationName) || {};
+    this.globalApis.set(integrationName, _extends({}, integrationApis, apis.reduce(function (acc, api) {
+      var _extends4;
+      return _extends({}, acc, (_extends4 = {}, _extends4[api.type] = api, _extends4));
+    }, {})));
+  };
+  _proto.availableIntegrations = function availableIntegrations() {
+    return Array.from(this.integrations.entries()).map(function (_ref14) {
+      var name = _ref14[0],
+        integration = _ref14[1];
+      return {
+        name: name,
+        integration: integration
+      };
+    });
+  };
+  _proto.getIntegration = function getIntegration(name) {
+    return this.integrations.get(name);
+  };
+  _proto.getGlobalEvents = function getGlobalEvents() {
+    return this.globalEvents;
+  };
+  _proto.getSystemEvents = function getSystemEvents() {
+    var events = this.globalEvents.get(this.config.name);
+    return events || {};
+  };
+  _proto.getEventsByIntegration = function getEventsByIntegration(name) {
+    return this.globalEvents.get(name);
+  };
+  _proto.getGlobalEventHandlers = function getGlobalEventHandlers() {
+    var _this3 = this;
+    return Array.from(this.globalEvents.entries()).flatMap(function (_ref15) {
+      var integrationName = _ref15[0],
+        events = _ref15[1];
+      var groupedHandlers = Object.keys(events).map(function (eventKey) {
+        var _events$eventKey;
+        var eventHandler = (_events$eventKey = events[eventKey]) == null ? void 0 : _events$eventKey.handler;
+        if (!eventHandler) return null;
+        var isSystemEvent = integrationName === _this3.config.name;
+        var _this3$createRouter = _this3.createRouter(),
+          makeWebhookUrl = _this3$createRouter.makeWebhookUrl;
+        if (isSystemEvent) {
+          return eventHandler({
+            getIntegration: function getIntegration(name) {
+              return _this3.getIntegration(name);
+            },
+            eventKey: eventKey,
+            getVectorProvider: function getVectorProvider(name) {
+              return _this3.agentsConfig.vectorProvider.find(function (provider) {
+                return provider.name === name;
+              });
+            },
+            makeWebhookUrl: makeWebhookUrl
+          });
+        } else {
+          var integration = _this3.getIntegration(integrationName);
+          return eventHandler({
+            integrationInstance: integration,
+            eventKey: eventKey,
+            makeWebhookUrl: makeWebhookUrl
+          });
+        }
+      }).filter(Boolean);
+      return groupedHandlers;
+    });
+  };
+  _proto.getApis = function getApis() {
+    return this.globalApis;
+  };
+  _proto.getSystemApis = function getSystemApis() {
+    return this.globalApis.get(this.config.name);
+  };
+  _proto.getApisByIntegration = function getApisByIntegration(name, includeHidden) {
+    var integrationApis = this.globalApis.get(name);
+    if (includeHidden) {
+      return integrationApis;
+    }
+    return omitBy(integrationApis, function (value) {
+      return value.isHidden;
+    });
+  };
+  _proto.authenticatableIntegrations = function authenticatableIntegrations() {
+    return this.availableIntegrations().filter(function (_ref16) {
+      var integration = _ref16.integration;
+      try {
+        integration.getAuthenticator();
+        return true;
+      } catch (e) {
+        return false;
+      }
+    });
+  };
+  _proto.authenticator = function authenticator(name) {
+    var _int = this.getIntegration(name);
+    if (!_int) {
+      throw new Error("No integration exists for " + name);
     }
     return _int.getAuthenticator();
   };
   _proto.connectIntegrationByCredential = /*#__PURE__*/function () {
-    var _connectIntegrationByCredential = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref16) {
+    var _connectIntegrationByCredential = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref17) {
       var name, connectionId, credential, authenticator, integration;
       return _regeneratorRuntime().wrap(function _callee4$(_context4) {
         while (1) switch (_context4.prev = _context4.next) {
           case 0:
-            name = _ref16.name, connectionId = _ref16.connectionId, credential = _ref16.credential;
+            name = _ref17.name, connectionId = _ref17.connectionId, credential = _ref17.credential;
             authenticator = this.authenticator(name);
             if (authenticator) {
               _context4.next = 4;
@@ -4511,13 +6238,13 @@ var Framework = /*#__PURE__*/function () {
     return connectIntegrationByCredential;
   }();
   _proto.disconnectIntegration = /*#__PURE__*/function () {
-    var _disconnectIntegration = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref17) {
+    var _disconnectIntegration = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(_ref18) {
       var _yield$this$dataLayer;
       var name, connectionId, integration, k_id, authenticator;
       return _regeneratorRuntime().wrap(function _callee5$(_context5) {
         while (1) switch (_context5.prev = _context5.next) {
           case 0:
-            name = _ref17.name, connectionId = _ref17.connectionId;
+            name = _ref18.name, connectionId = _ref18.connectionId;
             integration = this.getIntegration(name);
             _context5.next = 4;
             return this.dataLayer.getConnection({
@@ -4578,14 +6305,14 @@ var Framework = /*#__PURE__*/function () {
     return disconnectIntegration;
   }();
   _proto.callApi = /*#__PURE__*/function () {
-    var _callApi = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref18) {
+    var _callApi = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(_ref19) {
       var _int2$getApis;
-      var _ref18$integrationNam, integrationName, api, payload, _this$globalApis$get, _apiExecutor, _int2, apiExecutor;
+      var _ref19$integrationNam, integrationName, api, payload, _this$globalApis$get, _apiExecutor, _int2, apiExecutor;
       return _regeneratorRuntime().wrap(function _callee6$(_context6) {
         while (1) switch (_context6.prev = _context6.next) {
           case 0:
-            _ref18$integrationNam = _ref18.integrationName, integrationName = _ref18$integrationNam === void 0 ? this.config.name : _ref18$integrationNam, api = _ref18.api, payload = _ref18.payload;
-            if (!(integrationName === this.config.name)) {
+            _ref19$integrationNam = _ref19.integrationName, integrationName = _ref19$integrationNam === void 0 ? this.config.name : _ref19$integrationNam, api = _ref19.api, payload = _ref19.payload;
+            if (!(integrationName === this.config.name || integrationName === 'SYSTEM')) {
               _context6.next = 6;
               break;
             }
@@ -4625,19 +6352,19 @@ var Framework = /*#__PURE__*/function () {
     return callApi;
   }();
   _proto.subscribeEvent = /*#__PURE__*/function () {
-    var _subscribeEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref19) {
-      var _this$config$runner$u, _this$config$runner, _ref20, _this$config$runner$s, _this$config$runner2;
-      var id, _ref19$interval, interval, _ref19$timeout, timeout, inngestApiUrl, inngestApiToken, startTime, _poll;
+    var _subscribeEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref20) {
+      var _this$config$runner$u, _this$config$runner, _ref21, _this$config$runner$s, _this$config$runner2;
+      var id, _ref20$interval, interval, _ref20$timeout, timeout, inngestApiUrl, inngestApiToken, startTime, _poll;
       return _regeneratorRuntime().wrap(function _callee8$(_context8) {
         while (1) switch (_context8.prev = _context8.next) {
           case 0:
-            id = _ref19.id, _ref19$interval = _ref19.interval, interval = _ref19$interval === void 0 ? 5000 : _ref19$interval, _ref19$timeout = _ref19.timeout, timeout = _ref19$timeout === void 0 ? 60000 : _ref19$timeout;
+            id = _ref20.id, _ref20$interval = _ref20.interval, interval = _ref20$interval === void 0 ? 5000 : _ref20$interval, _ref20$timeout = _ref20.timeout, timeout = _ref20$timeout === void 0 ? 60000 : _ref20$timeout;
             inngestApiUrl = (_this$config$runner$u = (_this$config$runner = this.config.runner) == null ? void 0 : _this$config$runner.uri) != null ? _this$config$runner$u : process.env.INNGEST_URL;
-            inngestApiToken = (_ref20 = (_this$config$runner$s = (_this$config$runner2 = this.config.runner) == null ? void 0 : _this$config$runner2.signingKey) != null ? _this$config$runner$s : process.env.INNGEST_SIGNING_KEY) != null ? _ref20 : '123';
+            inngestApiToken = (_ref21 = (_this$config$runner$s = (_this$config$runner2 = this.config.runner) == null ? void 0 : _this$config$runner2.signingKey) != null ? _this$config$runner$s : process.env.INNGEST_SIGNING_KEY) != null ? _ref21 : '123';
             startTime = Date.now();
             _poll = /*#__PURE__*/function () {
-              var _ref21 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
-                var response, _yield$response$json, data, error, lastRun, elapsedTime;
+              var _ref22 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
+                var response, obj, data, error, lastRun, elapsedTime;
                 return _regeneratorRuntime().wrap(function _callee7$(_context7) {
                   while (1) switch (_context7.prev = _context7.next) {
                     case 0:
@@ -4652,65 +6379,93 @@ var Framework = /*#__PURE__*/function () {
                       });
                     case 3:
                       response = _context7.sent;
+                      console.log("Polling for event " + id + "...", response.ok);
                       if (!response.ok) {
-                        _context7.next = 16;
+                        _context7.next = 28;
                         break;
                       }
-                      _context7.next = 7;
+                      _context7.next = 8;
                       return response.json();
-                    case 7:
-                      _yield$response$json = _context7.sent;
-                      data = _yield$response$json.data;
-                      error = _yield$response$json.error;
+                    case 8:
+                      obj = _context7.sent;
+                      console.log({
+                        obj: obj
+                      });
+                      data = obj.data, error = obj.error;
+                      console.log("Got data for event " + id + "...", data, error);
                       if (!error) {
-                        _context7.next = 12;
+                        _context7.next = 15;
                         break;
                       }
+                      console.error(error);
                       return _context7.abrupt("return", null);
-                    case 12:
+                    case 15:
+                      if (!((data == null ? void 0 : data.length) === 0)) {
+                        _context7.next = 19;
+                        break;
+                      }
+                      _context7.next = 18;
+                      return new Promise(function (resolve) {
+                        return setTimeout(resolve, interval);
+                      });
+                    case 18:
+                      return _context7.abrupt("return", _poll());
+                    case 19:
                       lastRun = data == null ? void 0 : data[0];
+                      console.log(lastRun);
                       if (lastRun) {
-                        _context7.next = 15;
+                        _context7.next = 23;
                         break;
                       }
                       return _context7.abrupt("return", null);
-                    case 15:
+                    case 23:
+                      if (!(lastRun.status === 'Running')) {
+                        _context7.next = 27;
+                        break;
+                      }
+                      _context7.next = 26;
+                      return new Promise(function (resolve) {
+                        return setTimeout(resolve, interval);
+                      });
+                    case 26:
+                      return _context7.abrupt("return", _poll());
+                    case 27:
                       return _context7.abrupt("return", {
                         status: lastRun.status,
                         startedAt: lastRun.run_started_at,
                         endedAt: lastRun.ended_at
                       });
-                    case 16:
-                      _context7.next = 21;
+                    case 28:
+                      _context7.next = 33;
                       break;
-                    case 18:
-                      _context7.prev = 18;
+                    case 30:
+                      _context7.prev = 30;
                       _context7.t0 = _context7["catch"](0);
                       console.error("Request failed: " + _context7.t0);
-                    case 21:
+                    case 33:
                       // Check if timeout has been reached
                       elapsedTime = Date.now() - startTime;
                       if (!(elapsedTime >= timeout)) {
-                        _context7.next = 25;
+                        _context7.next = 37;
                         break;
                       }
                       console.log('Polling timeout reached.');
                       return _context7.abrupt("return", null);
-                    case 25:
-                      _context7.next = 27;
+                    case 37:
+                      _context7.next = 39;
                       return new Promise(function (resolve) {
                         return setTimeout(resolve, interval);
                       });
-                    case 27:
+                    case 39:
                       return _context7.abrupt("return", _poll());
-                    case 28:
+                    case 40:
                     case "end":
                       return _context7.stop();
                   }
-                }, _callee7, null, [[0, 18]]);
+                }, _callee7, null, [[0, 30]]);
               }));
               return function poll() {
-                return _ref21.apply(this, arguments);
+                return _ref22.apply(this, arguments);
               };
             }();
             return _context8.abrupt("return", _poll());
@@ -4726,13 +6481,13 @@ var Framework = /*#__PURE__*/function () {
     return subscribeEvent;
   }();
   _proto.triggerEvent = /*#__PURE__*/function () {
-    var _triggerEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(_ref22) {
-      var _this3 = this;
-      var key, data, user, _ref22$integrationNam, integrationName, returnObj, integrationEvents, integrationEvent, event, workflowEvent;
+    var _triggerEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(_ref23) {
+      var _this4 = this;
+      var key, data, user, _ref23$integrationNam, integrationName, returnObj, integrationEvents, integrationEvent, event, workflowEvent;
       return _regeneratorRuntime().wrap(function _callee11$(_context11) {
         while (1) switch (_context11.prev = _context11.next) {
           case 0:
-            key = _ref22.key, data = _ref22.data, user = _ref22.user, _ref22$integrationNam = _ref22.integrationName, integrationName = _ref22$integrationNam === void 0 ? this.config.name : _ref22$integrationNam;
+            key = _ref23.key, data = _ref23.data, user = _ref23.user, _ref23$integrationNam = _ref23.integrationName, integrationName = _ref23$integrationNam === void 0 ? this.config.name : _ref23$integrationNam;
             _context11.prev = 1;
             returnObj = {
               event: {}
@@ -4752,7 +6507,7 @@ var Framework = /*#__PURE__*/function () {
             throw new Error("No event exists for " + key + " in " + integrationName);
           case 9:
             _context11.next = 11;
-            return client.send({
+            return client$1.send({
               name: key,
               data: data,
               user: user
@@ -4763,12 +6518,12 @@ var Framework = /*#__PURE__*/function () {
               subscribe: function () {
                 var _subscribe = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_temp) {
                   var _event$ids;
-                  var _ref23, interval, timeout;
+                  var _ref24, interval, timeout;
                   return _regeneratorRuntime().wrap(function _callee9$(_context9) {
                     while (1) switch (_context9.prev = _context9.next) {
                       case 0:
-                        _ref23 = _temp === void 0 ? {} : _temp, interval = _ref23.interval, timeout = _ref23.timeout;
-                        return _context9.abrupt("return", _this3.subscribeEvent({
+                        _ref24 = _temp === void 0 ? {} : _temp, interval = _ref24.interval, timeout = _ref24.timeout;
+                        return _context9.abrupt("return", _this4.subscribeEvent({
                           id: (_event$ids = event.ids) == null ? void 0 : _event$ids[0],
                           interval: interval,
                           timeout: timeout
@@ -4786,7 +6541,7 @@ var Framework = /*#__PURE__*/function () {
               }()
             });
             _context11.next = 15;
-            return client.send({
+            return client$1.send({
               name: 'workflow/run-automations',
               data: {
                 trigger: key,
@@ -4800,12 +6555,12 @@ var Framework = /*#__PURE__*/function () {
               subscribe: function () {
                 var _subscribe2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(_temp2) {
                   var _workflowEvent$ids;
-                  var _ref24, interval, timeout;
+                  var _ref25, interval, timeout;
                   return _regeneratorRuntime().wrap(function _callee10$(_context10) {
                     while (1) switch (_context10.prev = _context10.next) {
                       case 0:
-                        _ref24 = _temp2 === void 0 ? {} : _temp2, interval = _ref24.interval, timeout = _ref24.timeout;
-                        return _context10.abrupt("return", _this3.subscribeEvent({
+                        _ref25 = _temp2 === void 0 ? {} : _temp2, interval = _ref25.interval, timeout = _ref25.timeout;
+                        return _context10.abrupt("return", _this4.subscribeEvent({
                           id: (_workflowEvent$ids = workflowEvent.ids) == null ? void 0 : _workflowEvent$ids[0],
                           interval: interval,
                           timeout: timeout
@@ -4842,14 +6597,14 @@ var Framework = /*#__PURE__*/function () {
   _proto.triggerSystemEvent =
   /*#__PURE__*/
   function () {
-    var _triggerSystemEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(_ref25) {
+    var _triggerSystemEvent = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(_ref26) {
       var key, data, user, event, systemEvent;
       return _regeneratorRuntime().wrap(function _callee12$(_context12) {
         while (1) switch (_context12.prev = _context12.next) {
           case 0:
-            key = _ref25.key, data = _ref25.data, user = _ref25.user;
+            key = _ref26.key, data = _ref26.data, user = _ref26.user;
             _context12.next = 3;
-            return client.send({
+            return client$1.send({
               name: key,
               data: data,
               user: user
@@ -4862,7 +6617,7 @@ var Framework = /*#__PURE__*/function () {
               break;
             }
             _context12.next = 8;
-            return client.send({
+            return client$1.send({
               name: 'workflow/run-automations',
               data: {
                 trigger: key,
@@ -4885,10 +6640,10 @@ var Framework = /*#__PURE__*/function () {
   }();
   _proto.createRouter = function createRouter() {
     var self = this;
-    var makeWebhookUrl = function makeWebhookUrl(_ref26) {
+    var makeWebhookUrl = function makeWebhookUrl(_ref27) {
       var _self$config;
-      var event = _ref26.event,
-        name = _ref26.name;
+      var event = _ref27.event,
+        name = _ref27.name;
       return encodeURI("" + (self == null || (_self$config = self.config) == null ? void 0 : _self$config.systemHostURL) + self.routes.webhook + "?name=" + name + "&event=" + event);
     };
     var makeRedirectURI = function makeRedirectURI() {
@@ -4918,25 +6673,60 @@ var Framework = /*#__PURE__*/function () {
       registerRoutes: registerRoutes
     };
   };
-  return _createClass(Framework, [{
+  _proto.getAgent = /*#__PURE__*/function () {
+    var _getAgent2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(_ref28) {
+      var connectionId, agentId, agentBlueprint, arrMap, finalApis;
+      return _regeneratorRuntime().wrap(function _callee13$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
+          case 0:
+            connectionId = _ref28.connectionId, agentId = _ref28.agentId;
+            _context13.next = 3;
+            return getAgentBlueprint({
+              agentDir: this.config.agents.agentDirPath,
+              agentId: agentId
+            });
+          case 3:
+            agentBlueprint = _context13.sent;
+            arrMap = Array.from(this.getApis());
+            finalApis = arrMap.reduce(function (acc, _ref29) {
+              var v = _ref29[1];
+              return _extends({}, acc, v);
+            }, {});
+            return _context13.abrupt("return", getAgent({
+              connectionId: connectionId,
+              agent: agentBlueprint,
+              apis: finalApis
+            }));
+          case 7:
+          case "end":
+            return _context13.stop();
+        }
+      }, _callee13, this);
+    }));
+    function getAgent$1(_x12) {
+      return _getAgent2.apply(this, arguments);
+    }
+    return getAgent$1;
+  }();
+  return _createClass(Mastra, [{
     key: "routes",
     get: function get() {
-      var _this4 = this;
+      var _this5 = this;
       var registry = {
         connect: '/connect',
         callback: '/connect/callback',
         inngest: '/inngest',
         webhook: '/webhook'
       };
-      return Object.entries(registry).reduce(function (acc, _ref27) {
-        var _extends3;
-        var key = _ref27[0],
-          value = _ref27[1];
-        return _extends({}, acc, (_extends3 = {}, _extends3[key] = "" + _this4.config.routeRegistrationPath + value, _extends3));
+      return Object.entries(registry).reduce(function (acc, _ref30) {
+        var _extends5;
+        var key = _ref30[0],
+          value = _ref30[1];
+        return _extends({}, acc, (_extends5 = {}, _extends5[key] = "" + _this5.config.routeRegistrationPath + value, _extends5));
       }, {});
     }
   }]);
 }();
 
-export { DataLayer, FieldTypePrimitiveMap, FilterOperatorToSQL, FilterOperators, Framework, Integration, IntegrationAuth, IntegrationCredentialType, IntegrationError, IntegrationErrors, IntegrationFieldTypeEnum, RunStatus, SORT_MODE_TO_ABBR, WorkflowStatusEnum, allowedKey, apiKeyConnectionOptions, buildQueryString, callbackParams, connectParams, constructObjFromStringPath, extractSchemaOptions, fieldsWithCommaSeparatedValues, filterQuerySchema, flattenObject, generateSyncs, getEntityKey, getJSONField, getPath, getResponseDataKey, isLiteralObject, isObjectEmpty, mergeWithDefinedOnly, nextHeaders, oauthState, parseQueryParams, recordHasData, registerRoutes, sortQuerySchema, splitAndStripQuotes, transformFilterValueArray, transformFilterValueBoolean, webhookQueryParams };
+export { DataLayer, FieldTypePrimitiveMap, FilterOperatorToSQL, FilterOperators, Integration, IntegrationAuth, IntegrationCredentialType, IntegrationError, IntegrationErrors, IntegrationFieldTypeEnum, Mastra, RunStatus, SORT_MODE_TO_ABBR, WorkflowStatusEnum, allowedKey, apiKeyConnectionOptions, buildQueryString, callbackParams, connectParams, constructObjFromStringPath, extractSchemaOptions, fieldsWithCommaSeparatedValues, filterQuerySchema, flattenObject, generateSyncs, getEntityKey, getJSONField, getPath, getResponseDataKey, isLiteralObject, isObjectEmpty, mergeWithDefinedOnly, nextHeaders, oauthState, parseQueryParams, recordHasData, registerRoutes, sortQuerySchema, splitAndStripQuotes, transformFilterValueArray, transformFilterValueBoolean, webhookQueryParams };
 //# sourceMappingURL=core.esm.js.map