@mastra/mongodb 0.14.9 → 1.0.0-beta.0

package/dist/index.js

@@ -3,15 +3,15 @@ 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, resolveMessageLimit, TABLE_THREADS, TABLE_RESOURCES, TracesStorage, TABLE_TRACES, LegacyEvalsStorage, TABLE_EVALS, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ObservabilityStorage, TABLE_AI_SPANS } from '@mastra/core/storage';
+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/scores';
+import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/vector/index.ts
 
 // package.json
 var package_default = {
-  version: "0.14.9"};
+  version: "1.0.0-beta.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -114,8 +114,8 @@ var MongoDBVector = class extends MastraVector {
     euclidean: "euclidean",
     dotproduct: "dotProduct"
   };
-  constructor({ uri, dbName, options }) {
-    super();
+  constructor({ id, uri, dbName, options }) {
+    super({ id });
     const client = new MongoClient(uri, {
       ...options,
       driverInfo: {
@@ -693,155 +693,6 @@ var MongoDBConnector = class _MongoDBConnector {
     }
   }
 };
-function transformEvalRow(row) {
-  let testInfoValue = null;
-  if (row.test_info) {
-    try {
-      testInfoValue = typeof row.test_info === "string" ? safelyParseJSON(row.test_info) : row.test_info;
-    } catch (e) {
-      console.warn("Failed to parse test_info:", e);
-    }
-  }
-  let resultValue;
-  try {
-    resultValue = typeof row.result === "string" ? safelyParseJSON(row.result) : row.result;
-  } catch (e) {
-    console.warn("Failed to parse result:", e);
-    throw new Error("Invalid result format");
-  }
-  return {
-    agentName: row.agent_name,
-    input: row.input,
-    output: row.output,
-    result: resultValue,
-    metricName: row.metric_name,
-    instructions: row.instructions,
-    testInfo: testInfoValue,
-    globalRunId: row.global_run_id,
-    runId: row.run_id,
-    createdAt: row.createdAt
-  };
-}
-var LegacyEvalsMongoDB = class extends LegacyEvalsStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
-  }
-  /** @deprecated use getEvals instead */
-  async getEvalsByAgentName(agentName, type) {
-    try {
-      const query = {
-        agent_name: agentName
-      };
-      if (type === "test") {
-        query["test_info"] = { $ne: null };
-      }
-      if (type === "live") {
-        query["test_info"] = null;
-      }
-      const collection = await this.operations.getCollection(TABLE_EVALS);
-      const documents = await collection.find(query).sort({ created_at: "desc" }).toArray();
-      const result = documents.map((row) => transformEvalRow(row));
-      return result.filter((row) => {
-        if (type === "live") {
-          return !Boolean(row.testInfo?.testPath);
-        }
-        if (type === "test") {
-          return row.testInfo?.testPath !== null;
-        }
-        return true;
-      });
-    } catch (error) {
-      if (error instanceof Error && error.message.includes("no such table")) {
-        return [];
-      }
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { agentName }
-        },
-        error
-      );
-    }
-  }
-  async getEvals(options = {}) {
-    const { agentName, type, page = 0, perPage = 100, dateRange } = options;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const currentOffset = page * perPage;
-    const query = {};
-    if (agentName) {
-      query["agent_name"] = agentName;
-    }
-    if (type === "test") {
-      query["test_info"] = { $ne: null };
-    } else if (type === "live") {
-      query["test_info"] = null;
-    }
-    if (fromDate || toDate) {
-      query["createdAt"] = {};
-      if (fromDate) {
-        query["createdAt"]["$gte"] = fromDate;
-      }
-      if (toDate) {
-        query["createdAt"]["$lte"] = toDate;
-      }
-    }
-    try {
-      const collection = await this.operations.getCollection(TABLE_EVALS);
-      let total = 0;
-      if (page === 0 || perPage < 1e3) {
-        total = await collection.countDocuments(query);
-      }
-      if (total === 0) {
-        return {
-          evals: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const documents = await collection.find(query).sort({ created_at: "desc" }).skip(currentOffset).limit(perPage).toArray();
-      const evals = documents.map((row) => transformEvalRow(row));
-      const filteredEvals = evals.filter((row) => {
-        if (type === "live") {
-          return !Boolean(row.testInfo?.testPath);
-        }
-        if (type === "test") {
-          return row.testInfo?.testPath !== null;
-        }
-        return true;
-      });
-      const hasMore = currentOffset + filteredEvals.length < total;
-      return {
-        evals: filteredEvals,
-        total,
-        page,
-        perPage,
-        hasMore
-      };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_GET_EVALS_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            agentName: agentName || "all",
-            type: type || "all",
-            page,
-            perPage
-          }
-        },
-        error
-      );
-    }
-  }
-};
 
 // src/storage/domains/utils.ts
 function formatDateForMongoDB(date) {
@@ -876,10 +727,9 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
   }
   async _getIncludedMessages({
     threadId,
-    selectBy
+    include
   }) {
     if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    const include = selectBy?.include;
     if (!include) return null;
     const collection = await this.operations.getCollection(TABLE_MESSAGES);
     const includedMessages = [];
@@ -903,151 +753,147 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
     });
     return dedupedMessages.map((row) => this.parseRow(row));
   }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy,
-    format
-  }) {
+  async listMessagesById({ messageIds }) {
+    if (messageIds.length === 0) return { messages: [] };
     try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      const messages = [];
-      const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
-      if (selectBy?.include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
-        if (includeMessages) {
-          messages.push(...includeMessages);
-        }
-      }
-      const excludeIds = messages.map((m) => m.id);
       const collection = await this.operations.getCollection(TABLE_MESSAGES);
-      const query = { thread_id: threadId };
-      if (excludeIds.length > 0) {
-        query.id = { $nin: excludeIds };
-      }
-      if (limit > 0) {
-        const remainingMessages = await collection.find(query).sort({ createdAt: -1 }).limit(limit).toArray();
-        messages.push(...remainingMessages.map((row) => this.parseRow(row)));
-      }
-      messages.sort((a, b) => a.createdAt.getTime() - b.createdAt.getTime());
-      const list = new MessageList().add(messages, "memory");
-      if (format === "v2") return list.get.all.v2();
-      return list.get.all.v1();
+      const rawMessages = await collection.find({ id: { $in: messageIds } }).sort({ createdAt: -1 }).toArray();
+      const list = new MessageList().add(
+        rawMessages.map(this.parseRow),
+        "memory"
+      );
+      return { messages: list.get.all.db() };
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_MESSAGES_FAILED",
+          id: "MONGODB_STORE_LIST_MESSAGES_BY_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId, resourceId: resourceId ?? "" }
+          details: { messageIds: JSON.stringify(messageIds) }
         },
         error
       );
     }
   }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    if (messageIds.length === 0) return [];
-    try {
-      const collection = await this.operations.getCollection(TABLE_MESSAGES);
-      const rawMessages = await collection.find({ id: { $in: messageIds } }).sort({ createdAt: -1 }).toArray();
-      const list = new MessageList().add(rawMessages.map(this.parseRow), "memory");
-      if (format === `v1`) return list.get.all.v1();
-      return list.get.all.v2();
-    } catch (error) {
+  async listMessages(args) {
+    const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    if (!threadId.trim()) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_MESSAGES_BY_ID_FAILED",
+          id: "STORAGE_MONGODB_LIST_MESSAGES_INVALID_THREAD_ID",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { messageIds: JSON.stringify(messageIds) }
+          details: { threadId }
         },
-        error
+        new Error("threadId must be a non-empty string")
       );
     }
-  }
-  async getMessagesPaginated(args) {
-    const { threadId, resourceId, format, selectBy } = args;
-    const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
-    const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const messages = [];
-    if (selectBy?.include?.length) {
-      try {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
-        if (includeMessages) {
-          messages.push(...includeMessages);
-        }
-      } catch (error) {
-        throw new MastraError(
-          {
-            id: "MONGODB_STORE_GET_MESSAGES_PAGINATED_GET_INCLUDE_MESSAGES_FAILED",
-            domain: ErrorDomain.STORAGE,
-            category: ErrorCategory.THIRD_PARTY,
-            details: { threadId, resourceId: resourceId ?? "" }
-          },
-          error
-        );
-      }
+    if (page < 0) {
+      throw new MastraError(
+        {
+          id: "STORAGE_MONGODB_LIST_MESSAGES_INVALID_PAGE",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { page }
+        },
+        new Error("page must be >= 0")
+      );
     }
+    const perPage = normalizePerPage(perPageInput, 40);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      const currentOffset = page * perPage;
+      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 };
-      if (fromDate) {
-        query.createdAt = { ...query.createdAt, $gte: fromDate };
+      if (resourceId) {
+        query.resourceId = resourceId;
       }
-      if (toDate) {
-        query.createdAt = { ...query.createdAt, $lte: toDate };
+      if (filter?.dateRange?.start) {
+        query.createdAt = { ...query.createdAt, $gte: filter.dateRange.start };
+      }
+      if (filter?.dateRange?.end) {
+        query.createdAt = { ...query.createdAt, $lte: filter.dateRange.end };
       }
       const total = await collection.countDocuments(query);
-      if (total === 0 && messages.length === 0) {
+      const messages = [];
+      if (perPage !== 0) {
+        const sortObj = { [field]: sortOrder };
+        let cursor = collection.find(query).sort(sortObj).skip(offset);
+        if (perPageInput !== false) {
+          cursor = cursor.limit(perPage);
+        }
+        const dataResult = await cursor.toArray();
+        messages.push(...dataResult.map((row) => this.parseRow(row)));
+      }
+      if (total === 0 && messages.length === 0 && (!include || include.length === 0)) {
         return {
           messages: [],
           total: 0,
           page,
-          perPage,
+          perPage: perPageForResponse,
           hasMore: false
         };
       }
-      const excludeIds = messages.map((m) => m.id);
-      if (excludeIds.length > 0) {
-        query.id = { $nin: excludeIds };
+      const messageIds = new Set(messages.map((m) => m.id));
+      if (include && include.length > 0) {
+        const includeMessages = await this._getIncludedMessages({ threadId, include });
+        if (includeMessages) {
+          for (const includeMsg of includeMessages) {
+            if (!messageIds.has(includeMsg.id)) {
+              messages.push(includeMsg);
+              messageIds.add(includeMsg.id);
+            }
+          }
+        }
       }
-      const dataResult = await collection.find(query).sort({ createdAt: -1 }).skip(currentOffset).limit(perPage).toArray();
-      messages.push(...dataResult.map((row) => this.parseRow(row)));
-      const messagesToReturn = format === "v1" ? new MessageList().add(messages, "memory").get.all.v1() : new MessageList().add(messages, "memory").get.all.v2();
+      const list = new MessageList().add(messages, "memory");
+      let finalMessages = list.get.all.db();
+      finalMessages = finalMessages.sort((a, b) => {
+        const isDateField = field === "createdAt" || field === "updatedAt";
+        const aValue = isDateField ? new Date(a[field]).getTime() : a[field];
+        const bValue = isDateField ? new Date(b[field]).getTime() : b[field];
+        if (typeof aValue === "number" && typeof bValue === "number") {
+          return direction === "ASC" ? aValue - bValue : bValue - aValue;
+        }
+        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 allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
+      const hasMore = perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
       return {
-        messages: messagesToReturn,
+        messages: finalMessages,
         total,
         page,
-        perPage,
-        hasMore: (page + 1) * perPage < total
+        perPage: perPageForResponse,
+        hasMore
       };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MONGODB_STORE_GET_MESSAGES_PAGINATED_FAILED",
+          id: "MONGODB_STORE_LIST_MESSAGES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId, resourceId: resourceId ?? "" }
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
         },
         error
       );
-      this.logger?.trackException?.(mastraError);
       this.logger?.error?.(mastraError.toString());
-      return { messages: [], total: 0, page, perPage, hasMore: false };
+      this.logger?.trackException?.(mastraError);
+      return {
+        messages: [],
+        total: 0,
+        page,
+        perPage: perPageForResponse,
+        hasMore: false
+      };
     }
   }
-  async saveMessages({
-    messages,
-    format
-  }) {
-    if (messages.length === 0) return messages;
+  async saveMessages({ messages }) {
+    if (messages.length === 0) return { messages: [] };
     try {
       const threadId = messages[0]?.threadId;
       if (!threadId) {
@@ -1090,8 +936,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
         threadsCollection.updateOne({ id: threadId }, { $set: { updatedAt: /* @__PURE__ */ new Date() } })
       ]);
       const list = new MessageList().add(messages, "memory");
-      if (format === "v2") return list.get.all.v2();
-      return list.get.all.v1();
+      return { messages: list.get.all.db() };
     } catch (error) {
       throw new MastraError(
         {
@@ -1296,36 +1141,41 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       );
     }
   }
-  async getThreadsByResourceId({ resourceId }) {
+  async listThreadsByResourceId(args) {
     try {
-      const collection = await this.operations.getCollection(TABLE_THREADS);
-      const results = await collection.find({ resourceId }).sort({ updatedAt: -1 }).toArray();
-      if (!results.length) {
-        return [];
+      const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
+      if (page < 0) {
+        throw new MastraError(
+          {
+            id: "STORAGE_MONGODB_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            details: { page }
+          },
+          new Error("page must be >= 0")
+        );
       }
-      return results.map((result) => ({
-        ...result,
-        metadata: typeof result.metadata === "string" ? safelyParseJSON(result.metadata) : result.metadata
-      }));
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { resourceId }
-        },
-        error
-      );
-    }
-  }
-  async getThreadsByResourceIdPaginated(args) {
-    try {
-      const { resourceId, page, perPage } = args;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const { field, direction } = this.parseOrderBy(orderBy);
       const collection = await this.operations.getCollection(TABLE_THREADS);
       const query = { resourceId };
       const total = await collection.countDocuments(query);
-      const threads = await collection.find(query).sort({ updatedAt: -1 }).skip(page * perPage).limit(perPage).toArray();
+      if (perPage === 0) {
+        return {
+          threads: [],
+          total,
+          page,
+          perPage: perPageForResponse,
+          hasMore: offset < total
+        };
+      }
+      const sortOrder = direction === "ASC" ? 1 : -1;
+      let cursor = collection.find(query).sort({ [field]: sortOrder }).skip(offset);
+      if (perPageInput !== false) {
+        cursor = cursor.limit(perPage);
+      }
+      const threads = await cursor.toArray();
       return {
         threads: threads.map((thread) => ({
           id: thread.id,
@@ -1337,13 +1187,13 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
         })),
         total,
         page,
-        perPage,
-        hasMore: (page + 1) * perPage < total
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
       };
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+          id: "MONGODB_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId: args.resourceId }
@@ -1450,13 +1300,13 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     super();
     this.operations = operations;
   }
-  get aiTracingStrategy() {
+  get tracingStrategy() {
     return {
       preferred: "batch-with-updates",
       supported: ["batch-with-updates", "insert-only"]
     };
   }
-  async createAISpan(span) {
+  async createSpan(span) {
     try {
       const startedAt = span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt;
       const endedAt = span.endedAt instanceof Date ? span.endedAt.toISOString() : span.endedAt;
@@ -1467,11 +1317,11 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
         createdAt: (/* @__PURE__ */ new Date()).toISOString(),
         updatedAt: (/* @__PURE__ */ new Date()).toISOString()
       };
-      return this.operations.insert({ tableName: TABLE_AI_SPANS, record });
+      return this.operations.insert({ tableName: TABLE_SPANS, record });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_CREATE_AI_SPAN_FAILED",
+          id: "MONGODB_STORE_CREATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1485,9 +1335,9 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITrace(traceId) {
+  async getTrace(traceId) {
     try {
-      const collection = await this.operations.getCollection(TABLE_AI_SPANS);
+      const collection = await this.operations.getCollection(TABLE_SPANS);
       const spans = await collection.find({ traceId }).sort({ startedAt: -1 }).toArray();
       if (!spans || spans.length === 0) {
         return null;
@@ -1499,7 +1349,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_AI_TRACE_FAILED",
+          id: "MONGODB_STORE_GET_TRACE_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1510,7 +1360,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
@@ -1528,14 +1378,14 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
         updatedAt: (/* @__PURE__ */ new Date()).toISOString()
       };
       await this.operations.update({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         keys: { spanId, traceId },
         data: updateData
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_UPDATE_AI_SPAN_FAILED",
+          id: "MONGODB_STORE_UPDATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1547,7 +1397,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITracesPaginated({
+  async getTracesPaginated({
     filters,
     pagination
   }) {
@@ -1555,7 +1405,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     const perPage = pagination?.perPage ?? 10;
     const { entityId, entityType, ...actualFilters } = filters || {};
     try {
-      const collection = await this.operations.getCollection(TABLE_AI_SPANS);
+      const collection = await this.operations.getCollection(TABLE_SPANS);
       const mongoFilter = {
         parentSpanId: null,
         // Only get root spans for traces
@@ -1581,7 +1431,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
           name = `agent run: '${entityId}'`;
         } else {
           const error = new MastraError({
-            id: "MONGODB_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+            id: "MONGODB_STORE_GET_TRACES_PAGINATED_FAILED",
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.USER,
             details: {
@@ -1618,7 +1468,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          id: "MONGODB_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1626,7 +1476,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async batchCreateAISpans(args) {
+  async batchCreateSpans(args) {
     try {
       const records = args.records.map((record) => {
         const startedAt = record.startedAt instanceof Date ? record.startedAt.toISOString() : record.startedAt;
@@ -1640,13 +1490,13 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
         };
       });
       return this.operations.batchInsert({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         records
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_CREATE_AI_SPANS_FAILED",
+          id: "MONGODB_STORE_BATCH_CREATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1654,10 +1504,10 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async batchUpdateAISpans(args) {
+  async batchUpdateSpans(args) {
     try {
       return this.operations.batchUpdate({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         updates: args.records.map((record) => {
           const data = { ...record.updates };
           if (data.endedAt instanceof Date) {
@@ -1679,7 +1529,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_UPDATE_AI_SPANS_FAILED",
+          id: "MONGODB_STORE_BATCH_UPDATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1687,16 +1537,16 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
       );
     }
   }
-  async batchDeleteAITraces(args) {
+  async batchDeleteTraces(args) {
     try {
-      const collection = await this.operations.getCollection(TABLE_AI_SPANS);
+      const collection = await this.operations.getCollection(TABLE_SPANS);
       await collection.deleteMany({
         traceId: { $in: args.traceIds }
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MONGODB_STORE_BATCH_DELETE_AI_TRACES_FAILED",
+          id: "MONGODB_STORE_BATCH_DELETE_TRACES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1705,7 +1555,7 @@ var ObservabilityMongoDB = class extends ObservabilityStorage {
     }
   }
   /**
-   * Transform MongoDB document to AISpanRecord format
+   * Transform MongoDB document to SpanRecord format
    */
   transformSpanFromMongo(doc) {
     const { _id, ...span } = doc;
@@ -1953,12 +1803,12 @@ function transformScoreRow(row) {
       console.warn("Failed to parse entity:", e);
     }
   }
-  let runtimeContextValue = null;
-  if (row.runtimeContext) {
+  let requestContextValue = null;
+  if (row.requestContext) {
     try {
-      runtimeContextValue = typeof row.runtimeContext === "string" ? safelyParseJSON(row.runtimeContext) : row.runtimeContext;
+      requestContextValue = typeof row.requestContext === "string" ? safelyParseJSON(row.requestContext) : row.requestContext;
     } catch (e) {
-      console.warn("Failed to parse runtimeContext:", e);
+      console.warn("Failed to parse requestContext:", e);
     }
   }
   let metadataValue = null;
@@ -1989,7 +1839,7 @@ function transformScoreRow(row) {
     input: inputValue,
     output: outputValue,
     additionalContext: row.additionalContext,
-    runtimeContext: runtimeContextValue,
+    requestContext: requestContextValue,
     entity: entityValue,
     source: row.source,
     resourceId: row.resourceId,
@@ -2061,7 +1911,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
         input: typeof validatedScore.input === "string" ? safelyParseJSON(validatedScore.input) : validatedScore.input,
         output: typeof validatedScore.output === "string" ? safelyParseJSON(validatedScore.output) : validatedScore.output,
         additionalContext: validatedScore.additionalContext,
-        runtimeContext: typeof validatedScore.runtimeContext === "string" ? safelyParseJSON(validatedScore.runtimeContext) : validatedScore.runtimeContext,
+        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 || "",
@@ -2090,7 +1940,7 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId,
     pagination,
     entityId,
@@ -2098,6 +1948,9 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     source
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const query = { scorerId };
       if (entityId) {
         query.entityId = entityId;
@@ -2110,28 +1963,31 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       }
       const collection = await this.operations.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments(query);
-      const currentOffset = pagination.page * pagination.perPage;
       if (total === 0) {
         return {
           scores: [],
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           }
         };
       }
-      const documents = await collection.find(query).sort({ createdAt: "desc" }).skip(currentOffset).limit(pagination.perPage).toArray();
+      const end = perPageInput === false ? total : start + perPage;
+      let cursor = collection.find(query).sort({ createdAt: "desc" }).skip(start);
+      if (perPageInput !== false) {
+        cursor = cursor.limit(perPage);
+      }
+      const documents = await cursor.toArray();
       const scores = documents.map((row) => transformScoreRow(row));
-      const hasMore = currentOffset + scores.length < total;
       return {
         scores,
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         }
       };
     } catch (error) {
@@ -2146,35 +2002,41 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const collection = await this.operations.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments({ runId });
-      const currentOffset = pagination.page * pagination.perPage;
       if (total === 0) {
         return {
           scores: [],
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           }
         };
       }
-      const documents = await collection.find({ runId }).sort({ createdAt: "desc" }).skip(currentOffset).limit(pagination.perPage).toArray();
+      const end = perPageInput === false ? total : start + perPage;
+      let cursor = collection.find({ runId }).sort({ createdAt: "desc" }).skip(start);
+      if (perPageInput !== false) {
+        cursor = cursor.limit(perPage);
+      }
+      const documents = await cursor.toArray();
       const scores = documents.map((row) => transformScoreRow(row));
-      const hasMore = currentOffset + scores.length < total;
       return {
         scores,
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         }
       };
     } catch (error) {
@@ -2189,36 +2051,42 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId,
     entityType,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const collection = await this.operations.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments({ entityId, entityType });
-      const currentOffset = pagination.page * pagination.perPage;
       if (total === 0) {
         return {
           scores: [],
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           }
         };
       }
-      const documents = await collection.find({ entityId, entityType }).sort({ createdAt: "desc" }).skip(currentOffset).limit(pagination.perPage).toArray();
+      const end = perPageInput === false ? total : start + perPage;
+      let cursor = collection.find({ entityId, entityType }).sort({ createdAt: "desc" }).skip(start);
+      if (perPageInput !== false) {
+        cursor = cursor.limit(perPage);
+      }
+      const documents = await cursor.toArray();
       const scores = documents.map((row) => transformScoreRow(row));
-      const hasMore = currentOffset + scores.length < total;
       return {
         scores,
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         }
       };
     } catch (error) {
@@ -2233,37 +2101,43 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
       );
     }
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination
   }) {
     try {
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const query = { traceId, spanId };
       const collection = await this.operations.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments(query);
-      const currentOffset = pagination.page * pagination.perPage;
       if (total === 0) {
         return {
           scores: [],
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           }
         };
       }
-      const documents = await collection.find(query).sort({ createdAt: "desc" }).skip(currentOffset).limit(pagination.perPage).toArray();
+      const end = perPageInput === false ? total : start + perPage;
+      let cursor = collection.find(query).sort({ createdAt: "desc" }).skip(start);
+      if (perPageInput !== false) {
+        cursor = cursor.limit(perPage);
+      }
+      const documents = await cursor.toArray();
       const scores = documents.map((row) => transformScoreRow(row));
-      const hasMore = currentOffset + scores.length < total;
       return {
         scores,
         pagination: {
           total,
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         }
       };
     } catch (error) {
@@ -2279,121 +2153,6 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
     }
   }
 };
-var TracesStorageMongoDB = class extends TracesStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
-  }
-  async getTraces(args) {
-    if (args.fromDate || args.toDate) {
-      args.dateRange = {
-        start: args.fromDate,
-        end: args.toDate
-      };
-    }
-    try {
-      const result = await this.getTracesPaginated(args);
-      return result.traces;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_GET_TRACES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
-        },
-        error
-      );
-    }
-  }
-  async getTracesPaginated(args) {
-    const { name, scope, page = 0, perPage = 100, attributes, filters, dateRange } = args;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const currentOffset = page * perPage;
-    const query = {};
-    if (name) {
-      query["name"] = new RegExp(name);
-    }
-    if (scope) {
-      query["scope"] = scope;
-    }
-    if (attributes) {
-      query["$and"] = Object.entries(attributes).map(([key, value]) => ({
-        [`attributes.${key}`]: value
-      }));
-    }
-    if (filters) {
-      Object.entries(filters).forEach(([key, value]) => {
-        query[key] = value;
-      });
-    }
-    if (fromDate || toDate) {
-      query["createdAt"] = {};
-      if (fromDate) {
-        query["createdAt"]["$gte"] = fromDate;
-      }
-      if (toDate) {
-        query["createdAt"]["$lte"] = toDate;
-      }
-    }
-    try {
-      const collection = await this.operations.getCollection(TABLE_TRACES);
-      const total = await collection.countDocuments(query);
-      if (total === 0) {
-        return {
-          traces: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const result = await collection.find(query, {
-        sort: { startTime: -1 }
-      }).limit(perPage).skip(currentOffset).toArray();
-      const traces = result.map((row) => ({
-        id: row.id,
-        parentSpanId: row.parentSpanId,
-        traceId: row.traceId,
-        name: row.name,
-        scope: row.scope,
-        kind: row.kind,
-        status: safelyParseJSON(row.status),
-        events: safelyParseJSON(row.events),
-        links: safelyParseJSON(row.links),
-        attributes: safelyParseJSON(row.attributes),
-        startTime: row.startTime,
-        endTime: row.endTime,
-        other: safelyParseJSON(row.other),
-        createdAt: row.createdAt
-      }));
-      return {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: currentOffset + traces.length < total
-      };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_GET_TRACES_PAGINATED_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
-        },
-        error
-      );
-    }
-  }
-  async batchTraceInsert({ records }) {
-    this.logger.debug("Batch inserting traces", { count: records.length });
-    await this.operations.batchInsert({
-      tableName: TABLE_TRACES,
-      records
-    });
-  }
-};
 var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
   operations;
   constructor({ operations }) {
@@ -2405,7 +2164,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
     // runId,
     // stepId,
     // result,
-    // runtimeContext,
+    // requestContext,
   }) {
     throw new Error("Method not implemented.");
   }
@@ -2478,7 +2237,7 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
       );
     }
   }
-  async getWorkflowRuns(args) {
+  async listWorkflowRuns(args) {
     const options = args || {};
     try {
       const query = {};
@@ -2499,24 +2258,39 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
         query["resourceId"] = options.resourceId;
       }
       const collection = await this.operations.getCollection(TABLE_WORKFLOW_SNAPSHOT);
-      const total = await collection.countDocuments(query);
+      let total = 0;
       let cursor = collection.find(query).sort({ createdAt: -1 });
-      if (options.offset) {
-        cursor = cursor.skip(options.offset);
-      }
-      if (options.limit) {
-        cursor = cursor.limit(options.limit);
+      if (options.page !== void 0 && typeof options.perPage === "number") {
+        if (options.page < 0) {
+          throw new MastraError(
+            {
+              id: "STORAGE_MONGODB_INVALID_PAGE",
+              domain: ErrorDomain.STORAGE,
+              category: ErrorCategory.USER,
+              details: { page: options.page }
+            },
+            new Error("page must be >= 0")
+          );
+        }
+        total = await collection.countDocuments(query);
+        const normalizedPerPage = normalizePerPage(options.perPage, Number.MAX_SAFE_INTEGER);
+        if (normalizedPerPage === 0) {
+          return { runs: [], total };
+        }
+        const offset = options.page * normalizedPerPage;
+        cursor = cursor.skip(offset);
+        cursor = cursor.limit(Math.min(normalizedPerPage, 2147483647));
       }
       const results = await cursor.toArray();
       const runs = results.map((row) => this.parseWorkflowRun(row));
       return {
         runs,
-        total
+        total: total || runs.length
       };
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_MONGODB_STORE_GET_WORKFLOW_RUNS_FAILED",
+          id: "STORAGE_MONGODB_STORE_LIST_WORKFLOW_RUNS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { workflowName: options.workflowName || "unknown" }
@@ -2591,6 +2365,7 @@ var loadConnector = (config) => {
   }
   try {
     return MongoDBConnector.fromDatabaseConfig({
+      id: config.id,
       options: config.options,
       url: config.url,
       dbName: config.dbName
@@ -2617,11 +2392,11 @@ var MongoDBStore = class extends MastraStorage {
       hasColumn: false,
       createTable: false,
       deleteMessages: false,
-      getScoresBySpan: true
+      listScoresBySpan: true
     };
   }
   constructor(config) {
-    super({ name: "MongoDBStore" });
+    super({ id: config.id, name: "MongoDBStore" });
     this.stores = {};
     this.#connector = loadConnector(config);
     const operations = new StoreOperationsMongoDB({
@@ -2630,12 +2405,6 @@ var MongoDBStore = class extends MastraStorage {
     const memory = new MemoryStorageMongoDB({
       operations
     });
-    const traces = new TracesStorageMongoDB({
-      operations
-    });
-    const legacyEvals = new LegacyEvalsMongoDB({
-      operations
-    });
     const scores = new ScoresStorageMongoDB({
       operations
     });
@@ -2648,8 +2417,6 @@ var MongoDBStore = class extends MastraStorage {
     this.stores = {
       operations,
       memory,
-      traces,
-      legacyEvals,
       scores,
       workflows,
       observability
@@ -2682,12 +2449,6 @@ var MongoDBStore = class extends MastraStorage {
   async getThreadById({ threadId }) {
     return this.stores.memory.getThreadById({ threadId });
   }
-  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 });
   }
@@ -2701,21 +2462,8 @@ var MongoDBStore = class extends MastraStorage {
   async deleteThread({ threadId }) {
     return this.stores.memory.deleteThread({ threadId });
   }
-  async getMessages({
-    threadId,
-    selectBy,
-    format
-  }) {
-    return this.stores.memory.getMessages({ threadId, selectBy, format });
-  }
-  async getMessagesPaginated(_args) {
-    return this.stores.memory.getMessagesPaginated(_args);
-  }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    return this.stores.memory.getMessagesById({ messageIds, format });
+  async listMessagesById({ messageIds }) {
+    return this.stores.memory.listMessagesById({ messageIds });
   }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
@@ -2723,32 +2471,17 @@ var MongoDBStore = class extends MastraStorage {
   async updateMessages(_args) {
     return this.stores.memory.updateMessages(_args);
   }
-  async getTraces(args) {
-    return this.stores.traces.getTraces(args);
-  }
-  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);
-  }
-  async getEvals(options = {}) {
-    return this.stores.legacyEvals.getEvals(options);
-  }
-  async getEvalsByAgentName(agentName, type) {
-    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
+  async listWorkflowRuns(args) {
+    return this.stores.workflows.listWorkflowRuns(args);
   }
   async updateWorkflowResults({
     workflowName,
     runId,
     stepId,
     result,
-    runtimeContext
+    requestContext
   }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
   }
   async updateWorkflowState({
     workflowName,
@@ -2800,34 +2533,34 @@ var MongoDBStore = class extends MastraStorage {
   async saveScore(score) {
     return this.stores.scores.saveScore(score);
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId,
     pagination
   }) {
-    return this.stores.scores.getScoresByRunId({ runId, pagination });
+    return this.stores.scores.listScoresByRunId({ runId, pagination });
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId,
     entityType,
     pagination
   }) {
-    return this.stores.scores.getScoresByEntityId({ entityId, entityType, pagination });
+    return this.stores.scores.listScoresByEntityId({ entityId, entityType, pagination });
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId,
     pagination,
     entityId,
     entityType,
     source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
+    return this.stores.scores.listScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination
   }) {
-    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+    return this.stores.scores.listScoresBySpan({ traceId, spanId, pagination });
   }
   /**
    * RESOURCES
@@ -2850,9 +2583,9 @@ var MongoDBStore = class extends MastraStorage {
     });
   }
   /**
-   * AI Tracing/Observability
+   * Tracing/Observability
    */
-  async createAISpan(span) {
+  async createSpan(span) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2861,9 +2594,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.createAISpan(span);
+    return this.stores.observability.createSpan(span);
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
@@ -2876,9 +2609,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.updateAISpan({ spanId, traceId, updates });
+    return this.stores.observability.updateSpan({ spanId, traceId, updates });
   }
-  async getAITrace(traceId) {
+  async getTrace(traceId) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2887,9 +2620,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.getAITrace(traceId);
+    return this.stores.observability.getTrace(traceId);
   }
-  async getAITracesPaginated(args) {
+  async getTracesPaginated(args) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2898,9 +2631,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.getAITracesPaginated(args);
+    return this.stores.observability.getTracesPaginated(args);
   }
-  async batchCreateAISpans(args) {
+  async batchCreateSpans(args) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2909,9 +2642,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.batchCreateAISpans(args);
+    return this.stores.observability.batchCreateSpans(args);
   }
-  async batchUpdateAISpans(args) {
+  async batchUpdateSpans(args) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2920,9 +2653,9 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.batchUpdateAISpans(args);
+    return this.stores.observability.batchUpdateSpans(args);
   }
-  async batchDeleteAITraces(args) {
+  async batchDeleteTraces(args) {
     if (!this.stores.observability) {
       throw new MastraError({
         id: "MONGODB_STORE_OBSERVABILITY_NOT_INITIALIZED",
@@ -2931,7 +2664,7 @@ var MongoDBStore = class extends MastraStorage {
         text: "Observability storage is not initialized"
       });
     }
-    return this.stores.observability.batchDeleteAITraces(args);
+    return this.stores.observability.batchDeleteTraces(args);
   }
 };