@mastra/mssql 0.0.0-share-agent-metadata-with-cloud-20250718110128 → 0.0.0-sidebar-window-undefined-fix-20251029233656

package/dist/index.js

@@ -1,124 +1,125 @@
-import { MessageList } from '@mastra/core/agent';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, TABLE_EVALS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES } from '@mastra/core/storage';
-import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
-import sql from 'mssql';
+import { MastraStorage, LegacyEvalsStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_EVALS, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import sql3 from 'mssql';
+import { parseSqlIdentifier } from '@mastra/core/utils';
+import { MessageList } from '@mastra/core/agent';
+import { randomUUID } from 'crypto';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // src/storage/index.ts
-var MSSQLStore = class extends MastraStorage {
-  pool;
-  schema;
-  setupSchemaPromise = null;
-  schemaSetupComplete = void 0;
-  isConnected = null;
-  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;
-      this.pool = "connectionString" in config ? new sql.ConnectionPool(config.connectionString) : new sql.ConnectionPool({
-        server: config.server,
-        database: config.database,
-        user: config.user,
-        password: config.password,
-        port: config.port,
-        options: config.options || { encrypt: true, trustServerCertificate: true }
-      });
-    } catch (e) {
-      throw new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_INITIALIZATION_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER
-        },
-        e
-      );
-    }
+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;
+}
+function buildDateRangeFilter(dateRange, fieldName) {
+  const filters = {};
+  if (dateRange?.start) {
+    filters[`${fieldName}_gte`] = dateRange.start;
   }
-  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
-      );
-    }
+  if (dateRange?.end) {
+    filters[`${fieldName}_lte`] = dateRange.end;
   }
-  async _performInitializationAndStore() {
-    try {
-      await this.pool.connect();
-      return true;
-    } catch (err) {
-      throw err;
+  return filters;
+}
+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 fieldName = key.slice(0, -4);
+      conditions.push(`[${parseSqlIdentifier(fieldName, "field name")}] >= @${paramName}`);
+      params[paramName] = value instanceof Date ? value.toISOString() : value;
+    } else if (key.endsWith("_lte")) {
+      const fieldName = key.slice(0, -4);
+      conditions.push(`[${parseSqlIdentifier(fieldName, "field name")}] <= @${paramName}`);
+      params[paramName] = value instanceof Date ? value.toISOString() : value;
+    } else if (value === null) {
+      conditions.push(`[${parseSqlIdentifier(key, "field name")}] IS NULL`);
+    } else {
+      conditions.push(`[${parseSqlIdentifier(key, "field name")}] = @${paramName}`);
+      params[paramName] = value instanceof Date ? value.toISOString() : value;
     }
-  }
-  get supports() {
-    return {
-      selectByIncludeResourceScope: true,
-      resourceWorkingMemory: true
-    };
-  }
-  getTableName(indexName) {
-    const parsedIndexName = parseSqlIdentifier(indexName, "index name");
-    const quotedIndexName = `[${parsedIndexName}]`;
-    const quotedSchemaName = this.getSchemaName();
-    return quotedSchemaName ? `${quotedSchemaName}.${quotedIndexName}` : quotedIndexName;
-  }
-  getSchemaName() {
-    return this.schema ? `[${parseSqlIdentifier(this.schema, "schema name")}]` : void 0;
-  }
-  transformEvalRow(row) {
-    let testInfoValue = null, resultValue = null;
-    if (row.test_info) {
+  });
+  return {
+    sql: conditions.length > 0 ? ` WHERE ${conditions.join(" AND ")}` : "",
+    params
+  };
+}
+function transformFromSqlRow({
+  tableName,
+  sqlRow
+}) {
+  const schema = TABLE_SCHEMAS[tableName];
+  const result = {};
+  Object.entries(sqlRow).forEach(([key, value]) => {
+    const columnSchema = schema?.[key];
+    if (columnSchema?.type === "jsonb" && typeof value === "string") {
       try {
-        testInfoValue = typeof row.test_info === "string" ? JSON.parse(row.test_info) : row.test_info;
+        result[key] = JSON.parse(value);
       } catch {
+        result[key] = value;
       }
+    } else if (columnSchema?.type === "timestamp" && value && typeof value === "string") {
+      result[key] = new Date(value);
+    } else if (columnSchema?.type === "timestamp" && value instanceof Date) {
+      result[key] = value;
+    } else if (columnSchema?.type === "boolean") {
+      result[key] = Boolean(value);
+    } else {
+      result[key] = value;
     }
-    if (row.test_info) {
-      try {
-        resultValue = typeof row.result === "string" ? JSON.parse(row.result) : row.result;
-      } catch {
-      }
+  });
+  return result;
+}
+
+// src/storage/domains/legacy-evals/index.ts
+function transformEvalRow(row) {
+  let testInfoValue = null, resultValue = null;
+  if (row.test_info) {
+    try {
+      testInfoValue = typeof row.test_info === "string" ? JSON.parse(row.test_info) : row.test_info;
+    } catch {
     }
-    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
-    };
+  }
+  if (row.result) {
+    try {
+      resultValue = typeof row.result === "string" ? JSON.parse(row.result) : row.result;
+    } catch {
+    }
+  }
+  return {
+    agentName: row.agent_name,
+    input: row.input,
+    output: row.output,
+    result: resultValue,
+    metricName: row.metric_name,
+    instructions: row.instructions,
+    testInfo: testInfoValue,
+    globalRunId: row.global_run_id,
+    runId: row.run_id,
+    createdAt: row.created_at
+  };
+}
+var LegacyEvalsMSSQL = class extends LegacyEvalsStorage {
+  pool;
+  schema;
+  constructor({ pool, schema }) {
+    super();
+    this.pool = pool;
+    this.schema = schema;
   }
   /** @deprecated use getEvals instead */
   async getEvalsByAgentName(agentName, type) {
     try {
-      let query = `SELECT * FROM ${this.getTableName(TABLE_EVALS)} WHERE agent_name = @p1`;
+      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") {
@@ -129,495 +130,164 @@ var MSSQLStore = class extends MastraStorage {
       request.input("p1", agentName);
       const result = await request.query(query);
       const rows = result.recordset;
-      return typeof this.transformEvalRow === "function" ? rows?.map((row) => this.transformEvalRow(row)) ?? [] : rows ?? [];
+      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);
+      this.logger?.error?.("Failed to get evals for the specified agent:", error);
       throw 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
-      );
-    }
-  }
-  /** @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;
+  async getEvals(options = {}) {
+    const { agentName, type, page = 0, perPage = 100, dateRange } = options;
     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;
-      });
+    const where = [];
+    const params = {};
+    if (agentName) {
+      where.push("agent_name = @agentName");
+      params["agentName"] = agentName;
     }
-    if (filters) {
-      Object.entries(filters).forEach(([key, value]) => {
-        const parsedKey = parseFieldKey(key);
-        const paramName = `p${paramIndex++}`;
-        conditions.push(`[${parsedKey}] = @${paramName}`);
-        paramMap[paramName] = value;
-      });
+    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())) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[createdAt] >= @${paramName}`);
-      paramMap[paramName] = fromDate.toISOString();
+      where.push(`[created_at] >= @fromDate`);
+      params[`fromDate`] = fromDate.toISOString();
     }
     if (toDate instanceof Date && !isNaN(toDate.getTime())) {
-      const paramName = `p${paramIndex++}`;
-      conditions.push(`[createdAt] <= @${paramName}`);
-      paramMap[paramName] = toDate.toISOString();
+      where.push(`[created_at] <= @toDate`);
+      params[`toDate`] = toDate.toISOString();
     }
-    const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
-    const countQuery = `SELECT COUNT(*) as total FROM ${this.getTableName(TABLE_TRACES)} ${whereClause}`;
-    let total = 0;
+    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 countRequest = this.pool.request();
-      Object.entries(paramMap).forEach(([key, value]) => {
+      const countReq = this.pool.request();
+      Object.entries(params).forEach(([key, value]) => {
         if (value instanceof Date) {
-          countRequest.input(key, sql.DateTime, value);
+          countReq.input(key, sql3.DateTime, value);
         } else {
-          countRequest.input(key, value);
+          countReq.input(key, value);
         }
       });
-      const countResult = await countRequest.query(countQuery);
-      total = parseInt(countResult.recordset[0].total, 10);
+      const countResult = await countReq.query(countQuery);
+      const total = countResult.recordset[0]?.total || 0;
+      if (total === 0) {
+        return {
+          evals: [],
+          total: 0,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      const req = this.pool.request();
+      Object.entries(params).forEach(([key, value]) => {
+        if (value instanceof Date) {
+          req.input(key, sql3.DateTime, value);
+        } else {
+          req.input(key, value);
+        }
+      });
+      req.input("offset", offset);
+      req.input("perPage", perPage);
+      const result = await req.query(dataQuery);
+      const rows = result.recordset;
+      return {
+        evals: rows?.map((row) => transformEvalRow(row)) ?? [],
+        total,
+        page,
+        perPage,
+        hasMore: offset + (rows?.length ?? 0) < total
+      };
     } catch (error) {
-      throw new MastraError(
+      const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_TRACES_PAGINATED_FAILED_TO_RETRIEVE_TOTAL_COUNT",
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_EVALS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
+            agentName: agentName || "all",
+            type: type || "all",
+            page,
+            perPage
           }
         },
         error
       );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException?.(mastraError);
+      throw mastraError;
     }
-    if (total === 0) {
-      return {
-        traces: [],
-        total: 0,
-        page,
-        perPage,
-        hasMore: false
-      };
-    }
-    const dataQuery = `SELECT * FROM ${this.getTableName(TABLE_TRACES)} ${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, sql.DateTime, value);
-      } else {
-        dataRequest.input(key, value);
+  }
+};
+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;
     });
-    dataRequest.input("offset", currentOffset);
-    dataRequest.input("limit", perPage);
+    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 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
-      }));
+      const sql6 = `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(sql6);
+      const thread = resultSet.recordset[0] || null;
+      if (!thread) {
+        return null;
+      }
       return {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: currentOffset + traces.length < total
+        ...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_TRACES_PAGINATED_FAILED_TO_RETRIEVE_TRACES",
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_THREAD_BY_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
-          }
-        },
-        error
-      );
-    }
-  }
-  async setupSchema() {
-    if (!this.schema || this.schemaSetupComplete) {
-      return;
-    }
-    if (!this.setupSchemaPromise) {
-      this.setupSchemaPromise = (async () => {
-        try {
-          const checkRequest = this.pool.request();
-          checkRequest.input("schemaName", this.schema);
-          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.schema}]`);
-              this.logger?.info?.(`Schema "${this.schema}" created successfully`);
-            } catch (error) {
-              this.logger?.error?.(`Failed to create schema "${this.schema}"`, { error });
-              throw new Error(
-                `Unable to create schema "${this.schema}". 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.schema}" is ready for use`);
-        } catch (error) {
-          this.schemaSetupComplete = void 0;
-          this.setupSchemaPromise = null;
-          throw error;
-        } finally {
-          this.setupSchemaPromise = null;
-        }
-      })();
-    }
-    await this.setupSchemaPromise;
-  }
-  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";
-      default:
-        throw new MastraError({
-          id: "MASTRA_STORAGE_MSSQL_STORE_TYPE_NOT_SUPPORTED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
-        });
-    }
-  }
-  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.schema) {
-        await this.setupSchema();
-      }
-      const checkTableRequest = this.pool.request();
-      checkTableRequest.input("tableName", this.getTableName(tableName).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.schema || "dbo");
-      const checkTableResult = await checkTableRequest.query(checkTableSql);
-      const tableExists = Array.isArray(checkTableResult.recordset) && checkTableResult.recordset.length > 0;
-      if (!tableExists) {
-        const createSql = `CREATE TABLE ${this.getTableName(tableName)} (
-${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.schema || "dbo");
-      checkColumnRequest.input("tableName", this.getTableName(tableName).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 ${this.getTableName(tableName)} 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 ${this.getTableName(tableName)} 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
-      );
-    }
-  }
-  getDefaultValue(type) {
-    switch (type) {
-      case "timestamp":
-        return "DEFAULT SYSDATETIMEOFFSET()";
-      case "jsonb":
-        return "DEFAULT N'{}'";
-      default:
-        return super.getDefaultValue(type);
-    }
-  }
-  async alterTable({
-    tableName,
-    schema,
-    ifNotExists
-  }) {
-    const fullTableName = this.getTableName(tableName);
-    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.schema || "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 clearTable({ tableName }) {
-    const fullTableName = this.getTableName(tableName);
-    try {
-      const fkQuery = `
-        SELECT
-          OBJECT_SCHEMA_NAME(fk.parent_object_id) AS schema_name,
-          OBJECT_NAME(fk.parent_object_id) AS table_name
-        FROM sys.foreign_keys fk
-        WHERE fk.referenced_object_id = OBJECT_ID(@fullTableName)
-      `;
-      const fkResult = await this.pool.request().input("fullTableName", fullTableName).query(fkQuery);
-      const childTables = fkResult.recordset || [];
-      for (const child of childTables) {
-        const childTableName = this.schema ? `[${child.schema_name}].[${child.table_name}]` : `[${child.table_name}]`;
-        await this.clearTable({ tableName: childTableName });
-      }
-      await this.pool.request().query(`TRUNCATE TABLE ${fullTableName}`);
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_CLEAR_TABLE_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            tableName
-          }
-        },
-        error
-      );
-    }
-  }
-  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 ${this.getTableName(tableName)} (${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}`, sql.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 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 sql2 = `SELECT * FROM ${this.getTableName(tableName)} WHERE ${conditions}`;
-      const request = this.pool.request();
-      values.forEach((value, i) => {
-        request.input(`param${i}`, value);
-      });
-      const resultSet = await request.query(sql2);
-      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 getThreadById({ threadId }) {
-    try {
-      const sql2 = `SELECT 
-        id,
-        [resourceId],
-        title,
-        metadata,
-        [createdAt],
-        [updatedAt]
-      FROM ${this.getTableName(TABLE_THREADS)}
-      WHERE id = @threadId`;
-      const request = this.pool.request();
-      request.input("threadId", threadId);
-      const resultSet = await request.query(sql2);
-      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
+            threadId
           }
         },
         error
@@ -625,11 +295,11 @@ ${columns}
     }
   }
   async getThreadsByResourceIdPaginated(args) {
-    const { resourceId, page = 0, perPage: perPageInput } = 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 ${this.getTableName(TABLE_THREADS)} WHERE [resourceId] = @resourceId`;
+      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);
@@ -644,7 +314,9 @@ ${columns}
           hasMore: false
         };
       }
-      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY [seq_id] DESC OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
+      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
+      const dir = (sortDirection || "DESC").toUpperCase() === "ASC" ? "ASC" : "DESC";
+      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${dir} OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
       const dataRequest = this.pool.request();
       dataRequest.input("resourceId", resourceId);
       dataRequest.input("perPage", perPage);
@@ -684,7 +356,7 @@ ${columns}
   }
   async saveThread({ thread }) {
     try {
-      const table = this.getTableName(TABLE_THREADS);
+      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)
@@ -693,7 +365,6 @@ ${columns}
             [resourceId] = @resourceId,
             title = @title,
             metadata = @metadata,
-            [createdAt] = @createdAt,
             [updatedAt] = @updatedAt
         WHEN NOT MATCHED THEN
           INSERT (id, [resourceId], title, metadata, [createdAt], [updatedAt])
@@ -702,9 +373,14 @@ ${columns}
       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", thread.createdAt);
-      req.input("updatedAt", thread.updatedAt);
+      const metadata = thread.metadata ? JSON.stringify(thread.metadata) : null;
+      if (metadata === null) {
+        req.input("metadata", sql3.NVarChar, null);
+      } else {
+        req.input("metadata", metadata);
+      }
+      req.input("createdAt", sql3.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql3.DateTime2, thread.updatedAt);
       await req.query(mergeSql);
       return thread;
     } catch (error) {
@@ -725,10 +401,12 @@ ${columns}
    * @deprecated use getThreadsByResourceIdPaginated instead
    */
   async getThreadsByResourceId(args) {
-    const { resourceId } = args;
+    const { resourceId, orderBy = "createdAt", sortDirection = "DESC" } = args;
     try {
-      const baseQuery = `FROM ${this.getTableName(TABLE_THREADS)} WHERE [resourceId] = @resourceId`;
-      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY [seq_id] DESC`;
+      const baseQuery = `FROM ${getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) })} WHERE [resourceId] = @resourceId`;
+      const orderByField = orderBy === "createdAt" ? "[createdAt]" : "[updatedAt]";
+      const dir = (sortDirection || "DESC").toUpperCase() === "ASC" ? "ASC" : "DESC";
+      const dataQuery = `SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt] ${baseQuery} ORDER BY ${orderByField} ${dir}`;
       const request = this.pool.request();
       request.input("resourceId", resourceId);
       const resultSet = await request.query(dataQuery);
@@ -770,8 +448,8 @@ ${columns}
       ...metadata
     };
     try {
-      const table = this.getTableName(TABLE_THREADS);
-      const sql2 = `UPDATE ${table}
+      const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
+      const sql6 = `UPDATE ${table}
         SET title = @title,
             metadata = @metadata,
             [updatedAt] = @updatedAt
@@ -781,8 +459,8 @@ ${columns}
       req.input("id", id);
       req.input("title", title);
       req.input("metadata", JSON.stringify(mergedMetadata));
-      req.input("updatedAt", (/* @__PURE__ */ new Date()).toISOString());
-      const result = await req.query(sql2);
+      req.input("updatedAt", /* @__PURE__ */ new Date());
+      const result = await req.query(sql6);
       let thread = result.recordset && result.recordset[0];
       if (thread && "seq_id" in thread) {
         const { seq_id, ...rest } = thread;
@@ -822,8 +500,8 @@ ${columns}
     }
   }
   async deleteThread({ threadId }) {
-    const messagesTable = this.getTableName(TABLE_MESSAGES);
-    const threadsTable = this.getTableName(TABLE_THREADS);
+    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();
@@ -855,6 +533,7 @@ ${columns}
     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 = [];
@@ -881,7 +560,7 @@ ${columns}
             m.seq_id
           FROM (
             SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
-            FROM ${this.getTableName(TABLE_MESSAGES)}
+            FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
             WHERE [thread_id] = ${pThreadId}
           ) AS m
           WHERE m.id = ${pId}
@@ -889,7 +568,7 @@ ${columns}
             SELECT 1
             FROM (
               SELECT *, ROW_NUMBER() OVER (${orderByStatement}) as row_num
-              FROM ${this.getTableName(TABLE_MESSAGES)}
+              FROM ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })}
               WHERE [thread_id] = ${pThreadId}
             ) AS target
             WHERE target.id = ${pId}
@@ -926,11 +605,12 @@ ${columns}
     return dedupedRows;
   }
   async getMessages(args) {
-    const { threadId, format, selectBy } = args;
-    const selectStatement = `SELECT seq_id, id, content, role, type, [createdAt], thread_id AS threadId`;
+    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 = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
+    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) {
@@ -940,7 +620,7 @@ ${columns}
         }
       }
       const excludeIds = rows.map((m) => m.id).filter(Boolean);
-      let query = `${selectStatement} FROM ${this.getTableName(TABLE_MESSAGES)} WHERE [thread_id] = @threadId`;
+      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) {
@@ -960,30 +640,7 @@ ${columns}
         return timeDiff;
       });
       rows = rows.map(({ seq_id, ...rest }) => rest);
-      const fetchedMessages = (rows || []).map((message) => {
-        if (typeof message.content === "string") {
-          try {
-            message.content = JSON.parse(message.content);
-          } catch {
-          }
-        }
-        if (format === "v1") {
-          if (Array.isArray(message.content)) ; else if (typeof message.content === "object" && message.content && Array.isArray(message.content.parts)) {
-            message.content = message.content.parts;
-          } else {
-            message.content = [{ type: "text", text: "" }];
-          }
-        } else {
-          if (typeof message.content !== "object" || !message.content || !("parts" in message.content)) {
-            message.content = { format: 2, parts: [{ type: "text", text: "" }] };
-          }
-        }
-        if (message.type === "v2") delete message.type;
-        return message;
-      });
-      return format === "v2" ? fetchedMessages.map(
-        (m) => ({ ...m, content: m.content || { format: 2, parts: [{ type: "text", text: "" }] } })
-      ) : fetchedMessages;
+      return this._parseAndFormatMessages(rows, format);
     } catch (error) {
       const mastraError = new MastraError(
         {
@@ -991,40 +648,76 @@ ${columns}
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            threadId
+            threadId,
+            resourceId: resourceId ?? ""
           }
         },
         error
       );
       this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException(mastraError);
+      this.logger?.trackException?.(mastraError);
       return [];
     }
   }
-  async getMessagesPaginated(args) {
-    const { threadId, selectBy } = args;
-    const { page = 0, perPage: perPageInput } = selectBy?.pagination || {};
+  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`;
-    if (selectBy?.include?.length) {
-      await this._getIncludedMessages({ threadId, selectBy, orderByStatement });
+    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 {
-      const { threadId: threadId2, format, selectBy: selectBy2 } = args;
-      const { page: page2 = 0, perPage: perPageInput2, dateRange } = selectBy2?.pagination || {};
+      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`;
-      const orderByStatement2 = `ORDER BY [seq_id] DESC`;
-      let messages2 = [];
-      if (selectBy2?.include?.length) {
-        const includeMessages = await this._getIncludedMessages({ threadId: threadId2, selectBy: selectBy2, orderByStatement: orderByStatement2 });
-        if (includeMessages) messages2.push(...includeMessages);
-      }
-      const perPage = perPageInput2 !== void 0 ? perPageInput2 : this.resolveMessageLimit({ last: selectBy2?.last, defaultLimit: 40 });
-      const currentOffset = page2 * perPage;
+      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", threadId2);
+      request.input("threadId", threadId);
       if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
         conditions.push("[createdAt] >= @fromDate");
         request.input("fromDate", fromDate.toISOString());
@@ -1034,38 +727,38 @@ ${columns}
         request.input("toDate", toDate.toISOString());
       }
       const whereClause = `WHERE ${conditions.join(" AND ")}`;
-      const countQuery = `SELECT COUNT(*) as total FROM ${this.getTableName(TABLE_MESSAGES)} ${whereClause}`;
+      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 && messages2.length > 0) {
-        const parsedIncluded = this._parseAndFormatMessages(messages2, format);
+      if (total === 0 && messages.length > 0) {
+        const parsedIncluded = this._parseAndFormatMessages(messages, format);
         return {
           messages: parsedIncluded,
           total: parsedIncluded.length,
-          page: page2,
+          page,
           perPage,
           hasMore: false
         };
       }
-      const excludeIds = messages2.map((m) => m.id);
+      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 ${this.getTableName(TABLE_MESSAGES)} ${finalWhereClause} ${orderByStatement2} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`;
+      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);
-      messages2.push(...rows);
-      const parsed = this._parseAndFormatMessages(messages2, format);
+      messages.push(...rows);
+      const parsed = this._parseAndFormatMessages(messages, format);
       return {
         messages: parsed,
-        total: total + excludeIds.length,
-        page: page2,
+        total,
+        page,
         perPage,
         hasMore: currentOffset + rows.length < total
       };
@@ -1077,41 +770,17 @@ ${columns}
           category: ErrorCategory.THIRD_PARTY,
           details: {
             threadId,
+            resourceId: resourceId ?? "",
             page
           }
         },
         error
       );
       this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException(mastraError);
+      this.logger?.trackException?.(mastraError);
       return { messages: [], total: 0, page, perPage: perPageInput || 40, hasMore: false };
     }
   }
-  _parseAndFormatMessages(messages, format) {
-    const parsedMessages = messages.map((message) => {
-      let parsed = message;
-      if (typeof parsed.content === "string") {
-        try {
-          parsed = { ...parsed, content: JSON.parse(parsed.content) };
-        } catch {
-        }
-      }
-      if (format === "v1") {
-        if (Array.isArray(parsed.content)) ; else if (parsed.content?.parts) {
-          parsed.content = parsed.content.parts;
-        } else {
-          parsed.content = [{ type: "text", text: "" }];
-        }
-      } else {
-        if (!parsed.content?.parts) {
-          parsed = { ...parsed, content: { format: 2, parts: [{ type: "text", text: "" }] } };
-        }
-      }
-      return parsed;
-    });
-    const list = new MessageList().add(parsedMessages, "memory");
-    return format === "v2" ? list.get.all.v2() : list.get.all.v1();
-  }
   async saveMessages({
     messages,
     format
@@ -1136,8 +805,8 @@ ${columns}
         details: { threadId }
       });
     }
-    const tableMessages = this.getTableName(TABLE_MESSAGES);
-    const tableThreads = this.getTableName(TABLE_THREADS);
+    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();
@@ -1160,7 +829,7 @@ ${columns}
             "content",
             typeof message.content === "string" ? message.content : JSON.stringify(message.content)
           );
-          request.input("createdAt", message.createdAt.toISOString() || (/* @__PURE__ */ new Date()).toISOString());
+          request.input("createdAt", sql3.DateTime2, message.createdAt);
           request.input("role", message.role);
           request.input("type", message.type || "v2");
           request.input("resourceId", message.resourceId);
@@ -1179,7 +848,7 @@ ${columns}
           await request.query(mergeSql);
         }
         const threadReq = transaction.request();
-        threadReq.input("updatedAt", (/* @__PURE__ */ new Date()).toISOString());
+        threadReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
         threadReq.input("id", threadId);
         await threadReq.query(`UPDATE ${tableThreads} SET [updatedAt] = @updatedAt WHERE id = @id`);
         await transaction.commit();
@@ -1212,216 +881,6 @@ ${columns}
       );
     }
   }
-  async persistWorkflowSnapshot({
-    workflowName,
-    runId,
-    snapshot
-  }) {
-    const table = this.getTableName(TABLE_WORKFLOW_SNAPSHOT);
-    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", now);
-      request.input("updatedAt", 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.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 hasColumn(table, column) {
-    const schema = this.schema || "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;
-  }
-  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
-    };
-  }
-  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.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 = this.getTableName(TABLE_WORKFLOW_SNAPSHOT);
-      const request = this.pool.request();
-      Object.entries(paramMap).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          request.input(key, sql.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) => this.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
-      );
-    }
-  }
-  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 = this.getTableName(TABLE_WORKFLOW_SNAPSHOT);
-      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 this.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 updateMessages({
     messages
   }) {
@@ -1430,7 +889,7 @@ ${columns}
     }
     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 ${this.getTableName(TABLE_MESSAGES)}`;
+    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 {
@@ -1487,7 +946,7 @@ ${columns}
           }
         }
         if (setClauses.length > 0) {
-          const updateSql = `UPDATE ${this.getTableName(TABLE_MESSAGES)} SET ${setClauses.join(", ")} WHERE id = @id`;
+          const updateSql = `UPDATE ${getTableName({ indexName: TABLE_MESSAGES, schemaName: getSchemaName(this.schema) })} SET ${setClauses.join(", ")} WHERE id = @id`;
           await req.query(updateSql);
         }
       }
@@ -1496,7 +955,7 @@ ${columns}
         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 ${this.getTableName(TABLE_THREADS)} SET updatedAt = @updatedAt WHERE id IN (${threadIdParams})`;
+        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();
@@ -1524,137 +983,98 @@ ${columns}
       return message;
     });
   }
-  async close() {
-    if (this.pool) {
+  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 {
-        if (this.pool.connected) {
-          await this.pool.close();
-        } else if (this.pool.connecting) {
-          await this.pool.connect();
-          await this.pool.close();
+        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`);
+          }
         }
-      } catch (err) {
-        if (err.message && err.message.includes("Cannot close a pool while it is connecting")) ; else {
-          throw err;
-        }
-      }
-    }
-  }
-  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 = this.getTableName(TABLE_EVALS);
-    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, sql.DateTime, value);
-        } else {
-          countReq.input(key, value);
+        await transaction.commit();
+      } catch (error) {
+        try {
+          await transaction.rollback();
+        } catch {
         }
-      });
-      const countResult = await countReq.query(countQuery);
-      const total = countResult.recordset[0]?.total || 0;
-      if (total === 0) {
-        return {
-          evals: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
+        throw error;
       }
-      const req = this.pool.request();
-      Object.entries(params).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          req.input(key, sql.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) => this.transformEvalRow(row)) ?? [],
-        total,
-        page,
-        perPage,
-        hasMore: offset + (rows?.length ?? 0) < total
-      };
     } catch (error) {
-      const mastraError = new MastraError(
+      throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_EVALS_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_STORE_DELETE_MESSAGES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: {
-            agentName: agentName || "all",
-            type: type || "all",
-            page,
-            perPage
-          }
+          details: { messageIds: messageIds.join(", ") }
         },
         error
       );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      throw mastraError;
     }
   }
-  async saveResource({ resource }) {
-    const tableName = this.getTableName(TABLE_RESOURCES);
+  async getResourceById({ resourceId }) {
+    const tableName = getTableName({ indexName: TABLE_RESOURCES, schemaName: getSchemaName(this.schema) });
     try {
       const req = this.pool.request();
-      req.input("id", resource.id);
-      req.input("workingMemory", resource.workingMemory);
-      req.input("metadata", JSON.stringify(resource.metadata));
-      req.input("createdAt", resource.createdAt.toISOString());
-      req.input("updatedAt", resource.updatedAt.toISOString());
-      await req.query(
-        `INSERT INTO ${tableName} (id, workingMemory, metadata, createdAt, updatedAt) VALUES (@id, @workingMemory, @metadata, @createdAt, @updatedAt)`
-      );
-      return resource;
+      req.input("resourceId", resourceId);
+      const result = (await req.query(`SELECT * FROM ${tableName} WHERE id = @resourceId`)).recordset[0];
+      if (!result) {
+        return null;
+      }
+      return {
+        id: result.id,
+        createdAt: result.createdAt,
+        updatedAt: result.updatedAt,
+        workingMemory: result.workingMemory,
+        metadata: typeof result.metadata === "string" ? JSON.parse(result.metadata) : result.metadata
+      };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_SAVE_RESOURCE_FAILED",
+          id: "MASTRA_STORAGE_MSSQL_GET_RESOURCE_BY_ID_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { resourceId: resource.id }
+          details: { resourceId }
         },
         error
       );
       this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException(mastraError);
+      this.logger?.trackException?.(mastraError);
       throw mastraError;
     }
   }
+  async saveResource({ resource }) {
+    await this.operations.insert({
+      tableName: TABLE_RESOURCES,
+      record: {
+        ...resource,
+        metadata: resource.metadata
+      }
+    });
+    return resource;
+  }
   async updateResource({
     resourceId,
     workingMemory,
@@ -1681,7 +1101,7 @@ ${columns}
         },
         updatedAt: /* @__PURE__ */ new Date()
       };
-      const tableName = this.getTableName(TABLE_RESOURCES);
+      const tableName = getTableName({ indexName: TABLE_RESOURCES, schemaName: getSchemaName(this.schema) });
       const updates = [];
       const req = this.pool.request();
       if (workingMemory !== void 0) {
@@ -1708,39 +1128,2326 @@ ${columns}
         error
       );
       this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException(mastraError);
+      this.logger?.trackException?.(mastraError);
       throw mastraError;
     }
   }
-  async getResourceById({ resourceId }) {
-    const tableName = this.getTableName(TABLE_RESOURCES);
+};
+var ObservabilityMSSQL = class extends ObservabilityStorage {
+  pool;
+  operations;
+  schema;
+  constructor({
+    pool,
+    operations,
+    schema
+  }) {
+    super();
+    this.pool = pool;
+    this.operations = operations;
+    this.schema = schema;
+  }
+  get aiTracingStrategy() {
+    return {
+      preferred: "batch-with-updates",
+      supported: ["batch-with-updates", "insert-only"]
+    };
+  }
+  async createAISpan(span) {
     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) {
+      const startedAt = span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt;
+      const endedAt = span.endedAt instanceof Date ? span.endedAt.toISOString() : span.endedAt;
+      const record = {
+        ...span,
+        startedAt,
+        endedAt
+        // Note: createdAt/updatedAt will be set by default values
+      };
+      return this.operations.insert({ tableName: TABLE_AI_SPANS, record });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_CREATE_AI_SPAN_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            spanId: span.spanId,
+            traceId: span.traceId,
+            spanType: span.spanType,
+            spanName: span.name
+          }
+        },
+        error
+      );
+    }
+  }
+  async getAITrace(traceId) {
+    try {
+      const tableName = getTableName({
+        indexName: TABLE_AI_SPANS,
+        schemaName: getSchemaName(this.schema)
+      });
+      const request = this.pool.request();
+      request.input("traceId", traceId);
+      const result = await request.query(
+        `SELECT
+          [traceId], [spanId], [parentSpanId], [name], [scope], [spanType],
+          [attributes], [metadata], [links], [input], [output], [error], [isEvent],
+          [startedAt], [endedAt], [createdAt], [updatedAt]
+        FROM ${tableName}
+        WHERE [traceId] = @traceId
+        ORDER BY [startedAt] DESC`
+      );
+      if (!result.recordset || result.recordset.length === 0) {
         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
+        traceId,
+        spans: result.recordset.map(
+          (span) => transformFromSqlRow({
+            tableName: TABLE_AI_SPANS,
+            sqlRow: span
+          })
+        )
       };
     } catch (error) {
-      const mastraError = new MastraError(
+      throw new MastraError(
         {
-          id: "MASTRA_STORAGE_MSSQL_GET_RESOURCE_BY_ID_FAILED",
+          id: "MSSQL_STORE_GET_AI_TRACE_FAILED",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { resourceId }
+          category: ErrorCategory.USER,
+          details: {
+            traceId
+          }
+        },
+        error
+      );
+    }
+  }
+  async updateAISpan({
+    spanId,
+    traceId,
+    updates
+  }) {
+    try {
+      const data = { ...updates };
+      if (data.endedAt instanceof Date) {
+        data.endedAt = data.endedAt.toISOString();
+      }
+      if (data.startedAt instanceof Date) {
+        data.startedAt = data.startedAt.toISOString();
+      }
+      await this.operations.update({
+        tableName: TABLE_AI_SPANS,
+        keys: { spanId, traceId },
+        data
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_UPDATE_AI_SPAN_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            spanId,
+            traceId
+          }
+        },
+        error
+      );
+    }
+  }
+  async getAITracesPaginated({
+    filters,
+    pagination
+  }) {
+    const page = pagination?.page ?? 0;
+    const perPage = pagination?.perPage ?? 10;
+    const { entityId, entityType, ...actualFilters } = filters || {};
+    const filtersWithDateRange = {
+      ...actualFilters,
+      ...buildDateRangeFilter(pagination?.dateRange, "startedAt"),
+      parentSpanId: null
+      // Only get root spans for traces
+    };
+    const whereClause = prepareWhereClause(filtersWithDateRange);
+    let actualWhereClause = whereClause.sql;
+    const params = { ...whereClause.params };
+    let currentParamIndex = Object.keys(params).length + 1;
+    if (entityId && entityType) {
+      let name = "";
+      if (entityType === "workflow") {
+        name = `workflow run: '${entityId}'`;
+      } else if (entityType === "agent") {
+        name = `agent run: '${entityId}'`;
+      } else {
+        const error = new MastraError({
+          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            entityType
+          },
+          text: `Cannot filter by entity type: ${entityType}`
+        });
+        throw error;
+      }
+      const entityParam = `p${currentParamIndex++}`;
+      if (actualWhereClause) {
+        actualWhereClause += ` AND [name] = @${entityParam}`;
+      } else {
+        actualWhereClause = ` WHERE [name] = @${entityParam}`;
+      }
+      params[entityParam] = name;
+    }
+    const tableName = getTableName({
+      indexName: TABLE_AI_SPANS,
+      schemaName: getSchemaName(this.schema)
+    });
+    try {
+      const countRequest = this.pool.request();
+      Object.entries(params).forEach(([key, value]) => {
+        countRequest.input(key, value);
+      });
+      const countResult = await countRequest.query(
+        `SELECT COUNT(*) as count FROM ${tableName}${actualWhereClause}`
+      );
+      const total = countResult.recordset[0]?.count ?? 0;
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page,
+            perPage,
+            hasMore: false
+          },
+          spans: []
+        };
+      }
+      const dataRequest = this.pool.request();
+      Object.entries(params).forEach(([key, value]) => {
+        dataRequest.input(key, value);
+      });
+      dataRequest.input("offset", page * perPage);
+      dataRequest.input("limit", perPage);
+      const dataResult = await dataRequest.query(
+        `SELECT * FROM ${tableName}${actualWhereClause} ORDER BY [startedAt] DESC OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`
+      );
+      const spans = dataResult.recordset.map(
+        (row) => transformFromSqlRow({
+          tableName: TABLE_AI_SPANS,
+          sqlRow: row
+        })
+      );
+      return {
+        pagination: {
+          total,
+          page,
+          perPage,
+          hasMore: (page + 1) * perPage < total
+        },
+        spans
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER
+        },
+        error
+      );
+    }
+  }
+  async batchCreateAISpans(args) {
+    if (!args.records || args.records.length === 0) {
+      return;
+    }
+    try {
+      await this.operations.batchInsert({
+        tableName: TABLE_AI_SPANS,
+        records: args.records.map((span) => ({
+          ...span,
+          startedAt: span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt,
+          endedAt: span.endedAt instanceof Date ? span.endedAt.toISOString() : span.endedAt
+        }))
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_BATCH_CREATE_AI_SPANS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            count: args.records.length
+          }
+        },
+        error
+      );
+    }
+  }
+  async batchUpdateAISpans(args) {
+    if (!args.records || args.records.length === 0) {
+      return;
+    }
+    try {
+      const updates = args.records.map(({ traceId, spanId, updates: data }) => {
+        const processedData = { ...data };
+        if (processedData.endedAt instanceof Date) {
+          processedData.endedAt = processedData.endedAt.toISOString();
+        }
+        if (processedData.startedAt instanceof Date) {
+          processedData.startedAt = processedData.startedAt.toISOString();
+        }
+        return {
+          keys: { spanId, traceId },
+          data: processedData
+        };
+      });
+      await this.operations.batchUpdate({
+        tableName: TABLE_AI_SPANS,
+        updates
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_BATCH_UPDATE_AI_SPANS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            count: args.records.length
+          }
+        },
+        error
+      );
+    }
+  }
+  async batchDeleteAITraces(args) {
+    if (!args.traceIds || args.traceIds.length === 0) {
+      return;
+    }
+    try {
+      const keys = args.traceIds.map((traceId) => ({ traceId }));
+      await this.operations.batchDelete({
+        tableName: TABLE_AI_SPANS,
+        keys
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MSSQL_STORE_BATCH_DELETE_AI_TRACES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            count: args.traceIds.length
+          }
         },
         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, useLargeStorage = false) {
+    switch (type) {
+      case "text":
+        if (useLargeStorage) {
+          return "NVARCHAR(MAX)";
+        }
+        return isPrimaryKey ? "NVARCHAR(255)" : "NVARCHAR(400)";
+      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";
+      case "boolean":
+        return "BIT";
+      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,
+    transaction
+  }) {
+    try {
+      const columns = Object.keys(record);
+      const parsedColumns = columns.map((col) => parseSqlIdentifier(col, "column name"));
+      const paramNames = columns.map((_, i) => `@param${i}`);
+      const insertSql = `INSERT INTO ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} (${parsedColumns.map((c) => `[${c}]`).join(", ")}) VALUES (${paramNames.join(", ")})`;
+      const request = transaction ? transaction.request() : this.pool.request();
+      columns.forEach((col, i) => {
+        const value = record[col];
+        const preparedValue = this.prepareValue(value, col, tableName);
+        if (preparedValue instanceof Date) {
+          request.input(`param${i}`, sql3.DateTime2, preparedValue);
+        } else if (preparedValue === null || preparedValue === void 0) {
+          request.input(`param${i}`, this.getMssqlType(tableName, col), null);
+        } else {
+          request.input(`param${i}`, preparedValue);
+        }
+      });
+      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?.number === 4712) {
+          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 SYSUTCDATETIME()";
+      case "jsonb":
+        return "DEFAULT N'{}'";
+      case "boolean":
+        return "DEFAULT 0";
+      default:
+        return super.getDefaultValue(type);
+    }
+  }
+  async createTable({
+    tableName,
+    schema
+  }) {
+    try {
+      const uniqueConstraintColumns = tableName === TABLE_WORKFLOW_SNAPSHOT ? ["workflow_name", "run_id"] : [];
+      const largeDataColumns = [
+        "workingMemory",
+        "snapshot",
+        "metadata",
+        "content",
+        // messages.content - can be very long conversation content
+        "input",
+        // evals.input - test input data
+        "output",
+        // evals.output - test output data
+        "instructions",
+        // evals.instructions - evaluation instructions
+        "other"
+        // traces.other - additional trace data
+      ];
+      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);
+        const useLargeStorage = largeDataColumns.includes(name);
+        return `[${parsedName}] ${this.getSqlType(def.type, isIndexed, useLargeStorage)} ${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 largeDataColumns = [
+              "workingMemory",
+              "snapshot",
+              "metadata",
+              "content",
+              "input",
+              "output",
+              "instructions",
+              "other"
+            ];
+            const useLargeStorage = largeDataColumns.includes(columnName);
+            const isIndexed = !!columnDef.primaryKey;
+            const sqlType = this.getSqlType(columnDef.type, isIndexed, useLargeStorage);
+            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 sql6 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
+      const request = this.pool.request();
+      keyEntries.forEach(([key, value], i) => {
+        const preparedValue = this.prepareValue(value, key, tableName);
+        if (preparedValue === null || preparedValue === void 0) {
+          request.input(`param${i}`, this.getMssqlType(tableName, key), null);
+        } else {
+          request.input(`param${i}`, preparedValue);
+        }
+      });
+      const resultSet = await request.query(sql6);
+      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, transaction });
+      }
+      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
+      );
+    }
+  }
+  /**
+   * Prepares a value for database operations, handling Date objects and JSON serialization
+   */
+  prepareValue(value, columnName, tableName) {
+    if (value === null || value === void 0) {
+      return value;
+    }
+    if (value instanceof Date) {
+      return value;
+    }
+    const schema = TABLE_SCHEMAS[tableName];
+    const columnSchema = schema?.[columnName];
+    if (columnSchema?.type === "boolean") {
+      return value ? 1 : 0;
+    }
+    if (columnSchema?.type === "jsonb") {
+      return JSON.stringify(value);
+    }
+    if (typeof value === "object") {
+      return JSON.stringify(value);
+    }
+    return value;
+  }
+  /**
+   * Maps TABLE_SCHEMAS types to mssql param types (used when value is null)
+   */
+  getMssqlType(tableName, columnName) {
+    const col = TABLE_SCHEMAS[tableName]?.[columnName];
+    switch (col?.type) {
+      case "text":
+        return sql3.NVarChar;
+      case "timestamp":
+        return sql3.DateTime2;
+      case "uuid":
+        return sql3.UniqueIdentifier;
+      case "jsonb":
+        return sql3.NVarChar;
+      case "integer":
+        return sql3.Int;
+      case "bigint":
+        return sql3.BigInt;
+      case "float":
+        return sql3.Float;
+      case "boolean":
+        return sql3.Bit;
+      default:
+        return sql3.NVarChar;
+    }
+  }
+  /**
+   * Update a single record in the database
+   */
+  async update({
+    tableName,
+    keys,
+    data,
+    transaction
+  }) {
+    try {
+      if (!data || Object.keys(data).length === 0) {
+        throw new MastraError({
+          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_DATA",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: "Cannot update with empty data payload"
+        });
+      }
+      if (!keys || Object.keys(keys).length === 0) {
+        throw new MastraError({
+          id: "MASTRA_STORAGE_MSSQL_UPDATE_EMPTY_KEYS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: "Cannot update without keys to identify records"
+        });
+      }
+      const setClauses = [];
+      const request = transaction ? transaction.request() : this.pool.request();
+      let paramIndex = 0;
+      Object.entries(data).forEach(([key, value]) => {
+        const parsedKey = parseSqlIdentifier(key, "column name");
+        const paramName = `set${paramIndex++}`;
+        setClauses.push(`[${parsedKey}] = @${paramName}`);
+        const preparedValue = this.prepareValue(value, key, tableName);
+        if (preparedValue === null || preparedValue === void 0) {
+          request.input(paramName, this.getMssqlType(tableName, key), null);
+        } else {
+          request.input(paramName, preparedValue);
+        }
+      });
+      const whereConditions = [];
+      Object.entries(keys).forEach(([key, value]) => {
+        const parsedKey = parseSqlIdentifier(key, "column name");
+        const paramName = `where${paramIndex++}`;
+        whereConditions.push(`[${parsedKey}] = @${paramName}`);
+        const preparedValue = this.prepareValue(value, key, tableName);
+        if (preparedValue === null || preparedValue === void 0) {
+          request.input(paramName, this.getMssqlType(tableName, key), null);
+        } else {
+          request.input(paramName, preparedValue);
+        }
+      });
+      const tableName_ = getTableName({
+        indexName: tableName,
+        schemaName: getSchemaName(this.schemaName)
+      });
+      const updateSql = `UPDATE ${tableName_} SET ${setClauses.join(", ")} WHERE ${whereConditions.join(" AND ")}`;
+      await request.query(updateSql);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Update multiple records in a single batch transaction
+   */
+  async batchUpdate({
+    tableName,
+    updates
+  }) {
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      for (const { keys, data } of updates) {
+        await this.update({ tableName, keys, data, transaction });
+      }
+      await transaction.commit();
+    } catch (error) {
+      await transaction.rollback();
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_UPDATE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName,
+            numberOfRecords: updates.length
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Delete multiple records by keys
+   */
+  async batchDelete({ tableName, keys }) {
+    if (keys.length === 0) {
+      return;
+    }
+    const tableName_ = getTableName({
+      indexName: tableName,
+      schemaName: getSchemaName(this.schemaName)
+    });
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      for (const keySet of keys) {
+        const conditions = [];
+        const request = transaction.request();
+        let paramIndex = 0;
+        Object.entries(keySet).forEach(([key, value]) => {
+          const parsedKey = parseSqlIdentifier(key, "column name");
+          const paramName = `p${paramIndex++}`;
+          conditions.push(`[${parsedKey}] = @${paramName}`);
+          const preparedValue = this.prepareValue(value, key, tableName);
+          if (preparedValue === null || preparedValue === void 0) {
+            request.input(paramName, this.getMssqlType(tableName, key), null);
+          } else {
+            request.input(paramName, preparedValue);
+          }
+        });
+        const deleteSql = `DELETE FROM ${tableName_} WHERE ${conditions.join(" AND ")}`;
+        await request.query(deleteSql);
+      }
+      await transaction.commit();
+    } catch (error) {
+      await transaction.rollback();
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_BATCH_DELETE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName,
+            numberOfRecords: keys.length
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Create a new index on a table
+   */
+  async createIndex(options) {
+    try {
+      const { name, table, columns, unique = false, where } = options;
+      const schemaName = this.schemaName || "dbo";
+      const fullTableName = getTableName({
+        indexName: table,
+        schemaName: getSchemaName(this.schemaName)
+      });
+      const indexNameSafe = parseSqlIdentifier(name, "index name");
+      const checkRequest = this.pool.request();
+      checkRequest.input("indexName", indexNameSafe);
+      checkRequest.input("schemaName", schemaName);
+      checkRequest.input("tableName", table);
+      const indexExists = await checkRequest.query(`
+        SELECT 1 as found
+        FROM sys.indexes i
+        INNER JOIN sys.tables t ON i.object_id = t.object_id
+        INNER JOIN sys.schemas s ON t.schema_id = s.schema_id
+        WHERE i.name = @indexName
+          AND s.name = @schemaName
+          AND t.name = @tableName
+      `);
+      if (indexExists.recordset && indexExists.recordset.length > 0) {
+        return;
+      }
+      const uniqueStr = unique ? "UNIQUE " : "";
+      const columnsStr = columns.map((col) => {
+        if (col.includes(" DESC") || col.includes(" ASC")) {
+          const [colName, ...modifiers] = col.split(" ");
+          if (!colName) {
+            throw new Error(`Invalid column specification: ${col}`);
+          }
+          return `[${parseSqlIdentifier(colName, "column name")}] ${modifiers.join(" ")}`;
+        }
+        return `[${parseSqlIdentifier(col, "column name")}]`;
+      }).join(", ");
+      const whereStr = where ? ` WHERE ${where}` : "";
+      const createIndexSql = `CREATE ${uniqueStr}INDEX [${indexNameSafe}] ON ${fullTableName} (${columnsStr})${whereStr}`;
+      await this.pool.request().query(createIndexSql);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_INDEX_CREATE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            indexName: options.name,
+            tableName: options.table
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Drop an existing index
+   */
+  async dropIndex(indexName) {
+    try {
+      const schemaName = this.schemaName || "dbo";
+      const indexNameSafe = parseSqlIdentifier(indexName, "index name");
+      const checkRequest = this.pool.request();
+      checkRequest.input("indexName", indexNameSafe);
+      checkRequest.input("schemaName", schemaName);
+      const result = await checkRequest.query(`
+        SELECT t.name as table_name
+        FROM sys.indexes i
+        INNER JOIN sys.tables t ON i.object_id = t.object_id
+        INNER JOIN sys.schemas s ON t.schema_id = s.schema_id
+        WHERE i.name = @indexName
+          AND s.name = @schemaName
+      `);
+      if (!result.recordset || result.recordset.length === 0) {
+        return;
+      }
+      if (result.recordset.length > 1) {
+        const tables = result.recordset.map((r) => r.table_name).join(", ");
+        throw new MastraError({
+          id: "MASTRA_STORAGE_MSSQL_INDEX_AMBIGUOUS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: `Index "${indexNameSafe}" exists on multiple tables (${tables}) in schema "${schemaName}". Please drop indexes manually or ensure unique index names.`
+        });
+      }
+      const tableName = result.recordset[0].table_name;
+      const fullTableName = getTableName({
+        indexName: tableName,
+        schemaName: getSchemaName(this.schemaName)
+      });
+      const dropSql = `DROP INDEX [${indexNameSafe}] ON ${fullTableName}`;
+      await this.pool.request().query(dropSql);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_INDEX_DROP_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            indexName
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * List indexes for a specific table or all tables
+   */
+  async listIndexes(tableName) {
+    try {
+      const schemaName = this.schemaName || "dbo";
+      let query;
+      const request = this.pool.request();
+      request.input("schemaName", schemaName);
+      if (tableName) {
+        query = `
+          SELECT 
+            i.name as name,
+            o.name as [table],
+            i.is_unique as is_unique,
+            CAST(SUM(s.used_page_count) * 8 / 1024.0 AS VARCHAR(50)) + ' MB' as size
+          FROM sys.indexes i
+          INNER JOIN sys.objects o ON i.object_id = o.object_id
+          INNER JOIN sys.schemas sch ON o.schema_id = sch.schema_id
+          LEFT JOIN sys.dm_db_partition_stats s ON i.object_id = s.object_id AND i.index_id = s.index_id
+          WHERE sch.name = @schemaName
+          AND o.name = @tableName
+          AND i.name IS NOT NULL
+          GROUP BY i.name, o.name, i.is_unique
+        `;
+        request.input("tableName", tableName);
+      } else {
+        query = `
+          SELECT 
+            i.name as name,
+            o.name as [table],
+            i.is_unique as is_unique,
+            CAST(SUM(s.used_page_count) * 8 / 1024.0 AS VARCHAR(50)) + ' MB' as size
+          FROM sys.indexes i
+          INNER JOIN sys.objects o ON i.object_id = o.object_id
+          INNER JOIN sys.schemas sch ON o.schema_id = sch.schema_id
+          LEFT JOIN sys.dm_db_partition_stats s ON i.object_id = s.object_id AND i.index_id = s.index_id
+          WHERE sch.name = @schemaName
+          AND i.name IS NOT NULL
+          GROUP BY i.name, o.name, i.is_unique
+        `;
+      }
+      const result = await request.query(query);
+      const indexes = [];
+      for (const row of result.recordset) {
+        const colRequest = this.pool.request();
+        colRequest.input("indexName", row.name);
+        colRequest.input("schemaName", schemaName);
+        const colResult = await colRequest.query(`
+          SELECT c.name as column_name
+          FROM sys.indexes i
+          INNER JOIN sys.index_columns ic ON i.object_id = ic.object_id AND i.index_id = ic.index_id
+          INNER JOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id
+          INNER JOIN sys.objects o ON i.object_id = o.object_id
+          INNER JOIN sys.schemas s ON o.schema_id = s.schema_id
+          WHERE i.name = @indexName
+          AND s.name = @schemaName
+          ORDER BY ic.key_ordinal
+        `);
+        indexes.push({
+          name: row.name,
+          table: row.table,
+          columns: colResult.recordset.map((c) => c.column_name),
+          unique: row.is_unique || false,
+          size: row.size || "0 MB",
+          definition: ""
+          // MSSQL doesn't store definition like PG
+        });
+      }
+      return indexes;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_INDEX_LIST_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: tableName ? {
+            tableName
+          } : {}
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Get detailed statistics for a specific index
+   */
+  async describeIndex(indexName) {
+    try {
+      const schemaName = this.schemaName || "dbo";
+      const request = this.pool.request();
+      request.input("indexName", indexName);
+      request.input("schemaName", schemaName);
+      const query = `
+        SELECT 
+          i.name as name,
+          o.name as [table],
+          i.is_unique as is_unique,
+          CAST(SUM(s.used_page_count) * 8 / 1024.0 AS VARCHAR(50)) + ' MB' as size,
+          i.type_desc as method,
+          ISNULL(us.user_scans, 0) as scans,
+          ISNULL(us.user_seeks + us.user_scans, 0) as tuples_read,
+          ISNULL(us.user_lookups, 0) as tuples_fetched
+        FROM sys.indexes i
+        INNER JOIN sys.objects o ON i.object_id = o.object_id
+        INNER JOIN sys.schemas sch ON o.schema_id = sch.schema_id
+        LEFT JOIN sys.dm_db_partition_stats s ON i.object_id = s.object_id AND i.index_id = s.index_id
+        LEFT JOIN sys.dm_db_index_usage_stats us ON i.object_id = us.object_id AND i.index_id = us.index_id
+        WHERE i.name = @indexName
+        AND sch.name = @schemaName
+        GROUP BY i.name, o.name, i.is_unique, i.type_desc, us.user_seeks, us.user_scans, us.user_lookups
+      `;
+      const result = await request.query(query);
+      if (!result.recordset || result.recordset.length === 0) {
+        throw new Error(`Index "${indexName}" not found in schema "${schemaName}"`);
+      }
+      const row = result.recordset[0];
+      const colRequest = this.pool.request();
+      colRequest.input("indexName", indexName);
+      colRequest.input("schemaName", schemaName);
+      const colResult = await colRequest.query(`
+        SELECT c.name as column_name
+        FROM sys.indexes i
+        INNER JOIN sys.index_columns ic ON i.object_id = ic.object_id AND i.index_id = ic.index_id
+        INNER JOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id
+        INNER JOIN sys.objects o ON i.object_id = o.object_id
+        INNER JOIN sys.schemas s ON o.schema_id = s.schema_id
+        WHERE i.name = @indexName
+        AND s.name = @schemaName
+        ORDER BY ic.key_ordinal
+      `);
+      return {
+        name: row.name,
+        table: row.table,
+        columns: colResult.recordset.map((c) => c.column_name),
+        unique: row.is_unique || false,
+        size: row.size || "0 MB",
+        definition: "",
+        method: row.method?.toLowerCase() || "nonclustered",
+        scans: Number(row.scans) || 0,
+        tuples_read: Number(row.tuples_read) || 0,
+        tuples_fetched: Number(row.tuples_fetched) || 0
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_INDEX_DESCRIBE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            indexName
+          }
+        },
+        error
+      );
+    }
+  }
+  /**
+   * Returns definitions for automatic performance indexes
+   * IMPORTANT: Uses seq_id DESC instead of createdAt DESC for MSSQL due to millisecond accuracy limitations
+   * NOTE: Using NVARCHAR(400) for text columns (800 bytes) leaves room for composite indexes
+   */
+  getAutomaticIndexDefinitions() {
+    const schemaPrefix = this.schemaName ? `${this.schemaName}_` : "";
+    return [
+      // Composite indexes for optimal filtering + sorting performance
+      // NVARCHAR(400) = 800 bytes, plus BIGINT (8 bytes) = 808 bytes total (under 900-byte limit)
+      {
+        name: `${schemaPrefix}mastra_threads_resourceid_seqid_idx`,
+        table: TABLE_THREADS,
+        columns: ["resourceId", "seq_id DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_messages_thread_id_seqid_idx`,
+        table: TABLE_MESSAGES,
+        columns: ["thread_id", "seq_id DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_traces_name_seqid_idx`,
+        table: TABLE_TRACES,
+        columns: ["name", "seq_id DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_evals_agent_name_seqid_idx`,
+        table: TABLE_EVALS,
+        columns: ["agent_name", "seq_id DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_scores_trace_id_span_id_seqid_idx`,
+        table: TABLE_SCORERS,
+        columns: ["traceId", "spanId", "seq_id DESC"]
+      },
+      // AI Spans indexes for optimal trace querying
+      {
+        name: `${schemaPrefix}mastra_ai_spans_traceid_startedat_idx`,
+        table: TABLE_AI_SPANS,
+        columns: ["traceId", "startedAt DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_ai_spans_parentspanid_startedat_idx`,
+        table: TABLE_AI_SPANS,
+        columns: ["parentSpanId", "startedAt DESC"]
+      },
+      {
+        name: `${schemaPrefix}mastra_ai_spans_name_idx`,
+        table: TABLE_AI_SPANS,
+        columns: ["name"]
+      },
+      {
+        name: `${schemaPrefix}mastra_ai_spans_spantype_startedat_idx`,
+        table: TABLE_AI_SPANS,
+        columns: ["spanType", "startedAt DESC"]
+      }
+    ];
+  }
+  /**
+   * Creates automatic indexes for optimal query performance
+   * Uses getAutomaticIndexDefinitions() to determine which indexes to create
+   */
+  async createAutomaticIndexes() {
+    try {
+      const indexes = this.getAutomaticIndexDefinitions();
+      for (const indexOptions of indexes) {
+        try {
+          await this.createIndex(indexOptions);
+        } catch (error) {
+          this.logger?.warn?.(`Failed to create index ${indexOptions.name}:`, error);
+        }
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_CREATE_PERFORMANCE_INDEXES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+};
+function transformScoreRow(row) {
+  return {
+    ...row,
+    input: safelyParseJSON(row.input),
+    scorer: safelyParseJSON(row.scorer),
+    preprocessStepResult: safelyParseJSON(row.preprocessStepResult),
+    analyzeStepResult: safelyParseJSON(row.analyzeStepResult),
+    metadata: safelyParseJSON(row.metadata),
+    output: safelyParseJSON(row.output),
+    additionalContext: safelyParseJSON(row.additionalContext),
+    runtimeContext: safelyParseJSON(row.runtimeContext),
+    entity: safelyParseJSON(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) {
+    let validatedScore;
+    try {
+      validatedScore = saveScorePayloadSchema.parse(score);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_SAVE_SCORE_VALIDATION_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+    try {
+      const scoreId = randomUUID();
+      const {
+        scorer,
+        preprocessStepResult,
+        analyzeStepResult,
+        metadata,
+        input,
+        output,
+        additionalContext,
+        runtimeContext,
+        entity,
+        ...rest
+      } = validatedScore;
+      await this.operations.insert({
+        tableName: TABLE_SCORERS,
+        record: {
+          id: scoreId,
+          ...rest,
+          input: input || "",
+          output: output || "",
+          preprocessStepResult: preprocessStepResult || null,
+          analyzeStepResult: analyzeStepResult || null,
+          metadata: metadata || null,
+          additionalContext: additionalContext || null,
+          runtimeContext: runtimeContext || null,
+          entity: entity || null,
+          scorer: 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,
+    entityId,
+    entityType,
+    source
+  }) {
+    try {
+      const conditions = ["[scorerId] = @p1"];
+      const params = { p1: scorerId };
+      let paramIndex = 2;
+      if (entityId) {
+        conditions.push(`[entityId] = @p${paramIndex}`);
+        params[`p${paramIndex}`] = entityId;
+        paramIndex++;
+      }
+      if (entityType) {
+        conditions.push(`[entityType] = @p${paramIndex}`);
+        params[`p${paramIndex}`] = entityType;
+        paramIndex++;
+      }
+      if (source) {
+        conditions.push(`[source] = @p${paramIndex}`);
+        params[`p${paramIndex}`] = source;
+        paramIndex++;
+      }
+      const whereClause = conditions.join(" AND ");
+      const tableName = getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) });
+      const countRequest = this.pool.request();
+      Object.entries(params).forEach(([key, value]) => {
+        countRequest.input(key, value);
+      });
+      const totalResult = await countRequest.query(`SELECT COUNT(*) as count FROM ${tableName} WHERE ${whereClause}`);
+      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();
+      Object.entries(params).forEach(([key, value]) => {
+        dataRequest.input(key, value);
+      });
+      dataRequest.input("perPage", pagination.perPage);
+      dataRequest.input("offset", pagination.page * pagination.perPage);
+      const dataQuery = `SELECT * FROM ${tableName} WHERE ${whereClause} ORDER BY [createdAt] DESC OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
+      const result = await dataRequest.query(dataQuery);
+      return {
+        pagination: {
+          total: Number(total),
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: Number(total) > (pagination.page + 1) * pagination.perPage
+        },
+        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
+      );
+    }
+  }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const request = this.pool.request();
+      request.input("p1", traceId);
+      request.input("p2", spanId);
+      const totalResult = await request.query(
+        `SELECT COUNT(*) as count FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [traceId] = @p1 AND [spanId] = @p2`
+      );
+      const total = totalResult.recordset[0]?.count || 0;
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          },
+          scores: []
+        };
+      }
+      const limit = pagination.perPage + 1;
+      const dataRequest = this.pool.request();
+      dataRequest.input("p1", traceId);
+      dataRequest.input("p2", spanId);
+      dataRequest.input("p3", limit);
+      dataRequest.input("p4", pagination.page * pagination.perPage);
+      const result = await dataRequest.query(
+        `SELECT * FROM ${getTableName({ indexName: TABLE_SCORERS, schemaName: getSchemaName(this.schema) })} WHERE [traceId] = @p1 AND [spanId] = @p2 ORDER BY [createdAt] DESC OFFSET @p4 ROWS FETCH NEXT @p3 ROWS ONLY`
+      );
+      return {
+        pagination: {
+          total: Number(total),
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: result.recordset.length > pagination.perPage
+        },
+        scores: result.recordset.slice(0, pagination.perPage).map((row) => transformScoreRow(row))
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_GET_SCORES_BY_SPAN_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { traceId, spanId }
+        },
+        error
+      );
+    }
+  }
+};
+var WorkflowsMSSQL = class extends WorkflowsStorage {
+  pool;
+  operations;
+  schema;
+  constructor({
+    pool,
+    operations,
+    schema
+  }) {
+    super();
+    this.pool = pool;
+    this.operations = operations;
+    this.schema = schema;
+  }
+  parseWorkflowRun(row) {
+    let parsedSnapshot = row.snapshot;
+    if (typeof parsedSnapshot === "string") {
+      try {
+        parsedSnapshot = JSON.parse(row.snapshot);
+      } catch (e) {
+        this.logger?.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
+    };
+  }
+  async updateWorkflowResults({
+    workflowName,
+    runId,
+    stepId,
+    result,
+    runtimeContext
+  }) {
+    const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      const selectRequest = new sql3.Request(transaction);
+      selectRequest.input("workflow_name", workflowName);
+      selectRequest.input("run_id", runId);
+      const existingSnapshotResult = await selectRequest.query(
+        `SELECT snapshot FROM ${table} WITH (UPDLOCK, HOLDLOCK) WHERE workflow_name = @workflow_name AND run_id = @run_id`
+      );
+      let snapshot;
+      if (!existingSnapshotResult.recordset || existingSnapshotResult.recordset.length === 0) {
+        snapshot = {
+          context: {},
+          activePaths: [],
+          timestamp: Date.now(),
+          suspendedPaths: {},
+          resumeLabels: {},
+          serializedStepGraph: [],
+          value: {},
+          waitingPaths: {},
+          status: "pending",
+          runId,
+          runtimeContext: {}
+        };
+      } else {
+        const existingSnapshot = existingSnapshotResult.recordset[0].snapshot;
+        snapshot = typeof existingSnapshot === "string" ? JSON.parse(existingSnapshot) : existingSnapshot;
+      }
+      snapshot.context[stepId] = result;
+      snapshot.runtimeContext = { ...snapshot.runtimeContext, ...runtimeContext };
+      const upsertReq = new sql3.Request(transaction);
+      upsertReq.input("workflow_name", workflowName);
+      upsertReq.input("run_id", runId);
+      upsertReq.input("snapshot", JSON.stringify(snapshot));
+      upsertReq.input("createdAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      await upsertReq.query(
+        `MERGE ${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 transaction.commit();
+      return snapshot.context;
+    } catch (error) {
+      try {
+        await transaction.rollback();
+      } catch {
+      }
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_RESULTS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            workflowName,
+            runId,
+            stepId
+          }
+        },
+        error
+      );
+    }
+  }
+  async updateWorkflowState({
+    workflowName,
+    runId,
+    opts
+  }) {
+    const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      const selectRequest = new sql3.Request(transaction);
+      selectRequest.input("workflow_name", workflowName);
+      selectRequest.input("run_id", runId);
+      const existingSnapshotResult = await selectRequest.query(
+        `SELECT snapshot FROM ${table} WITH (UPDLOCK, HOLDLOCK) WHERE workflow_name = @workflow_name AND run_id = @run_id`
+      );
+      if (!existingSnapshotResult.recordset || existingSnapshotResult.recordset.length === 0) {
+        await transaction.rollback();
+        return void 0;
+      }
+      const existingSnapshot = existingSnapshotResult.recordset[0].snapshot;
+      const snapshot = typeof existingSnapshot === "string" ? JSON.parse(existingSnapshot) : existingSnapshot;
+      if (!snapshot || !snapshot?.context) {
+        await transaction.rollback();
+        throw new MastraError(
+          {
+            id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_SNAPSHOT_NOT_FOUND",
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.SYSTEM,
+            details: {
+              workflowName,
+              runId
+            }
+          },
+          new Error(`Snapshot not found for runId ${runId}`)
+        );
+      }
+      const updatedSnapshot = { ...snapshot, ...opts };
+      const updateRequest = new sql3.Request(transaction);
+      updateRequest.input("snapshot", JSON.stringify(updatedSnapshot));
+      updateRequest.input("workflow_name", workflowName);
+      updateRequest.input("run_id", runId);
+      updateRequest.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+      await updateRequest.query(
+        `UPDATE ${table} SET snapshot = @snapshot, [updatedAt] = @updatedAt WHERE workflow_name = @workflow_name AND run_id = @run_id`
+      );
+      await transaction.commit();
+      return updatedSnapshot;
+    } catch (error) {
+      try {
+        await transaction.rollback();
+      } catch {
+      }
+      throw new MastraError(
+        {
+          id: "MASTRA_STORAGE_MSSQL_STORE_UPDATE_WORKFLOW_STATE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            workflowName,
+            runId
+          }
+        },
+        error
+      );
+    }
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    resourceId,
+    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("resourceId", resourceId);
+      request.input("snapshot", JSON.stringify(snapshot));
+      request.input("createdAt", sql3.DateTime2, new Date(now));
+      request.input("updatedAt", sql3.DateTime2, new Date(now));
+      const mergeSql = `MERGE INTO ${table} AS target
+        USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
+        ON target.workflow_name = src.workflow_name AND target.run_id = src.run_id
+        WHEN MATCHED THEN UPDATE SET
+          resourceId = @resourceId,
+          snapshot = @snapshot,
+          [updatedAt] = @updatedAt
+        WHEN NOT MATCHED THEN INSERT (workflow_name, run_id, resourceId, snapshot, [createdAt], [updatedAt])
+          VALUES (@workflow_name, @run_id, @resourceId, @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 this.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 {
+          this.logger?.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, sql3.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) => this.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 sql3.ConnectionPool(config.connectionString) : new sql3.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 workflows = new WorkflowsMSSQL({ pool: this.pool, operations, schema: this.schema });
+      const memory = new MemoryMSSQL({ pool: this.pool, schema: this.schema, operations });
+      const observability = new ObservabilityMSSQL({ pool: this.pool, operations, schema: this.schema });
+      this.stores = {
+        operations,
+        scores,
+        workflows,
+        legacyEvals,
+        memory,
+        observability
+      };
+    } 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();
+      try {
+        await this.stores.operations.createAutomaticIndexes();
+      } catch (indexError) {
+        this.logger?.warn?.("Failed to create indexes:", indexError);
+      }
+    } 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,
+      getScoresBySpan: true,
+      aiTracing: true,
+      indexManagement: true
+    };
+  }
+  /** @deprecated use getEvals instead */
+  async getEvalsByAgentName(agentName, type) {
+    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
+  }
+  async getEvals(options = {}) {
+    return this.stores.legacyEvals.getEvals(options);
+  }
+  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,
+    resourceId,
+    snapshot
+  }) {
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, 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();
+  }
+  /**
+   * Index Management
+   */
+  async createIndex(options) {
+    return this.stores.operations.createIndex(options);
+  }
+  async listIndexes(tableName) {
+    return this.stores.operations.listIndexes(tableName);
+  }
+  async describeIndex(indexName) {
+    return this.stores.operations.describeIndex(indexName);
+  }
+  async dropIndex(indexName) {
+    return this.stores.operations.dropIndex(indexName);
+  }
+  /**
+   * AI Tracing / Observability
+   */
+  getObservabilityStore() {
+    if (!this.stores.observability) {
+      throw new MastraError({
+        id: "MSSQL_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.SYSTEM,
+        text: "Observability storage is not initialized"
+      });
+    }
+    return this.stores.observability;
+  }
+  async createAISpan(span) {
+    return this.getObservabilityStore().createAISpan(span);
+  }
+  async updateAISpan({
+    spanId,
+    traceId,
+    updates
+  }) {
+    return this.getObservabilityStore().updateAISpan({ spanId, traceId, updates });
+  }
+  async getAITrace(traceId) {
+    return this.getObservabilityStore().getAITrace(traceId);
+  }
+  async getAITracesPaginated(args) {
+    return this.getObservabilityStore().getAITracesPaginated(args);
+  }
+  async batchCreateAISpans(args) {
+    return this.getObservabilityStore().batchCreateAISpans(args);
+  }
+  async batchUpdateAISpans(args) {
+    return this.getObservabilityStore().batchUpdateAISpans(args);
+  }
+  async batchDeleteAITraces(args) {
+    return this.getObservabilityStore().batchDeleteAITraces(args);
+  }
+  /**
+   * Scorers
+   */
+  async getScoreById({ id: _id }) {
+    return this.stores.scores.getScoreById({ id: _id });
+  }
+  async getScoresByScorerId({
+    scorerId: _scorerId,
+    pagination: _pagination,
+    entityId: _entityId,
+    entityType: _entityType,
+    source: _source
+  }) {
+    return this.stores.scores.getScoresByScorerId({
+      scorerId: _scorerId,
+      pagination: _pagination,
+      entityId: _entityId,
+      entityType: _entityType,
+      source: _source
+    });
+  }
+  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
+    });
+  }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination: _pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination: _pagination });
+  }
+};
 
 export { MSSQLStore };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map