@mastra/upstash 1.0.0-beta.4 → 1.0.0-beta.6

package/dist/index.js

@@ -32,7 +32,7 @@ function processRecord(tableName, record) {
       ...record.resourceId ? { resourceId: record.resourceId } : {}
     });
   } else if (tableName === TABLE_SCORERS) {
-    key = getKey(tableName, { runId: record.runId });
+    key = getKey(tableName, { id: record.id });
   } else {
     key = getKey(tableName, { id: record.id });
   }
@@ -1123,22 +1123,39 @@ var ScoresUpstash = class extends ScoresStorage {
         {
           id: createStorageErrorId("UPSTASH", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.USER,
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
-    const { key, processedRecord } = processRecord(TABLE_SCORERS, validatedScore);
+    const now = /* @__PURE__ */ new Date();
+    const id = crypto.randomUUID();
+    const createdAt = now;
+    const updatedAt = now;
+    const scoreWithId = {
+      ...validatedScore,
+      id,
+      createdAt,
+      updatedAt
+    };
+    const { key, processedRecord } = processRecord(TABLE_SCORERS, scoreWithId);
     try {
       await this.client.set(key, processedRecord);
-      return { score };
+      return { score: { ...validatedScore, id, createdAt, updatedAt } };
     } catch (error) {
       throw new MastraError(
         {
           id: createStorageErrorId("UPSTASH", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { id: score.id }
+          details: { id }
         },
         error
       );
@@ -1418,6 +1435,26 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
       );
     }
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    const key = getKey(TABLE_WORKFLOW_SNAPSHOT, { namespace: "workflows", workflow_name: workflowName, run_id: runId });
+    try {
+      await this.client.del(key);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("UPSTASH", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            namespace: "workflows",
+            runId,
+            workflowName
+          }
+        },
+        error
+      );
+    }
+  }
   async listWorkflowRuns({
     workflowName,
     fromDate,
@@ -1614,6 +1651,9 @@ var UpstashStore = class extends MastraStorage {
   async loadWorkflowSnapshot(params) {
     return this.stores.workflows.loadWorkflowSnapshot(params);
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async listWorkflowRuns(args = {}) {
     return this.stores.workflows.listWorkflowRuns(args);
   }