@mastra/mssql 0.0.0-add-libsql-changeset-20250910154739

package/dist/index.js

@@ -0,0 +1,2361 @@
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
+import { MastraStorage, LegacyEvalsStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, ScoresStorage, TABLE_SCORERS, TracesStorage, TABLE_TRACES, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, TABLE_EVALS, TABLE_THREADS, TABLE_MESSAGES } from '@mastra/core/storage';
+import sql2 from 'mssql';
+import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
+import { MessageList } from '@mastra/core/agent';
+
+// src/storage/index.ts
+function getSchemaName(schema) {
+  return schema ? `[${parseSqlIdentifier(schema, "schema name")}]` : void 0;
+}
+function getTableName({ indexName, schemaName }) {
+  const parsedIndexName = parseSqlIdentifier(indexName, "index name");
+  const quotedIndexName = `[${parsedIndexName}]`;
+  const quotedSchemaName = schemaName;
+  return quotedSchemaName ? `${quotedSchemaName}.${quotedIndexName}` : quotedIndexName;
+}
+
+// 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.test_info) {
+    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 [];
+      }
+      console.error("Failed to get evals for the specified agent: " + error?.message);
+      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, sql2.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, sql2.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;
+  operations;
+  _parseAndFormatMessages(messages, format) {
+    const messagesWithParsedContent = messages.map((message) => {
+      if (typeof message.content === "string") {
+        try {
+          return { ...message, content: JSON.parse(message.content) };
+        } catch {
+          return message;
+        }
+      }
+      return message;
+    });
+    const cleanMessages = messagesWithParsedContent.map(({ seq_id, ...rest }) => rest);
+    const list = new MessageList().add(cleanMessages, "memory");
+    return format === "v2" ? list.get.all.v2() : list.get.all.v1();
+  }
+  constructor({
+    pool,
+    schema,
+    operations
+  }) {
+    super();
+    this.pool = pool;
+    this.schema = schema;
+    this.operations = operations;
+  }
+  async getThreadById({ threadId }) {
+    try {
+      const sql7 = `SELECT 
+        id,
+        [resourceId],
+        title,
+        metadata,
+        [createdAt],
+        [updatedAt]
+      FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })}
+      WHERE id = @threadId`;
+      const request = this.pool.request();
+      request.input("threadId", threadId);
+      const resultSet = await request.query(sql7);
+      const thread = resultSet.recordset[0] || null;
+      if (!thread) {
+        return null;
+      }
+      return {
+        ...thread,
+        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+        createdAt: thread.createdAt,
+        updatedAt: thread.updatedAt
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREAD_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId
+          }
+        },
+        error
+      );
+    }
+  }
+  async getThreadsByResourceIdPaginated(args) {
+    const { resourceId, page = 0, perPage: perPageInput, orderBy = "createdAt", sortDirection = "DESC" } = args;
+    try {
+      const perPage = perPageInput !== void 0 ? perPageInput : 100;
+      const currentOffset = page * perPage;
+      const baseQuery = `FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} WHERE [resourceId] = @resourceId`;
+      const countQuery = `SELECT COUNT(*) as count ${baseQuery}`;
+      const countRequest = this.pool.request();
+      countRequest.input("resourceId", resourceId);
+      const countResult = await countRequest.query(countQuery);
+      const total = parseInt(countResult.recordset[0]?.count ?? "0", 10);
+      if (total === 0) {
+        return {
+          threads: [],
+          total: 0,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
+      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${sortDirection} OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
+      const dataRequest = this.pool.request();
+      dataRequest.input("resourceId", resourceId);
+      dataRequest.input("perPage", perPage);
+      dataRequest.input("offset", currentOffset);
+      const rowsResult = await dataRequest.query(dataQuery);
+      const rows = rowsResult.recordset || [];
+      const threads = rows.map((thread) => ({
+        ...thread,
+        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+        createdAt: thread.createdAt,
+        updatedAt: thread.updatedAt
+      }));
+      return {
+        threads,
+        total,
+        page,
+        perPage,
+        hasMore: currentOffset + threads.length < total
+      };
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            resourceId,
+            page
+          }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException?.(mastraError);
+      return { threads: [], total: 0, page, perPage: perPageInput || 100, hasMore: false };
+    }
+  }
+  async saveThread({ thread }) {
+    try {
+      const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+      const mergeSql = `MERGE INTO ${table} WITH (HOLDLOCK) AS target
+        USING (SELECT @id AS id) AS source
+        ON (target.id = source.id)
+        WHEN MATCHED THEN
+          UPDATE SET
+            [resourceId] = @resourceId,
+            title = @title,
+            metadata = @metadata,
+            [updatedAt] = @updatedAt
+        WHEN NOT MATCHED THEN
+          INSERT (id, [resourceId], title, metadata, [createdAt], [updatedAt])
+          VALUES (@id, @resourceId, @title, @metadata, @createdAt, @updatedAt);`;
+      const req = this.pool.request();
+      req.input("id", thread.id);
+      req.input("resourceId", thread.resourceId);
+      req.input("title", thread.title);
+      req.input("metadata", thread.metadata ? JSON.stringify(thread.metadata) : null);
+      req.input("createdAt", sql2.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql2.DateTime2, thread.updatedAt);
+      await req.query(mergeSql);
+      return thread;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId: thread.id
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * @deprecated use getThreadsByResourceIdPaginated instead
+   */
+  async getThreadsByResourceId(args) {
+    const { resourceId, orderBy = "createdAt", sortDirection = "DESC" } = args;
+    try {
+      const baseQuery = `FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} WHERE [resourceId] = @resourceId`;
+      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
+      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${sortDirection}`;
+      const request = this.pool.request();
+      request.input("resourceId", resourceId);
+      const resultSet = await request.query(dataQuery);
+      const rows = resultSet.recordset || [];
+      return rows.map((thread) => ({
+        ...thread,
+        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+        createdAt: thread.createdAt,
+        updatedAt: thread.updatedAt
+      }));
+    } catch (error) {
+      this.logger?.error?.(`Error getting threads for resource ${resourceId}:`, error);
+      return [];
+    }
+  }
+  /**
+   * Updates a thread's title and metadata, merging with existing metadata. Returns the updated thread.
+   */
+  async updateThread({
+    id,
+    title,
+    metadata
+  }) {
+    const existingThread = await this.getThreadById({ threadId: id });
+    if (!existingThread) {
+      throw new MastraError({
+        id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: `Thread ${id} not found`,
+        details: {
+          threadId: id,
+          title
+        }
+      });
+    }
+    const mergedMetadata = {
+      ...existingThread.metadata,
+      ...metadata
+    };
+    try {
+      const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+      const sql7 = `UPDATE ${table}
+        SET title = @title,
+            metadata = @metadata,
+            [updatedAt] = @updatedAt
+        OUTPUT INSERTED.*
+        WHERE id = @id`;
+      const req = this.pool.request();
+      req.input("id", id);
+      req.input("title", title);
+      req.input("metadata", JSON.stringify(mergedMetadata));
+      req.input("updatedAt", /* @__PURE__ */ new Date());
+      const result = await req.query(sql7);
+      let thread = result.recordset && result.recordset[0];
+      if (thread && "seq_id" in thread) {
+        const { seq_id, ...rest } = thread;
+        thread = rest;
+      }
+      if (!thread) {
+        throw new MastraError({
+          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: `Thread ${id} not found after update`,
+          details: {
+            threadId: id,
+            title
+          }
+        });
+      }
+      return {
+        ...thread,
+        metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+        createdAt: thread.createdAt,
+        updatedAt: thread.updatedAt
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId: id,
+            title
+          }
+        },
+        error
+      );
+    }
+  }
+  async deleteThread({ threadId }) {
+    const messagesTable = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
+    const threadsTable = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+    const deleteMessagesSql = `DELETE FROM ${messagesTable} WHERE [thread_id] = @threadId`;
+    const deleteThreadSql = `DELETE FROM ${threadsTable} WHERE id = @threadId`;
+    const tx = this.pool.transaction();
+    try {
+      await tx.begin();
+      const req = tx.request();
+      req.input("threadId", threadId);
+      await req.query(deleteMessagesSql);
+      await req.query(deleteThreadSql);
+      await tx.commit();
+    } catch (error) {
+      await tx.rollback().catch(() => {
+      });
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId
+          }
+        },
+        error
+      );
+    }
+  }
+  async _getIncludedMessages({
+    threadId,
+    selectBy,
+    orderByStatement
+  }) {
+    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+    const include = selectBy?.include;
+    if (!include) return null;
+    const unionQueries = [];
+    const paramValues = [];
+    let paramIdx = 1;
+    const paramNames = [];
+    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}`;
+      unionQueries.push(
+        `
+          SELECT
+            m.id, 
+            m.content, 
+            m.role, 
+            m.type,
+            m.[createdAt], 
+            m.thread_id AS threadId,
+            m.[resourceId],
+            m.seq_id
+          FROM (
+            SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
+            FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
+            WHERE [thread_id] = ${pThreadId}
+          ) AS m
+          WHERE m.id = ${pId}
+          OR EXISTS (
+            SELECT 1
+            FROM (
+              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 (
+              (m.row_num <= target.row_num + ${pPrev} AND m.row_num > target.row_num)
+              OR
+              (m.row_num >= target.row_num - ${pNext} AND m.row_num < target.row_num)
+            )
+          )
+        `
+      );
+      paramValues.push(searchId, id, withPreviousMessages, withNextMessages);
+      paramNames.push(`p${paramIdx}`, `p${paramIdx + 1}`, `p${paramIdx + 2}`, `p${paramIdx + 3}`);
+      paramIdx += 4;
+    }
+    const finalQuery = `
+      SELECT * FROM (
+        ${unionQueries.join(" UNION ALL ")}
+      ) AS union_result
+      ORDER BY [seq_id] ASC
+    `;
+    const req = this.pool.request();
+    for (let i = 0; i < paramValues.length; ++i) {
+      req.input(paramNames[i], paramValues[i]);
+    }
+    const result = await req.query(finalQuery);
+    const includedRows = result.recordset || [];
+    const seen = /* @__PURE__ */ new Set();
+    const dedupedRows = includedRows.filter((row) => {
+      if (seen.has(row.id)) return false;
+      seen.add(row.id);
+      return true;
+    });
+    return dedupedRows;
+  }
+  async getMessages(args) {
+    const { threadId, resourceId, format, selectBy } = args;
+    const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
+    const orderByStatement = `ORDER BY [seq_id] DESC`;
+    const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
+    try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+      let rows = [];
+      const include = selectBy?.include || [];
+      if (include?.length) {
+        const includeMessages = await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
+        if (includeMessages) {
+          rows.push(...includeMessages);
+        }
+      }
+      const excludeIds = rows.map((m) => m.id).filter(Boolean);
+      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [thread_id] = @threadId`;
+      const request = this.pool.request();
+      request.input("threadId", threadId);
+      if (excludeIds.length > 0) {
+        const excludeParams = excludeIds.map((_, idx) => `@id${idx}`);
+        query += ` AND id NOT IN (${excludeParams.join(", ")})`;
+        excludeIds.forEach((id, idx) => {
+          request.input(`id${idx}`, id);
+        });
+      }
+      query += ` ${orderByStatement} OFFSET 0 ROWS FETCH NEXT @limit ROWS ONLY`;
+      request.input("limit", limit);
+      const result = await request.query(query);
+      const remainingRows = result.recordset || [];
+      rows.push(...remainingRows);
+      rows.sort((a, b) => {
+        const timeDiff = a.seq_id - b.seq_id;
+        return timeDiff;
+      });
+      rows = rows.map(({ seq_id, ...rest }) => rest);
+      return this._parseAndFormatMessages(rows, format);
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return [];
+    }
+  }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
+    const orderByStatement = `ORDER BY [seq_id] DESC`;
+    try {
+      let rows = [];
+      let query = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} WHERE [id] IN (${messageIds.map((_, i) => `@id${i}`).join(", ")})`;
+      const request = this.pool.request();
+      messageIds.forEach((id, i) => request.input(`id${i}`, id));
+      query += ` ${orderByStatement}`;
+      const result = await request.query(query);
+      const remainingRows = result.recordset || [];
+      rows.push(...remainingRows);
+      rows.sort((a, b) => {
+        const timeDiff = a.seq_id - b.seq_id;
+        return timeDiff;
+      });
+      rows = rows.map(({ seq_id, ...rest }) => rest);
+      if (format === `v1`) return this._parseAndFormatMessages(rows, format);
+      return this._parseAndFormatMessages(rows, `v2`);
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return [];
+    }
+  }
+  async getMessagesPaginated(args) {
+    const { threadId, resourceId, format, selectBy } = args;
+    const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
+    try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+      const fromDate = dateRange?.start;
+      const toDate = dateRange?.end;
+      const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId, resourceId`;
+      const orderByStatement = `ORDER BY [seq_id] DESC`;
+      let messages = [];
+      if (selectBy?.include?.length) {
+        const includeMessages = await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
+        if (includeMessages) messages.push(...includeMessages);
+      }
+      const perPage = perPageInput !== void 0 ? perPageInput : resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
+      const currentOffset = page * perPage;
+      const conditions = ["[thread_id] = @threadId"];
+      const request = this.pool.request();
+      request.input("threadId", threadId);
+      if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
+        conditions.push("[createdAt] >= @fromDate");
+        request.input("fromDate", fromDate.toISOString());
+      }
+      if (toDate instanceof Date && !isNaN(toDate.getTime())) {
+        conditions.push("[createdAt] <= @toDate");
+        request.input("toDate", toDate.toISOString());
+      }
+      const whereClause = `WHERE ${conditions.join(" AND ")}`;
+      const countQuery = `SELECT COUNT(*) as total FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} ${whereClause}`;
+      const countResult = await request.query(countQuery);
+      const total = parseInt(countResult.recordset[0]?.total, 10) || 0;
+      if (total === 0 && messages.length > 0) {
+        const parsedIncluded = this._parseAndFormatMessages(messages, format);
+        return {
+          messages: parsedIncluded,
+          total: parsedIncluded.length,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      const excludeIds = messages.map((m) => m.id);
+      if (excludeIds.length > 0) {
+        const excludeParams = excludeIds.map((_, idx) => `@id${idx}`);
+        conditions.push(`id NOT IN (${excludeParams.join(", ")})`);
+        excludeIds.forEach((id, idx) => request.input(`id${idx}`, id));
+      }
+      const finalWhereClause = `WHERE ${conditions.join(" AND ")}`;
+      const dataQuery = `${selectStatement} FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} ${finalWhereClause} ${orderByStatement} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
+      request.input("offset", currentOffset);
+      request.input("limit", perPage);
+      const rowsResult = await request.query(dataQuery);
+      const rows = rowsResult.recordset || [];
+      rows.sort((a, b) => a.seq_id - b.seq_id);
+      messages.push(...rows);
+      const parsed = this._parseAndFormatMessages(messages, format);
+      return {
+        messages: parsed,
+        total: total + excludeIds.length,
+        page,
+        perPage,
+        hasMore: currentOffset + rows.length < total
+      };
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_MESSAGES_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId,
+            resourceId: resourceId ?? "",
+            page
+          }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return { messages: [], total: 0, page, perPage: perPageInput || 40, hasMore: false };
+    }
+  }
+  async saveMessages({
+    messages,
+    format
+  }) {
+    if (messages.length === 0) return messages;
+    const threadId = messages[0]?.threadId;
+    if (!threadId) {
+      throw new MastraError({
+        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY,
+        text: `Thread ID is required`
+      });
+    }
+    const thread = await this.getThreadById({ threadId });
+    if (!thread) {
+      throw new MastraError({
+        id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY,
+        text: `Thread ${threadId} not found`,
+        details: { threadId }
+      });
+    }
+    const tableMessages = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
+    const tableThreads = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+    try {
+      const transaction = this.pool.transaction();
+      await transaction.begin();
+      try {
+        for (const message of messages) {
+          if (!message.threadId) {
+            throw new Error(
+              `Expected to find a threadId for message, but couldn't find one. An unexpected error has occurred.`
+            );
+          }
+          if (!message.resourceId) {
+            throw new Error(
+              `Expected to find a resourceId for message, but couldn't find one. An unexpected error has occurred.`
+            );
+          }
+          const request = transaction.request();
+          request.input("id", message.id);
+          request.input("thread_id", message.threadId);
+          request.input(
+            "content",
+            typeof message.content === "string" ? message.content : JSON.stringify(message.content)
+          );
+          request.input("createdAt", sql2.DateTime2, message.createdAt);
+          request.input("role", message.role);
+          request.input("type", message.type || "v2");
+          request.input("resourceId", message.resourceId);
+          const mergeSql = `MERGE INTO ${tableMessages} AS target
+            USING (SELECT @id AS id) AS src
+            ON target.id = src.id
+            WHEN MATCHED THEN UPDATE SET
+              thread_id = @thread_id,
+              content = @content,
+              [createdAt] = @createdAt,
+              role = @role,
+              type = @type,
+              resourceId = @resourceId
+            WHEN NOT MATCHED THEN INSERT (id, thread_id, content, [createdAt], role, type, resourceId)
+              VALUES (@id, @thread_id, @content, @createdAt, @role, @type, @resourceId);`;
+          await request.query(mergeSql);
+        }
+        const threadReq = transaction.request();
+        threadReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+        threadReq.input("id", threadId);
+        await threadReq.query(`UPDATE ${tableThreads} SET [updatedAt] = @updatedAt WHERE id = @id`);
+        await transaction.commit();
+      } catch (error) {
+        await transaction.rollback();
+        throw error;
+      }
+      const messagesWithParsedContent = messages.map((message) => {
+        if (typeof message.content === "string") {
+          try {
+            return { ...message, content: JSON.parse(message.content) };
+          } catch {
+            return message;
+          }
+        }
+        return message;
+      });
+      const list = new MessageList().add(messagesWithParsedContent, "memory");
+      if (format === "v2") return list.get.all.v2();
+      return list.get.all.v1();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error
+      );
+    }
+  }
+  async updateMessages({
+    messages
+  }) {
+    if (!messages || messages.length === 0) {
+      return [];
+    }
+    const messageIds = messages.map((m) => m.id);
+    const idParams = messageIds.map((_, i) => `@id${i}`).join(", ");
+    let selectQuery = `SELECT id, content, role, type, createdAt, thread_id AS threadId, resourceId FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}`;
+    if (idParams.length > 0) {
+      selectQuery += ` WHERE id IN (${idParams})`;
+    } else {
+      return [];
+    }
+    const selectReq = this.pool.request();
+    messageIds.forEach((id, i) => selectReq.input(`id${i}`, id));
+    const existingMessagesDb = (await selectReq.query(selectQuery)).recordset;
+    if (!existingMessagesDb || existingMessagesDb.length === 0) {
+      return [];
+    }
+    const existingMessages = existingMessagesDb.map((msg) => {
+      if (typeof msg.content === "string") {
+        try {
+          msg.content = JSON.parse(msg.content);
+        } catch {
+        }
+      }
+      return msg;
+    });
+    const threadIdsToUpdate = /* @__PURE__ */ new Set();
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      for (const existingMessage of existingMessages) {
+        const updatePayload = messages.find((m) => m.id === existingMessage.id);
+        if (!updatePayload) continue;
+        const { id, ...fieldsToUpdate } = updatePayload;
+        if (Object.keys(fieldsToUpdate).length === 0) continue;
+        threadIdsToUpdate.add(existingMessage.threadId);
+        if (updatePayload.threadId && updatePayload.threadId !== existingMessage.threadId) {
+          threadIdsToUpdate.add(updatePayload.threadId);
+        }
+        const setClauses = [];
+        const req = transaction.request();
+        req.input("id", id);
+        const columnMapping = { threadId: "thread_id" };
+        const updatableFields = { ...fieldsToUpdate };
+        if (updatableFields.content) {
+          const newContent = {
+            ...existingMessage.content,
+            ...updatableFields.content,
+            ...existingMessage.content?.metadata && updatableFields.content.metadata ? { metadata: { ...existingMessage.content.metadata, ...updatableFields.content.metadata } } : {}
+          };
+          setClauses.push(`content = @content`);
+          req.input("content", JSON.stringify(newContent));
+          delete updatableFields.content;
+        }
+        for (const key in updatableFields) {
+          if (Object.prototype.hasOwnProperty.call(updatableFields, key)) {
+            const dbColumn = columnMapping[key] || key;
+            setClauses.push(`[${dbColumn}] = @${dbColumn}`);
+            req.input(dbColumn, updatableFields[key]);
+          }
+        }
+        if (setClauses.length > 0) {
+          const updateSql = `UPDATE ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} SET ${setClauses.join(", ")} WHERE id = @id`;
+          await req.query(updateSql);
+        }
+      }
+      if (threadIdsToUpdate.size > 0) {
+        const threadIdParams = Array.from(threadIdsToUpdate).map((_, i) => `@tid${i}`).join(", ");
+        const threadReq = transaction.request();
+        Array.from(threadIdsToUpdate).forEach((tid, i) => threadReq.input(`tid${i}`, tid));
+        threadReq.input("updatedAt", (/* @__PURE__ */ new Date()).toISOString());
+        const threadSql = `UPDATE ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} SET updatedAt = @updatedAt WHERE id IN (${threadIdParams})`;
+        await threadReq.query(threadSql);
+      }
+      await transaction.commit();
+    } catch (error) {
+      await transaction.rollback();
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_UPDATE_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+    const refetchReq = this.pool.request();
+    messageIds.forEach((id, i) => refetchReq.input(`id${i}`, id));
+    const updatedMessages = (await refetchReq.query(selectQuery)).recordset;
+    return (updatedMessages || []).map((message) => {
+      if (typeof message.content === "string") {
+        try {
+          message.content = JSON.parse(message.content);
+        } catch {
+        }
+      }
+      return message;
+    });
+  }
+  async deleteMessages(messageIds) {
+    if (!messageIds || messageIds.length === 0) {
+      return;
+    }
+    try {
+      const messageTableName = getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
+      const threadTableName = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+      const placeholders = messageIds.map((_, idx) => `@p${idx + 1}`).join(",");
+      const request = this.pool.request();
+      messageIds.forEach((id, idx) => {
+        request.input(`p${idx + 1}`, id);
+      });
+      const messages = await request.query(
+        `SELECT DISTINCT [thread_id] FROM ${messageTableName} WHERE [id] IN (${placeholders})`
+      );
+      const threadIds = messages.recordset?.map((msg) => msg.thread_id).filter(Boolean) || [];
+      const transaction = this.pool.transaction();
+      await transaction.begin();
+      try {
+        const deleteRequest = transaction.request();
+        messageIds.forEach((id, idx) => {
+          deleteRequest.input(`p${idx + 1}`, id);
+        });
+        await deleteRequest.query(`DELETE FROM ${messageTableName} WHERE [id] IN (${placeholders})`);
+        if (threadIds.length > 0) {
+          for (const threadId of threadIds) {
+            const updateRequest = transaction.request();
+            updateRequest.input("p1", threadId);
+            await updateRequest.query(`UPDATE ${threadTableName} SET [updatedAt] = GETDATE() WHERE [id] = @p1`);
+          }
+        }
+        await transaction.commit();
+      } catch (error) {
+        try {
+          await transaction.rollback();
+        } catch {
+        }
+        throw error;
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { messageIds: messageIds.join(", ") }
+        },
+        error
+      );
+    }
+  }
+  async getResourceById({ resourceId }) {
+    const tableName = getTableName({ indexName: TABLE_RESOURCES, schemaName: getSchemaName(this.schema) });
+    try {
+      const req = this.pool.request();
+      req.input("resourceId", resourceId);
+      const result = (await req.query(`SELECT * FROM ${tableName} WHERE id = @resourceId`)).recordset[0];
+      if (!result) {
+        return null;
+      }
+      return {
+        ...result,
+        workingMemory: typeof result.workingMemory === "object" ? JSON.stringify(result.workingMemory) : result.workingMemory,
+        metadata: typeof result.metadata === "string" ? JSON.parse(result.metadata) : result.metadata
+      };
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_GET_RESOURCE_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { resourceId }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      throw mastraError;
+    }
+  }
+  async saveResource({ resource }) {
+    await this.operations.insert({
+      tableName: TABLE_RESOURCES,
+      record: {
+        ...resource,
+        metadata: JSON.stringify(resource.metadata)
+      }
+    });
+    return resource;
+  }
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata
+  }) {
+    try {
+      const existingResource = await this.getResourceById({ resourceId });
+      if (!existingResource) {
+        const newResource = {
+          id: resourceId,
+          workingMemory,
+          metadata: metadata || {},
+          createdAt: /* @__PURE__ */ new Date(),
+          updatedAt: /* @__PURE__ */ new Date()
+        };
+        return this.saveResource({ resource: newResource });
+      }
+      const updatedResource = {
+        ...existingResource,
+        workingMemory: workingMemory !== void 0 ? workingMemory : existingResource.workingMemory,
+        metadata: {
+          ...existingResource.metadata,
+          ...metadata
+        },
+        updatedAt: /* @__PURE__ */ new Date()
+      };
+      const tableName = getTableName({ indexName: TABLE_RESOURCES, schemaName: getSchemaName(this.schema) });
+      const updates = [];
+      const req = this.pool.request();
+      if (workingMemory !== void 0) {
+        updates.push("workingMemory = @workingMemory");
+        req.input("workingMemory", workingMemory);
+      }
+      if (metadata) {
+        updates.push("metadata = @metadata");
+        req.input("metadata", JSON.stringify(updatedResource.metadata));
+      }
+      updates.push("updatedAt = @updatedAt");
+      req.input("updatedAt", updatedResource.updatedAt.toISOString());
+      req.input("id", resourceId);
+      await req.query(`UPDATE ${tableName} SET ${updates.join(", ")} WHERE id = @id`);
+      return updatedResource;
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_UPDATE_RESOURCE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { resourceId }
+        },
+        error
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      throw mastraError;
+    }
+  }
+};
+var StoreOperationsMSSQL = class extends StoreOperations {
+  pool;
+  schemaName;
+  setupSchemaPromise = null;
+  schemaSetupComplete = void 0;
+  getSqlType(type, isPrimaryKey = false) {
+    switch (type) {
+      case "text":
+        return isPrimaryKey ? "NVARCHAR(255)" : "NVARCHAR(MAX)";
+      case "timestamp":
+        return "DATETIME2(7)";
+      case "uuid":
+        return "UNIQUEIDENTIFIER";
+      case "jsonb":
+        return "NVARCHAR(MAX)";
+      case "integer":
+        return "INT";
+      case "bigint":
+        return "BIGINT";
+      case "float":
+        return "FLOAT";
+      default:
+        throw new MastraError({
+          id: "MASTRA_STORAGE_MSSQL_STORE_TYPE_NOT_SUPPORTED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        });
+    }
+  }
+  constructor({ pool, schemaName }) {
+    super();
+    this.pool = pool;
+    this.schemaName = schemaName;
+  }
+  async hasColumn(table, column) {
+    const schema = this.schemaName || "dbo";
+    const request = this.pool.request();
+    request.input("schema", schema);
+    request.input("table", table);
+    request.input("column", column);
+    request.input("columnLower", column.toLowerCase());
+    const result = await request.query(
+      `SELECT 1 FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @table AND (COLUMN_NAME = @column OR COLUMN_NAME = @columnLower)`
+    );
+    return result.recordset.length > 0;
+  }
+  async setupSchema() {
+    if (!this.schemaName || this.schemaSetupComplete) {
+      return;
+    }
+    if (!this.setupSchemaPromise) {
+      this.setupSchemaPromise = (async () => {
+        try {
+          const checkRequest = this.pool.request();
+          checkRequest.input("schemaName", this.schemaName);
+          const checkResult = await checkRequest.query(`
+            SELECT 1 AS found FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = @schemaName
+          `);
+          const schemaExists = Array.isArray(checkResult.recordset) && checkResult.recordset.length > 0;
+          if (!schemaExists) {
+            try {
+              await this.pool.request().query(`CREATE SCHEMA [${this.schemaName}]`);
+              this.logger?.info?.(`Schema "${this.schemaName}" created successfully`);
+            } catch (error) {
+              this.logger?.error?.(`Failed to create schema "${this.schemaName}"`, { error });
+              throw new Error(
+                `Unable to create schema "${this.schemaName}". This requires CREATE privilege on the database. Either create the schema manually or grant CREATE privilege to the user.`
+              );
+            }
+          }
+          this.schemaSetupComplete = true;
+          this.logger?.debug?.(`Schema "${this.schemaName}" is ready for use`);
+        } catch (error) {
+          this.schemaSetupComplete = void 0;
+          this.setupSchemaPromise = null;
+          throw error;
+        } finally {
+          this.setupSchemaPromise = null;
+        }
+      })();
+    }
+    await this.setupSchemaPromise;
+  }
+  async insert({ tableName, record }) {
+    try {
+      const columns = Object.keys(record).map((col) => parseSqlIdentifier(col, "column name"));
+      const values = Object.values(record);
+      const paramNames = values.map((_, i) => `@param${i}`);
+      const insertSql = `INSERT INTO ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} (${columns.map((c) => `[${c}]`).join(", ")}) VALUES (${paramNames.join(", ")})`;
+      const request = this.pool.request();
+      values.forEach((value, i) => {
+        if (value instanceof Date) {
+          request.input(`param${i}`, sql2.DateTime2, value);
+        } else if (typeof value === "object" && value !== null) {
+          request.input(`param${i}`, JSON.stringify(value));
+        } else {
+          request.input(`param${i}`, value);
+        }
+      });
+      await request.query(insertSql);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  async clearTable({ tableName }) {
+    const fullTableName = getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) });
+    try {
+      try {
+        await this.pool.request().query(`TRUNCATE TABLE ${fullTableName}`);
+      } catch (truncateError) {
+        if (truncateError.message && truncateError.message.includes("foreign key")) {
+          await this.pool.request().query(`DELETE FROM ${fullTableName}`);
+        } else {
+          throw truncateError;
+        }
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_CLEAR_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  getDefaultValue(type) {
+    switch (type) {
+      case "timestamp":
+        return "DEFAULT SYSDATETIMEOFFSET()";
+      case "jsonb":
+        return "DEFAULT N'{}'";
+      default:
+        return super.getDefaultValue(type);
+    }
+  }
+  async createTable({
+    tableName,
+    schema
+  }) {
+    try {
+      const uniqueConstraintColumns = tableName === TABLE_WORKFLOW_SNAPSHOT ? ["workflow_name", "run_id"] : [];
+      const columns = Object.entries(schema).map(([name, def]) => {
+        const parsedName = parseSqlIdentifier(name, "column name");
+        const constraints = [];
+        if (def.primaryKey) constraints.push("PRIMARY KEY");
+        if (!def.nullable) constraints.push("NOT NULL");
+        const isIndexed = !!def.primaryKey || uniqueConstraintColumns.includes(name);
+        return `[${parsedName}] ${this.getSqlType(def.type, isIndexed)} ${constraints.join(" ")}`.trim();
+      }).join(",\n");
+      if (this.schemaName) {
+        await this.setupSchema();
+      }
+      const checkTableRequest = this.pool.request();
+      checkTableRequest.input(
+        "tableName",
+        getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) }).replace(/[[\]]/g, "").split(".").pop()
+      );
+      const checkTableSql = `SELECT 1 AS found FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @tableName`;
+      checkTableRequest.input("schema", this.schemaName || "dbo");
+      const checkTableResult = await checkTableRequest.query(checkTableSql);
+      const tableExists = Array.isArray(checkTableResult.recordset) && checkTableResult.recordset.length > 0;
+      if (!tableExists) {
+        const createSql = `CREATE TABLE ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} (
+${columns}
+)`;
+        await this.pool.request().query(createSql);
+      }
+      const columnCheckSql = `
+        SELECT 1 AS found
+        FROM INFORMATION_SCHEMA.COLUMNS
+        WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @tableName AND COLUMN_NAME = 'seq_id'
+      `;
+      const checkColumnRequest = this.pool.request();
+      checkColumnRequest.input("schema", this.schemaName || "dbo");
+      checkColumnRequest.input(
+        "tableName",
+        getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) }).replace(/[[\]]/g, "").split(".").pop()
+      );
+      const columnResult = await checkColumnRequest.query(columnCheckSql);
+      const columnExists = Array.isArray(columnResult.recordset) && columnResult.recordset.length > 0;
+      if (!columnExists) {
+        const alterSql = `ALTER TABLE ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} ADD seq_id BIGINT IDENTITY(1,1)`;
+        await this.pool.request().query(alterSql);
+      }
+      if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
+        const constraintName = "mastra_workflow_snapshot_workflow_name_run_id_key";
+        const checkConstraintSql = `SELECT 1 AS found FROM sys.key_constraints WHERE name = @constraintName`;
+        const checkConstraintRequest = this.pool.request();
+        checkConstraintRequest.input("constraintName", constraintName);
+        const constraintResult = await checkConstraintRequest.query(checkConstraintSql);
+        const constraintExists = Array.isArray(constraintResult.recordset) && constraintResult.recordset.length > 0;
+        if (!constraintExists) {
+          const addConstraintSql = `ALTER TABLE ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} ADD CONSTRAINT ${constraintName} UNIQUE ([workflow_name], [run_id])`;
+          await this.pool.request().query(addConstraintSql);
+        }
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Alters table schema to add columns if they don't exist
+   * @param tableName Name of the table
+   * @param schema Schema of the table
+   * @param ifNotExists Array of column names to add if they don't exist
+   */
+  async alterTable({
+    tableName,
+    schema,
+    ifNotExists
+  }) {
+    const fullTableName = getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) });
+    try {
+      for (const columnName of ifNotExists) {
+        if (schema[columnName]) {
+          const columnCheckRequest = this.pool.request();
+          columnCheckRequest.input("tableName", fullTableName.replace(/[[\]]/g, "").split(".").pop());
+          columnCheckRequest.input("columnName", columnName);
+          columnCheckRequest.input("schema", this.schemaName || "dbo");
+          const checkSql = `SELECT 1 AS found FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @tableName AND COLUMN_NAME = @columnName`;
+          const checkResult = await columnCheckRequest.query(checkSql);
+          const columnExists = Array.isArray(checkResult.recordset) && checkResult.recordset.length > 0;
+          if (!columnExists) {
+            const columnDef = schema[columnName];
+            const sqlType = this.getSqlType(columnDef.type);
+            const nullable = columnDef.nullable === false ? "NOT NULL" : "";
+            const defaultValue = columnDef.nullable === false ? this.getDefaultValue(columnDef.type) : "";
+            const parsedColumnName = parseSqlIdentifier(columnName, "column name");
+            const alterSql = `ALTER TABLE ${fullTableName} ADD [${parsedColumnName}] ${sqlType} ${nullable} ${defaultValue}`.trim();
+            await this.pool.request().query(alterSql);
+            this.logger?.debug?.(`Ensured column ${parsedColumnName} exists in table ${fullTableName}`);
+          }
+        }
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_ALTER_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  async load({ tableName, keys }) {
+    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 values = keyEntries.map(([_, value]) => value);
+      const sql7 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
+      const request = this.pool.request();
+      values.forEach((value, i) => {
+        request.input(`param${i}`, value);
+      });
+      const resultSet = await request.query(sql7);
+      const result = resultSet.recordset[0] || null;
+      if (!result) {
+        return null;
+      }
+      if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
+        const snapshot = result;
+        if (typeof snapshot.snapshot === "string") {
+          snapshot.snapshot = JSON.parse(snapshot.snapshot);
+        }
+        return snapshot;
+      }
+      return result;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  async batchInsert({ tableName, records }) {
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      for (const record of records) {
+        await this.insert({ tableName, record });
+      }
+      await transaction.commit();
+    } catch (error) {
+      await transaction.rollback();
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName,
+            numberOfRecords: records.length
+          }
+        },
+        error
+      );
+    }
+  }
+  async dropTable({ tableName }) {
+    try {
+      const tableNameWithSchema = getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) });
+      await this.pool.request().query(`DROP TABLE IF EXISTS ${tableNameWithSchema}`);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_DROP_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+};
+function parseJSON(jsonString) {
+  try {
+    return JSON.parse(jsonString);
+  } catch {
+    return jsonString;
+  }
+}
+function transformScoreRow(row) {
+  return {
+    ...row,
+    input: parseJSON(row.input),
+    scorer: parseJSON(row.scorer),
+    preprocessStepResult: parseJSON(row.preprocessStepResult),
+    analyzeStepResult: parseJSON(row.analyzeStepResult),
+    metadata: parseJSON(row.metadata),
+    output: parseJSON(row.output),
+    additionalContext: parseJSON(row.additionalContext),
+    runtimeContext: parseJSON(row.runtimeContext),
+    entity: parseJSON(row.entity),
+    createdAt: row.createdAt,
+    updatedAt: row.updatedAt
+  };
+}
+var ScoresMSSQL = class extends ScoresStorage {
+  pool;
+  operations;
+  schema;
+  constructor({
+    pool,
+    operations,
+    schema
+  }) {
+    super();
+    this.pool = pool;
+    this.operations = operations;
+    this.schema = schema;
+  }
+  async getScoreById({ id }) {
+    try {
+      const request = this.pool.request();
+      request.input("p1", id);
+      const result = await request.query(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE id = @p1`
+      );
+      if (result.recordset.length === 0) {
+        return null;
+      }
+      return transformScoreRow(result.recordset[0]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORE_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id }
+        },
+        error
+      );
+    }
+  }
+  async saveScore(score) {
+    try {
+      const scoreId = crypto.randomUUID();
+      const {
+        scorer,
+        preprocessStepResult,
+        analyzeStepResult,
+        metadata,
+        input,
+        output,
+        additionalContext,
+        runtimeContext,
+        entity,
+        ...rest
+      } = score;
+      await this.operations.insert({
+        tableName: TABLE_SCORERS,
+        record: {
+          id: scoreId,
+          ...rest,
+          input: JSON.stringify(input) || "",
+          output: JSON.stringify(output) || "",
+          preprocessStepResult: preprocessStepResult ? JSON.stringify(preprocessStepResult) : null,
+          analyzeStepResult: analyzeStepResult ? JSON.stringify(analyzeStepResult) : null,
+          metadata: metadata ? JSON.stringify(metadata) : null,
+          additionalContext: additionalContext ? JSON.stringify(additionalContext) : null,
+          runtimeContext: runtimeContext ? JSON.stringify(runtimeContext) : null,
+          entity: entity ? JSON.stringify(entity) : null,
+          scorer: scorer ? JSON.stringify(scorer) : null,
+          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+        }
+      });
+      const scoreFromDb = await this.getScoreById({ id: scoreId });
+      return { score: scoreFromDb };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async getScoresByScorerId({
+    scorerId,
+    pagination
+  }) {
+    try {
+      const request = this.pool.request();
+      request.input("p1", scorerId);
+      const totalResult = await request.query(
+        `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [scorerId] = @p1`
+      );
+      const total = totalResult.recordset[0]?.count || 0;
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          },
+          scores: []
+        };
+      }
+      const dataRequest = this.pool.request();
+      dataRequest.input("p1", scorerId);
+      dataRequest.input("p2", pagination.perPage);
+      dataRequest.input("p3", pagination.page * pagination.perPage);
+      const result = await dataRequest.query(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [scorerId] = @p1 ORDER BY [createdAt] DESC OFFSET @p3 ROWS FETCH NEXT @p2 ROWS ONLY`
+      );
+      return {
+        pagination: {
+          total: Number(total),
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+        },
+        scores: result.recordset.map((row) => transformScoreRow(row))
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SCORER_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { scorerId }
+        },
+        error
+      );
+    }
+  }
+  async getScoresByRunId({
+    runId,
+    pagination
+  }) {
+    try {
+      const request = this.pool.request();
+      request.input("p1", runId);
+      const totalResult = await request.query(
+        `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [runId] = @p1`
+      );
+      const total = totalResult.recordset[0]?.count || 0;
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          },
+          scores: []
+        };
+      }
+      const dataRequest = this.pool.request();
+      dataRequest.input("p1", runId);
+      dataRequest.input("p2", pagination.perPage);
+      dataRequest.input("p3", pagination.page * pagination.perPage);
+      const result = await dataRequest.query(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [runId] = @p1 ORDER BY [createdAt] DESC OFFSET @p3 ROWS FETCH NEXT @p2 ROWS ONLY`
+      );
+      return {
+        pagination: {
+          total: Number(total),
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+        },
+        scores: result.recordset.map((row) => transformScoreRow(row))
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_RUN_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { runId }
+        },
+        error
+      );
+    }
+  }
+  async getScoresByEntityId({
+    entityId,
+    entityType,
+    pagination
+  }) {
+    try {
+      const request = this.pool.request();
+      request.input("p1", entityId);
+      request.input("p2", entityType);
+      const totalResult = await request.query(
+        `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [entityId] = @p1 AND [entityType] = @p2`
+      );
+      const total = totalResult.recordset[0]?.count || 0;
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          },
+          scores: []
+        };
+      }
+      const dataRequest = this.pool.request();
+      dataRequest.input("p1", entityId);
+      dataRequest.input("p2", entityType);
+      dataRequest.input("p3", pagination.perPage);
+      dataRequest.input("p4", pagination.page * pagination.perPage);
+      const result = await dataRequest.query(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [entityId] = @p1 AND [entityType] = @p2 ORDER BY [createdAt] DESC OFFSET @p4 ROWS FETCH NEXT @p3 ROWS ONLY`
+      );
+      return {
+        pagination: {
+          total: Number(total),
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+        },
+        scores: result.recordset.map((row) => transformScoreRow(row))
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { entityId, entityType }
+        },
+        error
+      );
+    }
+  }
+};
+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, sql2.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, sql2.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
+    });
+  }
+};
+function parseWorkflowRun(row) {
+  let parsedSnapshot = row.snapshot;
+  if (typeof parsedSnapshot === "string") {
+    try {
+      parsedSnapshot = JSON.parse(row.snapshot);
+    } catch (e) {
+      console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+    }
+  }
+  return {
+    workflowName: row.workflow_name,
+    runId: row.run_id,
+    snapshot: parsedSnapshot,
+    createdAt: row.createdAt,
+    updatedAt: row.updatedAt,
+    resourceId: row.resourceId
+  };
+}
+var WorkflowsMSSQL = class extends WorkflowsStorage {
+  pool;
+  operations;
+  schema;
+  constructor({
+    pool,
+    operations,
+    schema
+  }) {
+    super();
+    this.pool = pool;
+    this.operations = operations;
+    this.schema = schema;
+  }
+  updateWorkflowResults({
+    // workflowName,
+    // runId,
+    // stepId,
+    // result,
+    // runtimeContext,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  updateWorkflowState({
+    // workflowName,
+    // runId,
+    // opts,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    snapshot
+  }) {
+    const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    try {
+      const request = this.pool.request();
+      request.input("workflow_name", workflowName);
+      request.input("run_id", runId);
+      request.input("snapshot", JSON.stringify(snapshot));
+      request.input("createdAt", sql2.DateTime2, new Date(now));
+      request.input("updatedAt", sql2.DateTime2, new Date(now));
+      const mergeSql = `MERGE INTO ${table} AS target
+        USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
+        ON target.workflow_name = src.workflow_name AND target.run_id = src.run_id
+        WHEN MATCHED THEN UPDATE SET
+          snapshot = @snapshot,
+          [updatedAt] = @updatedAt
+        WHEN NOT MATCHED THEN INSERT (workflow_name, run_id, snapshot, [createdAt], [updatedAt])
+          VALUES (@workflow_name, @run_id, @snapshot, @createdAt, @updatedAt);`;
+      await request.query(mergeSql);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            workflowName,
+            runId
+          }
+        },
+        error
+      );
+    }
+  }
+  async loadWorkflowSnapshot({
+    workflowName,
+    runId
+  }) {
+    try {
+      const result = await this.operations.load({
+        tableName: TABLE_WORKFLOW_SNAPSHOT,
+        keys: {
+          workflow_name: workflowName,
+          run_id: runId
+        }
+      });
+      if (!result) {
+        return null;
+      }
+      return result.snapshot;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            workflowName,
+            runId
+          }
+        },
+        error
+      );
+    }
+  }
+  async getWorkflowRunById({
+    runId,
+    workflowName
+  }) {
+    try {
+      const conditions = [];
+      const paramMap = {};
+      if (runId) {
+        conditions.push(`[run_id] = @runId`);
+        paramMap["runId"] = runId;
+      }
+      if (workflowName) {
+        conditions.push(`[workflow_name] = @workflowName`);
+        paramMap["workflowName"] = workflowName;
+      }
+      const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+      const tableName = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+      const query = `SELECT * FROM ${tableName} ${whereClause}`;
+      const request = this.pool.request();
+      Object.entries(paramMap).forEach(([key, value]) => request.input(key, value));
+      const result = await request.query(query);
+      if (!result.recordset || result.recordset.length === 0) {
+        return null;
+      }
+      return parseWorkflowRun(result.recordset[0]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            runId,
+            workflowName: workflowName || ""
+          }
+        },
+        error
+      );
+    }
+  }
+  async getWorkflowRuns({
+    workflowName,
+    fromDate,
+    toDate,
+    limit,
+    offset,
+    resourceId
+  } = {}) {
+    try {
+      const conditions = [];
+      const paramMap = {};
+      if (workflowName) {
+        conditions.push(`[workflow_name] = @workflowName`);
+        paramMap["workflowName"] = workflowName;
+      }
+      if (resourceId) {
+        const hasResourceId = await this.operations.hasColumn(TABLE_WORKFLOW_SNAPSHOT, "resourceId");
+        if (hasResourceId) {
+          conditions.push(`[resourceId] = @resourceId`);
+          paramMap["resourceId"] = resourceId;
+        } else {
+          console.warn(`[${TABLE_WORKFLOW_SNAPSHOT}] resourceId column not found. Skipping resourceId filter.`);
+        }
+      }
+      if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
+        conditions.push(`[createdAt] >= @fromDate`);
+        paramMap[`fromDate`] = fromDate.toISOString();
+      }
+      if (toDate instanceof Date && !isNaN(toDate.getTime())) {
+        conditions.push(`[createdAt] <= @toDate`);
+        paramMap[`toDate`] = toDate.toISOString();
+      }
+      const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+      let total = 0;
+      const tableName = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+      const request = this.pool.request();
+      Object.entries(paramMap).forEach(([key, value]) => {
+        if (value instanceof Date) {
+          request.input(key, sql2.DateTime, value);
+        } else {
+          request.input(key, value);
+        }
+      });
+      if (limit !== void 0 && offset !== void 0) {
+        const countQuery = `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`;
+        const countResult = await request.query(countQuery);
+        total = Number(countResult.recordset[0]?.count || 0);
+      }
+      let query = `SELECT * FROM ${tableName} ${whereClause} ORDER BY [seq_id] DESC`;
+      if (limit !== void 0 && offset !== void 0) {
+        query += ` OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
+        request.input("limit", limit);
+        request.input("offset", offset);
+      }
+      const result = await request.query(query);
+      const runs = (result.recordset || []).map((row) => parseWorkflowRun(row));
+      return { runs, total: total || runs.length };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_WORKFLOW_RUNS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            workflowName: workflowName || "all"
+          }
+        },
+        error
+      );
+    }
+  }
+};
+
+// src/storage/index.ts
+var MSSQLStore = class extends MastraStorage {
+  pool;
+  schema;
+  isConnected = null;
+  stores;
+  constructor(config) {
+    super({ name: "MSSQLStore" });
+    try {
+      if ("connectionString" in config) {
+        if (!config.connectionString || typeof config.connectionString !== "string" || config.connectionString.trim() === "") {
+          throw new Error("MSSQLStore: connectionString must be provided and cannot be empty.");
+        }
+      } else {
+        const required = ["server", "database", "user", "password"];
+        for (const key of required) {
+          if (!(key in config) || typeof config[key] !== "string" || config[key].trim() === "") {
+            throw new Error(`MSSQLStore: ${key} must be provided and cannot be empty.`);
+          }
+        }
+      }
+      this.schema = config.schemaName || "dbo";
+      this.pool = "connectionString" in config ? new sql2.ConnectionPool(config.connectionString) : new sql2.ConnectionPool({
+        server: config.server,
+        database: config.database,
+        user: config.user,
+        password: config.password,
+        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 });
+      this.stores = {
+        operations,
+        scores,
+        traces,
+        workflows,
+        legacyEvals,
+        memory
+      };
+    } catch (e) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_INITIALIZATION_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER
+        },
+        e
+      );
+    }
+  }
+  async init() {
+    if (this.isConnected === null) {
+      this.isConnected = this._performInitializationAndStore();
+    }
+    try {
+      await this.isConnected;
+      await super.init();
+    } catch (error) {
+      this.isConnected = null;
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_INIT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async _performInitializationAndStore() {
+    try {
+      await this.pool.connect();
+      return true;
+    } catch (err) {
+      throw err;
+    }
+  }
+  get supports() {
+    return {
+      selectByIncludeResourceScope: true,
+      resourceWorkingMemory: true,
+      hasColumn: true,
+      createTable: true,
+      deleteMessages: 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
+  }) {
+    return this.stores.operations.createTable({ tableName, schema });
+  }
+  async alterTable({
+    tableName,
+    schema,
+    ifNotExists
+  }) {
+    return this.stores.operations.alterTable({ tableName, schema, ifNotExists });
+  }
+  async clearTable({ tableName }) {
+    return this.stores.operations.clearTable({ tableName });
+  }
+  async dropTable({ tableName }) {
+    return this.stores.operations.dropTable({ tableName });
+  }
+  async insert({ tableName, record }) {
+    return this.stores.operations.insert({ tableName, record });
+  }
+  async batchInsert({ tableName, records }) {
+    return this.stores.operations.batchInsert({ tableName, records });
+  }
+  async load({ tableName, keys }) {
+    return this.stores.operations.load({ tableName, keys });
+  }
+  /**
+   * Memory
+   */
+  async getThreadById({ threadId }) {
+    return this.stores.memory.getThreadById({ threadId });
+  }
+  /**
+   * @deprecated use getThreadsByResourceIdPaginated instead
+   */
+  async getThreadsByResourceId(args) {
+    return this.stores.memory.getThreadsByResourceId(args);
+  }
+  async getThreadsByResourceIdPaginated(args) {
+    return this.stores.memory.getThreadsByResourceIdPaginated(args);
+  }
+  async saveThread({ thread }) {
+    return this.stores.memory.saveThread({ thread });
+  }
+  async updateThread({
+    id,
+    title,
+    metadata
+  }) {
+    return this.stores.memory.updateThread({ id, title, metadata });
+  }
+  async deleteThread({ threadId }) {
+    return this.stores.memory.deleteThread({ threadId });
+  }
+  async getMessages(args) {
+    return this.stores.memory.getMessages(args);
+  }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
+  async getMessagesPaginated(args) {
+    return this.stores.memory.getMessagesPaginated(args);
+  }
+  async saveMessages(args) {
+    return this.stores.memory.saveMessages(args);
+  }
+  async updateMessages({
+    messages
+  }) {
+    return this.stores.memory.updateMessages({ messages });
+  }
+  async deleteMessages(messageIds) {
+    return this.stores.memory.deleteMessages(messageIds);
+  }
+  async getResourceById({ resourceId }) {
+    return this.stores.memory.getResourceById({ resourceId });
+  }
+  async saveResource({ resource }) {
+    return this.stores.memory.saveResource({ resource });
+  }
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata
+  }) {
+    return this.stores.memory.updateResource({ resourceId, workingMemory, metadata });
+  }
+  /**
+   * Workflows
+   */
+  async updateWorkflowResults({
+    workflowName,
+    runId,
+    stepId,
+    result,
+    runtimeContext
+  }) {
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+  }
+  async updateWorkflowState({
+    workflowName,
+    runId,
+    opts
+  }) {
+    return this.stores.workflows.updateWorkflowState({ workflowName, runId, opts });
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    snapshot
+  }) {
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+  }
+  async loadWorkflowSnapshot({
+    workflowName,
+    runId
+  }) {
+    return this.stores.workflows.loadWorkflowSnapshot({ workflowName, runId });
+  }
+  async getWorkflowRuns({
+    workflowName,
+    fromDate,
+    toDate,
+    limit,
+    offset,
+    resourceId
+  } = {}) {
+    return this.stores.workflows.getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId });
+  }
+  async getWorkflowRunById({
+    runId,
+    workflowName
+  }) {
+    return this.stores.workflows.getWorkflowRunById({ runId, workflowName });
+  }
+  async close() {
+    await this.pool.close();
+  }
+  /**
+   * Scorers
+   */
+  async getScoreById({ id: _id }) {
+    return this.stores.scores.getScoreById({ id: _id });
+  }
+  async getScoresByScorerId({
+    scorerId: _scorerId,
+    pagination: _pagination
+  }) {
+    return this.stores.scores.getScoresByScorerId({ scorerId: _scorerId, pagination: _pagination });
+  }
+  async saveScore(_score) {
+    return this.stores.scores.saveScore(_score);
+  }
+  async getScoresByRunId({
+    runId: _runId,
+    pagination: _pagination
+  }) {
+    return this.stores.scores.getScoresByRunId({ runId: _runId, pagination: _pagination });
+  }
+  async getScoresByEntityId({
+    entityId: _entityId,
+    entityType: _entityType,
+    pagination: _pagination
+  }) {
+    return this.stores.scores.getScoresByEntityId({
+      entityId: _entityId,
+      entityType: _entityType,
+      pagination: _pagination
+    });
+  }
+};
+
+export { MSSQLStore };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map