@mastra/mssql 0.0.0-fix-issue-10434-concurrent-write-corruption-20251124213939 → 0.0.0-fix-request-context-as-query-key-20251209093005

package/dist/index.js

@@ -1,5 +1,5 @@
 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_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, createStorageErrorId, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import sql2 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
 import { parseSqlIdentifier } from '@mastra/core/utils';
@@ -26,24 +26,62 @@ function buildDateRangeFilter(dateRange, fieldName) {
   }
   return filters;
 }
+function isInOperator(value) {
+  return typeof value === "object" && value !== null && "$in" in value && Array.isArray(value.$in);
+}
 function prepareWhereClause(filters, _schema) {
   const conditions = [];
   const params = {};
   let paramIndex = 1;
   Object.entries(filters).forEach(([key, value]) => {
     if (value === void 0) return;
-    const paramName = `p${paramIndex++}`;
     if (key.endsWith("_gte")) {
+      const paramName = `p${paramIndex++}`;
       const fieldName = key.slice(0, -4);
       conditions.push(`[${parseSqlIdentifier(fieldName, "field name")}] >= @${paramName}`);
       params[paramName] = value instanceof Date ? value.toISOString() : value;
     } else if (key.endsWith("_lte")) {
+      const paramName = `p${paramIndex++}`;
       const fieldName = key.slice(0, -4);
       conditions.push(`[${parseSqlIdentifier(fieldName, "field name")}] <= @${paramName}`);
       params[paramName] = value instanceof Date ? value.toISOString() : value;
     } else if (value === null) {
       conditions.push(`[${parseSqlIdentifier(key, "field name")}] IS NULL`);
+    } else if (isInOperator(value)) {
+      const inValues = value.$in;
+      if (inValues.length === 0) {
+        conditions.push("1 = 0");
+      } else if (inValues.length === 1) {
+        const paramName = `p${paramIndex++}`;
+        conditions.push(`[${parseSqlIdentifier(key, "field name")}] = @${paramName}`);
+        params[paramName] = inValues[0] instanceof Date ? inValues[0].toISOString() : inValues[0];
+      } else {
+        const inParamNames = [];
+        for (const item of inValues) {
+          const paramName = `p${paramIndex++}`;
+          inParamNames.push(`@${paramName}`);
+          params[paramName] = item instanceof Date ? item.toISOString() : item;
+        }
+        conditions.push(`[${parseSqlIdentifier(key, "field name")}] IN (${inParamNames.join(", ")})`);
+      }
+    } else if (Array.isArray(value)) {
+      if (value.length === 0) {
+        conditions.push("1 = 0");
+      } else if (value.length === 1) {
+        const paramName = `p${paramIndex++}`;
+        conditions.push(`[${parseSqlIdentifier(key, "field name")}] = @${paramName}`);
+        params[paramName] = value[0] instanceof Date ? value[0].toISOString() : value[0];
+      } else {
+        const inParamNames = [];
+        for (const item of value) {
+          const paramName = `p${paramIndex++}`;
+          inParamNames.push(`@${paramName}`);
+          params[paramName] = item instanceof Date ? item.toISOString() : item;
+        }
+        conditions.push(`[${parseSqlIdentifier(key, "field name")}] IN (${inParamNames.join(", ")})`);
+      }
     } else {
+      const paramName = `p${paramIndex++}`;
       conditions.push(`[${parseSqlIdentifier(key, "field name")}] = @${paramName}`);
       params[paramName] = value instanceof Date ? value.toISOString() : value;
     }
@@ -137,7 +175,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREAD_BY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -152,7 +190,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
     if (page < 0) {
       throw new MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        id: createStorageErrorId("MSSQL", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: "Page number must be non-negative",
@@ -211,7 +249,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -264,7 +302,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_THREAD_FAILED",
+          id: createStorageErrorId("MSSQL", "SAVE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -286,7 +324,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     const existingThread = await this.getThreadById({ threadId: id });
     if (!existingThread) {
       throw new MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+        id: createStorageErrorId("MSSQL", "UPDATE_THREAD", "NOT_FOUND"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: `Thread ${id} not found`,
@@ -321,7 +359,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       }
       if (!thread) {
         throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_THREAD", "NOT_FOUND"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           text: `Thread ${id} not found after update`,
@@ -340,7 +378,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -370,7 +408,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       });
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_THREAD_FAILED",
+          id: createStorageErrorId("MSSQL", "DELETE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -381,23 +419,18 @@ var MemoryMSSQL = class extends MemoryStorage {
       );
     }
   }
-  async _getIncludedMessages({
-    threadId,
-    include
-  }) {
-    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    if (!include) return null;
+  async _getIncludedMessages({ include }) {
+    if (!include || include.length === 0) return null;
     const unionQueries = [];
     const paramValues = [];
     let paramIdx = 1;
     const paramNames = [];
+    const tableName = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
     for (const inc of include) {
       const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
-      const searchId = inc.threadId || threadId;
-      const pThreadId = `@p${paramIdx}`;
-      const pId = `@p${paramIdx + 1}`;
-      const pPrev = `@p${paramIdx + 2}`;
-      const pNext = `@p${paramIdx + 3}`;
+      const pId = `@p${paramIdx}`;
+      const pPrev = `@p${paramIdx + 1}`;
+      const pNext = `@p${paramIdx + 2}`;
       unionQueries.push(
         `
           SELECT
@@ -411,16 +444,16 @@ var MemoryMSSQL = class extends MemoryStorage {
             m.seq_id
           FROM (
             SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
-            FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
-            WHERE [thread_id] = ${pThreadId}
+            FROM ${tableName}
+            WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId})
           ) AS m
           WHERE m.id = ${pId}
           OR EXISTS (
             SELECT 1
             FROM (
               SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
-              FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
-              WHERE [thread_id] = ${pThreadId}
+              FROM ${tableName}
+              WHERE [thread_id] = (SELECT thread_id FROM ${tableName} WHERE id = ${pId})
             ) AS target
             WHERE target.id = ${pId}
             AND (
@@ -433,9 +466,9 @@ var MemoryMSSQL = class extends MemoryStorage {
           )
         `
       );
-      paramValues.push(searchId, id, withPreviousMessages, withNextMessages);
-      paramNames.push(`p${paramIdx}`, `p${paramIdx + 1}`, `p${paramIdx + 2}`, `p${paramIdx + 3}`);
-      paramIdx += 4;
+      paramValues.push(id, withPreviousMessages, withNextMessages);
+      paramNames.push(`p${paramIdx}`, `p${paramIdx + 1}`, `p${paramIdx + 2}`);
+      paramIdx += 3;
     }
     const finalQuery = `
       SELECT * FROM (
@@ -490,7 +523,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_BY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -506,25 +539,26 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    if (!threadId.trim()) {
+    const threadIds = Array.isArray(threadId) ? threadId : [threadId];
+    if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
-          id: "STORAGE_MSSQL_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("MSSQL", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     if (page < 0) {
       throw new MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        id: createStorageErrorId("MSSQL", "LIST_MESSAGES", "INVALID_PAGE"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.USER,
         text: "Page number must be non-negative",
         details: {
-          threadId,
+          threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
           page
         }
       });
@@ -537,7 +571,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       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,
+        thread_id: threadIds.length === 1 ? threadIds[0] : { $in: threadIds },
         ...resourceId ? { resourceId } : {},
         ...buildDateRangeFilter(filter?.dateRange, "createdAt")
       };
@@ -581,7 +615,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       }
       if (include?.length) {
         const messageIds = new Set(messages.map((m) => m.id));
-        const includeMessages = await this._getIncludedMessages({ threadId, include });
+        const includeMessages = await this._getIncludedMessages({ include });
         includeMessages?.forEach((msg) => {
           if (!messageIds.has(msg.id)) {
             messages.push(msg);
@@ -604,7 +638,8 @@ var MemoryMSSQL = class extends MemoryStorage {
         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 threadIdSet = new Set(threadIds);
+      const returnedThreadMessageCount = finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).length;
       const hasMore = perPageInput !== false && returnedThreadMessageCount < total && offset + perPage < total;
       return {
         messages: finalMessages,
@@ -616,11 +651,11 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -642,7 +677,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     const threadId = messages[0]?.threadId;
     if (!threadId) {
       throw new MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        id: createStorageErrorId("MSSQL", "SAVE_MESSAGES", "INVALID_THREAD_ID"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.THIRD_PARTY,
         text: `Thread ID is required`
@@ -651,7 +686,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     const thread = await this.getThreadById({ threadId });
     if (!thread) {
       throw new MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        id: createStorageErrorId("MSSQL", "SAVE_MESSAGES", "THREAD_NOT_FOUND"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.THIRD_PARTY,
         text: `Thread ${threadId} not found`,
@@ -724,7 +759,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+          id: createStorageErrorId("MSSQL", "SAVE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -815,7 +850,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       await transaction.rollback();
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_MESSAGES_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -877,7 +912,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_MESSAGES_FAILED",
+          id: createStorageErrorId("MSSQL", "DELETE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { messageIds: messageIds.join(", ") }
@@ -905,7 +940,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_GET_RESOURCE_BY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -972,7 +1007,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_RESOURCE_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1019,7 +1054,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_CREATE_SPAN_FAILED",
+          id: createStorageErrorId("MSSQL", "CREATE_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1065,7 +1100,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_TRACE_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_TRACE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1097,7 +1132,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_UPDATE_SPAN_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1134,7 +1169,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         name = `agent run: '${entityId}'`;
       } else {
         const error = new MastraError({
-          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_TRACES_PAGINATED", "INVALID_ENTITY_TYPE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1203,7 +1238,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_TRACES_PAGINATED", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1227,7 +1262,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_CREATE_SPANS_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_CREATE_SPANS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1263,7 +1298,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_UPDATE_SPANS_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_UPDATE_SPANS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1287,7 +1322,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_DELETE_TRACES_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_DELETE_TRACES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1327,7 +1362,7 @@ var StoreOperationsMSSQL = class extends StoreOperations {
         return "BIT";
       default:
         throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_STORE_TYPE_NOT_SUPPORTED",
+          id: createStorageErrorId("MSSQL", "TYPE", "NOT_SUPPORTED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         });
@@ -1413,7 +1448,7 @@ var StoreOperationsMSSQL = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INSERT_FAILED",
+          id: createStorageErrorId("MSSQL", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1439,7 +1474,7 @@ var StoreOperationsMSSQL = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CLEAR_TABLE_FAILED",
+          id: createStorageErrorId("MSSQL", "CLEAR_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1542,7 +1577,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_TABLE_FAILED",
+          id: createStorageErrorId("MSSQL", "CREATE_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1602,7 +1637,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_ALTER_TABLE_FAILED",
+          id: createStorageErrorId("MSSQL", "ALTER_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1643,7 +1678,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_FAILED",
+          id: createStorageErrorId("MSSQL", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1666,7 +1701,7 @@ ${columns}
       await transaction.rollback();
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_INSERT_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1685,7 +1720,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DROP_TABLE_FAILED",
+          id: createStorageErrorId("MSSQL", "DROP_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1771,7 +1806,7 @@ ${columns}
     try {
       if (!data || Object.keys(data).length === 0) {
         throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_DATA",
+          id: createStorageErrorId("MSSQL", "UPDATE", "EMPTY_DATA"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           text: "Cannot update with empty data payload"
@@ -1779,7 +1814,7 @@ ${columns}
       }
       if (!keys || Object.keys(keys).length === 0) {
         throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_KEYS",
+          id: createStorageErrorId("MSSQL", "UPDATE", "EMPTY_KEYS"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           text: "Cannot update without keys to identify records"
@@ -1820,7 +1855,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1849,7 +1884,7 @@ ${columns}
       await transaction.rollback();
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_UPDATE_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_UPDATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1898,7 +1933,7 @@ ${columns}
       await transaction.rollback();
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_DELETE_FAILED",
+          id: createStorageErrorId("MSSQL", "BATCH_DELETE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1955,7 +1990,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_CREATE_FAILED",
+          id: createStorageErrorId("MSSQL", "INDEX_CREATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -1991,7 +2026,7 @@ ${columns}
       if (result.recordset.length > 1) {
         const tables = result.recordset.map((r) => r.table_name).join(", ");
         throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_INDEX_AMBIGUOUS",
+          id: createStorageErrorId("MSSQL", "INDEX", "AMBIGUOUS"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           text: `Index "${indexNameSafe}" exists on multiple tables (${tables}) in schema "${schemaName}". Please drop indexes manually or ensure unique index names.`
@@ -2007,7 +2042,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_DROP_FAILED",
+          id: createStorageErrorId("MSSQL", "INDEX_DROP", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2091,7 +2126,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_LIST_FAILED",
+          id: createStorageErrorId("MSSQL", "INDEX_LIST", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: tableName ? {
@@ -2164,7 +2199,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_DESCRIBE_FAILED",
+          id: createStorageErrorId("MSSQL", "INDEX_DESCRIBE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2245,7 +2280,7 @@ ${columns}
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_PERFORMANCE_INDEXES_FAILED",
+          id: createStorageErrorId("MSSQL", "CREATE_PERFORMANCE_INDEXES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -2255,20 +2290,9 @@ ${columns}
   }
 };
 function transformScoreRow(row) {
-  return {
-    ...row,
-    input: safelyParseJSON(row.input),
-    scorer: safelyParseJSON(row.scorer),
-    preprocessStepResult: safelyParseJSON(row.preprocessStepResult),
-    analyzeStepResult: safelyParseJSON(row.analyzeStepResult),
-    metadata: safelyParseJSON(row.metadata),
-    output: safelyParseJSON(row.output),
-    additionalContext: safelyParseJSON(row.additionalContext),
-    requestContext: safelyParseJSON(row.requestContext),
-    entity: safelyParseJSON(row.entity),
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt
-  };
+  return transformScoreRow$1(row, {
+    convertTimestamps: true
+  });
 }
 var ScoresMSSQL = class extends ScoresStorage {
   pool;
@@ -2298,7 +2322,7 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORE_BY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_SCORE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { id }
@@ -2314,15 +2338,23 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_VALIDATION_FAILED",
+          id: createStorageErrorId("MSSQL", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.USER,
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
     try {
       const scoreId = randomUUID();
+      const now = /* @__PURE__ */ new Date();
       const {
         scorer,
         preprocessStepResult,
@@ -2349,16 +2381,15 @@ var ScoresMSSQL = class extends ScoresStorage {
           requestContext: requestContext || null,
           entity: entity || null,
           scorer: scorer || null,
-          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+          createdAt: now.toISOString(),
+          updatedAt: now.toISOString()
         }
       });
-      const scoreFromDb = await this.getScoreById({ id: scoreId });
-      return { score: scoreFromDb };
+      return { score: { ...validatedScore, id: scoreId, createdAt: now, updatedAt: now } };
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_FAILED",
+          id: createStorageErrorId("MSSQL", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -2436,7 +2467,7 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_SCORES_BY_SCORER_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { scorerId }
@@ -2491,7 +2522,7 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_RUN_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_SCORES_BY_RUN_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { runId }
@@ -2549,7 +2580,7 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { entityId, entityType }
@@ -2607,7 +2638,7 @@ var ScoresMSSQL = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SPAN_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_SCORES_BY_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { traceId, spanId }
@@ -2711,7 +2742,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       }
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_RESULTS_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2749,7 +2780,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         await transaction.rollback();
         throw new MastraError(
           {
-            id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_SNAPSHOT_NOT_FOUND",
+            id: createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_STATE", "SNAPSHOT_NOT_FOUND"),
             domain: ErrorDomain.STORAGE,
             category: ErrorCategory.SYSTEM,
             details: {
@@ -2776,9 +2807,10 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         await transaction.rollback();
       } catch {
       }
+      if (error instanceof MastraError) throw error;
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_FAILED",
+          id: createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_STATE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2819,7 +2851,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("MSSQL", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2850,7 +2882,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: createStorageErrorId("MSSQL", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2890,7 +2922,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: createStorageErrorId("MSSQL", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2970,7 +3002,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_WORKFLOW_RUNS_FAILED",
+          id: createStorageErrorId("MSSQL", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
@@ -2993,7 +3025,7 @@ var MSSQLStore = class extends MastraStorage {
     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" });
+    super({ id: config.id, name: "MSSQLStore", disableInit: config.disableInit });
     try {
       if ("connectionString" in config) {
         if (!config.connectionString || typeof config.connectionString !== "string" || config.connectionString.trim() === "") {
@@ -3031,7 +3063,7 @@ var MSSQLStore = class extends MastraStorage {
     } catch (e) {
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INITIALIZATION_FAILED",
+          id: createStorageErrorId("MSSQL", "INITIALIZATION", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -3055,7 +3087,7 @@ var MSSQLStore = class extends MastraStorage {
       this.isConnected = null;
       throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INIT_FAILED",
+          id: createStorageErrorId("MSSQL", "INIT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -3223,7 +3255,7 @@ var MSSQLStore = class extends MastraStorage {
   getObservabilityStore() {
     if (!this.stores.observability) {
       throw new MastraError({
-        id: "MSSQL_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: createStorageErrorId("MSSQL", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: ErrorDomain.STORAGE,
         category: ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -3277,8 +3309,8 @@ var MSSQLStore = class extends MastraStorage {
       source: _source
     });
   }
-  async saveScore(_score) {
-    return this.stores.scores.saveScore(_score);
+  async saveScore(score) {
+    return this.stores.scores.saveScore(score);
   }
   async listScoresByRunId({
     runId: _runId,