@mastra/mongodb 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/index.js

@@ -1,9 +1,9 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
+import { createVectorErrorId, MastraStorage, createStorageErrorId, StoreOperations, TABLE_SCHEMAS, safelyParseJSON, MemoryStorage, TABLE_MESSAGES, normalizePerPage, calculatePagination, TABLE_THREADS, TABLE_RESOURCES, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ObservabilityStorage, TABLE_SPANS, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
 import { MongoClient } from 'mongodb';
 import { v4 } from 'uuid';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
-import { MastraStorage, StoreOperations, TABLE_SCHEMAS, safelyParseJSON, MemoryStorage, TABLE_MESSAGES, normalizePerPage, calculatePagination, TABLE_THREADS, TABLE_RESOURCES, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ObservabilityStorage, TABLE_SPANS } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 
@@ -11,7 +11,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.0.0-beta.2"};
+  version: "1.0.0-beta.4"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -134,7 +134,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_CONNECT_FAILED",
+          id: createVectorErrorId("MONGODB", "CONNECT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -148,7 +148,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_DISCONNECT_FAILED",
+          id: createVectorErrorId("MONGODB", "DISCONNECT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -169,7 +169,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_CREATE_INDEX_INVALID_ARGS",
+          id: createVectorErrorId("MONGODB", "CREATE_INDEX", "INVALID_ARGS"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -222,7 +222,7 @@ var MongoDBVector = class extends MastraVector {
       if (error.codeName !== "IndexAlreadyExists") {
         throw new MastraError(
           {
-            id: "STORAGE_MONGODB_VECTOR_CREATE_INDEX_FAILED",
+            id: createVectorErrorId("MONGODB", "CREATE_INDEX", "FAILED"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.THIRD_PARTY
           },
@@ -235,7 +235,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_CREATE_INDEX_FAILED_STORE_METADATA",
+          id: createVectorErrorId("MONGODB", "CREATE_INDEX", "STORE_METADATA_FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -312,7 +312,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_UPSERT_FAILED",
+          id: createVectorErrorId("MONGODB", "UPSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -393,7 +393,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_QUERY_FAILED",
+          id: createVectorErrorId("MONGODB", "QUERY", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -411,7 +411,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_LIST_INDEXES_FAILED",
+          id: createVectorErrorId("MONGODB", "LIST_INDEXES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -440,7 +440,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_DESCRIBE_INDEX_FAILED",
+          id: createVectorErrorId("MONGODB", "DESCRIBE_INDEX", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -463,7 +463,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_DELETE_INDEX_FAILED",
+          id: createVectorErrorId("MONGODB", "DELETE_INDEX", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -488,7 +488,7 @@ var MongoDBVector = class extends MastraVector {
     const { indexName, update } = params;
     if ("id" in params && params.id && "filter" in params && params.filter) {
       throw new MastraError({
-        id: "STORAGE_MONGODB_VECTOR_UPDATE_MUTUALLY_EXCLUSIVE_PARAMS",
+        id: createVectorErrorId("MONGODB", "UPDATE_VECTOR", "MUTUALLY_EXCLUSIVE"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         details: { indexName },
@@ -497,7 +497,7 @@ var MongoDBVector = class extends MastraVector {
     }
     if (!("id" in params || "filter" in params) || !params.id && !params.filter) {
       throw new MastraError({
-        id: "STORAGE_MONGODB_VECTOR_UPDATE_MISSING_PARAMS",
+        id: createVectorErrorId("MONGODB", "UPDATE_VECTOR", "NO_TARGET"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: "Either id or filter must be provided",
@@ -531,7 +531,7 @@ var MongoDBVector = class extends MastraVector {
         const filter = params.filter;
         if (!filter || Object.keys(filter).length === 0) {
           throw new MastraError({
-            id: "STORAGE_MONGODB_VECTOR_UPDATE_EMPTY_FILTER",
+            id: createVectorErrorId("MONGODB", "UPDATE_VECTOR", "EMPTY_FILTER"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { indexName },
@@ -542,7 +542,7 @@ var MongoDBVector = class extends MastraVector {
         const transformedFilter = this.transformMetadataFilter(mongoFilter);
         if (!transformedFilter || Object.keys(transformedFilter).length === 0) {
           throw new MastraError({
-            id: "STORAGE_MONGODB_VECTOR_UPDATE_INVALID_FILTER",
+            id: createVectorErrorId("MONGODB", "UPDATE_VECTOR", "INVALID_FILTER"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { indexName },
@@ -564,7 +564,7 @@ var MongoDBVector = class extends MastraVector {
       }
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_UPDATE_VECTOR_FAILED",
+          id: createVectorErrorId("MONGODB", "UPDATE_VECTOR", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: errorDetails
@@ -587,7 +587,7 @@ var MongoDBVector = class extends MastraVector {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_DELETE_VECTOR_FAILED",
+          id: createVectorErrorId("MONGODB", "DELETE_VECTOR", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -602,7 +602,7 @@ var MongoDBVector = class extends MastraVector {
   async deleteVectors({ indexName, filter, ids }) {
     if (!filter && !ids) {
       throw new MastraError({
-        id: "STORAGE_MONGODB_VECTOR_DELETE_MISSING_PARAMS",
+        id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "NO_TARGET"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         details: { indexName },
@@ -611,7 +611,7 @@ var MongoDBVector = class extends MastraVector {
     }
     if (filter && ids) {
       throw new MastraError({
-        id: "STORAGE_MONGODB_VECTOR_DELETE_CONFLICTING_PARAMS",
+        id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "MUTUALLY_EXCLUSIVE"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         details: { indexName },
@@ -623,7 +623,7 @@ var MongoDBVector = class extends MastraVector {
       if (ids) {
         if (ids.length === 0) {
           throw new MastraError({
-            id: "STORAGE_MONGODB_VECTOR_DELETE_EMPTY_IDS",
+            id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "EMPTY_IDS"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { indexName },
@@ -634,7 +634,7 @@ var MongoDBVector = class extends MastraVector {
       } else {
         if (!filter || Object.keys(filter).length === 0) {
           throw new MastraError({
-            id: "STORAGE_MONGODB_VECTOR_DELETE_EMPTY_FILTER",
+            id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "EMPTY_FILTER"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { indexName },
@@ -645,7 +645,7 @@ var MongoDBVector = class extends MastraVector {
         const transformedFilter = this.transformMetadataFilter(mongoFilter);
         if (!transformedFilter || Object.keys(transformedFilter).length === 0) {
           throw new MastraError({
-            id: "STORAGE_MONGODB_VECTOR_DELETE_INVALID_FILTER",
+            id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "INVALID_FILTER"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { indexName },
@@ -663,7 +663,7 @@ var MongoDBVector = class extends MastraVector {
       }
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_VECTOR_DELETE_VECTORS_FAILED",
+          id: createVectorErrorId("MONGODB", "DELETE_VECTORS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -836,8 +836,6 @@ var MongoDBConnector = class _MongoDBConnector {
     }
   }
 };
-
-// src/storage/domains/utils.ts
 function formatDateForMongoDB(date) {
   return typeof date === "string" ? new Date(date) : date;
 }
@@ -868,18 +866,16 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     if (row.type && row.type !== "v2") result.type = row.type;
     return result;
   }
-  async _getIncludedMessages({
-    threadId,
-    include
-  }) {
-    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    if (!include) return null;
+  async _getIncludedMessages({ include }) {
+    if (!include || include.length === 0) return null;
     const collection = await this.operations.getCollection(TABLE_MESSAGES);
     const includedMessages = [];
     for (const inc of include) {
       const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
-      const searchThreadId = inc.threadId || threadId;
-      const allMessages = await collection.find({ thread_id: searchThreadId }).sort({ createdAt: 1 }).toArray();
+      const targetMessage = await collection.findOne({ id });
+      if (!targetMessage) continue;
+      const messageThreadId = targetMessage.thread_id;
+      const allMessages = await collection.find({ thread_id: messageThreadId }).sort({ createdAt: 1 }).toArray();
       const targetIndex = allMessages.findIndex((msg) => msg.id === id);
       if (targetIndex === -1) continue;
       const startIndex = Math.max(0, targetIndex - withPreviousMessages);
@@ -909,7 +905,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_LIST_MESSAGES_BY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { messageIds: JSON.stringify(messageIds) }
@@ -920,21 +916,22 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    if (!threadId.trim()) {
+    const threadIds = Array.isArray(threadId) ? threadId : [threadId];
+    if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("MONGODB", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_LIST_MESSAGES_INVALID_PAGE",
+          id: createStorageErrorId("MONGODB", "LIST_MESSAGES", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -948,7 +945,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       const sortOrder = direction === "ASC" ? 1 : -1;
       const collection = await this.operations.getCollection(TABLE_MESSAGES);
-      const query = { thread_id: threadId };
+      const query = { thread_id: threadIds.length === 1 ? threadIds[0] : { $in: threadIds } };
       if (resourceId) {
         query.resourceId = resourceId;
       }
@@ -980,7 +977,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       }
       const messageIds = new Set(messages.map((m) => m.id));
       if (include && include.length > 0) {
-        const includeMessages = await this._getIncludedMessages({ threadId, include });
+        const includeMessages = await this._getIncludedMessages({ include });
         if (includeMessages) {
           for (const includeMsg of includeMessages) {
             if (!messageIds.has(includeMsg.id)) {
@@ -1001,7 +998,10 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
         }
         return direction === "ASC" ? String(aValue).localeCompare(String(bValue)) : String(bValue).localeCompare(String(aValue));
       });
-      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
+      const threadIdSet = new Set(threadIds);
+      const returnedThreadMessageIds = new Set(
+        finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)
+      );
       const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
       const hasMore = perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
       return {
@@ -1014,11 +1014,11 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MONGODB_STORE_LIST_MESSAGES_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -1083,7 +1083,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_SAVE_MESSAGES_FAILED",
+          id: createStorageErrorId("MONGODB", "SAVE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1181,7 +1181,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_RESOURCE_BY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1207,7 +1207,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_SAVE_RESOURCE_FAILED",
+          id: createStorageErrorId("MONGODB", "SAVE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId: resource.id }
@@ -1252,7 +1252,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_UPDATE_RESOURCE_FAILED",
+          id: createStorageErrorId("MONGODB", "UPDATE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1275,7 +1275,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_THREAD_BY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -1290,7 +1290,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       if (page < 0) {
         throw new MastraError(
           {
-            id: "STORAGE_MONGODB_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
+            id: createStorageErrorId("MONGODB", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: { page }
@@ -1336,7 +1336,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId: args.resourceId }
@@ -1362,7 +1362,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_SAVE_THREAD_FAILED",
+          id: createStorageErrorId("MONGODB", "SAVE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: thread.id }
@@ -1379,7 +1379,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     const thread = await this.getThreadById({ threadId: id });
     if (!thread) {
       throw new MastraError({
-        id: "STORAGE_MONGODB_STORE_UPDATE_THREAD_NOT_FOUND",
+        id: createStorageErrorId("MONGODB", "UPDATE_THREAD", "NOT_FOUND"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.THIRD_PARTY,
         details: { threadId: id, status: 404 },
@@ -1408,7 +1408,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_UPDATE_THREAD_FAILED",
+          id: createStorageErrorId("MONGODB", "UPDATE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: id }
@@ -1427,7 +1427,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_DELETE_THREAD_FAILED",
+          id: createStorageErrorId("MONGODB", "DELETE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -1464,7 +1464,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_CREATE_SPAN_FAILED",
+          id: createStorageErrorId("MONGODB", "CREATE_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1492,7 +1492,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_TRACE_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_TRACE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1528,7 +1528,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_UPDATE_SPAN_FAILED",
+          id: createStorageErrorId("MONGODB", "UPDATE_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1574,7 +1574,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
           name = `agent run: '${entityId}'`;
         } else {
           const error = new MastraError({
-            id: "MONGODB_STORE_GET_TRACES_PAGINATED_FAILED",
+            id: createStorageErrorId("MONGODB", "GET_TRACES_PAGINATED", "INVALID_ENTITY_TYPE"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: {
@@ -1611,7 +1611,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_TRACES_PAGINATED", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1639,7 +1639,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_CREATE_SPANS_FAILED",
+          id: createStorageErrorId("MONGODB", "BATCH_CREATE_SPANS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1672,7 +1672,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_UPDATE_SPANS_FAILED",
+          id: createStorageErrorId("MONGODB", "BATCH_UPDATE_SPANS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1689,7 +1689,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_DELETE_TRACES_FAILED",
+          id: createStorageErrorId("MONGODB", "BATCH_DELETE_TRACES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1740,7 +1740,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_CLEAR_TABLE_FAILED",
+          id: createStorageErrorId("MONGODB", "CLEAR_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1762,7 +1762,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
       }
       throw new MastraError(
         {
-          id: "MONGODB_STORE_DROP_TABLE_FAILED",
+          id: createStorageErrorId("MONGODB", "DROP_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1793,7 +1793,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_INSERT_FAILED",
+          id: createStorageErrorId("MONGODB", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1816,7 +1816,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_BATCH_INSERT_FAILED",
+          id: createStorageErrorId("MONGODB", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1833,7 +1833,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_LOAD_FAILED",
+          id: createStorageErrorId("MONGODB", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1855,7 +1855,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_UPDATE_FAILED",
+          id: createStorageErrorId("MONGODB", "UPDATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1887,7 +1887,7 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_BATCH_UPDATE_FAILED",
+          id: createStorageErrorId("MONGODB", "BATCH_UPDATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1898,98 +1898,9 @@ var StoreOperationsMongoDB = class extends StoreOperations {
   }
 };
 function transformScoreRow(row) {
-  let scorerValue = null;
-  if (row.scorer) {
-    try {
-      scorerValue = typeof row.scorer === "string" ? safelyParseJSON(row.scorer) : row.scorer;
-    } catch (e) {
-      console.warn("Failed to parse scorer:", e);
-    }
-  }
-  let preprocessStepResultValue = null;
-  if (row.preprocessStepResult) {
-    try {
-      preprocessStepResultValue = typeof row.preprocessStepResult === "string" ? safelyParseJSON(row.preprocessStepResult) : row.preprocessStepResult;
-    } catch (e) {
-      console.warn("Failed to parse preprocessStepResult:", e);
-    }
-  }
-  let analyzeStepResultValue = null;
-  if (row.analyzeStepResult) {
-    try {
-      analyzeStepResultValue = typeof row.analyzeStepResult === "string" ? safelyParseJSON(row.analyzeStepResult) : row.analyzeStepResult;
-    } catch (e) {
-      console.warn("Failed to parse analyzeStepResult:", e);
-    }
-  }
-  let inputValue = null;
-  if (row.input) {
-    try {
-      inputValue = typeof row.input === "string" ? safelyParseJSON(row.input) : row.input;
-    } catch (e) {
-      console.warn("Failed to parse input:", e);
-    }
-  }
-  let outputValue = null;
-  if (row.output) {
-    try {
-      outputValue = typeof row.output === "string" ? safelyParseJSON(row.output) : row.output;
-    } catch (e) {
-      console.warn("Failed to parse output:", e);
-    }
-  }
-  let entityValue = null;
-  if (row.entity) {
-    try {
-      entityValue = typeof row.entity === "string" ? safelyParseJSON(row.entity) : row.entity;
-    } catch (e) {
-      console.warn("Failed to parse entity:", e);
-    }
-  }
-  let requestContextValue = null;
-  if (row.requestContext) {
-    try {
-      requestContextValue = typeof row.requestContext === "string" ? safelyParseJSON(row.requestContext) : row.requestContext;
-    } catch (e) {
-      console.warn("Failed to parse requestContext:", e);
-    }
-  }
-  let metadataValue = null;
-  if (row.metadata) {
-    try {
-      metadataValue = typeof row.metadata === "string" ? safelyParseJSON(row.metadata) : row.metadata;
-    } catch (e) {
-      console.warn("Failed to parse metadata:", e);
-    }
-  }
-  return {
-    id: row.id,
-    entityId: row.entityId,
-    entityType: row.entityType,
-    scorerId: row.scorerId,
-    traceId: row.traceId,
-    spanId: row.spanId,
-    runId: row.runId,
-    scorer: scorerValue,
-    preprocessStepResult: preprocessStepResultValue,
-    preprocessPrompt: row.preprocessPrompt,
-    analyzeStepResult: analyzeStepResultValue,
-    generateScorePrompt: row.generateScorePrompt,
-    score: row.score,
-    analyzePrompt: row.analyzePrompt,
-    reasonPrompt: row.reasonPrompt,
-    metadata: metadataValue,
-    input: inputValue,
-    output: outputValue,
-    additionalContext: row.additionalContext,
-    requestContext: requestContextValue,
-    entity: entityValue,
-    source: row.source,
-    resourceId: row.resourceId,
-    threadId: row.threadId,
-    createdAt: new Date(row.createdAt),
-    updatedAt: new Date(row.updatedAt)
-  };
+  return transformScoreRow$1(row, {
+    convertTimestamps: true
+  });
 }
 var ScoresStorageMongoDB = class extends ScoresStorage {
   operations;
@@ -2008,7 +1919,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_SCORE_BY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_SCORE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { id }
@@ -2024,7 +1935,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_SAVE_SCORE_VALIDATION_FAILED",
+          id: createStorageErrorId("MONGODB", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -2034,36 +1945,29 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     try {
       const now = /* @__PURE__ */ new Date();
       const scoreId = `score-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
-      const scoreData = {
-        id: scoreId,
-        entityId: validatedScore.entityId,
-        entityType: validatedScore.entityType,
-        scorerId: validatedScore.scorerId,
-        traceId: validatedScore.traceId || "",
-        spanId: validatedScore.spanId || "",
-        runId: validatedScore.runId,
-        scorer: typeof validatedScore.scorer === "string" ? safelyParseJSON(validatedScore.scorer) : validatedScore.scorer,
-        preprocessStepResult: typeof validatedScore.preprocessStepResult === "string" ? safelyParseJSON(validatedScore.preprocessStepResult) : validatedScore.preprocessStepResult,
-        analyzeStepResult: typeof validatedScore.analyzeStepResult === "string" ? safelyParseJSON(validatedScore.analyzeStepResult) : validatedScore.analyzeStepResult,
-        score: validatedScore.score,
-        reason: validatedScore.reason,
-        preprocessPrompt: validatedScore.preprocessPrompt,
-        generateScorePrompt: validatedScore.generateScorePrompt,
-        generateReasonPrompt: validatedScore.generateReasonPrompt,
-        analyzePrompt: validatedScore.analyzePrompt,
-        input: typeof validatedScore.input === "string" ? safelyParseJSON(validatedScore.input) : validatedScore.input,
-        output: typeof validatedScore.output === "string" ? safelyParseJSON(validatedScore.output) : validatedScore.output,
-        additionalContext: validatedScore.additionalContext,
-        requestContext: typeof validatedScore.requestContext === "string" ? safelyParseJSON(validatedScore.requestContext) : validatedScore.requestContext,
-        entity: typeof validatedScore.entity === "string" ? safelyParseJSON(validatedScore.entity) : validatedScore.entity,
-        source: validatedScore.source,
-        resourceId: validatedScore.resourceId || "",
-        threadId: validatedScore.threadId || "",
-        createdAt: now,
-        updatedAt: now
+      const scorer = typeof validatedScore.scorer === "string" ? safelyParseJSON(validatedScore.scorer) : validatedScore.scorer;
+      const preprocessStepResult = typeof validatedScore.preprocessStepResult === "string" ? safelyParseJSON(validatedScore.preprocessStepResult) : validatedScore.preprocessStepResult;
+      const analyzeStepResult = typeof validatedScore.analyzeStepResult === "string" ? safelyParseJSON(validatedScore.analyzeStepResult) : validatedScore.analyzeStepResult;
+      const input = typeof validatedScore.input === "string" ? safelyParseJSON(validatedScore.input) : validatedScore.input;
+      const output = typeof validatedScore.output === "string" ? safelyParseJSON(validatedScore.output) : validatedScore.output;
+      const requestContext = typeof validatedScore.requestContext === "string" ? safelyParseJSON(validatedScore.requestContext) : validatedScore.requestContext;
+      const entity = typeof validatedScore.entity === "string" ? safelyParseJSON(validatedScore.entity) : validatedScore.entity;
+      const createdAt = now;
+      const updatedAt = now;
+      const dataToSave = {
+        ...validatedScore,
+        scorer,
+        preprocessStepResult,
+        analyzeStepResult,
+        input,
+        output,
+        requestContext,
+        entity,
+        createdAt,
+        updatedAt
       };
       const collection = await this.operations.getCollection(TABLE_SCORERS);
-      await collection.insertOne(scoreData);
+      await collection.insertOne(dataToSave);
       const savedScore = {
         ...score,
         id: scoreId,
@@ -2074,7 +1978,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_SAVE_SCORE_FAILED",
+          id: createStorageErrorId("MONGODB", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scorerId: score.scorerId, runId: score.runId }
@@ -2136,7 +2040,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_SCORES_BY_SCORER_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scorerId, page: pagination.page, perPage: pagination.perPage }
@@ -2185,7 +2089,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_SCORES_BY_RUN_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_SCORES_BY_RUN_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { runId, page: pagination.page, perPage: pagination.perPage }
@@ -2235,7 +2139,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { entityId, entityType, page: pagination.page, perPage: pagination.perPage }
@@ -2286,7 +2190,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_SCORES_BY_SPAN_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_SCORES_BY_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { traceId, spanId, page: pagination.page, perPage: pagination.perPage }
@@ -2343,7 +2247,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("MONGODB", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName, runId }
@@ -2371,7 +2275,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("MONGODB", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName, runId }
@@ -2410,7 +2314,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
         if (options.page < 0) {
           throw new MastraError(
             {
-              id: "STORAGE_MONGODB_INVALID_PAGE",
+              id: createStorageErrorId("MONGODB", "LIST_WORKFLOW_RUNS", "INVALID_PAGE"),
               domain: ErrorDomain.STORAGE,
               category: ErrorCategory.USER,
               details: { page: options.page }
@@ -2436,7 +2340,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_LIST_WORKFLOW_RUNS_FAILED",
+          id: createStorageErrorId("MONGODB", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName: options.workflowName || "unknown" }
@@ -2463,7 +2367,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: createStorageErrorId("MONGODB", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { runId: args.runId }
@@ -2501,7 +2405,7 @@ var loadConnector = (config) => {
   } catch (error) {
     throw new MastraError(
       {
-        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        id: createStorageErrorId("MONGODB", "CONSTRUCTOR", "FAILED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         details: { connectionHandler: true }
@@ -2519,7 +2423,7 @@ var loadConnector = (config) => {
   } catch (error) {
     throw new MastraError(
       {
-        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        id: createStorageErrorId("MONGODB", "CONSTRUCTOR", "FAILED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         details: { url: config?.url, dbName: config?.dbName }
@@ -2542,7 +2446,7 @@ var MongoDBStore = class extends MastraStorage {
     };
   }
   constructor(config) {
-    super({ id: config.id, name: "MongoDBStore" });
+    super({ id: config.id, name: "MongoDBStore", disableInit: config.disableInit });
     this.stores = {};
     this.#connector = loadConnector(config);
     const operations = new StoreOperationsMongoDB({
@@ -2662,7 +2566,7 @@ var MongoDBStore = class extends MastraStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_CLOSE_FAILED",
+          id: createStorageErrorId("MONGODB", "CLOSE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -2734,7 +2638,7 @@ var MongoDBStore = class extends MastraStorage {
   async createSpan(span) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2749,7 +2653,7 @@ var MongoDBStore = class extends MastraStorage {
   }) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2760,7 +2664,7 @@ var MongoDBStore = class extends MastraStorage {
   async getTrace(traceId) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2771,7 +2675,7 @@ var MongoDBStore = class extends MastraStorage {
   async getTracesPaginated(args) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2782,7 +2686,7 @@ var MongoDBStore = class extends MastraStorage {
   async batchCreateSpans(args) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2793,7 +2697,7 @@ var MongoDBStore = class extends MastraStorage {
   async batchUpdateSpans(args) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -2804,7 +2708,7 @@ var MongoDBStore = class extends MastraStorage {
   async batchDeleteTraces(args) {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MONGODB", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"