@mastra/core 0.1.27-alpha.50 → 0.1.27-alpha.52

package/dist/core.cjs.development.js

@@ -3560,6 +3560,345 @@ var MastraEngine = /*#__PURE__*/function () {
   return MastraEngine;
 }();
 
+var MockMastraEngine = /*#__PURE__*/function (_MastraEngine) {
+  function MockMastraEngine(config) {
+    var _this;
+    _this = _MastraEngine.call(this, config) || this;
+    _this.entities = [];
+    _this.records = [];
+    return _this;
+  }
+  _inheritsLoose(MockMastraEngine, _MastraEngine);
+  var _proto = MockMastraEngine.prototype;
+  _proto.createEntity = /*#__PURE__*/function () {
+    var _createEntity = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(params) {
+      var entity;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            entity = {
+              id: "entity_" + Date.now(),
+              name: params.name,
+              connectionId: params.connectionId,
+              createdAt: new Date(),
+              updatedAt: new Date(),
+              lastSyncId: null
+            };
+            this.entities.push(entity);
+            return _context.abrupt("return", entity);
+          case 3:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function createEntity(_x) {
+      return _createEntity.apply(this, arguments);
+    }
+    return createEntity;
+  }();
+  _proto.getEntityById = /*#__PURE__*/function () {
+    var _getEntityById = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(params) {
+      var entity;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            entity = this.entities.find(function (e) {
+              return e.id === params.id;
+            });
+            if (entity) {
+              _context2.next = 3;
+              break;
+            }
+            throw new Error("Entity with id " + params.id + " not found");
+          case 3:
+            return _context2.abrupt("return", entity);
+          case 4:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, this);
+    }));
+    function getEntityById(_x2) {
+      return _getEntityById.apply(this, arguments);
+    }
+    return getEntityById;
+  }();
+  _proto.getEntity = /*#__PURE__*/function () {
+    var _getEntity = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(_ref) {
+      var connectionId, name;
+      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
+          case 0:
+            connectionId = _ref.connectionId, name = _ref.name;
+            return _context3.abrupt("return", this.entities.find(function (e) {
+              return (name ? e.name === name : true) && (connectionId ? e.connectionId === connectionId : true);
+            }));
+          case 2:
+          case "end":
+            return _context3.stop();
+        }
+      }, _callee3, this);
+    }));
+    function getEntity(_x3) {
+      return _getEntity.apply(this, arguments);
+    }
+    return getEntity;
+  }();
+  _proto.deleteEntityById = /*#__PURE__*/function () {
+    var _deleteEntityById = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(_ref2) {
+      var id, entityIndex, _this$entities$splice, deletedEntity;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            id = _ref2.id;
+            entityIndex = this.entities.findIndex(function (e) {
+              return e.id === id;
+            });
+            if (!(entityIndex === -1)) {
+              _context4.next = 4;
+              break;
+            }
+            throw new Error("Entity with id " + id + " not found");
+          case 4:
+            _this$entities$splice = this.entities.splice(entityIndex, 1), deletedEntity = _this$entities$splice[0]; // Also delete associated records
+            this.records = this.records.filter(function (r) {
+              return r.entityId !== id;
+            });
+            return _context4.abrupt("return", deletedEntity);
+          case 7:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4, this);
+    }));
+    function deleteEntityById(_x4) {
+      return _deleteEntityById.apply(this, arguments);
+    }
+    return deleteEntityById;
+  }();
+  _proto.upsertRecords = /*#__PURE__*/function () {
+    var _upsertRecords = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(params) {
+      var _this2 = this;
+      var _loop, _iterator, _step;
+      return _regeneratorRuntime().wrap(function _callee5$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            _context6.next = 2;
+            return this.getEntityById({
+              id: params.entityId
+            });
+          case 2:
+            _loop = /*#__PURE__*/_regeneratorRuntime().mark(function _loop() {
+              var record, existingRecordIndex, fullRecord;
+              return _regeneratorRuntime().wrap(function _loop$(_context5) {
+                while (1) switch (_context5.prev = _context5.next) {
+                  case 0:
+                    record = _step.value;
+                    existingRecordIndex = _this2.records.findIndex(function (r) {
+                      return r.entityId === params.entityId && r.externalId === record.externalId;
+                    });
+                    fullRecord = {
+                      id: "record_" + Date.now() + "_" + Math.random(),
+                      entityId: params.entityId,
+                      externalId: record.externalId,
+                      data: record.data,
+                      entityType: record.entityType,
+                      createdAt: new Date(),
+                      updatedAt: new Date()
+                    };
+                    if (existingRecordIndex !== -1) {
+                      _this2.records[existingRecordIndex] = fullRecord;
+                    } else {
+                      _this2.records.push(fullRecord);
+                    }
+                  case 4:
+                  case "end":
+                    return _context5.stop();
+                }
+              }, _loop);
+            });
+            _iterator = _createForOfIteratorHelperLoose(params.records);
+          case 4:
+            if ((_step = _iterator()).done) {
+              _context6.next = 8;
+              break;
+            }
+            return _context6.delegateYield(_loop(), "t0", 6);
+          case 6:
+            _context6.next = 4;
+            break;
+          case 8:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee5, this);
+    }));
+    function upsertRecords(_x5) {
+      return _upsertRecords.apply(this, arguments);
+    }
+    return upsertRecords;
+  }();
+  _proto.getRecordsByEntityId = /*#__PURE__*/function () {
+    var _getRecordsByEntityId = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(params) {
+      return _regeneratorRuntime().wrap(function _callee6$(_context7) {
+        while (1) switch (_context7.prev = _context7.next) {
+          case 0:
+            return _context7.abrupt("return", this.records.filter(function (r) {
+              return r.entityId === params.entityId;
+            }));
+          case 1:
+          case "end":
+            return _context7.stop();
+        }
+      }, _callee6, this);
+    }));
+    function getRecordsByEntityId(_x6) {
+      return _getRecordsByEntityId.apply(this, arguments);
+    }
+    return getRecordsByEntityId;
+  }();
+  _proto.getRecordsByEntityName = /*#__PURE__*/function () {
+    var _getRecordsByEntityName = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(_ref3) {
+      var name, connectionId, entity;
+      return _regeneratorRuntime().wrap(function _callee7$(_context8) {
+        while (1) switch (_context8.prev = _context8.next) {
+          case 0:
+            name = _ref3.name, connectionId = _ref3.connectionId;
+            _context8.next = 3;
+            return this.getEntity({
+              name: name,
+              connectionId: connectionId
+            });
+          case 3:
+            entity = _context8.sent;
+            if (entity) {
+              _context8.next = 6;
+              break;
+            }
+            return _context8.abrupt("return", []);
+          case 6:
+            return _context8.abrupt("return", this.getRecordsByEntityId({
+              entityId: entity.id
+            }));
+          case 7:
+          case "end":
+            return _context8.stop();
+        }
+      }, _callee7, this);
+    }));
+    function getRecordsByEntityName(_x7) {
+      return _getRecordsByEntityName.apply(this, arguments);
+    }
+    return getRecordsByEntityName;
+  }();
+  _proto.getRecords = /*#__PURE__*/function () {
+    var _getRecords = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(_ref4) {
+      var entityName, connectionId, options, entity, records, start;
+      return _regeneratorRuntime().wrap(function _callee8$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            entityName = _ref4.entityName, connectionId = _ref4.connectionId, options = _ref4.options;
+            _context9.next = 3;
+            return this.getEntity({
+              name: entityName,
+              connectionId: connectionId
+            });
+          case 3:
+            entity = _context9.sent;
+            if (entity) {
+              _context9.next = 6;
+              break;
+            }
+            return _context9.abrupt("return", []);
+          case 6:
+            _context9.next = 8;
+            return this.getRecordsByEntityId({
+              entityId: entity.id
+            });
+          case 8:
+            records = _context9.sent;
+            // Apply basic filtering based on options
+            if (options.filters) {
+              records = records.filter(function () {
+                // Implement your filtering logic here
+                return true; // Placeholder
+              });
+            }
+            // Apply sorting
+            if (options.sort) {
+              records = records.sort(function () {
+                // Implement your sorting logic here
+                return 0; // Placeholder
+              });
+            }
+            // Apply pagination
+            if (options.limit) {
+              start = options.offset || 0;
+              records = records.slice(start, start + options.limit);
+            }
+            return _context9.abrupt("return", records);
+          case 13:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee8, this);
+    }));
+    function getRecords(_x8) {
+      return _getRecords.apply(this, arguments);
+    }
+    return getRecords;
+  }();
+  _proto.syncRecords = /*#__PURE__*/function () {
+    var _syncRecords = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(_ref5) {
+      var connectionId, name, records, lastSyncId, entity;
+      return _regeneratorRuntime().wrap(function _callee9$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            connectionId = _ref5.connectionId, name = _ref5.name, records = _ref5.records, lastSyncId = _ref5.lastSyncId;
+            _context10.next = 3;
+            return this.getEntity({
+              name: name,
+              connectionId: connectionId
+            });
+          case 3:
+            entity = _context10.sent;
+            if (entity) {
+              _context10.next = 8;
+              break;
+            }
+            _context10.next = 7;
+            return this.createEntity({
+              name: name,
+              connectionId: connectionId
+            });
+          case 7:
+            entity = _context10.sent;
+          case 8:
+            _context10.next = 10;
+            return this.upsertRecords({
+              entityId: entity.id,
+              records: records.map(function (record) {
+                return _extends({}, record, {
+                  lastSyncId: lastSyncId,
+                  entityType: name // Using name as entityType for simplicity
+                });
+              })
+            });
+          case 10:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee9, this);
+    }));
+    function syncRecords(_x9) {
+      return _syncRecords.apply(this, arguments);
+    }
+    return syncRecords;
+  }();
+  return MockMastraEngine;
+}(MastraEngine);
+
 var _telemetry = /*#__PURE__*/_classPrivateFieldLooseKey("telemetry");
 var MastraVector = /*#__PURE__*/function () {
   function MastraVector() {
@@ -4964,7 +5303,7 @@ function _resolveVariables2(_ref30) {
       continue;
     }
     // If path is empty or '.', return the entire source data
-    var value = variable.path === '' || variable.path === '.' ? sourceData[key] : radash.get(sourceData, variable.path);
+    var value = variable.path === '' || variable.path === '.' ? sourceData : radash.get(sourceData, variable.path);
     this.log(LogLevel.DEBUG, "Resolved variable " + key, {
       value: value,
       runId: _classPrivateFieldLooseBase(this, _runId)[_runId]
@@ -5481,6 +5820,7 @@ exports.LogLevel = LogLevel;
 exports.MastraEngine = MastraEngine;
 exports.MastraMemory = MastraMemory;
 exports.MastraVector = MastraVector;
+exports.MockMastraEngine = MockMastraEngine;
 exports.MultiLogger = MultiLogger;
 exports.OpenAPIToolset = OpenAPIToolset;
 exports.RegisteredLogger = RegisteredLogger;