@mastra/mongodb 0.14.6-alpha.0 → 0.14.7-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @mastra/mongodb
 
+## 0.14.7-alpha.0
+
+### Patch Changes
+
+- Adds MongoDB Observability support, and MongoDB Storage documentation, examples, and telemetry. ([#8426](https://github.com/mastra-ai/mastra/pull/8426))
+
+- Updated dependencies [[`f743dbb`](https://github.com/mastra-ai/mastra/commit/f743dbb8b40d1627b5c10c0e6fc154f4ebb6e394), [`5df9cce`](https://github.com/mastra-ai/mastra/commit/5df9cce1a753438413f64c11eeef8f845745c2a8), [`2060766`](https://github.com/mastra-ai/mastra/commit/20607667bf78ea104cca3e15dfb93ae0b62c9d18), [`2c4438b`](https://github.com/mastra-ai/mastra/commit/2c4438b87817ab7eed818c7990fef010475af1a3)]:
+  - @mastra/core@0.23.0-alpha.0
+
+## 0.14.6
+
+### Patch Changes
+
+- Update peerdeps to 0.23.0-0 ([#9043](https://github.com/mastra-ai/mastra/pull/9043))
+
+- Updated dependencies [[`c67ca32`](https://github.com/mastra-ai/mastra/commit/c67ca32e3c2cf69bfc146580770c720220ca44ac), [`efb5ed9`](https://github.com/mastra-ai/mastra/commit/efb5ed946ae7f410bc68c9430beb4b010afd25ec), [`dbc9e12`](https://github.com/mastra-ai/mastra/commit/dbc9e1216ba575ba59ead4afb727a01215f7de4f), [`99e41b9`](https://github.com/mastra-ai/mastra/commit/99e41b94957cdd25137d3ac12e94e8b21aa01b68), [`c28833c`](https://github.com/mastra-ai/mastra/commit/c28833c5b6d8e10eeffd7f7d39129d53b8bca240), [`8ea07b4`](https://github.com/mastra-ai/mastra/commit/8ea07b4bdc73e4218437dbb6dcb0f4b23e745a44), [`ba201b8`](https://github.com/mastra-ai/mastra/commit/ba201b8f8feac4c72350f2dbd52c13c7297ba7b0), [`f053e89`](https://github.com/mastra-ai/mastra/commit/f053e89160dbd0bd3333fc3492f68231b5c7c349), [`4fc4136`](https://github.com/mastra-ai/mastra/commit/4fc413652866a8d2240694fddb2562e9edbb70df), [`b78e04d`](https://github.com/mastra-ai/mastra/commit/b78e04d935a16ecb1e59c5c96e564903527edddd), [`d10baf5`](https://github.com/mastra-ai/mastra/commit/d10baf5a3c924f2a6654e23a3e318ed03f189b76), [`038c55a`](https://github.com/mastra-ai/mastra/commit/038c55a7090fc1b1513a966386d3072617f836ac), [`182f045`](https://github.com/mastra-ai/mastra/commit/182f0458f25bd70aa774e64fd923c8a483eddbf1), [`9a1a485`](https://github.com/mastra-ai/mastra/commit/9a1a4859b855e37239f652bf14b1ecd1029b8c4e), [`9257233`](https://github.com/mastra-ai/mastra/commit/9257233c4ffce09b2bedc2a9adbd70d7a83fa8e2), [`7620d2b`](https://github.com/mastra-ai/mastra/commit/7620d2bddeb4fae4c3c0a0b4e672969795fca11a), [`b2365f0`](https://github.com/mastra-ai/mastra/commit/b2365f038dd4c5f06400428b224af963f399ad50), [`0f1a4c9`](https://github.com/mastra-ai/mastra/commit/0f1a4c984fb4b104b2f0b63ba18c9fa77f567700), [`9029ba3`](https://github.com/mastra-ai/mastra/commit/9029ba34459c8859fed4c6b73efd8e2d0021e7ba), [`426cc56`](https://github.com/mastra-ai/mastra/commit/426cc561c85ae76a112ded2385532a91f9f9f074), [`00931fb`](https://github.com/mastra-ai/mastra/commit/00931fb1a21aa42c4fbc20c2c40dd62466b8fc8f), [`e473bfe`](https://github.com/mastra-ai/mastra/commit/e473bfe416c0b8e876973c2b6a6f13c394b7a93f), [`b78e04d`](https://github.com/mastra-ai/mastra/commit/b78e04d935a16ecb1e59c5c96e564903527edddd), [`2db6160`](https://github.com/mastra-ai/mastra/commit/2db6160e2022ff8827c15d30157e684683b934b5), [`8aeea37`](https://github.com/mastra-ai/mastra/commit/8aeea37efdde347c635a67fed56794943b7f74ec), [`02fe153`](https://github.com/mastra-ai/mastra/commit/02fe15351d6021d214da48ec982a0e9e4150bcee), [`648e2ca`](https://github.com/mastra-ai/mastra/commit/648e2ca42da54838c6ccbdaadc6fadd808fa6b86), [`74567b3`](https://github.com/mastra-ai/mastra/commit/74567b3d237ae3915cd0bca3cf55fa0a64e4e4a4), [`b65c5e0`](https://github.com/mastra-ai/mastra/commit/b65c5e0fe6f3c390a9a8bbcf69304d972c3a4afb), [`15a1733`](https://github.com/mastra-ai/mastra/commit/15a1733074cee8bd37370e1af34cd818e89fa7ac), [`fc2a774`](https://github.com/mastra-ai/mastra/commit/fc2a77468981aaddc3e77f83f0c4ad4a4af140da), [`4e08933`](https://github.com/mastra-ai/mastra/commit/4e08933625464dfde178347af5b6278fcf34188e)]:
+  - @mastra/core@0.22.0
+
 ## 0.14.6-alpha.0
 
 ### Patch Changes

package/dist/index.cjs

@@ -10,6 +10,10 @@ var agent = require('@mastra/core/agent');
 var scores = require('@mastra/core/scores');
 
 // src/vector/index.ts
+
+// package.json
+var package_default = {
+  version: "0.14.7-alpha.0"};
 var MongoDBFilterTranslator = class extends filter.BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -114,7 +118,14 @@ var MongoDBVector = class extends vector.MastraVector {
   };
   constructor({ uri, dbName, options }) {
     super();
-    this.client = new mongodb.MongoClient(uri, options);
+    const client = new mongodb.MongoClient(uri, {
+      ...options,
+      driverInfo: {
+        name: "mastra-vector",
+        version: package_default.version
+      }
+    });
+    this.client = client;
     this.db = this.client.db(dbName);
     this.collections = /* @__PURE__ */ new Map();
   }
@@ -634,8 +645,15 @@ var MongoDBConnector = class _MongoDBConnector {
         "MongoDBStore: dbName must be provided and cannot be empty. Passing an empty string may cause fallback to local MongoDB defaults."
       );
     }
+    const client = new mongodb.MongoClient(config.url, {
+      ...config.options,
+      driverInfo: {
+        name: "mastra-storage",
+        version: package_default.version
+      }
+    });
     return new _MongoDBConnector({
-      client: new mongodb.MongoClient(config.url, config.options),
+      client,
       dbName: config.dbName,
       handler: void 0
     });
@@ -1428,6 +1446,286 @@ var MemoryStorageMongoDB = class extends storage.MemoryStorage {
     }
   }
 };
+var ObservabilityMongoDB = class extends storage.ObservabilityStorage {
+  operations;
+  constructor({ operations }) {
+    super();
+    this.operations = operations;
+  }
+  get aiTracingStrategy() {
+    return {
+      preferred: "batch-with-updates",
+      supported: ["batch-with-updates", "insert-only"]
+    };
+  }
+  async createAISpan(span) {
+    try {
+      const startedAt = span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt;
+      const endedAt = span.endedAt instanceof Date ? span.endedAt.toISOString() : span.endedAt;
+      const record = {
+        ...span,
+        startedAt,
+        endedAt,
+        createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+        updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      return this.operations.insert({ tableName: storage.TABLE_AI_SPANS, record });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_CREATE_AI_SPAN_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: {
+            spanId: span.spanId,
+            traceId: span.traceId,
+            spanType: span.spanType,
+            spanName: span.name
+          }
+        },
+        error$1
+      );
+    }
+  }
+  async getAITrace(traceId) {
+    try {
+      const collection = await this.operations.getCollection(storage.TABLE_AI_SPANS);
+      const spans = await collection.find({ traceId }).sort({ startedAt: -1 }).toArray();
+      if (!spans || spans.length === 0) {
+        return null;
+      }
+      return {
+        traceId,
+        spans: spans.map((span) => this.transformSpanFromMongo(span))
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_GET_AI_TRACE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: {
+            traceId
+          }
+        },
+        error$1
+      );
+    }
+  }
+  async updateAISpan({
+    spanId,
+    traceId,
+    updates
+  }) {
+    try {
+      const data = { ...updates };
+      if (data.endedAt instanceof Date) {
+        data.endedAt = data.endedAt.toISOString();
+      }
+      if (data.startedAt instanceof Date) {
+        data.startedAt = data.startedAt.toISOString();
+      }
+      const updateData = {
+        ...data,
+        updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      await this.operations.update({
+        tableName: storage.TABLE_AI_SPANS,
+        keys: { spanId, traceId },
+        data: updateData
+      });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_UPDATE_AI_SPAN_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: {
+            spanId,
+            traceId
+          }
+        },
+        error$1
+      );
+    }
+  }
+  async getAITracesPaginated({
+    filters,
+    pagination
+  }) {
+    const page = pagination?.page ?? 0;
+    const perPage = pagination?.perPage ?? 10;
+    const { entityId, entityType, ...actualFilters } = filters || {};
+    try {
+      const collection = await this.operations.getCollection(storage.TABLE_AI_SPANS);
+      const mongoFilter = {
+        parentSpanId: null,
+        // Only get root spans for traces
+        ...actualFilters
+      };
+      if (pagination?.dateRange) {
+        const dateFilter = {};
+        if (pagination.dateRange.start) {
+          dateFilter.$gte = pagination.dateRange.start instanceof Date ? pagination.dateRange.start.toISOString() : pagination.dateRange.start;
+        }
+        if (pagination.dateRange.end) {
+          dateFilter.$lte = pagination.dateRange.end instanceof Date ? pagination.dateRange.end.toISOString() : pagination.dateRange.end;
+        }
+        if (Object.keys(dateFilter).length > 0) {
+          mongoFilter.startedAt = dateFilter;
+        }
+      }
+      if (entityId && entityType) {
+        let name = "";
+        if (entityType === "workflow") {
+          name = `workflow run: '${entityId}'`;
+        } else if (entityType === "agent") {
+          name = `agent run: '${entityId}'`;
+        } else {
+          const error$1 = new error.MastraError({
+            id: "MONGODB_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.USER,
+            details: {
+              entityType
+            },
+            text: `Cannot filter by entity type: ${entityType}`
+          });
+          throw error$1;
+        }
+        mongoFilter.name = name;
+      }
+      const count = await collection.countDocuments(mongoFilter);
+      if (count === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page,
+            perPage,
+            hasMore: false
+          },
+          spans: []
+        };
+      }
+      const spans = await collection.find(mongoFilter).sort({ startedAt: -1 }).skip(page * perPage).limit(perPage).toArray();
+      return {
+        pagination: {
+          total: count,
+          page,
+          perPage,
+          hasMore: spans.length === perPage
+        },
+        spans: spans.map((span) => this.transformSpanFromMongo(span))
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        error$1
+      );
+    }
+  }
+  async batchCreateAISpans(args) {
+    try {
+      const records = args.records.map((record) => {
+        const startedAt = record.startedAt instanceof Date ? record.startedAt.toISOString() : record.startedAt;
+        const endedAt = record.endedAt instanceof Date ? record.endedAt.toISOString() : record.endedAt;
+        return {
+          ...record,
+          startedAt,
+          endedAt,
+          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+        };
+      });
+      return this.operations.batchInsert({
+        tableName: storage.TABLE_AI_SPANS,
+        records
+      });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_BATCH_CREATE_AI_SPANS_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        error$1
+      );
+    }
+  }
+  async batchUpdateAISpans(args) {
+    try {
+      return this.operations.batchUpdate({
+        tableName: storage.TABLE_AI_SPANS,
+        updates: args.records.map((record) => {
+          const data = { ...record.updates };
+          if (data.endedAt instanceof Date) {
+            data.endedAt = data.endedAt.toISOString();
+          }
+          if (data.startedAt instanceof Date) {
+            data.startedAt = data.startedAt.toISOString();
+          }
+          const updateData = {
+            ...data,
+            updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+          };
+          return {
+            keys: { spanId: record.spanId, traceId: record.traceId },
+            data: updateData
+          };
+        })
+      });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_BATCH_UPDATE_AI_SPANS_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        error$1
+      );
+    }
+  }
+  async batchDeleteAITraces(args) {
+    try {
+      const collection = await this.operations.getCollection(storage.TABLE_AI_SPANS);
+      await collection.deleteMany({
+        traceId: { $in: args.traceIds }
+      });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "MONGODB_STORE_BATCH_DELETE_AI_TRACES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        error$1
+      );
+    }
+  }
+  /**
+   * Transform MongoDB document to AISpanRecord format
+   */
+  transformSpanFromMongo(doc) {
+    const { _id, ...span } = doc;
+    if (span.startedAt && typeof span.startedAt === "string") {
+      span.startedAt = span.startedAt;
+    }
+    if (span.endedAt && typeof span.endedAt === "string") {
+      span.endedAt = span.endedAt;
+    }
+    if (span.createdAt && typeof span.createdAt === "string") {
+      span.createdAt = new Date(span.createdAt);
+    }
+    if (span.updatedAt && typeof span.updatedAt === "string") {
+      span.updatedAt = new Date(span.updatedAt);
+    }
+    return span;
+  }
+};
 var StoreOperationsMongoDB = class extends storage.StoreOperations {
   #connector;
   constructor(config) {
@@ -1449,19 +1747,18 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
       const collection = await this.getCollection(tableName);
       await collection.deleteMany({});
     } catch (error$1) {
-      if (error$1 instanceof Error) {
-        const matstraError = new error.MastraError(
-          {
-            id: "STORAGE_MONGODB_STORE_CLEAR_TABLE_FAILED",
-            domain: error.ErrorDomain.STORAGE,
-            category: error.ErrorCategory.THIRD_PARTY,
-            details: { tableName }
-          },
-          error$1
-        );
-        this.logger.error(matstraError.message);
-        this.logger?.trackException(matstraError);
-      }
+      const mastraError = new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_CLEAR_TABLE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+      this.logger.error(mastraError.message);
+      this.logger?.trackException(mastraError);
+      throw mastraError;
     }
   }
   async dropTable({ tableName }) {
@@ -1485,6 +1782,9 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
   }
   processJsonbFields(tableName, record) {
     const schema = storage.TABLE_SCHEMAS[tableName];
+    if (!schema) {
+      return record;
+    }
     return Object.fromEntries(
       Object.entries(schema).map(([key, value]) => {
         if (value.type === "jsonb" && record[key] && typeof record[key] === "string") {
@@ -1500,19 +1800,18 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
       const recordToInsert = this.processJsonbFields(tableName, record);
       await collection.insertOne(recordToInsert);
     } catch (error$1) {
-      if (error$1 instanceof Error) {
-        const matstraError = new error.MastraError(
-          {
-            id: "STORAGE_MONGODB_STORE_INSERT_FAILED",
-            domain: error.ErrorDomain.STORAGE,
-            category: error.ErrorCategory.THIRD_PARTY,
-            details: { tableName }
-          },
-          error$1
-        );
-        this.logger.error(matstraError.message);
-        this.logger?.trackException(matstraError);
-      }
+      const mastraError = new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_INSERT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+      this.logger.error(mastraError.message);
+      this.logger?.trackException(mastraError);
+      throw mastraError;
     }
   }
   async batchInsert({ tableName, records }) {
@@ -1552,6 +1851,60 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
       );
     }
   }
+  async update({
+    tableName,
+    keys,
+    data
+  }) {
+    try {
+      const collection = await this.getCollection(tableName);
+      const processedData = this.processJsonbFields(tableName, data);
+      const cleanData = Object.fromEntries(Object.entries(processedData).filter(([_, value]) => value !== void 0));
+      await collection.updateOne(keys, { $set: cleanData });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_UPDATE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+  }
+  async batchUpdate({
+    tableName,
+    updates
+  }) {
+    if (!updates.length) {
+      return;
+    }
+    try {
+      const collection = await this.getCollection(tableName);
+      const bulkOps = updates.map(({ keys, data }) => {
+        const processedData = this.processJsonbFields(tableName, data);
+        const cleanData = Object.fromEntries(Object.entries(processedData).filter(([_, value]) => value !== void 0));
+        return {
+          updateOne: {
+            filter: keys,
+            update: { $set: cleanData }
+          }
+        };
+      });
+      await collection.bulkWrite(bulkOps);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_BATCH_UPDATE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+  }
 };
 function transformScoreRow(row) {
   let scorerValue = null;
@@ -2291,13 +2644,17 @@ var MongoDBStore = class extends storage.MastraStorage {
     const workflows = new WorkflowsStorageMongoDB({
       operations
     });
+    const observability = new ObservabilityMongoDB({
+      operations
+    });
     this.stores = {
       operations,
       memory,
       traces,
       legacyEvals,
       scores,
-      workflows
+      workflows,
+      observability
     };
   }
   async createTable({
@@ -2330,6 +2687,9 @@ var MongoDBStore = class extends storage.MastraStorage {
   async getThreadsByResourceId({ resourceId }) {
     return this.stores.memory.getThreadsByResourceId({ resourceId });
   }
+  async getThreadsByResourceIdPaginated(_args) {
+    return this.stores.memory.getThreadsByResourceIdPaginated(_args);
+  }
   async saveThread({ thread }) {
     return this.stores.memory.saveThread({ thread });
   }
@@ -2350,6 +2710,9 @@ var MongoDBStore = class extends storage.MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesPaginated(_args) {
+    return this.stores.memory.getMessagesPaginated(_args);
+  }
   async getMessagesById({
     messageIds,
     format
@@ -2359,12 +2722,6 @@ var MongoDBStore = class extends storage.MastraStorage {
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }
-  async getThreadsByResourceIdPaginated(_args) {
-    return this.stores.memory.getThreadsByResourceIdPaginated(_args);
-  }
-  async getMessagesPaginated(_args) {
-    return this.stores.memory.getMessagesPaginated(_args);
-  }
   async updateMessages(_args) {
     return this.stores.memory.updateMessages(_args);
   }
@@ -2374,6 +2731,9 @@ var MongoDBStore = class extends storage.MastraStorage {
   async getTracesPaginated(args) {
     return this.stores.traces.getTracesPaginated(args);
   }
+  async batchTraceInsert({ records }) {
+    return this.stores.traces.batchTraceInsert({ records });
+  }
   async getWorkflowRuns(args) {
     return this.stores.workflows.getWorkflowRuns(args);
   }
@@ -2491,6 +2851,90 @@ var MongoDBStore = class extends storage.MastraStorage {
       metadata
     });
   }
+  /**
+   * AI Tracing/Observability
+   */
+  async createAISpan(span) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.createAISpan(span);
+  }
+  async updateAISpan({
+    spanId,
+    traceId,
+    updates
+  }) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.updateAISpan({ spanId, traceId, updates });
+  }
+  async getAITrace(traceId) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.getAITrace(traceId);
+  }
+  async getAITracesPaginated(args) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.getAITracesPaginated(args);
+  }
+  async batchCreateAISpans(args) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.batchCreateAISpans(args);
+  }
+  async batchUpdateAISpans(args) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.batchUpdateAISpans(args);
+  }
+  async batchDeleteAITraces(args) {
+    if (!this.stores.observability) {
+      throw new error.MastraError({
+        id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability.batchDeleteAITraces(args);
+  }
 };
 
 // src/vector/prompt.ts