@mastra/mssql 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/index.cjs

@@ -32,24 +32,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(`[${utils.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(`[${utils.parseSqlIdentifier(fieldName, "field name")}] <= @${paramName}`);
       params[paramName] = value instanceof Date ? value.toISOString() : value;
     } else if (value === null) {
       conditions.push(`[${utils.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(`[${utils.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(`[${utils.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(`[${utils.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(`[${utils.parseSqlIdentifier(key, "field name")}] IN (${inParamNames.join(", ")})`);
+      }
     } else {
+      const paramName = `p${paramIndex++}`;
       conditions.push(`[${utils.parseSqlIdentifier(key, "field name")}] = @${paramName}`);
       params[paramName] = value instanceof Date ? value.toISOString() : value;
     }
@@ -143,7 +181,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREAD_BY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_THREAD_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -158,7 +196,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
     if (page < 0) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        id: storage.createStorageErrorId("MSSQL", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: "Page number must be non-negative",
@@ -217,7 +255,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -270,7 +308,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_THREAD_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "SAVE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -292,7 +330,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     const existingThread = await this.getThreadById({ threadId: id });
     if (!existingThread) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+        id: storage.createStorageErrorId("MSSQL", "UPDATE_THREAD", "NOT_FOUND"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: `Thread ${id} not found`,
@@ -327,7 +365,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
       }
       if (!thread) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_THREAD", "NOT_FOUND"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           text: `Thread ${id} not found after update`,
@@ -346,7 +384,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -376,7 +414,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
       });
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_THREAD_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "DELETE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -387,23 +425,18 @@ var MemoryMSSQL = class extends storage.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: storage.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
@@ -417,16 +450,16 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
             m.seq_id
           FROM (
             SELECT *, ROW_NUMBER() OVER (ORDER BY [createdAt] ASC) as row_num
-            FROM ${getTableName({ indexName: storage.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: storage.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 (
@@ -439,9 +472,9 @@ var MemoryMSSQL = class extends storage.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 (
@@ -496,7 +529,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_BY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -512,25 +545,26 @@ var MemoryMSSQL = class extends storage.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 error.MastraError(
         {
-          id: "STORAGE_MSSQL_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: storage.createStorageErrorId("MSSQL", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: error.ErrorDomain.STORAGE,
           category: error.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 error.MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_INVALID_PAGE",
+        id: storage.createStorageErrorId("MSSQL", "LIST_MESSAGES", "INVALID_PAGE"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: "Page number must be non-negative",
         details: {
-          threadId,
+          threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
           page
         }
       });
@@ -543,7 +577,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
       const tableName = getTableName({ indexName: storage.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")
       };
@@ -587,7 +621,7 @@ var MemoryMSSQL = class extends storage.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);
@@ -610,7 +644,8 @@ var MemoryMSSQL = class extends storage.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,
@@ -622,11 +657,11 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -648,7 +683,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     const threadId = messages[0]?.threadId;
     if (!threadId) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        id: storage.createStorageErrorId("MSSQL", "SAVE_MESSAGES", "INVALID_THREAD_ID"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.THIRD_PARTY,
         text: `Thread ID is required`
@@ -657,7 +692,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     const thread = await this.getThreadById({ threadId });
     if (!thread) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        id: storage.createStorageErrorId("MSSQL", "SAVE_MESSAGES", "THREAD_NOT_FOUND"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.THIRD_PARTY,
         text: `Thread ${threadId} not found`,
@@ -730,7 +765,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "SAVE_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { threadId }
@@ -821,7 +856,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
       await transaction.rollback();
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -883,7 +918,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "DELETE_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { messageIds: messageIds.join(", ") }
@@ -911,7 +946,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_GET_RESOURCE_BY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -978,7 +1013,7 @@ var MemoryMSSQL = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_RESOURCE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_RESOURCE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { resourceId }
@@ -1025,7 +1060,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_CREATE_SPAN_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "CREATE_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1071,7 +1106,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_GET_TRACE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_TRACE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1103,7 +1138,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_UPDATE_SPAN_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1140,7 +1175,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
         name = `agent run: '${entityId}'`;
       } else {
         const error$1 = new error.MastraError({
-          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_TRACES_PAGINATED", "INVALID_ENTITY_TYPE"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1209,7 +1244,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_TRACES_PAGINATED", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -1233,7 +1268,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_BATCH_CREATE_SPANS_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_CREATE_SPANS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1269,7 +1304,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_BATCH_UPDATE_SPANS_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_UPDATE_SPANS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1293,7 +1328,7 @@ var ObservabilityMSSQL = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MSSQL_STORE_BATCH_DELETE_TRACES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_DELETE_TRACES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -1333,7 +1368,7 @@ var StoreOperationsMSSQL = class extends storage.StoreOperations {
         return "BIT";
       default:
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_MSSQL_STORE_TYPE_NOT_SUPPORTED",
+          id: storage.createStorageErrorId("MSSQL", "TYPE", "NOT_SUPPORTED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         });
@@ -1419,7 +1454,7 @@ var StoreOperationsMSSQL = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INSERT_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INSERT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1445,7 +1480,7 @@ var StoreOperationsMSSQL = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CLEAR_TABLE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "CLEAR_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1548,7 +1583,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_TABLE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "CREATE_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1608,7 +1643,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_ALTER_TABLE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "ALTER_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1649,7 +1684,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LOAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1672,7 +1707,7 @@ ${columns}
       await transaction.rollback();
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_INSERT_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_INSERT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1691,7 +1726,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_DROP_TABLE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "DROP_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1777,7 +1812,7 @@ ${columns}
     try {
       if (!data || Object.keys(data).length === 0) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_DATA",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE", "EMPTY_DATA"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           text: "Cannot update with empty data payload"
@@ -1785,7 +1820,7 @@ ${columns}
       }
       if (!keys || Object.keys(keys).length === 0) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_KEYS",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE", "EMPTY_KEYS"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           text: "Cannot update without keys to identify records"
@@ -1826,7 +1861,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1855,7 +1890,7 @@ ${columns}
       await transaction.rollback();
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_UPDATE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_UPDATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1904,7 +1939,7 @@ ${columns}
       await transaction.rollback();
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_DELETE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "BATCH_DELETE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1961,7 +1996,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_CREATE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INDEX_CREATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1997,7 +2032,7 @@ ${columns}
       if (result.recordset.length > 1) {
         const tables = result.recordset.map((r) => r.table_name).join(", ");
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_MSSQL_INDEX_AMBIGUOUS",
+          id: storage.createStorageErrorId("MSSQL", "INDEX", "AMBIGUOUS"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           text: `Index "${indexNameSafe}" exists on multiple tables (${tables}) in schema "${schemaName}". Please drop indexes manually or ensure unique index names.`
@@ -2013,7 +2048,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_DROP_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INDEX_DROP", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2097,7 +2132,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_LIST_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INDEX_LIST", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: tableName ? {
@@ -2170,7 +2205,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_INDEX_DESCRIBE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INDEX_DESCRIBE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2251,7 +2286,7 @@ ${columns}
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_PERFORMANCE_INDEXES_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "CREATE_PERFORMANCE_INDEXES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -2261,20 +2296,9 @@ ${columns}
   }
 };
 function transformScoreRow(row) {
-  return {
-    ...row,
-    input: storage.safelyParseJSON(row.input),
-    scorer: storage.safelyParseJSON(row.scorer),
-    preprocessStepResult: storage.safelyParseJSON(row.preprocessStepResult),
-    analyzeStepResult: storage.safelyParseJSON(row.analyzeStepResult),
-    metadata: storage.safelyParseJSON(row.metadata),
-    output: storage.safelyParseJSON(row.output),
-    additionalContext: storage.safelyParseJSON(row.additionalContext),
-    requestContext: storage.safelyParseJSON(row.requestContext),
-    entity: storage.safelyParseJSON(row.entity),
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt
-  };
+  return storage.transformScoreRow(row, {
+    convertTimestamps: true
+  });
 }
 var ScoresMSSQL = class extends storage.ScoresStorage {
   pool;
@@ -2304,7 +2328,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORE_BY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_SCORE_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { id }
@@ -2320,7 +2344,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_VALIDATION_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -2364,7 +2388,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "SAVE_SCORE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -2442,7 +2466,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_SCORES_BY_SCORER_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { scorerId }
@@ -2497,7 +2521,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_RUN_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_SCORES_BY_RUN_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { runId }
@@ -2555,7 +2579,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { entityId, entityType }
@@ -2613,7 +2637,7 @@ var ScoresMSSQL = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SPAN_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_SCORES_BY_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { traceId, spanId }
@@ -2717,7 +2741,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
       }
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_RESULTS_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2755,7 +2779,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
         await transaction.rollback();
         throw new error.MastraError(
           {
-            id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_SNAPSHOT_NOT_FOUND",
+            id: storage.createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_STATE", "SNAPSHOT_NOT_FOUND"),
             domain: error.ErrorDomain.STORAGE,
             category: error.ErrorCategory.SYSTEM,
             details: {
@@ -2782,9 +2806,10 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
         await transaction.rollback();
       } catch {
       }
+      if (error$1 instanceof error.MastraError) throw error$1;
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "UPDATE_WORKFLOW_STATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2825,7 +2850,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2856,7 +2881,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2896,7 +2921,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2976,7 +3001,7 @@ var WorkflowsMSSQL = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_LIST_WORKFLOW_RUNS_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2999,7 +3024,7 @@ var MSSQLStore = class extends storage.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() === "") {
@@ -3037,7 +3062,7 @@ var MSSQLStore = class extends storage.MastraStorage {
     } catch (e) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INITIALIZATION_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INITIALIZATION", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -3061,7 +3086,7 @@ var MSSQLStore = class extends storage.MastraStorage {
       this.isConnected = null;
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INIT_FAILED",
+          id: storage.createStorageErrorId("MSSQL", "INIT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3229,7 +3254,7 @@ var MSSQLStore = class extends storage.MastraStorage {
   getObservabilityStore() {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "MSSQL_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("MSSQL", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"