@mastra/mssql 0.5.1 → 1.0.0-beta.1

package/dist/index.js

@@ -1,10 +1,10 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, LegacyEvalsStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_EVALS, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, TracesStorage, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
-import sql3 from 'mssql';
-import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
+import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import sql2 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
+import { parseSqlIdentifier } from '@mastra/core/utils';
 import { randomUUID } from 'crypto';
-import { saveScorePayloadSchema } from '@mastra/core/scores';
+import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/storage/index.ts
 function getSchemaName(schema) {
@@ -80,153 +80,7 @@ function transformFromSqlRow({
   return result;
 }
 
-// src/storage/domains/legacy-evals/index.ts
-function transformEvalRow(row) {
-  let testInfoValue = null, resultValue = null;
-  if (row.test_info) {
-    try {
-      testInfoValue = typeof row.test_info === "string" ? JSON.parse(row.test_info) : row.test_info;
-    } catch {
-    }
-  }
-  if (row.result) {
-    try {
-      resultValue = typeof row.result === "string" ? JSON.parse(row.result) : row.result;
-    } catch {
-    }
-  }
-  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.created_at
-  };
-}
-var LegacyEvalsMSSQL = class extends LegacyEvalsStorage {
-  pool;
-  schema;
-  constructor({ pool, schema }) {
-    super();
-    this.pool = pool;
-    this.schema = schema;
-  }
-  /** @deprecated use getEvals instead */
-  async getEvalsByAgentName(agentName, type) {
-    try {
-      let query = `SELECT * FROM ${getTableName({ indexName: TABLE_EVALS, schemaName: getSchemaName(this.schema) })} WHERE agent_name = @p1`;
-      if (type === "test") {
-        query += " AND test_info IS NOT NULL AND JSON_VALUE(test_info, '$.testPath') IS NOT NULL";
-      } else if (type === "live") {
-        query += " AND (test_info IS NULL OR JSON_VALUE(test_info, '$.testPath') IS NULL)";
-      }
-      query += " ORDER BY created_at DESC";
-      const request = this.pool.request();
-      request.input("p1", agentName);
-      const result = await request.query(query);
-      const rows = result.recordset;
-      return typeof transformEvalRow === "function" ? rows?.map((row) => transformEvalRow(row)) ?? [] : rows ?? [];
-    } catch (error) {
-      if (error && error.number === 208 && error.message && error.message.includes("Invalid object name")) {
-        return [];
-      }
-      this.logger?.error?.("Failed to get evals for the specified agent:", error);
-      throw error;
-    }
-  }
-  async getEvals(options = {}) {
-    const { agentName, type, page = 0, perPage = 100, dateRange } = options;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const where = [];
-    const params = {};
-    if (agentName) {
-      where.push("agent_name = @agentName");
-      params["agentName"] = agentName;
-    }
-    if (type === "test") {
-      where.push("test_info IS NOT NULL AND JSON_VALUE(test_info, '$.testPath') IS NOT NULL");
-    } else if (type === "live") {
-      where.push("(test_info IS NULL OR JSON_VALUE(test_info, '$.testPath') IS NULL)");
-    }
-    if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
-      where.push(`[created_at] >= @fromDate`);
-      params[`fromDate`] = fromDate.toISOString();
-    }
-    if (toDate instanceof Date && !isNaN(toDate.getTime())) {
-      where.push(`[created_at] <= @toDate`);
-      params[`toDate`] = toDate.toISOString();
-    }
-    const whereClause = where.length > 0 ? `WHERE ${where.join(" AND ")}` : "";
-    const tableName = getTableName({ indexName: TABLE_EVALS, schemaName: getSchemaName(this.schema) });
-    const offset = page * perPage;
-    const countQuery = `SELECT COUNT(*) as total FROM ${tableName} ${whereClause}`;
-    const dataQuery = `SELECT * FROM ${tableName} ${whereClause} ORDER BY seq_id DESC OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
-    try {
-      const countReq = this.pool.request();
-      Object.entries(params).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          countReq.input(key, sql3.DateTime, value);
-        } else {
-          countReq.input(key, value);
-        }
-      });
-      const countResult = await countReq.query(countQuery);
-      const total = countResult.recordset[0]?.total || 0;
-      if (total === 0) {
-        return {
-          evals: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const req = this.pool.request();
-      Object.entries(params).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          req.input(key, sql3.DateTime, value);
-        } else {
-          req.input(key, value);
-        }
-      });
-      req.input("offset", offset);
-      req.input("perPage", perPage);
-      const result = await req.query(dataQuery);
-      const rows = result.recordset;
-      return {
-        evals: rows?.map((row) => transformEvalRow(row)) ?? [],
-        total,
-        page,
-        perPage,
-        hasMore: offset + (rows?.length ?? 0) < total
-      };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_EVALS_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            agentName: agentName || "all",
-            type: type || "all",
-            page,
-            perPage
-          }
-        },
-        error
-      );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      throw mastraError;
-    }
-  }
-};
+// src/storage/domains/memory/index.ts
 var MemoryMSSQL = class extends MemoryStorage {
   pool;
   schema;
@@ -244,7 +98,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     });
     const cleanMessages = messagesWithParsedContent.map(({ seq_id, ...rest }) => rest);
     const list = new MessageList().add(cleanMessages, "memory");
-    return format === "v2" ? list.get.all.v2() : list.get.all.v1();
+    return format === "v2" ? list.get.all.db() : list.get.all.v1();
   }
   constructor({
     pool,
@@ -258,7 +112,7 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async getThreadById({ threadId }) {
     try {
-      const sql7 = `SELECT 
+      const sql5 = `SELECT 
         id,
         [resourceId],
         title,
@@ -269,7 +123,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       WHERE id = @threadId`;
       const request = this.pool.request();
       request.input("threadId", threadId);
-      const resultSet = await request.query(sql7);
+      const resultSet = await request.query(sql5);
       const thread = resultSet.recordset[0] || null;
       if (!thread) {
         return null;
@@ -294,11 +148,24 @@ var MemoryMSSQL = class extends MemoryStorage {
       );
     }
   }
-  async getThreadsByResourceIdPaginated(args) {
-    const { resourceId, page = 0, perPage: perPageInput, orderBy = "createdAt", sortDirection = "DESC" } = args;
+  async listThreadsByResourceId(args) {
+    const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
+    if (page < 0) {
+      throw new MastraError({
+        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "Page number must be non-negative",
+        details: {
+          resourceId,
+          page
+        }
+      });
+    }
+    const perPage = normalizePerPage(perPageInput, 100);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+    const { field, direction } = this.parseOrderBy(orderBy);
     try {
-      const perPage = perPageInput !== void 0 ? perPageInput : 100;
-      const currentOffset = page * perPage;
       const baseQuery = `FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} WHERE [resourceId] = @resourceId`;
       const countQuery = `SELECT COUNT(*) as count ${baseQuery}`;
       const countRequest = this.pool.request();
@@ -310,17 +177,22 @@ var MemoryMSSQL = class extends MemoryStorage {
           threads: [],
           total: 0,
           page,
-          perPage,
+          perPage: perPageForResponse,
           hasMore: false
         };
       }
-      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
-      const dir = (sortDirection || "DESC").toUpperCase() === "ASC" ? "ASC" : "DESC";
+      const orderByField = field === "createdAt" ? "[createdAt]" : "[updatedAt]";
+      const dir = (direction || "DESC").toUpperCase() === "ASC" ? "ASC" : "DESC";
+      const limitValue = perPageInput === false ? total : perPage;
       const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${dir} OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
       const dataRequest = this.pool.request();
       dataRequest.input("resourceId", resourceId);
-      dataRequest.input("perPage", perPage);
-      dataRequest.input("offset", currentOffset);
+      dataRequest.input("offset", offset);
+      if (limitValue > 2147483647) {
+        dataRequest.input("perPage", sql2.BigInt, limitValue);
+      } else {
+        dataRequest.input("perPage", limitValue);
+      }
       const rowsResult = await dataRequest.query(dataQuery);
       const rows = rowsResult.recordset || [];
       const threads = rows.map((thread) => ({
@@ -333,13 +205,13 @@ var MemoryMSSQL = class extends MemoryStorage {
         threads,
         total,
         page,
-        perPage,
-        hasMore: currentOffset + threads.length < total
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
       };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -351,7 +223,13 @@ var MemoryMSSQL = class extends MemoryStorage {
       );
       this.logger?.error?.(mastraError.toString());
       this.logger?.trackException?.(mastraError);
-      return { threads: [], total: 0, page, perPage: perPageInput || 100, hasMore: false };
+      return {
+        threads: [],
+        total: 0,
+        page,
+        perPage: perPageForResponse,
+        hasMore: false
+      };
     }
   }
   async saveThread({ thread }) {
@@ -375,12 +253,12 @@ var MemoryMSSQL = class extends MemoryStorage {
       req.input("title", thread.title);
       const metadata = thread.metadata ? JSON.stringify(thread.metadata) : null;
       if (metadata === null) {
-        req.input("metadata", sql3.NVarChar, null);
+        req.input("metadata", sql2.NVarChar, null);
       } else {
         req.input("metadata", metadata);
       }
-      req.input("createdAt", sql3.DateTime2, thread.createdAt);
-      req.input("updatedAt", sql3.DateTime2, thread.updatedAt);
+      req.input("createdAt", sql2.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql2.DateTime2, thread.updatedAt);
       await req.query(mergeSql);
       return thread;
     } catch (error) {
@@ -397,31 +275,6 @@ var MemoryMSSQL = class extends MemoryStorage {
       );
     }
   }
-  /**
-   * @deprecated use getThreadsByResourceIdPaginated instead
-   */
-  async getThreadsByResourceId(args) {
-    const { resourceId, orderBy = "createdAt", sortDirection = "DESC" } = args;
-    try {
-      const baseQuery = `FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} WHERE [resourceId] = @resourceId`;
-      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
-      const dir = (sortDirection || "DESC").toUpperCase() === "ASC" ? "ASC" : "DESC";
-      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${dir}`;
-      const request = this.pool.request();
-      request.input("resourceId", resourceId);
-      const resultSet = await request.query(dataQuery);
-      const rows = resultSet.recordset || [];
-      return rows.map((thread) => ({
-        ...thread,
-        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
-        createdAt: thread.createdAt,
-        updatedAt: thread.updatedAt
-      }));
-    } catch (error) {
-      this.logger?.error?.(`Error getting threads for resource ${resourceId}:`, error);
-      return [];
-    }
-  }
   /**
    * Updates a thread's title and metadata, merging with existing metadata. Returns the updated thread.
    */
@@ -449,7 +302,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     };
     try {
       const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
-      const sql7 = `UPDATE ${table}
+      const sql5 = `UPDATE ${table}
         SET title = @title,
             metadata = @metadata,
             [updatedAt] = @updatedAt
@@ -460,7 +313,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       req.input("title", title);
       req.input("metadata", JSON.stringify(mergedMetadata));
       req.input("updatedAt", /* @__PURE__ */ new Date());
-      const result = await req.query(sql7);
+      const result = await req.query(sql5);
       let thread = result.recordset && result.recordset[0];
       if (thread && "seq_id" in thread) {
         const { seq_id, ...rest } = thread;
@@ -530,11 +383,9 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async _getIncludedMessages({
     threadId,
-    selectBy,
-    orderByStatement
+    include
   }) {
     if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    const include = selectBy?.include;
     if (!include) return null;
     const unionQueries = [];
     const paramValues = [];
@@ -559,7 +410,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             m.[resourceId],
             m.seq_id
           FROM (
-            SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
+            SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
             FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
             WHERE [thread_id] = ${pThreadId}
           ) AS m
@@ -567,15 +418,17 @@ var MemoryMSSQL = class extends MemoryStorage {
           OR EXISTS (
             SELECT 1
             FROM (
-              SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
+              SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
               FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
               WHERE [thread_id] = ${pThreadId}
             ) AS target
             WHERE target.id = ${pId}
             AND (
-              (m.row_num <= target.row_num + ${pPrev} AND m.row_num > target.row_num)
+              -- Get previous messages (messages that come BEFORE the target)
+              (m.row_num < target.row_num AND m.row_num >= target.row_num - ${pPrev})
               OR
-              (m.row_num >= target.row_num - ${pNext} AND m.row_num < target.row_num)
+              -- Get next messages (messages that come AFTER the target)
+              (m.row_num > target.row_num AND m.row_num <= target.row_num + ${pNext})
             )
           )
         `
@@ -604,34 +457,16 @@ var MemoryMSSQL = class extends MemoryStorage {
     });
     return dedupedRows;
   }
-  async getMessages(args) {
-    const { threadId, resourceId, format, selectBy } = args;
+  async listMessagesById({ messageIds }) {
+    if (messageIds.length === 0) return { messages: [] };
     const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
     const orderByStatement = `ORDER BY [seq_id] DESC`;
-    const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
     try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       let rows = [];
-      const include = selectBy?.include || [];
-      if (include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
-        if (includeMessages) {
-          rows.push(...includeMessages);
-        }
-      }
-      const excludeIds = rows.map((m) => m.id).filter(Boolean);
-      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [thread_id] = @threadId`;
+      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [id] IN (${messageIds.map((_, i) => `@id${i}`).join(", ")})`;
       const request = this.pool.request();
-      request.input("threadId", threadId);
-      if (excludeIds.length > 0) {
-        const excludeParams = excludeIds.map((_, idx) => `@id${idx}`);
-        query += ` AND id NOT IN (${excludeParams.join(", ")})`;
-        excludeIds.forEach((id, idx) => {
-          request.input(`id${idx}`, id);
-        });
-      }
-      query += ` ${orderByStatement} OFFSET 0 ROWS FETCH NEXT @limit ROWS ONLY`;
-      request.input("limit", limit);
+      messageIds.forEach((id, i) => request.input(`id${i}`, id));
+      query += ` ${orderByStatement}`;
       const result = await request.query(query);
       const remainingRows = result.recordset || [];
       rows.push(...remainingRows);
@@ -639,153 +474,171 @@ var MemoryMSSQL = class extends MemoryStorage {
         const timeDiff = a.seq_id - b.seq_id;
         return timeDiff;
       });
-      rows = rows.map(({ seq_id, ...rest }) => rest);
-      return this._parseAndFormatMessages(rows, format);
+      const messagesWithParsedContent = rows.map((row) => {
+        if (typeof row.content === "string") {
+          try {
+            return { ...row, content: JSON.parse(row.content) };
+          } catch {
+            return row;
+          }
+        }
+        return row;
+      });
+      const cleanMessages = messagesWithParsedContent.map(({ seq_id, ...rest }) => rest);
+      const list = new MessageList().add(cleanMessages, "memory");
+      return { messages: list.get.all.db() };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_BY_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
-            resourceId: resourceId ?? ""
+            messageIds: JSON.stringify(messageIds)
           }
         },
         error
       );
       this.logger?.error?.(mastraError.toString());
       this.logger?.trackException?.(mastraError);
-      return [];
+      return { messages: [] };
     }
   }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    if (messageIds.length === 0) return [];
-    const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
-    const orderByStatement = `ORDER BY [seq_id] DESC`;
-    try {
-      let rows = [];
-      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [id] IN (${messageIds.map((_, i) => `@id${i}`).join(", ")})`;
-      const request = this.pool.request();
-      messageIds.forEach((id, i) => request.input(`id${i}`, id));
-      query += ` ${orderByStatement}`;
-      const result = await request.query(query);
-      const remainingRows = result.recordset || [];
-      rows.push(...remainingRows);
-      rows.sort((a, b) => {
-        const timeDiff = a.seq_id - b.seq_id;
-        return timeDiff;
-      });
-      rows = rows.map(({ seq_id, ...rest }) => rest);
-      if (format === `v1`) return this._parseAndFormatMessages(rows, format);
-      return this._parseAndFormatMessages(rows, `v2`);
-    } catch (error) {
-      const mastraError = new MastraError(
+  async listMessages(args) {
+    const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    if (!threadId.trim()) {
+      throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_BY_ID_FAILED",
+          id: "STORAGE_MSSQL_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")
       );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      return [];
     }
-  }
-  async getMessagesPaginated(args) {
-    const { threadId, resourceId, format, selectBy } = args;
-    const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
+    if (page < 0) {
+      throw new MastraError({
+        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "Page number must be non-negative",
+        details: {
+          threadId,
+          page
+        }
+      });
+    }
+    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 fromDate = dateRange?.start;
-      const toDate = dateRange?.end;
-      const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
-      const orderByStatement = `ORDER BY [seq_id] DESC`;
-      let messages = [];
-      if (selectBy?.include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
-        if (includeMessages) messages.push(...includeMessages);
-      }
-      const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
-      const currentOffset = page * perPage;
-      const conditions = ["[thread_id] = @threadId"];
-      const request = this.pool.request();
-      request.input("threadId", threadId);
-      if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
-        conditions.push("[createdAt] >= @fromDate");
-        request.input("fromDate", fromDate.toISOString());
-      }
-      if (toDate instanceof Date && !isNaN(toDate.getTime())) {
-        conditions.push("[createdAt] <= @toDate");
-        request.input("toDate", toDate.toISOString());
-      }
-      const whereClause = `WHERE ${conditions.join(" AND ")}`;
-      const countQuery = `SELECT COUNT(*) as total FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} ${whereClause}`;
-      const countResult = await request.query(countQuery);
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
+      const orderByStatement = `ORDER BY [${field}] ${direction}, [seq_id] ${direction}`;
+      const tableName = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
+      const baseQuery = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId FROM ${tableName}`;
+      const filters = {
+        thread_id: threadId,
+        ...resourceId ? { resourceId } : {},
+        ...buildDateRangeFilter(filter?.dateRange, "createdAt")
+      };
+      const { sql: actualWhereClause = "", params: whereParams } = prepareWhereClause(
+        filters);
+      const bindWhereParams = (req) => {
+        Object.entries(whereParams).forEach(([paramName, paramValue]) => req.input(paramName, paramValue));
+      };
+      const countRequest = this.pool.request();
+      bindWhereParams(countRequest);
+      const countResult = await countRequest.query(`SELECT COUNT(*) as total FROM ${tableName}${actualWhereClause}`);
       const total = parseInt(countResult.recordset[0]?.total, 10) || 0;
-      if (total === 0 && messages.length > 0) {
-        const parsedIncluded = this._parseAndFormatMessages(messages, format);
+      const fetchBaseMessages = async () => {
+        const request = this.pool.request();
+        bindWhereParams(request);
+        if (perPageInput === false) {
+          const result2 = await request.query(`${baseQuery}${actualWhereClause} ${orderByStatement}`);
+          return result2.recordset || [];
+        }
+        request.input("offset", offset);
+        request.input("limit", perPage > 2147483647 ? sql2.BigInt : sql2.Int, perPage);
+        const result = await request.query(
+          `${baseQuery}${actualWhereClause} ${orderByStatement} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`
+        );
+        return result.recordset || [];
+      };
+      const baseRows = perPage === 0 ? [] : await fetchBaseMessages();
+      const messages = [...baseRows];
+      const seqById = /* @__PURE__ */ new Map();
+      messages.forEach((msg) => {
+        if (typeof msg.seq_id === "number") seqById.set(msg.id, msg.seq_id);
+      });
+      if (total === 0 && messages.length === 0 && (!include || include.length === 0)) {
         return {
-          messages: parsedIncluded,
-          total: parsedIncluded.length,
+          messages: [],
+          total: 0,
           page,
-          perPage,
+          perPage: perPageForResponse,
           hasMore: false
         };
       }
-      const excludeIds = messages.map((m) => m.id);
-      if (excludeIds.length > 0) {
-        const excludeParams = excludeIds.map((_, idx) => `@id${idx}`);
-        conditions.push(`id NOT IN (${excludeParams.join(", ")})`);
-        excludeIds.forEach((id, idx) => request.input(`id${idx}`, id));
-      }
-      const finalWhereClause = `WHERE ${conditions.join(" AND ")}`;
-      const dataQuery = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} ${finalWhereClause} ${orderByStatement} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
-      request.input("offset", currentOffset);
-      request.input("limit", perPage);
-      const rowsResult = await request.query(dataQuery);
-      const rows = rowsResult.recordset || [];
-      rows.sort((a, b) => a.seq_id - b.seq_id);
-      messages.push(...rows);
-      const parsed = this._parseAndFormatMessages(messages, format);
+      if (include?.length) {
+        const messageIds = new Set(messages.map((m) => m.id));
+        const includeMessages = await this._getIncludedMessages({ threadId, include });
+        includeMessages?.forEach((msg) => {
+          if (!messageIds.has(msg.id)) {
+            messages.push(msg);
+            messageIds.add(msg.id);
+            if (typeof msg.seq_id === "number") seqById.set(msg.id, msg.seq_id);
+          }
+        });
+      }
+      const parsed = this._parseAndFormatMessages(messages, "v2");
+      const mult = direction === "ASC" ? 1 : -1;
+      const finalMessages = parsed.sort((a, b) => {
+        const aVal = field === "createdAt" ? new Date(a.createdAt).getTime() : a[field];
+        const bVal = field === "createdAt" ? new Date(b.createdAt).getTime() : b[field];
+        if (aVal == null || bVal == null) {
+          return aVal == null && bVal == null ? a.id.localeCompare(b.id) : aVal == null ? 1 : -1;
+        }
+        const diff = (typeof aVal === "number" && typeof bVal === "number" ? aVal - bVal : String(aVal).localeCompare(String(bVal))) * mult;
+        if (diff !== 0) return diff;
+        const seqA = seqById.get(a.id);
+        const seqB = seqById.get(b.id);
+        return seqA != null && seqB != null ? (seqA - seqB) * mult : a.id.localeCompare(b.id);
+      });
+      const returnedThreadMessageCount = finalMessages.filter((m) => m.threadId === threadId).length;
+      const hasMore = perPageInput !== false && returnedThreadMessageCount < total && offset + perPage < total;
       return {
-        messages: parsed,
+        messages: finalMessages,
         total,
         page,
-        perPage,
-        hasMore: currentOffset + rows.length < total
+        perPage: perPageForResponse,
+        hasMore
       };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_PAGINATED_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
             threadId,
-            resourceId: resourceId ?? "",
-            page
+            resourceId: resourceId ?? ""
           }
         },
         error
       );
       this.logger?.error?.(mastraError.toString());
       this.logger?.trackException?.(mastraError);
-      return { messages: [], total: 0, page, perPage: perPageInput || 40, hasMore: false };
+      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: [] };
     const threadId = messages[0]?.threadId;
     if (!threadId) {
       throw new MastraError({
@@ -829,7 +682,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             "content",
             typeof message.content === "string" ? message.content : JSON.stringify(message.content)
           );
-          request.input("createdAt", sql3.DateTime2, message.createdAt);
+          request.input("createdAt", sql2.DateTime2, message.createdAt);
           request.input("role", message.role);
           request.input("type", message.type || "v2");
           request.input("resourceId", message.resourceId);
@@ -848,7 +701,7 @@ var MemoryMSSQL = class extends MemoryStorage {
           await request.query(mergeSql);
         }
         const threadReq = transaction.request();
-        threadReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+        threadReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
         threadReq.input("id", threadId);
         await threadReq.query(`UPDATE ${tableThreads} SET [updatedAt] = @updatedAt WHERE id = @id`);
         await transaction.commit();
@@ -867,8 +720,7 @@ var MemoryMSSQL = class extends MemoryStorage {
         return message;
       });
       const list = new MessageList().add(messagesWithParsedContent, "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(
         {
@@ -1147,13 +999,13 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     this.operations = operations;
     this.schema = schema;
   }
-  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;
@@ -1163,11 +1015,11 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         endedAt
         // Note: createdAt/updatedAt will be set by default values
       };
-      return this.operations.insert({ tableName: TABLE_AI_SPANS, record });
+      return this.operations.insert({ tableName: TABLE_SPANS, record });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_CREATE_AI_SPAN_FAILED",
+          id: "MSSQL_STORE_CREATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1181,10 +1033,10 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITrace(traceId) {
+  async getTrace(traceId) {
     try {
       const tableName = getTableName({
-        indexName: TABLE_AI_SPANS,
+        indexName: TABLE_SPANS,
         schemaName: getSchemaName(this.schema)
       });
       const request = this.pool.request();
@@ -1205,7 +1057,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         traceId,
         spans: result.recordset.map(
           (span) => transformFromSqlRow({
-            tableName: TABLE_AI_SPANS,
+            tableName: TABLE_SPANS,
             sqlRow: span
           })
         )
@@ -1213,7 +1065,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_AI_TRACE_FAILED",
+          id: "MSSQL_STORE_GET_TRACE_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1224,7 +1076,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
@@ -1238,14 +1090,14 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         data.startedAt = data.startedAt.toISOString();
       }
       await this.operations.update({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         keys: { spanId, traceId },
         data
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_UPDATE_AI_SPAN_FAILED",
+          id: "MSSQL_STORE_UPDATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1257,7 +1109,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITracesPaginated({
+  async getTracesPaginated({
     filters,
     pagination
   }) {
@@ -1282,7 +1134,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         name = `agent run: '${entityId}'`;
       } else {
         const error = new MastraError({
-          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1301,7 +1153,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       params[entityParam] = name;
     }
     const tableName = getTableName({
-      indexName: TABLE_AI_SPANS,
+      indexName: TABLE_SPANS,
       schemaName: getSchemaName(this.schema)
     });
     try {
@@ -1335,7 +1187,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
       const spans = dataResult.recordset.map(
         (row) => transformFromSqlRow({
-          tableName: TABLE_AI_SPANS,
+          tableName: TABLE_SPANS,
           sqlRow: row
         })
       );
@@ -1351,7 +1203,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1359,13 +1211,13 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchCreateAISpans(args) {
+  async batchCreateSpans(args) {
     if (!args.records || args.records.length === 0) {
       return;
     }
     try {
       await this.operations.batchInsert({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         records: args.records.map((span) => ({
           ...span,
           startedAt: span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt,
@@ -1375,7 +1227,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_CREATE_AI_SPANS_FAILED",
+          id: "MSSQL_STORE_BATCH_CREATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1386,7 +1238,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchUpdateAISpans(args) {
+  async batchUpdateSpans(args) {
     if (!args.records || args.records.length === 0) {
       return;
     }
@@ -1405,13 +1257,13 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         };
       });
       await this.operations.batchUpdate({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         updates
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_UPDATE_AI_SPANS_FAILED",
+          id: "MSSQL_STORE_BATCH_UPDATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1422,20 +1274,20 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchDeleteAITraces(args) {
+  async batchDeleteTraces(args) {
     if (!args.traceIds || args.traceIds.length === 0) {
       return;
     }
     try {
       const keys = args.traceIds.map((traceId) => ({ traceId }));
       await this.operations.batchDelete({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         keys
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_DELETE_AI_TRACES_FAILED",
+          id: "MSSQL_STORE_BATCH_DELETE_TRACES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1550,7 +1402,7 @@ var StoreOperationsMSSQL = class extends StoreOperations {
         const value = record[col];
         const preparedValue = this.prepareValue(value, col, tableName);
         if (preparedValue instanceof Date) {
-          request.input(`param${i}`, sql3.DateTime2, preparedValue);
+          request.input(`param${i}`, sql2.DateTime2, preparedValue);
         } else if (preparedValue === null || preparedValue === void 0) {
           request.input(`param${i}`, this.getMssqlType(tableName, col), null);
         } else {
@@ -1765,7 +1617,7 @@ ${columns}
     try {
       const keyEntries = Object.entries(keys).map(([key, value]) => [parseSqlIdentifier(key, "column name"), value]);
       const conditions = keyEntries.map(([key], i) => `[${key}] = @param${i}`).join(" AND ");
-      const sql7 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
+      const sql5 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
       const request = this.pool.request();
       keyEntries.forEach(([key, value], i) => {
         const preparedValue = this.prepareValue(value, key, tableName);
@@ -1775,7 +1627,7 @@ ${columns}
           request.input(`param${i}`, preparedValue);
         }
       });
-      const resultSet = await request.query(sql7);
+      const resultSet = await request.query(sql5);
       const result = resultSet.recordset[0] || null;
       if (!result) {
         return null;
@@ -1860,6 +1712,20 @@ ${columns}
       return value ? 1 : 0;
     }
     if (columnSchema?.type === "jsonb") {
+      if (typeof value === "string") {
+        const trimmed = value.trim();
+        if (trimmed.length > 0) {
+          try {
+            JSON.parse(trimmed);
+            return trimmed;
+          } catch {
+          }
+        }
+        return JSON.stringify(value);
+      }
+      if (typeof value === "bigint") {
+        return value.toString();
+      }
       return JSON.stringify(value);
     }
     if (typeof value === "object") {
@@ -1874,23 +1740,23 @@ ${columns}
     const col = TABLE_SCHEMAS[tableName]?.[columnName];
     switch (col?.type) {
       case "text":
-        return sql3.NVarChar;
+        return sql2.NVarChar;
       case "timestamp":
-        return sql3.DateTime2;
+        return sql2.DateTime2;
       case "uuid":
-        return sql3.UniqueIdentifier;
+        return sql2.UniqueIdentifier;
       case "jsonb":
-        return sql3.NVarChar;
+        return sql2.NVarChar;
       case "integer":
-        return sql3.Int;
+        return sql2.Int;
       case "bigint":
-        return sql3.BigInt;
+        return sql2.BigInt;
       case "float":
-        return sql3.Float;
+        return sql2.Float;
       case "boolean":
-        return sql3.Bit;
+        return sql2.Bit;
       default:
-        return sql3.NVarChar;
+        return sql2.NVarChar;
     }
   }
   /**
@@ -2334,35 +2200,30 @@ ${columns}
         table: TABLE_TRACES,
         columns: ["name", "seq_id DESC"]
       },
-      {
-        name: `${schemaPrefix}mastra_evals_agent_name_seqid_idx`,
-        table: TABLE_EVALS,
-        columns: ["agent_name", "seq_id DESC"]
-      },
       {
         name: `${schemaPrefix}mastra_scores_trace_id_span_id_seqid_idx`,
         table: TABLE_SCORERS,
         columns: ["traceId", "spanId", "seq_id DESC"]
       },
-      // AI Spans indexes for optimal trace querying
+      // Spans indexes for optimal trace querying
       {
         name: `${schemaPrefix}mastra_ai_spans_traceid_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["traceId", "startedAt DESC"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_parentspanid_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["parentSpanId", "startedAt DESC"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_name_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["name"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_spantype_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["spanType", "startedAt DESC"]
       }
     ];
@@ -2403,7 +2264,7 @@ function transformScoreRow(row) {
     metadata: safelyParseJSON(row.metadata),
     output: safelyParseJSON(row.output),
     additionalContext: safelyParseJSON(row.additionalContext),
-    runtimeContext: safelyParseJSON(row.runtimeContext),
+    requestContext: safelyParseJSON(row.requestContext),
     entity: safelyParseJSON(row.entity),
     createdAt: row.createdAt,
     updatedAt: row.updatedAt
@@ -2470,7 +2331,7 @@ var ScoresMSSQL = class extends ScoresStorage {
         input,
         output,
         additionalContext,
-        runtimeContext,
+        requestContext,
         entity,
         ...rest
       } = validatedScore;
@@ -2485,7 +2346,7 @@ var ScoresMSSQL = class extends ScoresStorage {
           analyzeStepResult: analyzeStepResult || null,
           metadata: metadata || null,
           additionalContext: additionalContext || null,
-          runtimeContext: runtimeContext || null,
+          requestContext: requestContext || null,
           entity: entity || null,
           scorer: scorer || null,
           createdAt: (/* @__PURE__ */ new Date()).toISOString(),
@@ -2505,7 +2366,7 @@ var ScoresMSSQL = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId,
     pagination,
     entityId,
@@ -2539,31 +2400,36 @@ var ScoresMSSQL = class extends ScoresStorage {
       });
       const totalResult = await countRequest.query(`SELECT COUNT(*) as count FROM ${tableName} WHERE ${whereClause}`);
       const total = totalResult.recordset[0]?.count || 0;
+      const { page, perPage: perPageInput } = pagination;
       if (total === 0) {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           },
           scores: []
         };
       }
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const limitValue = perPageInput === false ? total : perPage;
+      const end = perPageInput === false ? total : start + perPage;
       const dataRequest = this.pool.request();
       Object.entries(params).forEach(([key, value]) => {
         dataRequest.input(key, value);
       });
-      dataRequest.input("perPage", pagination.perPage);
-      dataRequest.input("offset", pagination.page * pagination.perPage);
+      dataRequest.input("perPage", limitValue);
+      dataRequest.input("offset", start);
       const dataQuery = `SELECT * FROM ${tableName} WHERE ${whereClause} ORDER BY [createdAt] DESC OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
       const result = await dataRequest.query(dataQuery);
       return {
         pagination: {
           total: Number(total),
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores: result.recordset.map((row) => transformScoreRow(row))
       };
@@ -2579,7 +2445,7 @@ var ScoresMSSQL = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId,
     pagination
   }) {
@@ -2590,30 +2456,35 @@ var ScoresMSSQL = class extends ScoresStorage {
         `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [runId] = @p1`
       );
       const total = totalResult.recordset[0]?.count || 0;
+      const { page, perPage: perPageInput } = pagination;
       if (total === 0) {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageInput,
             hasMore: false
           },
           scores: []
         };
       }
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const limitValue = perPageInput === false ? total : perPage;
+      const end = perPageInput === false ? total : start + perPage;
       const dataRequest = this.pool.request();
       dataRequest.input("p1", runId);
-      dataRequest.input("p2", pagination.perPage);
-      dataRequest.input("p3", pagination.page * pagination.perPage);
+      dataRequest.input("p2", limitValue);
+      dataRequest.input("p3", start);
       const result = await dataRequest.query(
         `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [runId] = @p1 ORDER BY [createdAt] DESC OFFSET @p3 ROWS FETCH NEXT @p2 ROWS ONLY`
       );
       return {
         pagination: {
           total: Number(total),
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores: result.recordset.map((row) => transformScoreRow(row))
       };
@@ -2629,7 +2500,7 @@ var ScoresMSSQL = class extends ScoresStorage {
       );
     }
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId,
     entityType,
     pagination
@@ -2642,31 +2513,36 @@ var ScoresMSSQL = class extends ScoresStorage {
         `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [entityId] = @p1 AND [entityType] = @p2`
       );
       const total = totalResult.recordset[0]?.count || 0;
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       if (total === 0) {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
+      const limitValue = perPageInput === false ? total : perPage;
+      const end = perPageInput === false ? total : start + perPage;
       const dataRequest = this.pool.request();
       dataRequest.input("p1", entityId);
       dataRequest.input("p2", entityType);
-      dataRequest.input("p3", pagination.perPage);
-      dataRequest.input("p4", pagination.page * pagination.perPage);
+      dataRequest.input("p3", limitValue);
+      dataRequest.input("p4", start);
       const result = await dataRequest.query(
         `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [entityId] = @p1 AND [entityType] = @p2 ORDER BY [createdAt] DESC OFFSET @p4 ROWS FETCH NEXT @p3 ROWS ONLY`
       );
       return {
         pagination: {
           total: Number(total),
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
         scores: result.recordset.map((row) => transformScoreRow(row))
       };
@@ -2682,7 +2558,7 @@ var ScoresMSSQL = class extends ScoresStorage {
       );
     }
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination
@@ -2695,34 +2571,38 @@ var ScoresMSSQL = class extends ScoresStorage {
         `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [traceId] = @p1 AND [spanId] = @p2`
       );
       const total = totalResult.recordset[0]?.count || 0;
+      const { page, perPage: perPageInput } = pagination;
+      const perPage = normalizePerPage(perPageInput, 100);
+      const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       if (total === 0) {
         return {
           pagination: {
             total: 0,
-            page: pagination.page,
-            perPage: pagination.perPage,
+            page,
+            perPage: perPageForResponse,
             hasMore: false
           },
           scores: []
         };
       }
-      const limit = pagination.perPage + 1;
+      const limitValue = perPageInput === false ? total : perPage;
+      const end = perPageInput === false ? total : start + perPage;
       const dataRequest = this.pool.request();
       dataRequest.input("p1", traceId);
       dataRequest.input("p2", spanId);
-      dataRequest.input("p3", limit);
-      dataRequest.input("p4", pagination.page * pagination.perPage);
+      dataRequest.input("p3", limitValue);
+      dataRequest.input("p4", start);
       const result = await dataRequest.query(
         `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [traceId] = @p1 AND [spanId] = @p2 ORDER BY [createdAt] DESC OFFSET @p4 ROWS FETCH NEXT @p3 ROWS ONLY`
       );
       return {
         pagination: {
           total: Number(total),
-          page: pagination.page,
-          perPage: pagination.perPage,
-          hasMore: result.recordset.length > pagination.perPage
+          page,
+          perPage: perPageForResponse,
+          hasMore: end < total
         },
-        scores: result.recordset.slice(0, pagination.perPage).map((row) => transformScoreRow(row))
+        scores: result.recordset.map((row) => transformScoreRow(row))
       };
     } catch (error) {
       throw new MastraError(
@@ -2737,173 +2617,6 @@ var ScoresMSSQL = class extends ScoresStorage {
     }
   }
 };
-var TracesMSSQL = class extends TracesStorage {
-  pool;
-  operations;
-  schema;
-  constructor({
-    pool,
-    operations,
-    schema
-  }) {
-    super();
-    this.pool = pool;
-    this.operations = operations;
-    this.schema = schema;
-  }
-  /** @deprecated use getTracesPaginated instead*/
-  async getTraces(args) {
-    if (args.fromDate || args.toDate) {
-      args.dateRange = {
-        start: args.fromDate,
-        end: args.toDate
-      };
-    }
-    const result = await this.getTracesPaginated(args);
-    return result.traces;
-  }
-  async getTracesPaginated(args) {
-    const { name, scope, page = 0, perPage: perPageInput, attributes, filters, dateRange } = args;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const perPage = perPageInput !== void 0 ? perPageInput : 100;
-    const currentOffset = page * perPage;
-    const paramMap = {};
-    const conditions = [];
-    let paramIndex = 1;
-    if (name) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[name] LIKE @${paramName}`);
-      paramMap[paramName] = `${name}%`;
-    }
-    if (scope) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[scope] = @${paramName}`);
-      paramMap[paramName] = scope;
-    }
-    if (attributes) {
-      Object.entries(attributes).forEach(([key, value]) => {
-        const parsedKey = parseFieldKey(key);
-        const paramName = `p${paramIndex++}`;
-        conditions.push(`JSON_VALUE([attributes], '$.${parsedKey}') = @${paramName}`);
-        paramMap[paramName] = value;
-      });
-    }
-    if (filters) {
-      Object.entries(filters).forEach(([key, value]) => {
-        const parsedKey = parseFieldKey(key);
-        const paramName = `p${paramIndex++}`;
-        conditions.push(`[${parsedKey}] = @${paramName}`);
-        paramMap[paramName] = value;
-      });
-    }
-    if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[createdAt] >= @${paramName}`);
-      paramMap[paramName] = fromDate.toISOString();
-    }
-    if (toDate instanceof Date && !isNaN(toDate.getTime())) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[createdAt] <= @${paramName}`);
-      paramMap[paramName] = toDate.toISOString();
-    }
-    const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
-    const countQuery = `SELECT COUNT(*) as total FROM ${getTableName({ indexName: TABLE_TRACES, schemaName: getSchemaName(this.schema) })} ${whereClause}`;
-    let total = 0;
-    try {
-      const countRequest = this.pool.request();
-      Object.entries(paramMap).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          countRequest.input(key, sql3.DateTime, value);
-        } else {
-          countRequest.input(key, value);
-        }
-      });
-      const countResult = await countRequest.query(countQuery);
-      total = parseInt(countResult.recordset[0].total, 10);
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_TRACES_PAGINATED_FAILED_TO_RETRIEVE_TOTAL_COUNT",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
-          }
-        },
-        error
-      );
-    }
-    if (total === 0) {
-      return {
-        traces: [],
-        total: 0,
-        page,
-        perPage,
-        hasMore: false
-      };
-    }
-    const dataQuery = `SELECT * FROM ${getTableName({ indexName: TABLE_TRACES, schemaName: getSchemaName(this.schema) })} ${whereClause} ORDER BY [seq_id] DESC OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
-    const dataRequest = this.pool.request();
-    Object.entries(paramMap).forEach(([key, value]) => {
-      if (value instanceof Date) {
-        dataRequest.input(key, sql3.DateTime, value);
-      } else {
-        dataRequest.input(key, value);
-      }
-    });
-    dataRequest.input("offset", currentOffset);
-    dataRequest.input("limit", perPage);
-    try {
-      const rowsResult = await dataRequest.query(dataQuery);
-      const rows = rowsResult.recordset;
-      const traces = rows.map((row) => ({
-        id: row.id,
-        parentSpanId: row.parentSpanId,
-        traceId: row.traceId,
-        name: row.name,
-        scope: row.scope,
-        kind: row.kind,
-        status: JSON.parse(row.status),
-        events: JSON.parse(row.events),
-        links: JSON.parse(row.links),
-        attributes: JSON.parse(row.attributes),
-        startTime: row.startTime,
-        endTime: row.endTime,
-        other: row.other,
-        createdAt: row.createdAt
-      }));
-      return {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: currentOffset + traces.length < total
-      };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_TRACES_PAGINATED_FAILED_TO_RETRIEVE_TRACES",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
-          }
-        },
-        error
-      );
-    }
-  }
-  async batchTraceInsert({ records }) {
-    this.logger.debug("Batch inserting traces", { count: records.length });
-    await this.operations.batchInsert({
-      tableName: TABLE_TRACES,
-      records
-    });
-  }
-};
 var WorkflowsMSSQL = class extends WorkflowsStorage {
   pool;
   operations;
@@ -2941,13 +2654,13 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     runId,
     stepId,
     result,
-    runtimeContext
+    requestContext
   }) {
     const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql3.Request(transaction);
+      const selectRequest = new sql2.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2966,20 +2679,20 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
           waitingPaths: {},
           status: "pending",
           runId,
-          runtimeContext: {}
+          requestContext: {}
         };
       } else {
         const existingSnapshot = existingSnapshotResult.recordset[0].snapshot;
         snapshot = typeof existingSnapshot === "string" ? JSON.parse(existingSnapshot) : existingSnapshot;
       }
       snapshot.context[stepId] = result;
-      snapshot.runtimeContext = { ...snapshot.runtimeContext, ...runtimeContext };
-      const upsertReq = new sql3.Request(transaction);
+      snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
+      const upsertReq = new sql2.Request(transaction);
       upsertReq.input("workflow_name", workflowName);
       upsertReq.input("run_id", runId);
       upsertReq.input("snapshot", JSON.stringify(snapshot));
-      upsertReq.input("createdAt", sql3.DateTime2, /* @__PURE__ */ new Date());
-      upsertReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("createdAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
       await upsertReq.query(
         `MERGE ${table} AS target
          USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
@@ -3019,7 +2732,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql3.Request(transaction);
+      const selectRequest = new sql2.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -3047,11 +2760,11 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         );
       }
       const updatedSnapshot = { ...snapshot, ...opts };
-      const updateRequest = new sql3.Request(transaction);
+      const updateRequest = new sql2.Request(transaction);
       updateRequest.input("snapshot", JSON.stringify(updatedSnapshot));
       updateRequest.input("workflow_name", workflowName);
       updateRequest.input("run_id", runId);
-      updateRequest.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      updateRequest.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
       await updateRequest.query(
         `UPDATE ${table} SET snapshot = @snapshot, [updatedAt] = @updatedAt WHERE workflow_name = @workflow_name AND run_id = @run_id`
       );
@@ -3090,8 +2803,8 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       request.input("run_id", runId);
       request.input("resourceId", resourceId);
       request.input("snapshot", JSON.stringify(snapshot));
-      request.input("createdAt", sql3.DateTime2, new Date(now));
-      request.input("updatedAt", sql3.DateTime2, new Date(now));
+      request.input("createdAt", sql2.DateTime2, new Date(now));
+      request.input("updatedAt", sql2.DateTime2, new Date(now));
       const mergeSql = `MERGE INTO ${table} AS target
         USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
         ON target.workflow_name = src.workflow_name AND target.run_id = src.run_id
@@ -3188,12 +2901,12 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       );
     }
   }
-  async getWorkflowRuns({
+  async listWorkflowRuns({
     workflowName,
     fromDate,
     toDate,
-    limit,
-    offset,
+    page,
+    perPage,
     resourceId
   } = {}) {
     try {
@@ -3226,20 +2939,23 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       const request = this.pool.request();
       Object.entries(paramMap).forEach(([key, value]) => {
         if (value instanceof Date) {
-          request.input(key, sql3.DateTime, value);
+          request.input(key, sql2.DateTime, value);
         } else {
           request.input(key, value);
         }
       });
-      if (limit !== void 0 && offset !== void 0) {
+      const usePagination = typeof perPage === "number" && typeof page === "number";
+      if (usePagination) {
         const countQuery = `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`;
         const countResult = await request.query(countQuery);
         total = Number(countResult.recordset[0]?.count || 0);
       }
       let query = `SELECT * FROM ${tableName} ${whereClause} ORDER BY [seq_id] DESC`;
-      if (limit !== void 0 && offset !== void 0) {
-        query += ` OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
-        request.input("limit", limit);
+      if (usePagination) {
+        const normalizedPerPage = normalizePerPage(perPage, Number.MAX_SAFE_INTEGER);
+        const offset = page * normalizedPerPage;
+        query += ` OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
+        request.input("perPage", normalizedPerPage);
         request.input("offset", offset);
       }
       const result = await request.query(query);
@@ -3248,7 +2964,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_WORKFLOW_RUNS_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_WORKFLOW_RUNS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -3268,7 +2984,10 @@ var MSSQLStore = class extends MastraStorage {
   isConnected = null;
   stores;
   constructor(config) {
-    super({ name: "MSSQLStore" });
+    if (!config.id || typeof config.id !== "string" || config.id.trim() === "") {
+      throw new Error("MSSQLStore: id must be provided and cannot be empty.");
+    }
+    super({ id: config.id, name: "MSSQLStore" });
     try {
       if ("connectionString" in config) {
         if (!config.connectionString || typeof config.connectionString !== "string" || config.connectionString.trim() === "") {
@@ -3283,7 +3002,7 @@ var MSSQLStore = class extends MastraStorage {
         }
       }
       this.schema = config.schemaName || "dbo";
-      this.pool = "connectionString" in config ? new sql3.ConnectionPool(config.connectionString) : new sql3.ConnectionPool({
+      this.pool = "connectionString" in config ? new sql2.ConnectionPool(config.connectionString) : new sql2.ConnectionPool({
         server: config.server,
         database: config.database,
         user: config.user,
@@ -3291,19 +3010,15 @@ var MSSQLStore = class extends MastraStorage {
         port: config.port,
         options: config.options || { encrypt: true, trustServerCertificate: true }
       });
-      const legacyEvals = new LegacyEvalsMSSQL({ pool: this.pool, schema: this.schema });
       const operations = new StoreOperationsMSSQL({ pool: this.pool, schemaName: this.schema });
       const scores = new ScoresMSSQL({ pool: this.pool, operations, schema: this.schema });
-      const traces = new TracesMSSQL({ pool: this.pool, operations, schema: this.schema });
       const workflows = new WorkflowsMSSQL({ pool: this.pool, operations, schema: this.schema });
       const memory = new MemoryMSSQL({ pool: this.pool, schema: this.schema, operations });
       const observability = new ObservabilityMSSQL({ pool: this.pool, operations, schema: this.schema });
       this.stores = {
         operations,
         scores,
-        traces,
         workflows,
-        legacyEvals,
         memory,
         observability
       };
@@ -3357,30 +3072,11 @@ var MSSQLStore = class extends MastraStorage {
       hasColumn: true,
       createTable: true,
       deleteMessages: true,
-      getScoresBySpan: true,
-      aiTracing: true,
+      listScoresBySpan: true,
+      observabilityInstance: true,
       indexManagement: true
     };
   }
-  /** @deprecated use getEvals instead */
-  async getEvalsByAgentName(agentName, type) {
-    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
-  }
-  async getEvals(options = {}) {
-    return this.stores.legacyEvals.getEvals(options);
-  }
-  /**
-   * @deprecated use getTracesPaginated instead
-   */
-  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 createTable({
     tableName,
     schema
@@ -3415,15 +3111,6 @@ var MSSQLStore = class extends MastraStorage {
   async getThreadById({ threadId }) {
     return this.stores.memory.getThreadById({ threadId });
   }
-  /**
-   * @deprecated use getThreadsByResourceIdPaginated instead
-   */
-  async getThreadsByResourceId(args) {
-    return this.stores.memory.getThreadsByResourceId(args);
-  }
-  async getThreadsByResourceIdPaginated(args) {
-    return this.stores.memory.getThreadsByResourceIdPaginated(args);
-  }
   async saveThread({ thread }) {
     return this.stores.memory.saveThread({ thread });
   }
@@ -3437,17 +3124,8 @@ var MSSQLStore = class extends MastraStorage {
   async deleteThread({ threadId }) {
     return this.stores.memory.deleteThread({ threadId });
   }
-  async getMessages(args) {
-    return this.stores.memory.getMessages(args);
-  }
-  async getMessagesById({
-    messageIds,
-    format
-  }) {
-    return this.stores.memory.getMessagesById({ messageIds, format });
-  }
-  async getMessagesPaginated(args) {
-    return this.stores.memory.getMessagesPaginated(args);
+  async listMessagesById({ messageIds }) {
+    return this.stores.memory.listMessagesById({ messageIds });
   }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
@@ -3481,9 +3159,9 @@ var MSSQLStore = class extends MastraStorage {
     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,
@@ -3506,15 +3184,15 @@ var MSSQLStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.loadWorkflowSnapshot({ workflowName, runId });
   }
-  async getWorkflowRuns({
+  async listWorkflowRuns({
     workflowName,
     fromDate,
     toDate,
-    limit,
-    offset,
+    perPage,
+    page,
     resourceId
   } = {}) {
-    return this.stores.workflows.getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId });
+    return this.stores.workflows.listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId });
   }
   async getWorkflowRunById({
     runId,
@@ -3541,7 +3219,7 @@ var MSSQLStore = class extends MastraStorage {
     return this.stores.operations.dropIndex(indexName);
   }
   /**
-   * AI Tracing / Observability
+   * Tracing / Observability
    */
   getObservabilityStore() {
     if (!this.stores.observability) {
@@ -3554,30 +3232,30 @@ var MSSQLStore = class extends MastraStorage {
     }
     return this.stores.observability;
   }
-  async createAISpan(span) {
-    return this.getObservabilityStore().createAISpan(span);
+  async createSpan(span) {
+    return this.getObservabilityStore().createSpan(span);
   }
-  async updateAISpan({
+  async updateSpan({
     spanId,
     traceId,
     updates
   }) {
-    return this.getObservabilityStore().updateAISpan({ spanId, traceId, updates });
+    return this.getObservabilityStore().updateSpan({ spanId, traceId, updates });
   }
-  async getAITrace(traceId) {
-    return this.getObservabilityStore().getAITrace(traceId);
+  async getTrace(traceId) {
+    return this.getObservabilityStore().getTrace(traceId);
   }
-  async getAITracesPaginated(args) {
-    return this.getObservabilityStore().getAITracesPaginated(args);
+  async getTracesPaginated(args) {
+    return this.getObservabilityStore().getTracesPaginated(args);
   }
-  async batchCreateAISpans(args) {
-    return this.getObservabilityStore().batchCreateAISpans(args);
+  async batchCreateSpans(args) {
+    return this.getObservabilityStore().batchCreateSpans(args);
   }
-  async batchUpdateAISpans(args) {
-    return this.getObservabilityStore().batchUpdateAISpans(args);
+  async batchUpdateSpans(args) {
+    return this.getObservabilityStore().batchUpdateSpans(args);
   }
-  async batchDeleteAITraces(args) {
-    return this.getObservabilityStore().batchDeleteAITraces(args);
+  async batchDeleteTraces(args) {
+    return this.getObservabilityStore().batchDeleteTraces(args);
   }
   /**
    * Scorers
@@ -3585,14 +3263,14 @@ var MSSQLStore = class extends MastraStorage {
   async getScoreById({ id: _id }) {
     return this.stores.scores.getScoreById({ id: _id });
   }
-  async getScoresByScorerId({
+  async listScoresByScorerId({
     scorerId: _scorerId,
     pagination: _pagination,
     entityId: _entityId,
     entityType: _entityType,
     source: _source
   }) {
-    return this.stores.scores.getScoresByScorerId({
+    return this.stores.scores.listScoresByScorerId({
       scorerId: _scorerId,
       pagination: _pagination,
       entityId: _entityId,
@@ -3603,29 +3281,29 @@ var MSSQLStore = class extends MastraStorage {
   async saveScore(_score) {
     return this.stores.scores.saveScore(_score);
   }
-  async getScoresByRunId({
+  async listScoresByRunId({
     runId: _runId,
     pagination: _pagination
   }) {
-    return this.stores.scores.getScoresByRunId({ runId: _runId, pagination: _pagination });
+    return this.stores.scores.listScoresByRunId({ runId: _runId, pagination: _pagination });
   }
-  async getScoresByEntityId({
+  async listScoresByEntityId({
     entityId: _entityId,
     entityType: _entityType,
     pagination: _pagination
   }) {
-    return this.stores.scores.getScoresByEntityId({
+    return this.stores.scores.listScoresByEntityId({
       entityId: _entityId,
       entityType: _entityType,
       pagination: _pagination
     });
   }
-  async getScoresBySpan({
+  async listScoresBySpan({
     traceId,
     spanId,
     pagination: _pagination
   }) {
-    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination: _pagination });
+    return this.stores.scores.listScoresBySpan({ traceId, spanId, pagination: _pagination });
   }
 };