@mastra/mssql 0.0.0-remove-unused-model-providers-api-20251030210744 → 0.0.0-safe-stringify-telemetry-20251205024938

package/dist/index.js

@@ -1,8 +1,8 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
-import sql2 from 'mssql';
+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 { MessageList } from '@mastra/core/agent';
-import { parseSqlIdentifier } from '@mastra/core/utils';
 import { randomUUID } from 'crypto';
 import { saveScorePayloadSchema } from '@mastra/core/scores';
 
@@ -80,7 +80,153 @@ function transformFromSqlRow({
   return result;
 }
 
-// src/storage/domains/memory/index.ts
+// 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;
+    }
+  }
+};
 var MemoryMSSQL = class extends MemoryStorage {
   pool;
   schema;
@@ -112,7 +258,7 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async getThreadById({ threadId }) {
     try {
-      const sql5 = `SELECT 
+      const sql7 = `SELECT 
         id,
         [resourceId],
         title,
@@ -123,7 +269,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       WHERE id = @threadId`;
       const request = this.pool.request();
       request.input("threadId", threadId);
-      const resultSet = await request.query(sql5);
+      const resultSet = await request.query(sql7);
       const thread = resultSet.recordset[0] || null;
       if (!thread) {
         return null;
@@ -229,12 +375,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", sql2.NVarChar, null);
+        req.input("metadata", sql3.NVarChar, null);
       } else {
         req.input("metadata", metadata);
       }
-      req.input("createdAt", sql2.DateTime2, thread.createdAt);
-      req.input("updatedAt", sql2.DateTime2, thread.updatedAt);
+      req.input("createdAt", sql3.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql3.DateTime2, thread.updatedAt);
       await req.query(mergeSql);
       return thread;
     } catch (error) {
@@ -303,7 +449,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     };
     try {
       const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
-      const sql5 = `UPDATE ${table}
+      const sql7 = `UPDATE ${table}
         SET title = @title,
             metadata = @metadata,
             [updatedAt] = @updatedAt
@@ -314,7 +460,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(sql5);
+      const result = await req.query(sql7);
       let thread = result.recordset && result.recordset[0];
       if (thread && "seq_id" in thread) {
         const { seq_id, ...rest } = thread;
@@ -384,7 +530,8 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async _getIncludedMessages({
     threadId,
-    selectBy
+    selectBy,
+    orderByStatement
   }) {
     if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
     const include = selectBy?.include;
@@ -412,7 +559,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             m.[resourceId],
             m.seq_id
           FROM (
-            SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
+            SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
             FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
             WHERE [thread_id] = ${pThreadId}
           ) AS m
@@ -420,17 +567,15 @@ var MemoryMSSQL = class extends MemoryStorage {
           OR EXISTS (
             SELECT 1
             FROM (
-              SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
+              SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
               FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
               WHERE [thread_id] = ${pThreadId}
             ) AS target
             WHERE target.id = ${pId}
             AND (
-              -- Get previous messages (messages that come BEFORE the target)
-              (m.row_num < target.row_num AND m.row_num >= target.row_num - ${pPrev})
+              (m.row_num <= target.row_num + ${pPrev} AND m.row_num > target.row_num)
               OR
-              -- Get next messages (messages that come AFTER the target)
-              (m.row_num > target.row_num AND m.row_num <= target.row_num + ${pNext})
+              (m.row_num >= target.row_num - ${pNext} AND m.row_num < target.row_num)
             )
           )
         `
@@ -469,7 +614,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       let rows = [];
       const include = selectBy?.include || [];
       if (include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
+        const includeMessages = await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
         if (includeMessages) {
           rows.push(...includeMessages);
         }
@@ -514,7 +659,10 @@ var MemoryMSSQL = class extends MemoryStorage {
       return [];
     }
   }
-  async listMessagesById({ messageIds }) {
+  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`;
@@ -532,8 +680,8 @@ var MemoryMSSQL = class extends MemoryStorage {
         return timeDiff;
       });
       rows = rows.map(({ seq_id, ...rest }) => rest);
-      const messages = this._parseAndFormatMessages(rows, `v2`);
-      return messages;
+      if (format === `v1`) return this._parseAndFormatMessages(rows, format);
+      return this._parseAndFormatMessages(rows, `v2`);
     } catch (error) {
       const mastraError = new MastraError(
         {
@@ -551,135 +699,6 @@ var MemoryMSSQL = class extends MemoryStorage {
       return [];
     }
   }
-  async listMessages(args) {
-    const { threadId, resourceId, include, filter, limit, offset = 0, orderBy } = args;
-    if (!threadId.trim()) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MSSQL_LIST_MESSAGES_INVALID_THREAD_ID",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
-        },
-        new Error("threadId must be a non-empty string")
-      );
-    }
-    try {
-      let perPage = 40;
-      if (limit !== void 0) {
-        if (limit === false) {
-          perPage = Number.MAX_SAFE_INTEGER;
-        } else if (limit === 0) {
-          perPage = 0;
-        } else if (typeof limit === "number" && limit > 0) {
-          perPage = limit;
-        }
-      }
-      const page = perPage === 0 ? 0 : Math.floor(offset / perPage);
-      const sortField = orderBy?.field || "createdAt";
-      const sortDirection = orderBy?.direction || "DESC";
-      const orderByStatement = `ORDER BY [${sortField}] ${sortDirection}`;
-      const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
-      const tableName = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
-      const conditions = ["[thread_id] = @threadId"];
-      const request = this.pool.request();
-      request.input("threadId", threadId);
-      if (resourceId) {
-        conditions.push("[resourceId] = @resourceId");
-        request.input("resourceId", resourceId);
-      }
-      if (filter?.dateRange?.start) {
-        conditions.push("[createdAt] >= @fromDate");
-        request.input("fromDate", filter.dateRange.start);
-      }
-      if (filter?.dateRange?.end) {
-        conditions.push("[createdAt] <= @toDate");
-        request.input("toDate", filter.dateRange.end);
-      }
-      const whereClause = `WHERE ${conditions.join(" AND ")}`;
-      const countQuery = `SELECT COUNT(*) as total FROM ${tableName} ${whereClause}`;
-      const countResult = await request.query(countQuery);
-      const total = parseInt(countResult.recordset[0]?.total, 10) || 0;
-      const dataQuery = `${selectStatement} FROM ${tableName} ${whereClause} ${orderByStatement} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
-      request.input("offset", offset);
-      if (perPage > 2147483647) {
-        request.input("limit", sql2.BigInt, perPage);
-      } else {
-        request.input("limit", perPage);
-      }
-      const rowsResult = await request.query(dataQuery);
-      const rows = rowsResult.recordset || [];
-      const messages = [...rows];
-      if (total === 0 && messages.length === 0) {
-        return {
-          messages: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const messageIds = new Set(messages.map((m) => m.id));
-      if (include && include.length > 0) {
-        const selectBy = { include };
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
-        if (includeMessages) {
-          for (const includeMsg of includeMessages) {
-            if (!messageIds.has(includeMsg.id)) {
-              messages.push(includeMsg);
-              messageIds.add(includeMsg.id);
-            }
-          }
-        }
-      }
-      const parsed = this._parseAndFormatMessages(messages, "v2");
-      let finalMessages = parsed;
-      finalMessages = finalMessages.sort((a, b) => {
-        const aValue = sortField === "createdAt" ? new Date(a.createdAt).getTime() : a[sortField];
-        const bValue = sortField === "createdAt" ? new Date(b.createdAt).getTime() : b[sortField];
-        return sortDirection === "ASC" ? aValue - bValue : bValue - aValue;
-      });
-      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
-      const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
-      const hasMore = limit === false ? false : allThreadMessagesReturned ? false : offset + rows.length < total;
-      return {
-        messages: finalMessages,
-        total,
-        page,
-        perPage,
-        hasMore
-      };
-    } catch (error) {
-      const errorPerPage = limit === false ? Number.MAX_SAFE_INTEGER : limit === 0 ? 0 : limit || 40;
-      const mastraError = new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId,
-            resourceId: resourceId ?? ""
-          }
-        },
-        error
-      );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      return {
-        messages: [],
-        total: 0,
-        page: errorPerPage === 0 ? 0 : Math.floor(offset / errorPerPage),
-        perPage: errorPerPage,
-        hasMore: false
-      };
-    }
-  }
-  async listThreadsByResourceId(args) {
-    const { resourceId, limit, offset, orderBy, sortDirection } = args;
-    const page = Math.floor(offset / limit);
-    const perPage = limit;
-    return this.getThreadsByResourceIdPaginated({ resourceId, page, perPage, orderBy, sortDirection });
-  }
   async getMessagesPaginated(args) {
     const { threadId, resourceId, format, selectBy } = args;
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
@@ -691,7 +710,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       const orderByStatement = `ORDER BY [seq_id] DESC`;
       let messages = [];
       if (selectBy?.include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId, selectBy });
+        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 });
@@ -735,7 +754,8 @@ var MemoryMSSQL = class extends MemoryStorage {
       const rows = rowsResult.recordset || [];
       rows.sort((a, b) => a.seq_id - b.seq_id);
       messages.push(...rows);
-      const parsed = this._parseAndFormatMessages(messages, format);
+      let parsed = this._parseAndFormatMessages(messages, format);
+      parsed = parsed.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
       return {
         messages: parsed,
         total,
@@ -810,7 +830,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             "content",
             typeof message.content === "string" ? message.content : JSON.stringify(message.content)
           );
-          request.input("createdAt", sql2.DateTime2, message.createdAt);
+          request.input("createdAt", sql3.DateTime2, message.createdAt);
           request.input("role", message.role);
           request.input("type", message.type || "v2");
           request.input("resourceId", message.resourceId);
@@ -829,7 +849,7 @@ var MemoryMSSQL = class extends MemoryStorage {
           await request.query(mergeSql);
         }
         const threadReq = transaction.request();
-        threadReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+        threadReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
         threadReq.input("id", threadId);
         await threadReq.query(`UPDATE ${tableThreads} SET [updatedAt] = @updatedAt WHERE id = @id`);
         await transaction.commit();
@@ -1531,7 +1551,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}`, sql2.DateTime2, preparedValue);
+          request.input(`param${i}`, sql3.DateTime2, preparedValue);
         } else if (preparedValue === null || preparedValue === void 0) {
           request.input(`param${i}`, this.getMssqlType(tableName, col), null);
         } else {
@@ -1746,7 +1766,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 sql5 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
+      const sql7 = `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);
@@ -1756,7 +1776,7 @@ ${columns}
           request.input(`param${i}`, preparedValue);
         }
       });
-      const resultSet = await request.query(sql5);
+      const resultSet = await request.query(sql7);
       const result = resultSet.recordset[0] || null;
       if (!result) {
         return null;
@@ -1841,6 +1861,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") {
@@ -1855,23 +1889,23 @@ ${columns}
     const col = TABLE_SCHEMAS[tableName]?.[columnName];
     switch (col?.type) {
       case "text":
-        return sql2.NVarChar;
+        return sql3.NVarChar;
       case "timestamp":
-        return sql2.DateTime2;
+        return sql3.DateTime2;
       case "uuid":
-        return sql2.UniqueIdentifier;
+        return sql3.UniqueIdentifier;
       case "jsonb":
-        return sql2.NVarChar;
+        return sql3.NVarChar;
       case "integer":
-        return sql2.Int;
+        return sql3.Int;
       case "bigint":
-        return sql2.BigInt;
+        return sql3.BigInt;
       case "float":
-        return sql2.Float;
+        return sql3.Float;
       case "boolean":
-        return sql2.Bit;
+        return sql3.Bit;
       default:
-        return sql2.NVarChar;
+        return sql3.NVarChar;
     }
   }
   /**
@@ -2315,6 +2349,11 @@ ${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,
@@ -2379,10 +2418,10 @@ function transformScoreRow(row) {
     metadata: safelyParseJSON(row.metadata),
     output: safelyParseJSON(row.output),
     additionalContext: safelyParseJSON(row.additionalContext),
-    requestContext: safelyParseJSON(row.requestContext),
+    runtimeContext: safelyParseJSON(row.runtimeContext),
     entity: safelyParseJSON(row.entity),
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt
+    createdAt: new Date(row.createdAt),
+    updatedAt: new Date(row.updatedAt)
   };
 }
 var ScoresMSSQL = class extends ScoresStorage {
@@ -2446,7 +2485,7 @@ var ScoresMSSQL = class extends ScoresStorage {
         input,
         output,
         additionalContext,
-        requestContext,
+        runtimeContext,
         entity,
         ...rest
       } = validatedScore;
@@ -2461,7 +2500,7 @@ var ScoresMSSQL = class extends ScoresStorage {
           analyzeStepResult: analyzeStepResult || null,
           metadata: metadata || null,
           additionalContext: additionalContext || null,
-          requestContext: requestContext || null,
+          runtimeContext: runtimeContext || null,
           entity: entity || null,
           scorer: scorer || null,
           createdAt: (/* @__PURE__ */ new Date()).toISOString(),
@@ -2713,6 +2752,173 @@ 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;
@@ -2750,13 +2956,13 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     runId,
     stepId,
     result,
-    requestContext
+    runtimeContext
   }) {
     const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql2.Request(transaction);
+      const selectRequest = new sql3.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2767,28 +2973,29 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         snapshot = {
           context: {},
           activePaths: [],
+          activeStepsPath: {},
           timestamp: Date.now(),
           suspendedPaths: {},
           resumeLabels: {},
           serializedStepGraph: [],
+          status: "pending",
           value: {},
           waitingPaths: {},
-          status: "pending",
           runId,
-          requestContext: {}
+          runtimeContext: {}
         };
       } else {
         const existingSnapshot = existingSnapshotResult.recordset[0].snapshot;
         snapshot = typeof existingSnapshot === "string" ? JSON.parse(existingSnapshot) : existingSnapshot;
       }
       snapshot.context[stepId] = result;
-      snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
-      const upsertReq = new sql2.Request(transaction);
+      snapshot.runtimeContext = { ...snapshot.runtimeContext, ...runtimeContext };
+      const upsertReq = new sql3.Request(transaction);
       upsertReq.input("workflow_name", workflowName);
       upsertReq.input("run_id", runId);
       upsertReq.input("snapshot", JSON.stringify(snapshot));
-      upsertReq.input("createdAt", sql2.DateTime2, /* @__PURE__ */ new Date());
-      upsertReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("createdAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("updatedAt", sql3.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
@@ -2828,7 +3035,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql2.Request(transaction);
+      const selectRequest = new sql3.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2856,11 +3063,11 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         );
       }
       const updatedSnapshot = { ...snapshot, ...opts };
-      const updateRequest = new sql2.Request(transaction);
+      const updateRequest = new sql3.Request(transaction);
       updateRequest.input("snapshot", JSON.stringify(updatedSnapshot));
       updateRequest.input("workflow_name", workflowName);
       updateRequest.input("run_id", runId);
-      updateRequest.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+      updateRequest.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
       await updateRequest.query(
         `UPDATE ${table} SET snapshot = @snapshot, [updatedAt] = @updatedAt WHERE workflow_name = @workflow_name AND run_id = @run_id`
       );
@@ -2899,8 +3106,8 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       request.input("run_id", runId);
       request.input("resourceId", resourceId);
       request.input("snapshot", JSON.stringify(snapshot));
-      request.input("createdAt", sql2.DateTime2, new Date(now));
-      request.input("updatedAt", sql2.DateTime2, new Date(now));
+      request.input("createdAt", sql3.DateTime2, new Date(now));
+      request.input("updatedAt", sql3.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
@@ -2997,13 +3204,14 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       );
     }
   }
-  async listWorkflowRuns({
+  async getWorkflowRuns({
     workflowName,
     fromDate,
     toDate,
     limit,
     offset,
-    resourceId
+    resourceId,
+    status
   } = {}) {
     try {
       const conditions = [];
@@ -3012,6 +3220,10 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         conditions.push(`[workflow_name] = @workflowName`);
         paramMap["workflowName"] = workflowName;
       }
+      if (status) {
+        conditions.push(`JSON_VALUE([snapshot], '$.status') = @status`);
+        paramMap["status"] = status;
+      }
       if (resourceId) {
         const hasResourceId = await this.operations.hasColumn(TABLE_WORKFLOW_SNAPSHOT, "resourceId");
         if (hasResourceId) {
@@ -3035,7 +3247,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       const request = this.pool.request();
       Object.entries(paramMap).forEach(([key, value]) => {
         if (value instanceof Date) {
-          request.input(key, sql2.DateTime, value);
+          request.input(key, sql3.DateTime, value);
         } else {
           request.input(key, value);
         }
@@ -3092,7 +3304,7 @@ var MSSQLStore = class extends MastraStorage {
         }
       }
       this.schema = config.schemaName || "dbo";
-      this.pool = "connectionString" in config ? new sql2.ConnectionPool(config.connectionString) : new sql2.ConnectionPool({
+      this.pool = "connectionString" in config ? new sql3.ConnectionPool(config.connectionString) : new sql3.ConnectionPool({
         server: config.server,
         database: config.database,
         user: config.user,
@@ -3100,15 +3312,19 @@ 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
       };
@@ -3167,6 +3383,25 @@ var MSSQLStore = class extends MastraStorage {
       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
@@ -3226,6 +3461,12 @@ var MSSQLStore = class extends MastraStorage {
   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);
   }
@@ -3261,9 +3502,9 @@ var MSSQLStore = class extends MastraStorage {
     runId,
     stepId,
     result,
-    requestContext
+    runtimeContext
   }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
   }
   async updateWorkflowState({
     workflowName,
@@ -3286,15 +3527,8 @@ var MSSQLStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.loadWorkflowSnapshot({ workflowName, runId });
   }
-  async listWorkflowRuns({
-    workflowName,
-    fromDate,
-    toDate,
-    limit,
-    offset,
-    resourceId
-  } = {}) {
-    return this.stores.workflows.listWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId });
+  async getWorkflowRuns(args = {}) {
+    return this.stores.workflows.getWorkflowRuns(args);
   }
   async getWorkflowRunById({
     runId,