@mastra/upstash 1.2.0 → 1.2.1-alpha.1

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @mastra/upstash
 
+## 1.2.1-alpha.1
+
+### Patch Changes
+
+- Fixed workflow runs preserving their original creation time when re-persisted in Upstash storage, including concurrent saves. ([#18004](https://github.com/mastra-ai/mastra/pull/18004))
+
+- Updated dependencies [[`bf3fe49`](https://github.com/mastra-ai/mastra/commit/bf3fe49f9467dbbdb8f9eaf74e0f7971ffb19559), [`24ceaea`](https://github.com/mastra-ai/mastra/commit/24ceaea0bdd8609cabbab764380608ca6621a194), [`6ccf67b`](https://github.com/mastra-ai/mastra/commit/6ccf67bf075753754927a57bc2e1734ba2c820c5), [`825d8de`](https://github.com/mastra-ai/mastra/commit/825d8def9fa64c2bcc3d8dd6b49e09342c3ac5c7), [`ffa09e7`](https://github.com/mastra-ai/mastra/commit/ffa09e772a5c92270eabe2090fc42d45bd8ec4b7), [`461a7c5`](https://github.com/mastra-ai/mastra/commit/461a7c501449295287f4f0ee4b0b42344f39fcf8), [`4211472`](https://github.com/mastra-ai/mastra/commit/4211472a5a2bd319c60cd2e42d9109c3eef7ac1c), [`9e45902`](https://github.com/mastra-ai/mastra/commit/9e4590208e745055cecca202e2db0e5c65e17d3c), [`5c0df77`](https://github.com/mastra-ai/mastra/commit/5c0df776c40efa420f8c07a2f3ee66010296618e)]:
+  - @mastra/core@1.47.0-alpha.3
+
+## 1.2.1-alpha.0
+
+### Patch Changes
+
+- Updated dependencies [[`7f9ae70`](https://github.com/mastra-ai/mastra/commit/7f9ae70826b047e5a66218f9e92f20e54a2d791f), [`bf026c5`](https://github.com/mastra-ai/mastra/commit/bf026c5b641019c97facfe5924d14dd363866cb0), [`1505c07`](https://github.com/mastra-ai/mastra/commit/1505c07603f6346bae12aa82f140e8b88ffea9ab), [`e940f09`](https://github.com/mastra-ai/mastra/commit/e940f099ef5d18b403e6f2b4937e086a4da857b1)]:
+  - @mastra/core@1.46.1-alpha.1
+  - @mastra/redis@1.2.1-alpha.0
+
 ## 1.2.0
 
 ### Minor Changes

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-upstash
 description: Documentation for @mastra/upstash. Use when working with @mastra/upstash APIs, configuration, or implementation.
 metadata:
   package: "@mastra/upstash"
-  version: "1.2.0"
+  version: "1.2.1-alpha.1"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.2.0",
+  "version": "1.2.1-alpha.1",
   "package": "@mastra/upstash",
   "exports": {
     "RedisServerCache": {

package/dist/index.cjs

@@ -1765,6 +1765,36 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
   async dangerouslyClearAll() {
     await this.#db.deleteData({ tableName: storage.TABLE_WORKFLOW_SNAPSHOT });
   }
+  async getWorkflowRunRecord({
+    namespace,
+    runId,
+    resourceId,
+    workflowName
+  }) {
+    if (workflowName) {
+      const keyVariants = [
+        ...resourceId ? [{ namespace, workflow_name: workflowName, run_id: runId, resourceId }] : [],
+        { namespace, workflow_name: workflowName, run_id: runId }
+      ];
+      for (const keys2 of keyVariants) {
+        const data = await this.#db.get({
+          tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+          keys: keys2
+        });
+        if (data) return data;
+      }
+    }
+    const key = `${getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace })}:*run_id:${runId}*`;
+    const keys = await this.#db.scanKeys(key);
+    const workflows = await Promise.all(
+      keys.map(async (key2) => {
+        return this.client.get(key2);
+      })
+    );
+    return workflows.find(
+      (w) => w?.run_id === runId && (!workflowName || w.workflow_name === workflowName) && (!resourceId || w.resourceId === resourceId)
+    ) ?? null;
+  }
   async updateWorkflowResults({
     workflowName,
     runId,
@@ -1968,18 +1998,38 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
   async persistWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId, resourceId, snapshot, createdAt, updatedAt } = params;
     try {
-      await this.#db.insert({
-        tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
-        record: {
-          namespace,
-          workflow_name: workflowName,
-          run_id: runId,
-          resourceId,
-          snapshot,
-          createdAt: createdAt ?? /* @__PURE__ */ new Date(),
-          updatedAt: updatedAt ?? /* @__PURE__ */ new Date()
-        }
+      const now = /* @__PURE__ */ new Date();
+      const key = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
+        namespace,
+        workflow_name: workflowName,
+        run_id: runId,
+        ...resourceId ? { resourceId } : {}
       });
+      const record = {
+        namespace,
+        workflow_name: workflowName,
+        run_id: runId,
+        resourceId,
+        snapshot,
+        createdAt: (createdAt ?? now).toISOString(),
+        updatedAt: (updatedAt ?? now).toISOString()
+      };
+      await this.client.eval(
+        `
+        local existing = redis.call("GET", KEYS[1])
+        local next = cjson.decode(ARGV[1])
+        if existing then
+          local current = cjson.decode(existing)
+          if current["createdAt"] then
+            next["createdAt"] = current["createdAt"]
+          end
+        end
+        redis.call("SET", KEYS[1], cjson.encode(next))
+        return 1
+        `,
+        [key],
+        [JSON.stringify(record)]
+      );
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -1998,15 +2048,10 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
   }
   async loadWorkflowSnapshot(params) {
     const { namespace = "workflows", workflowName, runId } = params;
-    const key = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
-      namespace,
-      workflow_name: workflowName,
-      run_id: runId
-    });
     try {
-      const data = await this.client.get(key);
+      const data = await this.getWorkflowRunRecord({ namespace, runId, workflowName });
       if (!data) return null;
-      return data.snapshot;
+      return typeof data.snapshot === "string" ? JSON.parse(data.snapshot) : data.snapshot;
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -2028,15 +2073,7 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
     workflowName
   }) {
     try {
-      const key = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows", workflow_name: workflowName, run_id: runId }) + "*";
-      const keys = await this.#db.scanKeys(key);
-      const workflows = await Promise.all(
-        keys.map(async (key2) => {
-          const data2 = await this.client.get(key2);
-          return data2;
-        })
-      );
-      const data = workflows.find((w) => w?.run_id === runId && w?.workflow_name === workflowName);
+      const data = await this.getWorkflowRunRecord({ namespace: "workflows", runId, workflowName });
       if (!data) return null;
       return this.parseWorkflowRun(data);
     } catch (error$1) {
@@ -2056,8 +2093,14 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
     }
   }
   async deleteWorkflowRunById({ runId, workflowName }) {
-    const key = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows", workflow_name: workflowName, run_id: runId });
     try {
+      const record = await this.getWorkflowRunRecord({ namespace: "workflows", runId, workflowName });
+      const key = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
+        namespace: "workflows",
+        workflow_name: workflowName,
+        run_id: runId,
+        ...record?.resourceId ? { resourceId: record.resourceId } : {}
+      });
       await this.client.del(key);
     } catch (error$1) {
       throw new error.MastraError(
@@ -2096,24 +2139,7 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
           new Error("page must be >= 0")
         );
       }
-      let pattern = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows" }) + ":*";
-      if (workflowName && resourceId) {
-        pattern = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
-          namespace: "workflows",
-          workflow_name: workflowName,
-          run_id: "*",
-          resourceId
-        });
-      } else if (workflowName) {
-        pattern = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows", workflow_name: workflowName }) + ":*";
-      } else if (resourceId) {
-        pattern = getKey(storage.TABLE_WORKFLOW_SNAPSHOT, {
-          namespace: "workflows",
-          workflow_name: "*",
-          run_id: "*",
-          resourceId
-        });
-      }
+      const pattern = `${getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows" })}:*`;
       const keys = await this.#db.scanKeys(pattern);
       if (keys.length === 0) {
         return { runs: [], total: 0 };
@@ -2123,7 +2149,7 @@ var WorkflowsUpstash = class extends storage.WorkflowsStorage {
       const results = await pipeline.exec();
       let runs = results.map((result) => result).filter(
         (record) => record !== null && record !== void 0 && typeof record === "object" && "workflow_name" in record
-      ).filter((record) => !workflowName || record.workflow_name === workflowName).map((w) => this.parseWorkflowRun(w)).filter((w) => {
+      ).filter((record) => !workflowName || record.workflow_name === workflowName).filter((record) => !resourceId || record.resourceId === resourceId).map((w) => this.parseWorkflowRun(w)).filter((w) => {
         if (fromDate && w.createdAt < fromDate) return false;
         if (toDate && w.createdAt > toDate) return false;
         if (status) {