@mastra/memory 0.0.1

package/dist/memory.esm.js

@@ -0,0 +1,2361 @@
+import { MastraMemory } from '@mastra/core';
+import { randomUUID } from 'crypto';
+import pg from 'pg';
+import { Redis } from '@upstash/redis';
+import { createClient } from 'redis';
+
+function _arrayLikeToArray(r, a) {
+  (null == a || a > r.length) && (a = r.length);
+  for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e];
+  return n;
+}
+function asyncGeneratorStep(n, t, e, r, o, a, c) {
+  try {
+    var i = n[a](c),
+      u = i.value;
+  } catch (n) {
+    return void e(n);
+  }
+  i.done ? t(u) : Promise.resolve(u).then(r, o);
+}
+function _asyncToGenerator(n) {
+  return function () {
+    var t = this,
+      e = arguments;
+    return new Promise(function (r, o) {
+      var a = n.apply(t, e);
+      function _next(n) {
+        asyncGeneratorStep(a, r, o, _next, _throw, "next", n);
+      }
+      function _throw(n) {
+        asyncGeneratorStep(a, r, o, _next, _throw, "throw", n);
+      }
+      _next(void 0);
+    });
+  };
+}
+function _createForOfIteratorHelperLoose(r, e) {
+  var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"];
+  if (t) return (t = t.call(r)).next.bind(t);
+  if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) {
+    t && (r = t);
+    var o = 0;
+    return function () {
+      return o >= r.length ? {
+        done: !0
+      } : {
+        done: !1,
+        value: r[o++]
+      };
+    };
+  }
+  throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
+}
+function _extends() {
+  return _extends = Object.assign ? Object.assign.bind() : function (n) {
+    for (var e = 1; e < arguments.length; e++) {
+      var t = arguments[e];
+      for (var r in t) ({}).hasOwnProperty.call(t, r) && (n[r] = t[r]);
+    }
+    return n;
+  }, _extends.apply(null, arguments);
+}
+function _inheritsLoose(t, o) {
+  t.prototype = Object.create(o.prototype), t.prototype.constructor = t, _setPrototypeOf(t, o);
+}
+function _regeneratorRuntime() {
+  _regeneratorRuntime = function () {
+    return e;
+  };
+  var t,
+    e = {},
+    r = Object.prototype,
+    n = r.hasOwnProperty,
+    o = Object.defineProperty || function (t, e, r) {
+      t[e] = r.value;
+    },
+    i = "function" == typeof Symbol ? Symbol : {},
+    a = i.iterator || "@@iterator",
+    c = i.asyncIterator || "@@asyncIterator",
+    u = i.toStringTag || "@@toStringTag";
+  function define(t, e, r) {
+    return Object.defineProperty(t, e, {
+      value: r,
+      enumerable: !0,
+      configurable: !0,
+      writable: !0
+    }), t[e];
+  }
+  try {
+    define({}, "");
+  } catch (t) {
+    define = function (t, e, r) {
+      return t[e] = r;
+    };
+  }
+  function wrap(t, e, r, n) {
+    var i = e && e.prototype instanceof Generator ? e : Generator,
+      a = Object.create(i.prototype),
+      c = new Context(n || []);
+    return o(a, "_invoke", {
+      value: makeInvokeMethod(t, r, c)
+    }), a;
+  }
+  function tryCatch(t, e, r) {
+    try {
+      return {
+        type: "normal",
+        arg: t.call(e, r)
+      };
+    } catch (t) {
+      return {
+        type: "throw",
+        arg: t
+      };
+    }
+  }
+  e.wrap = wrap;
+  var h = "suspendedStart",
+    l = "suspendedYield",
+    f = "executing",
+    s = "completed",
+    y = {};
+  function Generator() {}
+  function GeneratorFunction() {}
+  function GeneratorFunctionPrototype() {}
+  var p = {};
+  define(p, a, function () {
+    return this;
+  });
+  var d = Object.getPrototypeOf,
+    v = d && d(d(values([])));
+  v && v !== r && n.call(v, a) && (p = v);
+  var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p);
+  function defineIteratorMethods(t) {
+    ["next", "throw", "return"].forEach(function (e) {
+      define(t, e, function (t) {
+        return this._invoke(e, t);
+      });
+    });
+  }
+  function AsyncIterator(t, e) {
+    function invoke(r, o, i, a) {
+      var c = tryCatch(t[r], t, o);
+      if ("throw" !== c.type) {
+        var u = c.arg,
+          h = u.value;
+        return h && "object" == typeof h && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) {
+          invoke("next", t, i, a);
+        }, function (t) {
+          invoke("throw", t, i, a);
+        }) : e.resolve(h).then(function (t) {
+          u.value = t, i(u);
+        }, function (t) {
+          return invoke("throw", t, i, a);
+        });
+      }
+      a(c.arg);
+    }
+    var r;
+    o(this, "_invoke", {
+      value: function (t, n) {
+        function callInvokeWithMethodAndArg() {
+          return new e(function (e, r) {
+            invoke(t, n, e, r);
+          });
+        }
+        return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg();
+      }
+    });
+  }
+  function makeInvokeMethod(e, r, n) {
+    var o = h;
+    return function (i, a) {
+      if (o === f) throw Error("Generator is already running");
+      if (o === s) {
+        if ("throw" === i) throw a;
+        return {
+          value: t,
+          done: !0
+        };
+      }
+      for (n.method = i, n.arg = a;;) {
+        var c = n.delegate;
+        if (c) {
+          var u = maybeInvokeDelegate(c, n);
+          if (u) {
+            if (u === y) continue;
+            return u;
+          }
+        }
+        if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) {
+          if (o === h) throw o = s, n.arg;
+          n.dispatchException(n.arg);
+        } else "return" === n.method && n.abrupt("return", n.arg);
+        o = f;
+        var p = tryCatch(e, r, n);
+        if ("normal" === p.type) {
+          if (o = n.done ? s : l, p.arg === y) continue;
+          return {
+            value: p.arg,
+            done: n.done
+          };
+        }
+        "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg);
+      }
+    };
+  }
+  function maybeInvokeDelegate(e, r) {
+    var n = r.method,
+      o = e.iterator[n];
+    if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y;
+    var i = tryCatch(o, e.iterator, r.arg);
+    if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y;
+    var a = i.arg;
+    return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y);
+  }
+  function pushTryEntry(t) {
+    var e = {
+      tryLoc: t[0]
+    };
+    1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e);
+  }
+  function resetTryEntry(t) {
+    var e = t.completion || {};
+    e.type = "normal", delete e.arg, t.completion = e;
+  }
+  function Context(t) {
+    this.tryEntries = [{
+      tryLoc: "root"
+    }], t.forEach(pushTryEntry, this), this.reset(!0);
+  }
+  function values(e) {
+    if (e || "" === e) {
+      var r = e[a];
+      if (r) return r.call(e);
+      if ("function" == typeof e.next) return e;
+      if (!isNaN(e.length)) {
+        var o = -1,
+          i = function next() {
+            for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next;
+            return next.value = t, next.done = !0, next;
+          };
+        return i.next = i;
+      }
+    }
+    throw new TypeError(typeof e + " is not iterable");
+  }
+  return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", {
+    value: GeneratorFunctionPrototype,
+    configurable: !0
+  }), o(GeneratorFunctionPrototype, "constructor", {
+    value: GeneratorFunction,
+    configurable: !0
+  }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) {
+    var e = "function" == typeof t && t.constructor;
+    return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name));
+  }, e.mark = function (t) {
+    return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t;
+  }, e.awrap = function (t) {
+    return {
+      __await: t
+    };
+  }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () {
+    return this;
+  }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) {
+    void 0 === i && (i = Promise);
+    var a = new AsyncIterator(wrap(t, r, n, o), i);
+    return e.isGeneratorFunction(r) ? a : a.next().then(function (t) {
+      return t.done ? t.value : a.next();
+    });
+  }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () {
+    return this;
+  }), define(g, "toString", function () {
+    return "[object Generator]";
+  }), e.keys = function (t) {
+    var e = Object(t),
+      r = [];
+    for (var n in e) r.push(n);
+    return r.reverse(), function next() {
+      for (; r.length;) {
+        var t = r.pop();
+        if (t in e) return next.value = t, next.done = !1, next;
+      }
+      return next.done = !0, next;
+    };
+  }, e.values = values, Context.prototype = {
+    constructor: Context,
+    reset: function (e) {
+      if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t);
+    },
+    stop: function () {
+      this.done = !0;
+      var t = this.tryEntries[0].completion;
+      if ("throw" === t.type) throw t.arg;
+      return this.rval;
+    },
+    dispatchException: function (e) {
+      if (this.done) throw e;
+      var r = this;
+      function handle(n, o) {
+        return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o;
+      }
+      for (var o = this.tryEntries.length - 1; o >= 0; --o) {
+        var i = this.tryEntries[o],
+          a = i.completion;
+        if ("root" === i.tryLoc) return handle("end");
+        if (i.tryLoc <= this.prev) {
+          var c = n.call(i, "catchLoc"),
+            u = n.call(i, "finallyLoc");
+          if (c && u) {
+            if (this.prev < i.catchLoc) return handle(i.catchLoc, !0);
+            if (this.prev < i.finallyLoc) return handle(i.finallyLoc);
+          } else if (c) {
+            if (this.prev < i.catchLoc) return handle(i.catchLoc, !0);
+          } else {
+            if (!u) throw Error("try statement without catch or finally");
+            if (this.prev < i.finallyLoc) return handle(i.finallyLoc);
+          }
+        }
+      }
+    },
+    abrupt: function (t, e) {
+      for (var r = this.tryEntries.length - 1; r >= 0; --r) {
+        var o = this.tryEntries[r];
+        if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) {
+          var i = o;
+          break;
+        }
+      }
+      i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null);
+      var a = i ? i.completion : {};
+      return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a);
+    },
+    complete: function (t, e) {
+      if ("throw" === t.type) throw t.arg;
+      return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y;
+    },
+    finish: function (t) {
+      for (var e = this.tryEntries.length - 1; e >= 0; --e) {
+        var r = this.tryEntries[e];
+        if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y;
+      }
+    },
+    catch: function (t) {
+      for (var e = this.tryEntries.length - 1; e >= 0; --e) {
+        var r = this.tryEntries[e];
+        if (r.tryLoc === t) {
+          var n = r.completion;
+          if ("throw" === n.type) {
+            var o = n.arg;
+            resetTryEntry(r);
+          }
+          return o;
+        }
+      }
+      throw Error("illegal catch attempt");
+    },
+    delegateYield: function (e, r, n) {
+      return this.delegate = {
+        iterator: values(e),
+        resultName: r,
+        nextLoc: n
+      }, "next" === this.method && (this.arg = t), y;
+    }
+  }, e;
+}
+function _setPrototypeOf(t, e) {
+  return _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function (t, e) {
+    return t.__proto__ = e, t;
+  }, _setPrototypeOf(t, e);
+}
+function _unsupportedIterableToArray(r, a) {
+  if (r) {
+    if ("string" == typeof r) return _arrayLikeToArray(r, a);
+    var t = {}.toString.call(r).slice(8, -1);
+    return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0;
+  }
+}
+
+var CloudflareKVProvider = /*#__PURE__*/function () {
+  function CloudflareKVProvider(namespace) {
+    this.namespace = void 0;
+    this.namespace = namespace;
+  }
+  var _proto2 = CloudflareKVProvider.prototype;
+  _proto2.get = /*#__PURE__*/function () {
+    var _get2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(key) {
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
+          case 0:
+            return _context5.abrupt("return", this.namespace.get(key, 'json'));
+          case 1:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, this);
+    }));
+    function get(_x7) {
+      return _get2.apply(this, arguments);
+    }
+    return get;
+  }();
+  _proto2.set = /*#__PURE__*/function () {
+    var _set = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(key, value) {
+      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            _context6.next = 2;
+            return this.namespace.put(key, JSON.stringify(value));
+          case 2:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee6, this);
+    }));
+    function set(_x8, _x9) {
+      return _set.apply(this, arguments);
+    }
+    return set;
+  }();
+  _proto2.del = /*#__PURE__*/function () {
+    var _del = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(key) {
+      return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+        while (1) switch (_context7.prev = _context7.next) {
+          case 0:
+            _context7.next = 2;
+            return this.namespace["delete"](key);
+          case 2:
+          case "end":
+            return _context7.stop();
+        }
+      }, _callee7, this);
+    }));
+    function del(_x10) {
+      return _del.apply(this, arguments);
+    }
+    return del;
+  }();
+  _proto2.sadd = /*#__PURE__*/function () {
+    var _sadd = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(key, value) {
+      var set;
+      return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+        while (1) switch (_context8.prev = _context8.next) {
+          case 0:
+            _context8.next = 2;
+            return this.get(key);
+          case 2:
+            _context8.t0 = _context8.sent;
+            if (_context8.t0) {
+              _context8.next = 5;
+              break;
+            }
+            _context8.t0 = [];
+          case 5:
+            set = _context8.t0;
+            if (set.includes(value)) {
+              _context8.next = 11;
+              break;
+            }
+            set.push(value);
+            _context8.next = 10;
+            return this.set(key, set);
+          case 10:
+            return _context8.abrupt("return", 1);
+          case 11:
+            return _context8.abrupt("return", 0);
+          case 12:
+          case "end":
+            return _context8.stop();
+        }
+      }, _callee8, this);
+    }));
+    function sadd(_x11, _x12) {
+      return _sadd.apply(this, arguments);
+    }
+    return sadd;
+  }();
+  _proto2.srem = /*#__PURE__*/function () {
+    var _srem = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(key, value) {
+      var set, index;
+      return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            _context9.next = 2;
+            return this.get(key);
+          case 2:
+            _context9.t0 = _context9.sent;
+            if (_context9.t0) {
+              _context9.next = 5;
+              break;
+            }
+            _context9.t0 = [];
+          case 5:
+            set = _context9.t0;
+            index = set.indexOf(value);
+            if (!(index !== -1)) {
+              _context9.next = 12;
+              break;
+            }
+            set.splice(index, 1);
+            _context9.next = 11;
+            return this.set(key, set);
+          case 11:
+            return _context9.abrupt("return", 1);
+          case 12:
+            return _context9.abrupt("return", 0);
+          case 13:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee9, this);
+    }));
+    function srem(_x13, _x14) {
+      return _srem.apply(this, arguments);
+    }
+    return srem;
+  }();
+  _proto2.smembers = /*#__PURE__*/function () {
+    var _smembers = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(key) {
+      return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            _context10.next = 2;
+            return this.get(key);
+          case 2:
+            _context10.t0 = _context10.sent;
+            if (_context10.t0) {
+              _context10.next = 5;
+              break;
+            }
+            _context10.t0 = [];
+          case 5:
+            return _context10.abrupt("return", _context10.t0);
+          case 6:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee10, this);
+    }));
+    function smembers(_x15) {
+      return _smembers.apply(this, arguments);
+    }
+    return smembers;
+  }();
+  _proto2.flushall = /*#__PURE__*/function () {
+    var _flushall = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11() {
+      var _this = this;
+      var cursor, result;
+      return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
+          case 0:
+            _context11.next = 2;
+            return this.namespace.list({
+              cursor: cursor
+            });
+          case 2:
+            result = _context11.sent;
+            _context11.next = 5;
+            return Promise.all(result.keys.map(function (key) {
+              return _this.namespace["delete"](key.name);
+            }));
+          case 5:
+            cursor = result.cursor;
+          case 6:
+            if (cursor) {
+              _context11.next = 0;
+              break;
+            }
+          case 7:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee11, this);
+    }));
+    function flushall() {
+      return _flushall.apply(this, arguments);
+    }
+    return flushall;
+  }();
+  _proto2.getWithVersion = /*#__PURE__*/function () {
+    var _getWithVersion = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(key) {
+      var versionKey, _yield$Promise$all, data, version;
+      return _regeneratorRuntime().wrap(function _callee12$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
+          case 0:
+            versionKey = key + ":version";
+            _context12.next = 3;
+            return Promise.all([this.get(key), this.get(versionKey).then(function (v) {
+              return v || 0;
+            })]);
+          case 3:
+            _yield$Promise$all = _context12.sent;
+            data = _yield$Promise$all[0];
+            version = _yield$Promise$all[1];
+            return _context12.abrupt("return", {
+              data: data,
+              version: version
+            });
+          case 7:
+          case "end":
+            return _context12.stop();
+        }
+      }, _callee12, this);
+    }));
+    function getWithVersion(_x16) {
+      return _getWithVersion.apply(this, arguments);
+    }
+    return getWithVersion;
+  }();
+  _proto2.setWithVersion = /*#__PURE__*/function () {
+    var _setWithVersion = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(key, value, expectedVersion) {
+      var versionKey, currentVersion;
+      return _regeneratorRuntime().wrap(function _callee13$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
+          case 0:
+            versionKey = key + ":version";
+            _context13.next = 3;
+            return this.get(versionKey);
+          case 3:
+            _context13.t0 = _context13.sent;
+            if (_context13.t0) {
+              _context13.next = 6;
+              break;
+            }
+            _context13.t0 = 0;
+          case 6:
+            currentVersion = _context13.t0;
+            if (!(currentVersion !== expectedVersion)) {
+              _context13.next = 9;
+              break;
+            }
+            return _context13.abrupt("return", false);
+          case 9:
+            _context13.next = 11;
+            return Promise.all([this.set(key, value), this.set(versionKey, expectedVersion + 1)]);
+          case 11:
+            return _context13.abrupt("return", true);
+          case 12:
+          case "end":
+            return _context13.stop();
+        }
+      }, _callee13, this);
+    }));
+    function setWithVersion(_x17, _x18, _x19) {
+      return _setWithVersion.apply(this, arguments);
+    }
+    return setWithVersion;
+  }();
+  return CloudflareKVProvider;
+}();
+
+var CloudflareKVMemory = /*#__PURE__*/function (_MastraMemory) {
+  function CloudflareKVMemory(namespace) {
+    var _this;
+    _this = _MastraMemory.call(this) || this;
+    _this.kv = void 0;
+    _this.threadPrefix = 'thread:';
+    _this.messagePrefix = 'messages:';
+    _this.kv = new CloudflareKVProvider(namespace);
+    return _this;
+  }
+  _inheritsLoose(CloudflareKVMemory, _MastraMemory);
+  var _proto = CloudflareKVMemory.prototype;
+  _proto.getThreadById = /*#__PURE__*/function () {
+    var _getThreadById = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(threadId) {
+      var thread;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.next = 2;
+            return this.kv.get("" + this.threadPrefix + threadId);
+          case 2:
+            thread = _context.sent;
+            if (thread && typeof thread.createdAt === 'string') {
+              thread.createdAt = new Date(thread.createdAt);
+              thread.updatedAt = new Date(thread.updatedAt);
+            }
+            return _context.abrupt("return", thread);
+          case 5:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function getThreadById(_x) {
+      return _getThreadById.apply(this, arguments);
+    }
+    return getThreadById;
+  }();
+  _proto.saveThread = /*#__PURE__*/function () {
+    var _saveThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(thread) {
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            thread.updatedAt = new Date();
+            _context2.next = 3;
+            return this.kv.set("" + this.threadPrefix + thread.id, thread);
+          case 3:
+            _context2.next = 5;
+            return this.kv.sadd('threads', thread.id);
+          case 5:
+            return _context2.abrupt("return", thread);
+          case 6:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, this);
+    }));
+    function saveThread(_x2) {
+      return _saveThread.apply(this, arguments);
+    }
+    return saveThread;
+  }();
+  _proto.retryOperation = /*#__PURE__*/function () {
+    var _retryOperation = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(operation, maxRetries) {
+      var lastError, _loop, _ret, attempt;
+      return _regeneratorRuntime().wrap(function _callee3$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            if (maxRetries === void 0) {
+              maxRetries = 5;
+            }
+            _loop = /*#__PURE__*/_regeneratorRuntime().mark(function _loop(attempt) {
+              return _regeneratorRuntime().wrap(function _loop$(_context3) {
+                while (1) switch (_context3.prev = _context3.next) {
+                  case 0:
+                    _context3.prev = 0;
+                    _context3.next = 3;
+                    return operation();
+                  case 3:
+                    _context3.t0 = _context3.sent;
+                    return _context3.abrupt("return", {
+                      v: _context3.t0
+                    });
+                  case 7:
+                    _context3.prev = 7;
+                    _context3.t1 = _context3["catch"](0);
+                    lastError = _context3.t1;
+                    _context3.next = 12;
+                    return new Promise(function (resolve) {
+                      return setTimeout(resolve, Math.random() * 100 * Math.pow(2, attempt));
+                    });
+                  case 12:
+                  case "end":
+                    return _context3.stop();
+                }
+              }, _loop, null, [[0, 7]]);
+            });
+            attempt = 0;
+          case 3:
+            if (!(attempt < maxRetries)) {
+              _context4.next = 11;
+              break;
+            }
+            return _context4.delegateYield(_loop(attempt), "t0", 5);
+          case 5:
+            _ret = _context4.t0;
+            if (!_ret) {
+              _context4.next = 8;
+              break;
+            }
+            return _context4.abrupt("return", _ret.v);
+          case 8:
+            attempt++;
+            _context4.next = 3;
+            break;
+          case 11:
+            throw lastError || new Error("Operation failed after " + maxRetries + " attempts");
+          case 12:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee3);
+    }));
+    function retryOperation(_x3, _x4) {
+      return _retryOperation.apply(this, arguments);
+    }
+    return retryOperation;
+  }();
+  _proto.addMessage = /*#__PURE__*/function () {
+    var _addMessage = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(threadId, content, role) {
+      var _this2 = this;
+      var message;
+      return _regeneratorRuntime().wrap(function _callee5$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            message = {
+              id: this.generateId(),
+              content: content,
+              role: role,
+              createdAt: new Date(),
+              threadId: threadId
+            };
+            _context6.next = 3;
+            return this.retryOperation(/*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
+              var key, existingMessages, messageMap, updatedMessages;
+              return _regeneratorRuntime().wrap(function _callee4$(_context5) {
+                while (1) switch (_context5.prev = _context5.next) {
+                  case 0:
+                    key = "" + _this2.messagePrefix + threadId;
+                    _context5.next = 3;
+                    return _this2.kv.get(key);
+                  case 3:
+                    _context5.t0 = _context5.sent;
+                    if (_context5.t0) {
+                      _context5.next = 6;
+                      break;
+                    }
+                    _context5.t0 = [];
+                  case 6:
+                    existingMessages = _context5.t0;
+                    messageMap = new Map();
+                    existingMessages.forEach(function (msg) {
+                      messageMap.set(msg.id, _extends({}, msg, {
+                        createdAt: new Date(msg.createdAt)
+                      }));
+                    });
+                    messageMap.set(message.id, message);
+                    updatedMessages = Array.from(messageMap.values());
+                    updatedMessages.sort(function (a, b) {
+                      var timeCompare = a.createdAt.getTime() - b.createdAt.getTime();
+                      return timeCompare === 0 ? a.id.localeCompare(b.id) : timeCompare;
+                    });
+                    _context5.next = 14;
+                    return _this2.kv.set(key, updatedMessages);
+                  case 14:
+                  case "end":
+                    return _context5.stop();
+                }
+              }, _callee4);
+            })));
+          case 3:
+            return _context6.abrupt("return", message);
+          case 4:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee5, this);
+    }));
+    function addMessage(_x5, _x6, _x7) {
+      return _addMessage.apply(this, arguments);
+    }
+    return addMessage;
+  }();
+  _proto.saveMessages = /*#__PURE__*/function () {
+    var _saveMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(messages) {
+      var _this3 = this;
+      var messagesByThread, _iterator, _step, message, key;
+      return _regeneratorRuntime().wrap(function _callee7$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            if (messages.length) {
+              _context9.next = 2;
+              break;
+            }
+            return _context9.abrupt("return", []);
+          case 2:
+            messagesByThread = new Map();
+            for (_iterator = _createForOfIteratorHelperLoose(messages); !(_step = _iterator()).done;) {
+              message = _step.value;
+              key = "" + this.messagePrefix + message.threadId;
+              if (!messagesByThread.has(key)) {
+                messagesByThread.set(key, []);
+              }
+              messagesByThread.get(key).push(_extends({}, message, {
+                createdAt: new Date(message.createdAt)
+              }));
+            }
+            _context9.next = 6;
+            return Promise.all(Array.from(messagesByThread.entries()).map(function (_ref2) {
+              var key = _ref2[0],
+                threadMessages = _ref2[1];
+              return _this3.retryOperation(/*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6() {
+                var saved, _loop2;
+                return _regeneratorRuntime().wrap(function _callee6$(_context8) {
+                  while (1) switch (_context8.prev = _context8.next) {
+                    case 0:
+                      saved = false;
+                      _loop2 = /*#__PURE__*/_regeneratorRuntime().mark(function _loop2() {
+                        var _yield$_this3$kv$getW, existingMessages, version, messageMap, updatedMessages;
+                        return _regeneratorRuntime().wrap(function _loop2$(_context7) {
+                          while (1) switch (_context7.prev = _context7.next) {
+                            case 0:
+                              _context7.next = 2;
+                              return _this3.kv.getWithVersion(key);
+                            case 2:
+                              _yield$_this3$kv$getW = _context7.sent;
+                              existingMessages = _yield$_this3$kv$getW.data;
+                              version = _yield$_this3$kv$getW.version;
+                              console.log('Read version:', version, 'Messages:', (existingMessages == null ? void 0 : existingMessages.length) || 0);
+                              messageMap = new Map();
+                              (existingMessages || []).forEach(function (msg) {
+                                messageMap.set(msg.id, _extends({}, msg, {
+                                  createdAt: new Date(msg.createdAt)
+                                }));
+                              });
+                              threadMessages.forEach(function (msg) {
+                                messageMap.set(msg.id, msg);
+                              });
+                              updatedMessages = Array.from(messageMap.values());
+                              updatedMessages.sort(function (a, b) {
+                                var timeCompare = a.createdAt.getTime() - b.createdAt.getTime();
+                                return timeCompare === 0 ? a.id.localeCompare(b.id) : timeCompare;
+                              });
+                              _context7.next = 13;
+                              return _this3.kv.setWithVersion(key, updatedMessages, version);
+                            case 13:
+                              saved = _context7.sent;
+                              console.log('Save attempt with version:', version, 'Success:', saved);
+                              if (saved) {
+                                _context7.next = 18;
+                                break;
+                              }
+                              _context7.next = 18;
+                              return new Promise(function (resolve) {
+                                return setTimeout(resolve, Math.random() * 50);
+                              });
+                            case 18:
+                            case "end":
+                              return _context7.stop();
+                          }
+                        }, _loop2);
+                      });
+                    case 2:
+                      if (saved) {
+                        _context8.next = 6;
+                        break;
+                      }
+                      return _context8.delegateYield(_loop2(), "t0", 4);
+                    case 4:
+                      _context8.next = 2;
+                      break;
+                    case 6:
+                    case "end":
+                      return _context8.stop();
+                  }
+                }, _callee6);
+              })));
+            }));
+          case 6:
+            return _context9.abrupt("return", messages);
+          case 7:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee7, this);
+    }));
+    function saveMessages(_x8) {
+      return _saveMessages.apply(this, arguments);
+    }
+    return saveMessages;
+  }();
+  _proto.getMessages = /*#__PURE__*/function () {
+    var _getMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(threadId) {
+      var messages;
+      return _regeneratorRuntime().wrap(function _callee8$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            _context10.next = 2;
+            return this.kv.get("" + this.messagePrefix + threadId);
+          case 2:
+            _context10.t0 = _context10.sent;
+            if (_context10.t0) {
+              _context10.next = 5;
+              break;
+            }
+            _context10.t0 = [];
+          case 5:
+            messages = _context10.t0;
+            return _context10.abrupt("return", messages.map(function (msg) {
+              return _extends({}, msg, {
+                createdAt: new Date(msg.createdAt)
+              });
+            }));
+          case 7:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee8, this);
+    }));
+    function getMessages(_x9) {
+      return _getMessages.apply(this, arguments);
+    }
+    return getMessages;
+  }();
+  _proto.getAllThreadIds = /*#__PURE__*/function () {
+    var _getAllThreadIds = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9() {
+      return _regeneratorRuntime().wrap(function _callee9$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
+          case 0:
+            return _context11.abrupt("return", this.kv.smembers('threads'));
+          case 1:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee9, this);
+    }));
+    function getAllThreadIds() {
+      return _getAllThreadIds.apply(this, arguments);
+    }
+    return getAllThreadIds;
+  }();
+  _proto.deleteThread = /*#__PURE__*/function () {
+    var _deleteThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(threadId) {
+      return _regeneratorRuntime().wrap(function _callee10$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
+          case 0:
+            _context12.next = 2;
+            return Promise.all([this.kv.del("" + this.threadPrefix + threadId), this.kv.del("" + this.messagePrefix + threadId), this.kv.srem('threads', threadId)]);
+          case 2:
+          case "end":
+            return _context12.stop();
+        }
+      }, _callee10, this);
+    }));
+    function deleteThread(_x10) {
+      return _deleteThread.apply(this, arguments);
+    }
+    return deleteThread;
+  }();
+  _proto.getThreads = /*#__PURE__*/function () {
+    var _getThreads = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(threadIds) {
+      var _this4 = this;
+      var threads;
+      return _regeneratorRuntime().wrap(function _callee11$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
+          case 0:
+            _context13.next = 2;
+            return Promise.all(threadIds.map(function (id) {
+              return _this4.getThreadById(id);
+            }));
+          case 2:
+            threads = _context13.sent;
+            return _context13.abrupt("return", threads.filter(function (t) {
+              return t !== null;
+            }));
+          case 4:
+          case "end":
+            return _context13.stop();
+        }
+      }, _callee11);
+    }));
+    function getThreads(_x11) {
+      return _getThreads.apply(this, arguments);
+    }
+    return getThreads;
+  }();
+  _proto.cleanup = /*#__PURE__*/function () {
+    var _cleanup = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12() {
+      return _regeneratorRuntime().wrap(function _callee12$(_context14) {
+        while (1) switch (_context14.prev = _context14.next) {
+          case 0:
+            _context14.next = 2;
+            return this.kv.flushall();
+          case 2:
+          case "end":
+            return _context14.stop();
+        }
+      }, _callee12, this);
+    }));
+    function cleanup() {
+      return _cleanup.apply(this, arguments);
+    }
+    return cleanup;
+  }();
+  return CloudflareKVMemory;
+}(MastraMemory);
+
+var Pool = pg.Pool;
+var PgMemory = /*#__PURE__*/function (_MastraMemory) {
+  function PgMemory(connectionString) {
+    var _this;
+    _this = _MastraMemory.call(this) || this;
+    _this.pool = void 0;
+    _this.pool = new Pool({
+      connectionString: connectionString
+    });
+    return _this;
+  }
+  _inheritsLoose(PgMemory, _MastraMemory);
+  var _proto = PgMemory.prototype;
+  _proto.drop = /*#__PURE__*/function () {
+    var _drop = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
+      var client;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.next = 2;
+            return this.pool.connect();
+          case 2:
+            client = _context.sent;
+            _context.next = 5;
+            return client.query('DELETE FROM mastra_messages');
+          case 5:
+            _context.next = 7;
+            return client.query('DELETE FROM mastra_threads');
+          case 7:
+            client.release();
+            _context.next = 10;
+            return this.pool.end();
+          case 10:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function drop() {
+      return _drop.apply(this, arguments);
+    }
+    return drop;
+  }();
+  _proto.ensureTablesExist = /*#__PURE__*/function () {
+    var _ensureTablesExist = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2() {
+      var client, _threadsResult$rows, _messagesResult$rows, threadsResult, messagesResult;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            _context2.next = 2;
+            return this.pool.connect();
+          case 2:
+            client = _context2.sent;
+            _context2.prev = 3;
+            _context2.next = 6;
+            return client.query("\n                SELECT EXISTS (\n                    SELECT 1\n                    FROM information_schema.tables\n                    WHERE table_name = 'mastra_threads'\n                );\n            ");
+          case 6:
+            threadsResult = _context2.sent;
+            if (threadsResult != null && (_threadsResult$rows = threadsResult.rows) != null && (_threadsResult$rows = _threadsResult$rows[0]) != null && _threadsResult$rows.exists) {
+              _context2.next = 10;
+              break;
+            }
+            _context2.next = 10;
+            return client.query("\n                    CREATE TABLE IF NOT EXISTS mastra_threads (\n                        id UUID PRIMARY KEY,\n                        title TEXT,\n                        created_at TIMESTAMP WITH TIME ZONE NOT NULL,\n                        updated_at TIMESTAMP WITH TIME ZONE NOT NULL,\n                        metadata JSONB\n                    );\n                ");
+          case 10:
+            _context2.next = 12;
+            return client.query("\n                SELECT EXISTS (\n                    SELECT 1\n                    FROM information_schema.tables\n                    WHERE table_name = 'mastra_messages'\n                );\n            ");
+          case 12:
+            messagesResult = _context2.sent;
+            if (messagesResult != null && (_messagesResult$rows = messagesResult.rows) != null && (_messagesResult$rows = _messagesResult$rows[0]) != null && _messagesResult$rows.exists) {
+              _context2.next = 16;
+              break;
+            }
+            _context2.next = 16;
+            return client.query("\n                    CREATE TABLE IF NOT EXISTS mastra_messages (\n                        id UUID PRIMARY KEY,\n                        content TEXT NOT NULL,\n                        role VARCHAR(20) NOT NULL,\n                        created_at TIMESTAMP WITH TIME ZONE NOT NULL,\n                        thread_id UUID NOT NULL,\n                        FOREIGN KEY (thread_id) REFERENCES mastra_threads(id)\n                    );\n                ");
+          case 16:
+            _context2.prev = 16;
+            client.release();
+            return _context2.finish(16);
+          case 19:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, this, [[3,, 16, 19]]);
+    }));
+    function ensureTablesExist() {
+      return _ensureTablesExist.apply(this, arguments);
+    }
+    return ensureTablesExist;
+  }();
+  _proto.updateThread = /*#__PURE__*/function () {
+    var _updateThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(id, title, metadata) {
+      var client, _result$rows, result;
+      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
+          case 0:
+            _context3.next = 2;
+            return this.pool.connect();
+          case 2:
+            client = _context3.sent;
+            _context3.prev = 3;
+            _context3.next = 6;
+            return client.query("\n                UPDATE mastra_threads\n                SET title = $1, metadata = $2, updated_at = NOW()\n                WHERE id = $3\n                RETURNING *\n                ", [title, JSON.stringify(metadata), id]);
+          case 6:
+            result = _context3.sent;
+            return _context3.abrupt("return", result == null || (_result$rows = result.rows) == null ? void 0 : _result$rows[0]);
+          case 8:
+            _context3.prev = 8;
+            client.release();
+            return _context3.finish(8);
+          case 11:
+          case "end":
+            return _context3.stop();
+        }
+      }, _callee3, this, [[3,, 8, 11]]);
+    }));
+    function updateThread(_x, _x2, _x3) {
+      return _updateThread.apply(this, arguments);
+    }
+    return updateThread;
+  }();
+  _proto.deleteThread = /*#__PURE__*/function () {
+    var _deleteThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(id) {
+      var client;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            _context4.next = 2;
+            return this.pool.connect();
+          case 2:
+            client = _context4.sent;
+            _context4.prev = 3;
+            _context4.next = 6;
+            return client.query("\n                DELETE FROM mastra_messages\n                WHERE thread_id = $1\n                ", [id]);
+          case 6:
+            _context4.next = 8;
+            return client.query("\n                DELETE FROM mastra_threads\n                WHERE id = $1\n                ", [id]);
+          case 8:
+            _context4.prev = 8;
+            client.release();
+            return _context4.finish(8);
+          case 11:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4, this, [[3,, 8, 11]]);
+    }));
+    function deleteThread(_x4) {
+      return _deleteThread.apply(this, arguments);
+    }
+    return deleteThread;
+  }();
+  _proto.deleteMessage = /*#__PURE__*/function () {
+    var _deleteMessage = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(id) {
+      var client;
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
+          case 0:
+            _context5.next = 2;
+            return this.pool.connect();
+          case 2:
+            client = _context5.sent;
+            _context5.prev = 3;
+            _context5.next = 6;
+            return client.query("\n                DELETE FROM mastra_messages\n                WHERE id = $1\n                ", [id]);
+          case 6:
+            _context5.prev = 6;
+            client.release();
+            return _context5.finish(6);
+          case 9:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, this, [[3,, 6, 9]]);
+    }));
+    function deleteMessage(_x5) {
+      return _deleteMessage.apply(this, arguments);
+    }
+    return deleteMessage;
+  }();
+  _proto.getThreadById = /*#__PURE__*/function () {
+    var _getThreadById = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(threadId) {
+      var client, result;
+      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            _context6.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            _context6.next = 4;
+            return this.pool.connect();
+          case 4:
+            client = _context6.sent;
+            _context6.prev = 5;
+            _context6.next = 8;
+            return client.query("\n                SELECT id, title, created_at AS createdAt, updated_at AS updatedAt, metadata\n                FROM mastra_threads\n                WHERE id = $1\n            ", [threadId]);
+          case 8:
+            result = _context6.sent;
+            return _context6.abrupt("return", result.rows[0] || null);
+          case 10:
+            _context6.prev = 10;
+            client.release();
+            return _context6.finish(10);
+          case 13:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee6, this, [[5,, 10, 13]]);
+    }));
+    function getThreadById(_x6) {
+      return _getThreadById.apply(this, arguments);
+    }
+    return getThreadById;
+  }();
+  _proto.saveThread = /*#__PURE__*/function () {
+    var _saveThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(thread) {
+      var client, _result$rows2, id, title, createdAt, updatedAt, metadata, result;
+      return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+        while (1) switch (_context7.prev = _context7.next) {
+          case 0:
+            _context7.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            _context7.next = 4;
+            return this.pool.connect();
+          case 4:
+            client = _context7.sent;
+            _context7.prev = 5;
+            id = thread.id, title = thread.title, createdAt = thread.createdAt, updatedAt = thread.updatedAt, metadata = thread.metadata;
+            _context7.next = 9;
+            return client.query("\n                INSERT INTO mastra_threads (id, title, created_at, updated_at, metadata)\n                VALUES ($1, $2, $3, $4, $5)\n                ON CONFLICT (id) DO UPDATE SET title = $2, updated_at = $4, metadata = $5\n                RETURNING id, title, created_at AS createdAt, updated_at AS updatedAt, metadata\n            ", [id, title, createdAt, updatedAt, JSON.stringify(metadata)]);
+          case 9:
+            result = _context7.sent;
+            return _context7.abrupt("return", result == null || (_result$rows2 = result.rows) == null ? void 0 : _result$rows2[0]);
+          case 11:
+            _context7.prev = 11;
+            client.release();
+            return _context7.finish(11);
+          case 14:
+          case "end":
+            return _context7.stop();
+        }
+      }, _callee7, this, [[5,, 11, 14]]);
+    }));
+    function saveThread(_x7) {
+      return _saveThread.apply(this, arguments);
+    }
+    return saveThread;
+  }();
+  _proto.saveMessages = /*#__PURE__*/function () {
+    var _saveMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(messages) {
+      var client, _iterator, _step, message, id, content, role, createdAt, threadId;
+      return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+        while (1) switch (_context8.prev = _context8.next) {
+          case 0:
+            _context8.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            _context8.next = 4;
+            return this.pool.connect();
+          case 4:
+            client = _context8.sent;
+            _context8.prev = 5;
+            _context8.next = 8;
+            return client.query('BEGIN');
+          case 8:
+            _iterator = _createForOfIteratorHelperLoose(messages);
+          case 9:
+            if ((_step = _iterator()).done) {
+              _context8.next = 16;
+              break;
+            }
+            message = _step.value;
+            id = message.id, content = message.content, role = message.role, createdAt = message.createdAt, threadId = message.threadId;
+            _context8.next = 14;
+            return client.query("\n                    INSERT INTO mastra_messages (id, content, role, created_at, thread_id)\n                    VALUES ($1, $2, $3, $4, $5)\n                ", [id, content, role, createdAt.toISOString(), threadId]);
+          case 14:
+            _context8.next = 9;
+            break;
+          case 16:
+            _context8.next = 18;
+            return client.query('COMMIT');
+          case 18:
+            return _context8.abrupt("return", messages);
+          case 21:
+            _context8.prev = 21;
+            _context8.t0 = _context8["catch"](5);
+            _context8.next = 25;
+            return client.query('ROLLBACK');
+          case 25:
+            throw _context8.t0;
+          case 26:
+            _context8.prev = 26;
+            client.release();
+            return _context8.finish(26);
+          case 29:
+          case "end":
+            return _context8.stop();
+        }
+      }, _callee8, this, [[5, 21, 26, 29]]);
+    }));
+    function saveMessages(_x8) {
+      return _saveMessages.apply(this, arguments);
+    }
+    return saveMessages;
+  }();
+  _proto.getMessages = /*#__PURE__*/function () {
+    var _getMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(threadId) {
+      var client, result;
+      return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            _context9.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            _context9.next = 4;
+            return this.pool.connect();
+          case 4:
+            client = _context9.sent;
+            _context9.prev = 5;
+            _context9.next = 8;
+            return client.query("\n                SELECT \n                    id, \n                    content, \n                    role, \n                    created_at AS createdAt, \n                    thread_id AS threadId\n                FROM mastra_messages\n                WHERE thread_id = $1\n                ORDER BY created_at ASC\n            ", [threadId]);
+          case 8:
+            result = _context9.sent;
+            return _context9.abrupt("return", result.rows);
+          case 10:
+            _context9.prev = 10;
+            client.release();
+            return _context9.finish(10);
+          case 13:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee9, this, [[5,, 10, 13]]);
+    }));
+    function getMessages(_x9) {
+      return _getMessages.apply(this, arguments);
+    }
+    return getMessages;
+  }();
+  _proto.createThread = /*#__PURE__*/function () {
+    var _createThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(title, metadata) {
+      var id, now, thread;
+      return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            _context10.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            id = randomUUID();
+            now = new Date();
+            thread = {
+              id: id,
+              title: title,
+              createdAt: now,
+              updatedAt: now,
+              metadata: metadata
+            };
+            return _context10.abrupt("return", this.saveThread(thread));
+          case 6:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee10, this);
+    }));
+    function createThread(_x10, _x11) {
+      return _createThread.apply(this, arguments);
+    }
+    return createThread;
+  }();
+  _proto.addMessage = /*#__PURE__*/function () {
+    var _addMessage = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(threadId, content, role) {
+      var thread, id, message, _yield$this$saveMessa, savedMessage;
+      return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
+          case 0:
+            _context11.next = 2;
+            return this.ensureTablesExist();
+          case 2:
+            _context11.next = 4;
+            return this.getThreadById(threadId);
+          case 4:
+            thread = _context11.sent;
+            if (thread) {
+              _context11.next = 7;
+              break;
+            }
+            throw new Error("Thread with ID " + threadId + " does not exist.");
+          case 7:
+            id = randomUUID();
+            message = {
+              id: id,
+              content: content,
+              role: role,
+              createdAt: new Date(),
+              threadId: threadId
+            };
+            _context11.next = 11;
+            return this.saveMessages([message]);
+          case 11:
+            _yield$this$saveMessa = _context11.sent;
+            savedMessage = _yield$this$saveMessa[0];
+            return _context11.abrupt("return", savedMessage);
+          case 14:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee11, this);
+    }));
+    function addMessage(_x12, _x13, _x14) {
+      return _addMessage.apply(this, arguments);
+    }
+    return addMessage;
+  }();
+  return PgMemory;
+}(MastraMemory);
+
+// Helper functions for date handling
+function serializeData(data) {
+  if (data === null || data === undefined) return data;
+  if (data instanceof Date) {
+    return {
+      __type: 'Date',
+      value: data.toISOString()
+    };
+  }
+  if (Array.isArray(data)) {
+    return data.map(function (item) {
+      return serializeData(item);
+    });
+  }
+  if (typeof data === 'object') {
+    return Object.keys(data).reduce(function (acc, key) {
+      acc[key] = serializeData(data[key]);
+      return acc;
+    }, {});
+  }
+  return data;
+}
+function deserializeData(data) {
+  if (data === null || data === undefined) return data;
+  if (typeof data === 'object' && data.__type === 'Date') {
+    return new Date(data.value);
+  }
+  if (Array.isArray(data)) {
+    return data.map(function (item) {
+      return deserializeData(item);
+    });
+  }
+  if (typeof data === 'object') {
+    return Object.keys(data).reduce(function (acc, key) {
+      acc[key] = deserializeData(data[key]);
+      return acc;
+    }, {});
+  }
+  return data;
+}
+// Provider for Upstash/Vercel KV
+var UpstashProvider = /*#__PURE__*/function () {
+  function UpstashProvider(url, token) {
+    this.client = void 0;
+    this.client = new Redis({
+      url: url,
+      token: token
+    });
+  }
+  var _proto = UpstashProvider.prototype;
+  _proto.get = /*#__PURE__*/function () {
+    var _get = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(key) {
+      var data;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.next = 2;
+            return this.client.get(key);
+          case 2:
+            data = _context.sent;
+            return _context.abrupt("return", deserializeData(data));
+          case 4:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function get(_x) {
+      return _get.apply(this, arguments);
+    }
+    return get;
+  }();
+  _proto.set = /*#__PURE__*/function () {
+    var _set = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(key, value) {
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            return _context2.abrupt("return", this.client.set(key, serializeData(value)));
+          case 1:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, this);
+    }));
+    function set(_x2, _x3) {
+      return _set.apply(this, arguments);
+    }
+    return set;
+  }();
+  _proto.del = /*#__PURE__*/function () {
+    var _del = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3(key) {
+      return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
+          case 0:
+            return _context3.abrupt("return", this.client.del(key));
+          case 1:
+          case "end":
+            return _context3.stop();
+        }
+      }, _callee3, this);
+    }));
+    function del(_x4) {
+      return _del.apply(this, arguments);
+    }
+    return del;
+  }();
+  _proto.sadd = /*#__PURE__*/function () {
+    var _sadd = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(key, value) {
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            return _context4.abrupt("return", this.client.sadd(key, value));
+          case 1:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4, this);
+    }));
+    function sadd(_x5, _x6) {
+      return _sadd.apply(this, arguments);
+    }
+    return sadd;
+  }();
+  _proto.srem = /*#__PURE__*/function () {
+    var _srem = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5(key, value) {
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
+          case 0:
+            return _context5.abrupt("return", this.client.srem(key, value));
+          case 1:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, this);
+    }));
+    function srem(_x7, _x8) {
+      return _srem.apply(this, arguments);
+    }
+    return srem;
+  }();
+  _proto.smembers = /*#__PURE__*/function () {
+    var _smembers = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6(key) {
+      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            return _context6.abrupt("return", this.client.smembers(key));
+          case 1:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee6, this);
+    }));
+    function smembers(_x9) {
+      return _smembers.apply(this, arguments);
+    }
+    return smembers;
+  }();
+  _proto.flushall = /*#__PURE__*/function () {
+    var _flushall = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
+      return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+        while (1) switch (_context7.prev = _context7.next) {
+          case 0:
+            return _context7.abrupt("return", this.client.flushall());
+          case 1:
+          case "end":
+            return _context7.stop();
+        }
+      }, _callee7, this);
+    }));
+    function flushall() {
+      return _flushall.apply(this, arguments);
+    }
+    return flushall;
+  }();
+  _proto.pipeline = function pipeline() {
+    var multi = this.client.multi();
+    var pipeline = {
+      get: function get(key) {
+        multi.get(key);
+        return pipeline;
+      },
+      set: function set(key, value) {
+        multi.set(key, JSON.stringify(serializeData(value)));
+        return pipeline;
+      },
+      del: function del(key) {
+        multi.del(key);
+        return pipeline;
+      },
+      srem: function srem(key, value) {
+        multi.srem(key, value);
+        return pipeline;
+      },
+      exec: function () {
+        var _exec = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8() {
+          var results;
+          return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+            while (1) switch (_context8.prev = _context8.next) {
+              case 0:
+                _context8.next = 2;
+                return multi.exec();
+              case 2:
+                results = _context8.sent;
+                return _context8.abrupt("return", results.map(function (result) {
+                  try {
+                    // For get operations that return string data
+                    if (typeof result === 'string') {
+                      return deserializeData(JSON.parse(result));
+                    }
+                    // For array results (like from lists/sets)
+                    if (Array.isArray(result)) {
+                      return result.map(function (item) {
+                        try {
+                          return typeof item === 'string' ? deserializeData(JSON.parse(item)) : item;
+                        } catch (_unused) {
+                          return item;
+                        }
+                      });
+                    }
+                    return result;
+                  } catch (_unused2) {
+                    return result;
+                  }
+                }));
+              case 4:
+              case "end":
+                return _context8.stop();
+            }
+          }, _callee8);
+        }));
+        function exec() {
+          return _exec.apply(this, arguments);
+        }
+        return exec;
+      }()
+    };
+    return pipeline;
+  };
+  return UpstashProvider;
+}();
+var LocalRedisProvider = /*#__PURE__*/function () {
+  function LocalRedisProvider() {
+    this.client = void 0;
+    this.client = createClient({
+      url: 'redis://localhost:6379'
+    });
+    this.client.connect();
+  }
+  var _proto2 = LocalRedisProvider.prototype;
+  _proto2.get = /*#__PURE__*/function () {
+    var _get2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(key) {
+      var data;
+      return _regeneratorRuntime().wrap(function _callee9$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            _context9.next = 2;
+            return this.client.get(key);
+          case 2:
+            data = _context9.sent;
+            return _context9.abrupt("return", data ? deserializeData(JSON.parse(data)) : null);
+          case 4:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee9, this);
+    }));
+    function get(_x10) {
+      return _get2.apply(this, arguments);
+    }
+    return get;
+  }();
+  _proto2.set = /*#__PURE__*/function () {
+    var _set2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10(key, value) {
+      return _regeneratorRuntime().wrap(function _callee10$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            return _context10.abrupt("return", this.client.set(key, JSON.stringify(serializeData(value))));
+          case 1:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee10, this);
+    }));
+    function set(_x11, _x12) {
+      return _set2.apply(this, arguments);
+    }
+    return set;
+  }();
+  _proto2.del = /*#__PURE__*/function () {
+    var _del2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(key) {
+      return _regeneratorRuntime().wrap(function _callee11$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
+          case 0:
+            return _context11.abrupt("return", this.client.del(key));
+          case 1:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee11, this);
+    }));
+    function del(_x13) {
+      return _del2.apply(this, arguments);
+    }
+    return del;
+  }();
+  _proto2.sadd = /*#__PURE__*/function () {
+    var _sadd2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(key, value) {
+      return _regeneratorRuntime().wrap(function _callee12$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
+          case 0:
+            return _context12.abrupt("return", this.client.sAdd(key, value));
+          case 1:
+          case "end":
+            return _context12.stop();
+        }
+      }, _callee12, this);
+    }));
+    function sadd(_x14, _x15) {
+      return _sadd2.apply(this, arguments);
+    }
+    return sadd;
+  }();
+  _proto2.srem = /*#__PURE__*/function () {
+    var _srem2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee13(key, value) {
+      return _regeneratorRuntime().wrap(function _callee13$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
+          case 0:
+            return _context13.abrupt("return", this.client.sRem(key, value));
+          case 1:
+          case "end":
+            return _context13.stop();
+        }
+      }, _callee13, this);
+    }));
+    function srem(_x16, _x17) {
+      return _srem2.apply(this, arguments);
+    }
+    return srem;
+  }();
+  _proto2.smembers = /*#__PURE__*/function () {
+    var _smembers2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee14(key) {
+      return _regeneratorRuntime().wrap(function _callee14$(_context14) {
+        while (1) switch (_context14.prev = _context14.next) {
+          case 0:
+            return _context14.abrupt("return", this.client.sMembers(key));
+          case 1:
+          case "end":
+            return _context14.stop();
+        }
+      }, _callee14, this);
+    }));
+    function smembers(_x18) {
+      return _smembers2.apply(this, arguments);
+    }
+    return smembers;
+  }();
+  _proto2.flushall = /*#__PURE__*/function () {
+    var _flushall2 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee15() {
+      return _regeneratorRuntime().wrap(function _callee15$(_context15) {
+        while (1) switch (_context15.prev = _context15.next) {
+          case 0:
+            return _context15.abrupt("return", this.client.flushAll());
+          case 1:
+          case "end":
+            return _context15.stop();
+        }
+      }, _callee15, this);
+    }));
+    function flushall() {
+      return _flushall2.apply(this, arguments);
+    }
+    return flushall;
+  }();
+  _proto2.pipeline = function pipeline() {
+    var multi = this.client.multi();
+    var pipeline = {
+      get: function get(key) {
+        multi.get(key);
+        return pipeline;
+      },
+      set: function set(key, value) {
+        multi.set(key, JSON.stringify(value));
+        return pipeline;
+      },
+      del: function del(key) {
+        multi.del(key);
+        return pipeline;
+      },
+      srem: function srem(key, value) {
+        multi.sRem(key, value);
+        return pipeline;
+      },
+      exec: function () {
+        var _exec2 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee16() {
+          return _regeneratorRuntime().wrap(function _callee16$(_context16) {
+            while (1) switch (_context16.prev = _context16.next) {
+              case 0:
+                return _context16.abrupt("return", multi.exec());
+              case 1:
+              case "end":
+                return _context16.stop();
+            }
+          }, _callee16);
+        }));
+        function exec() {
+          return _exec2.apply(this, arguments);
+        }
+        return exec;
+      }()
+    };
+    return pipeline;
+  };
+  _proto2.quit = /*#__PURE__*/function () {
+    var _quit = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee17() {
+      return _regeneratorRuntime().wrap(function _callee17$(_context17) {
+        while (1) switch (_context17.prev = _context17.next) {
+          case 0:
+            _context17.next = 2;
+            return this.client.quit();
+          case 2:
+          case "end":
+            return _context17.stop();
+        }
+      }, _callee17, this);
+    }));
+    function quit() {
+      return _quit.apply(this, arguments);
+    }
+    return quit;
+  }();
+  return LocalRedisProvider;
+}();
+
+var RedisMemory = /*#__PURE__*/function (_MastraMemory) {
+  function RedisMemory(redis) {
+    var _this;
+    _this = _MastraMemory.call(this) || this;
+    _this.redis = void 0;
+    _this.threadPrefix = 'thread:';
+    _this.messagePrefix = 'messages:';
+    _this.lockTimeouts = new Map();
+    _this.redis = redis;
+    return _this;
+  }
+  _inheritsLoose(RedisMemory, _MastraMemory);
+  var _proto = RedisMemory.prototype;
+  _proto.getThreadById = /*#__PURE__*/function () {
+    var _getThreadById = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(threadId) {
+      var thread;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.next = 2;
+            return this.redis.get("" + this.threadPrefix + threadId);
+          case 2:
+            thread = _context.sent;
+            if (thread && typeof thread.createdAt === 'string') {
+              thread.createdAt = new Date(thread.createdAt);
+              thread.updatedAt = new Date(thread.updatedAt);
+            }
+            return _context.abrupt("return", thread);
+          case 5:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function getThreadById(_x) {
+      return _getThreadById.apply(this, arguments);
+    }
+    return getThreadById;
+  }();
+  _proto.saveThread = /*#__PURE__*/function () {
+    var _saveThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee2(thread) {
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            thread.updatedAt = new Date();
+            _context2.next = 3;
+            return this.redis.set("" + this.threadPrefix + thread.id, thread);
+          case 3:
+            _context2.next = 5;
+            return this.redis.sadd('threads', thread.id);
+          case 5:
+            return _context2.abrupt("return", thread);
+          case 6:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, this);
+    }));
+    function saveThread(_x2) {
+      return _saveThread.apply(this, arguments);
+    }
+    return saveThread;
+  }();
+  _proto.withLock = /*#__PURE__*/function () {
+    var _withLock = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee4(key, operation) {
+      var _this2 = this;
+      var lockKey, lockTimeout, locked, timeoutId, i;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            lockKey = "lock:" + key;
+            lockTimeout = 5000;
+            locked = false;
+            _context4.prev = 3;
+            i = 0;
+          case 5:
+            if (!(i < 3)) {
+              _context4.next = 16;
+              break;
+            }
+            _context4.next = 8;
+            return this.redis.sadd(lockKey, '1');
+          case 8:
+            locked = _context4.sent;
+            if (!locked) {
+              _context4.next = 11;
+              break;
+            }
+            return _context4.abrupt("break", 16);
+          case 11:
+            _context4.next = 13;
+            return new Promise(function (resolve) {
+              return setTimeout(resolve, Math.random() * 100);
+            });
+          case 13:
+            i++;
+            _context4.next = 5;
+            break;
+          case 16:
+            if (locked) {
+              _context4.next = 18;
+              break;
+            }
+            throw new Error('Could not acquire lock');
+          case 18:
+            // Set lock timeout
+            timeoutId = setTimeout(/*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee3() {
+              return _regeneratorRuntime().wrap(function _callee3$(_context3) {
+                while (1) switch (_context3.prev = _context3.next) {
+                  case 0:
+                    _context3.prev = 0;
+                    _context3.next = 3;
+                    return _this2.redis.del(lockKey);
+                  case 3:
+                    _context3.next = 7;
+                    break;
+                  case 5:
+                    _context3.prev = 5;
+                    _context3.t0 = _context3["catch"](0);
+                  case 7:
+                    _this2.lockTimeouts["delete"](lockKey);
+                  case 8:
+                  case "end":
+                    return _context3.stop();
+                }
+              }, _callee3, null, [[0, 5]]);
+            })), lockTimeout);
+            this.lockTimeouts.set(lockKey, timeoutId);
+            // Execute operation
+            _context4.next = 22;
+            return operation();
+          case 22:
+            return _context4.abrupt("return", _context4.sent);
+          case 23:
+            _context4.prev = 23;
+            if (timeoutId !== undefined) {
+              clearTimeout(timeoutId);
+              this.lockTimeouts["delete"](lockKey);
+            }
+            if (!locked) {
+              _context4.next = 33;
+              break;
+            }
+            _context4.prev = 26;
+            _context4.next = 29;
+            return this.redis.del(lockKey);
+          case 29:
+            _context4.next = 33;
+            break;
+          case 31:
+            _context4.prev = 31;
+            _context4.t0 = _context4["catch"](26);
+          case 33:
+            return _context4.finish(23);
+          case 34:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4, this, [[3,, 23, 34], [26, 31]]);
+    }));
+    function withLock(_x3, _x4) {
+      return _withLock.apply(this, arguments);
+    }
+    return withLock;
+  }() // Add cleanup method
+  ;
+  _proto.cleanup =
+  /*#__PURE__*/
+  function () {
+    var _cleanup = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
+      var _iterator, _step, timeout;
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
+          case 0:
+            // Clear all timeouts
+            for (_iterator = _createForOfIteratorHelperLoose(this.lockTimeouts.values()); !(_step = _iterator()).done;) {
+              timeout = _step.value;
+              clearTimeout(timeout);
+            }
+            this.lockTimeouts.clear();
+          case 2:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, this);
+    }));
+    function cleanup() {
+      return _cleanup.apply(this, arguments);
+    }
+    return cleanup;
+  }();
+  _proto.saveMessages = /*#__PURE__*/function () {
+    var _saveMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee7(messages) {
+      var _this3 = this;
+      var messagesByThread, _loop, _i, _Object$entries;
+      return _regeneratorRuntime().wrap(function _callee7$(_context8) {
+        while (1) switch (_context8.prev = _context8.next) {
+          case 0:
+            if (messages.length) {
+              _context8.next = 2;
+              break;
+            }
+            return _context8.abrupt("return", []);
+          case 2:
+            messagesByThread = messages.reduce(function (acc, message) {
+              var key = "" + _this3.messagePrefix + message.threadId;
+              if (!acc[key]) {
+                acc[key] = [];
+              }
+              acc[key].push(_extends({}, message, {
+                createdAt: new Date(message.createdAt)
+              }));
+              return acc;
+            }, {});
+            _loop = /*#__PURE__*/_regeneratorRuntime().mark(function _loop() {
+              var _Object$entries$_i, key, threadMessages;
+              return _regeneratorRuntime().wrap(function _loop$(_context7) {
+                while (1) switch (_context7.prev = _context7.next) {
+                  case 0:
+                    _Object$entries$_i = _Object$entries[_i], key = _Object$entries$_i[0], threadMessages = _Object$entries$_i[1];
+                    _context7.next = 3;
+                    return _this3.withLock(key, /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee6() {
+                      var _threadMessages$;
+                      var existingMessages, messageMap, updatedMessages, _threadMessages$2, thread;
+                      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+                        while (1) switch (_context6.prev = _context6.next) {
+                          case 0:
+                            _context6.next = 2;
+                            return _this3.redis.get(key);
+                          case 2:
+                            _context6.t0 = _context6.sent;
+                            if (_context6.t0) {
+                              _context6.next = 5;
+                              break;
+                            }
+                            _context6.t0 = [];
+                          case 5:
+                            existingMessages = _context6.t0;
+                            messageMap = new Map(); // Process existing messages
+                            existingMessages.forEach(function (msg) {
+                              messageMap.set(msg.id, _extends({}, msg, {
+                                createdAt: new Date(msg.createdAt)
+                              }));
+                            });
+                            // Add new messages
+                            threadMessages.forEach(function (msg) {
+                              messageMap.set(msg.id, msg);
+                            });
+                            updatedMessages = Array.from(messageMap.values());
+                            updatedMessages.sort(function (a, b) {
+                              var timeCompare = a.createdAt.getTime() - b.createdAt.getTime();
+                              return timeCompare === 0 ? a.id.localeCompare(b.id) : timeCompare;
+                            });
+                            _context6.next = 13;
+                            return _this3.redis.set(key, updatedMessages);
+                          case 13:
+                            if (!(threadMessages != null && (_threadMessages$ = threadMessages[0]) != null && _threadMessages$.threadId)) {
+                              _context6.next = 21;
+                              break;
+                            }
+                            _context6.next = 16;
+                            return _this3.getThreadById(threadMessages == null || (_threadMessages$2 = threadMessages[0]) == null ? void 0 : _threadMessages$2.threadId);
+                          case 16:
+                            thread = _context6.sent;
+                            if (!thread) {
+                              _context6.next = 21;
+                              break;
+                            }
+                            thread.updatedAt = new Date();
+                            _context6.next = 21;
+                            return _this3.redis.set("" + _this3.threadPrefix + thread.id, thread);
+                          case 21:
+                          case "end":
+                            return _context6.stop();
+                        }
+                      }, _callee6);
+                    })));
+                  case 3:
+                  case "end":
+                    return _context7.stop();
+                }
+              }, _loop);
+            });
+            _i = 0, _Object$entries = Object.entries(messagesByThread);
+          case 5:
+            if (!(_i < _Object$entries.length)) {
+              _context8.next = 10;
+              break;
+            }
+            return _context8.delegateYield(_loop(), "t0", 7);
+          case 7:
+            _i++;
+            _context8.next = 5;
+            break;
+          case 10:
+            return _context8.abrupt("return", messages);
+          case 11:
+          case "end":
+            return _context8.stop();
+        }
+      }, _callee7);
+    }));
+    function saveMessages(_x5) {
+      return _saveMessages.apply(this, arguments);
+    }
+    return saveMessages;
+  }();
+  _proto.addMessage = /*#__PURE__*/function () {
+    var _addMessage = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee8(threadId, content, role) {
+      var message;
+      return _regeneratorRuntime().wrap(function _callee8$(_context9) {
+        while (1) switch (_context9.prev = _context9.next) {
+          case 0:
+            message = {
+              id: this.generateId(),
+              content: content,
+              role: role,
+              createdAt: new Date(),
+              threadId: threadId
+            };
+            _context9.next = 3;
+            return this.saveMessages([message]);
+          case 3:
+            return _context9.abrupt("return", message);
+          case 4:
+          case "end":
+            return _context9.stop();
+        }
+      }, _callee8, this);
+    }));
+    function addMessage(_x6, _x7, _x8) {
+      return _addMessage.apply(this, arguments);
+    }
+    return addMessage;
+  }();
+  _proto.getMessages = /*#__PURE__*/function () {
+    var _getMessages = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee9(threadId) {
+      var messages;
+      return _regeneratorRuntime().wrap(function _callee9$(_context10) {
+        while (1) switch (_context10.prev = _context10.next) {
+          case 0:
+            _context10.next = 2;
+            return this.redis.get("" + this.messagePrefix + threadId);
+          case 2:
+            _context10.t0 = _context10.sent;
+            if (_context10.t0) {
+              _context10.next = 5;
+              break;
+            }
+            _context10.t0 = [];
+          case 5:
+            messages = _context10.t0;
+            return _context10.abrupt("return", messages.map(function (msg) {
+              return _extends({}, msg, {
+                createdAt: new Date(msg.createdAt)
+              });
+            }));
+          case 7:
+          case "end":
+            return _context10.stop();
+        }
+      }, _callee9, this);
+    }));
+    function getMessages(_x9) {
+      return _getMessages.apply(this, arguments);
+    }
+    return getMessages;
+  }();
+  _proto.getAllThreadIds = /*#__PURE__*/function () {
+    var _getAllThreadIds = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee10() {
+      return _regeneratorRuntime().wrap(function _callee10$(_context11) {
+        while (1) switch (_context11.prev = _context11.next) {
+          case 0:
+            return _context11.abrupt("return", this.redis.smembers('threads'));
+          case 1:
+          case "end":
+            return _context11.stop();
+        }
+      }, _callee10, this);
+    }));
+    function getAllThreadIds() {
+      return _getAllThreadIds.apply(this, arguments);
+    }
+    return getAllThreadIds;
+  }();
+  _proto.deleteThread = /*#__PURE__*/function () {
+    var _deleteThread = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee11(threadId) {
+      var pipeline;
+      return _regeneratorRuntime().wrap(function _callee11$(_context12) {
+        while (1) switch (_context12.prev = _context12.next) {
+          case 0:
+            pipeline = this.redis.pipeline();
+            pipeline.del("" + this.threadPrefix + threadId);
+            pipeline.del("" + this.messagePrefix + threadId);
+            pipeline.srem('threads', threadId);
+            _context12.next = 6;
+            return pipeline.exec();
+          case 6:
+          case "end":
+            return _context12.stop();
+        }
+      }, _callee11, this);
+    }));
+    function deleteThread(_x10) {
+      return _deleteThread.apply(this, arguments);
+    }
+    return deleteThread;
+  }();
+  _proto.getThreads = /*#__PURE__*/function () {
+    var _getThreads = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee12(threadIds) {
+      var _this4 = this;
+      var threads;
+      return _regeneratorRuntime().wrap(function _callee12$(_context13) {
+        while (1) switch (_context13.prev = _context13.next) {
+          case 0:
+            _context13.next = 2;
+            return Promise.all(threadIds.map(function (id) {
+              return _this4.getThreadById(id);
+            }));
+          case 2:
+            threads = _context13.sent;
+            return _context13.abrupt("return", threads.filter(function (t) {
+              return t !== null;
+            }));
+          case 4:
+          case "end":
+            return _context13.stop();
+        }
+      }, _callee12);
+    }));
+    function getThreads(_x11) {
+      return _getThreads.apply(this, arguments);
+    }
+    return getThreads;
+  }();
+  _proto.generateId = function generateId() {
+    return crypto.randomUUID();
+  };
+  return RedisMemory;
+}(MastraMemory);
+
+export { CloudflareKVMemory, LocalRedisProvider, PgMemory, RedisMemory, UpstashProvider };
+//# sourceMappingURL=memory.esm.js.map