@mastra/upstash 1.0.0-beta.3 → 1.0.0-beta.5

package/dist/index.js

@@ -1,4 +1,4 @@
-import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, normalizePerPage, calculatePagination, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MemoryStorage, TABLE_THREADS, TABLE_RESOURCES, TABLE_MESSAGES, serializeDate, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { MastraStorage, createVectorErrorId, StoreOperations, createStorageErrorId, ScoresStorage, TABLE_SCORERS, normalizePerPage, calculatePagination, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MemoryStorage, TABLE_THREADS, TABLE_RESOURCES, TABLE_MESSAGES, serializeDate, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import { Redis } from '@upstash/redis';
 import { MessageList } from '@mastra/core/agent';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
@@ -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 });
   }
@@ -79,7 +79,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_GET_THREAD_BY_ID_FAILED",
+          id: createStorageErrorId("UPSTASH", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -97,7 +97,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
+          id: createStorageErrorId("UPSTASH", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -139,7 +139,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: createStorageErrorId("UPSTASH", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -171,7 +171,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_SAVE_THREAD_FAILED",
+          id: createStorageErrorId("UPSTASH", "SAVE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -193,7 +193,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     const thread = await this.getThreadById({ threadId: id });
     if (!thread) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_STORAGE_UPDATE_THREAD_FAILED",
+        id: createStorageErrorId("UPSTASH", "UPDATE_THREAD", "FAILED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: `Thread ${id} not found`,
@@ -216,7 +216,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_UPDATE_THREAD_FAILED",
+          id: createStorageErrorId("UPSTASH", "UPDATE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -245,7 +245,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_DELETE_THREAD_FAILED",
+          id: createStorageErrorId("UPSTASH", "DELETE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -271,7 +271,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_SAVE_MESSAGES_INVALID_ARGS",
+          id: createStorageErrorId("UPSTASH", "SAVE_MESSAGES", "INVALID_ARGS"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -341,7 +341,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_SAVE_MESSAGES_FAILED",
+          id: createStorageErrorId("UPSTASH", "SAVE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -467,7 +467,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LIST_MESSAGES_BY_ID_FAILED",
+          id: createStorageErrorId("UPSTASH", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -484,7 +484,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("UPSTASH", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
@@ -498,7 +498,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
       if (page < 0) {
         throw new MastraError(
           {
-            id: "STORAGE_UPSTASH_LIST_MESSAGES_INVALID_PAGE",
+            id: createStorageErrorId("UPSTASH", "LIST_MESSAGES", "INVALID_PAGE"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { page }
@@ -601,7 +601,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LIST_MESSAGES_FAILED",
+          id: createStorageErrorId("UPSTASH", "LIST_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -792,7 +792,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_UPDATE_MESSAGES_FAILED",
+          id: createStorageErrorId("UPSTASH", "UPDATE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -871,7 +871,7 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_DELETE_MESSAGES_FAILED",
+          id: createStorageErrorId("UPSTASH", "DELETE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { messageIds: messageIds.join(", ") }
@@ -916,7 +916,7 @@ var StoreOperationsUpstash = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_CLEAR_TABLE_FAILED",
+          id: createStorageErrorId("UPSTASH", "CLEAR_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -937,7 +937,7 @@ var StoreOperationsUpstash = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_INSERT_FAILED",
+          id: createStorageErrorId("UPSTASH", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -965,7 +965,7 @@ var StoreOperationsUpstash = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_BATCH_INSERT_FAILED",
+          id: createStorageErrorId("UPSTASH", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -984,7 +984,7 @@ var StoreOperationsUpstash = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LOAD_FAILED",
+          id: createStorageErrorId("UPSTASH", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1050,7 +1050,7 @@ var ScoresUpstash = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_GET_SCORE_BY_ID_FAILED",
+          id: createStorageErrorId("UPSTASH", "GET_SCORE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1121,24 +1121,41 @@ var ScoresUpstash = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_SAVE_SCORE_VALIDATION_FAILED",
+          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: "STORAGE_UPSTASH_STORAGE_SAVE_SCORE_FAILED",
+          id: createStorageErrorId("UPSTASH", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { id: score.id }
+          details: { id }
         },
         error
       );
@@ -1346,7 +1363,7 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("UPSTASH", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1373,7 +1390,7 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("UPSTASH", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1405,7 +1422,7 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: createStorageErrorId("UPSTASH", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1431,7 +1448,7 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
       if (page !== void 0 && page < 0) {
         throw new MastraError(
           {
-            id: "UPSTASH_STORE_INVALID_PAGE",
+            id: createStorageErrorId("UPSTASH", "LIST_WORKFLOW_RUNS", "INVALID_PAGE"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { page }
@@ -1491,9 +1508,10 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
       }
       return { runs, total };
     } catch (error) {
+      if (error instanceof MastraError) throw error;
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_STORAGE_LIST_WORKFLOW_RUNS_FAILED",
+          id: createStorageErrorId("UPSTASH", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1513,7 +1531,7 @@ var UpstashStore = class extends MastraStorage {
   redis;
   stores;
   constructor(config) {
-    super({ id: config.id, name: "Upstash" });
+    super({ id: config.id, name: "Upstash", disableInit: config.disableInit });
     this.redis = new Redis({
       url: config.url,
       token: config.token
@@ -1919,7 +1937,7 @@ var UpstashVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_UPSERT_FAILED",
+          id: createVectorErrorId("UPSTASH", "UPSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { namespace, vectorCount: vectors.length }
@@ -1982,7 +2000,7 @@ var UpstashVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_QUERY_FAILED",
+          id: createVectorErrorId("UPSTASH", "QUERY", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { namespace, topK }
@@ -2002,7 +2020,7 @@ var UpstashVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_LIST_INDEXES_FAILED",
+          id: createVectorErrorId("UPSTASH", "LIST_INDEXES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -2027,7 +2045,7 @@ var UpstashVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_DESCRIBE_INDEX_FAILED",
+          id: createVectorErrorId("UPSTASH", "DESCRIBE_INDEX", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { namespace }
@@ -2052,7 +2070,7 @@ var UpstashVector = class extends MastraVector {
       }
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_DELETE_INDEX_FAILED",
+          id: createVectorErrorId("UPSTASH", "DELETE_INDEX", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { namespace }
@@ -2077,7 +2095,7 @@ var UpstashVector = class extends MastraVector {
     const sparseVector = upstashUpdate.sparseVector;
     if ("id" in params && params.id && "filter" in params && params.filter) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_UPDATE_MUTUALLY_EXCLUSIVE",
+        id: createVectorErrorId("UPSTASH", "UPDATE_VECTOR", "MUTUALLY_EXCLUSIVE"),
         text: "Cannot specify both id and filter - they are mutually exclusive",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2086,7 +2104,7 @@ var UpstashVector = class extends MastraVector {
     }
     if (!("id" in params && params.id) && !("filter" in params && params.filter)) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_UPDATE_NO_TARGET",
+        id: createVectorErrorId("UPSTASH", "UPDATE_VECTOR", "NO_TARGET"),
         text: "Either id or filter must be provided",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2095,7 +2113,7 @@ var UpstashVector = class extends MastraVector {
     }
     if (!update.vector && !update.metadata && !sparseVector) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_UPDATE_NO_PAYLOAD",
+        id: createVectorErrorId("UPSTASH", "UPDATE_VECTOR", "NO_PAYLOAD"),
         text: "No update data provided",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2104,7 +2122,7 @@ var UpstashVector = class extends MastraVector {
     }
     if ("filter" in params && params.filter && Object.keys(params.filter).length === 0) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_UPDATE_EMPTY_FILTER",
+        id: createVectorErrorId("UPSTASH", "UPDATE_VECTOR", "EMPTY_FILTER"),
         text: "Filter cannot be an empty filter object",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2172,7 +2190,7 @@ var UpstashVector = class extends MastraVector {
       if (error instanceof MastraError) throw error;
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_UPDATE_VECTOR_FAILED",
+          id: createVectorErrorId("UPSTASH", "UPDATE_VECTOR", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2199,7 +2217,7 @@ var UpstashVector = class extends MastraVector {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTOR_FAILED",
+          id: createVectorErrorId("UPSTASH", "DELETE_VECTOR", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2223,7 +2241,7 @@ var UpstashVector = class extends MastraVector {
   async deleteVectors({ indexName: namespace, filter, ids }) {
     if (ids && filter) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTORS_MUTUALLY_EXCLUSIVE",
+        id: createVectorErrorId("UPSTASH", "DELETE_VECTORS", "MUTUALLY_EXCLUSIVE"),
         text: "Cannot specify both ids and filter - they are mutually exclusive",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2232,7 +2250,7 @@ var UpstashVector = class extends MastraVector {
     }
     if (!ids && !filter) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTORS_NO_TARGET",
+        id: createVectorErrorId("UPSTASH", "DELETE_VECTORS", "NO_TARGET"),
         text: "Either filter or ids must be provided",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2241,7 +2259,7 @@ var UpstashVector = class extends MastraVector {
     }
     if (ids && ids.length === 0) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTORS_EMPTY_IDS",
+        id: createVectorErrorId("UPSTASH", "DELETE_VECTORS", "EMPTY_IDS"),
         text: "Cannot delete with empty ids array",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2250,7 +2268,7 @@ var UpstashVector = class extends MastraVector {
     }
     if (filter && Object.keys(filter).length === 0) {
       throw new MastraError({
-        id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTORS_EMPTY_FILTER",
+        id: createVectorErrorId("UPSTASH", "DELETE_VECTORS", "EMPTY_FILTER"),
         text: "Cannot delete with empty filter object",
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
@@ -2284,7 +2302,7 @@ var UpstashVector = class extends MastraVector {
       if (error instanceof MastraError) throw error;
       throw new MastraError(
         {
-          id: "STORAGE_UPSTASH_VECTOR_DELETE_VECTORS_FAILED",
+          id: createVectorErrorId("UPSTASH", "DELETE_VECTORS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {