@mastra/mssql 0.0.0-fix-multi-modal-for-cloud-20251028082043 → 0.0.0-fix-persist-session-cache-option-mcp-server-20251030161352

package/dist/index.js

@@ -1,8 +1,8 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-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 { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import sql2 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
+import { parseSqlIdentifier } from '@mastra/core/utils';
 import { randomUUID } from 'crypto';
 import { saveScorePayloadSchema } from '@mastra/core/scores';
 
@@ -80,153 +80,7 @@ function transformFromSqlRow({
   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 {
-    }
-  }
-  if (row.result) {
-    try {
-      resultValue = typeof row.result === "string" ? JSON.parse(row.result) : row.result;
-    } catch {
-    }
-  }
-  return {
-    agentName: row.agent_name,
-    input: row.input,
-    output: row.output,
-    result: resultValue,
-    metricName: row.metric_name,
-    instructions: row.instructions,
-    testInfo: testInfoValue,
-    globalRunId: row.global_run_id,
-    runId: row.run_id,
-    createdAt: row.created_at
-  };
-}
-var LegacyEvalsMSSQL = class extends LegacyEvalsStorage {
-  pool;
-  schema;
-  constructor({ pool, schema }) {
-    super();
-    this.pool = pool;
-    this.schema = schema;
-  }
-  /** @deprecated use getEvals instead */
-  async getEvalsByAgentName(agentName, type) {
-    try {
-      let query = `SELECT * FROM ${getTableName({ indexName: TABLE_EVALS, schemaName: getSchemaName(this.schema) })} WHERE agent_name = @p1`;
-      if (type === "test") {
-        query += " AND test_info IS NOT NULL AND JSON_VALUE(test_info, '$.testPath') IS NOT NULL";
-      } else if (type === "live") {
-        query += " AND (test_info IS NULL OR JSON_VALUE(test_info, '$.testPath') IS NULL)";
-      }
-      query += " ORDER BY created_at DESC";
-      const request = this.pool.request();
-      request.input("p1", agentName);
-      const result = await request.query(query);
-      const rows = result.recordset;
-      return typeof transformEvalRow === "function" ? rows?.map((row) => transformEvalRow(row)) ?? [] : rows ?? [];
-    } catch (error) {
-      if (error && error.number === 208 && error.message && error.message.includes("Invalid object name")) {
-        return [];
-      }
-      this.logger?.error?.("Failed to get evals for the specified agent:", error);
-      throw error;
-    }
-  }
-  async getEvals(options = {}) {
-    const { agentName, type, page = 0, perPage = 100, dateRange } = options;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const where = [];
-    const params = {};
-    if (agentName) {
-      where.push("agent_name = @agentName");
-      params["agentName"] = agentName;
-    }
-    if (type === "test") {
-      where.push("test_info IS NOT NULL AND JSON_VALUE(test_info, '$.testPath') IS NOT NULL");
-    } else if (type === "live") {
-      where.push("(test_info IS NULL OR JSON_VALUE(test_info, '$.testPath') IS NULL)");
-    }
-    if (fromDate instanceof Date && !isNaN(fromDate.getTime())) {
-      where.push(`[created_at] >= @fromDate`);
-      params[`fromDate`] = fromDate.toISOString();
-    }
-    if (toDate instanceof Date && !isNaN(toDate.getTime())) {
-      where.push(`[created_at] <= @toDate`);
-      params[`toDate`] = toDate.toISOString();
-    }
-    const whereClause = where.length > 0 ? `WHERE ${where.join(" AND ")}` : "";
-    const tableName = getTableName({ indexName: TABLE_EVALS, schemaName: getSchemaName(this.schema) });
-    const offset = page * perPage;
-    const countQuery = `SELECT COUNT(*) as total FROM ${tableName} ${whereClause}`;
-    const dataQuery = `SELECT * FROM ${tableName} ${whereClause} ORDER BY seq_id DESC OFFSET @offset ROWS FETCH NEXT @perPage ROWS ONLY`;
-    try {
-      const countReq = this.pool.request();
-      Object.entries(params).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          countReq.input(key, sql3.DateTime, value);
-        } else {
-          countReq.input(key, value);
-        }
-      });
-      const countResult = await countReq.query(countQuery);
-      const total = countResult.recordset[0]?.total || 0;
-      if (total === 0) {
-        return {
-          evals: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const req = this.pool.request();
-      Object.entries(params).forEach(([key, value]) => {
-        if (value instanceof Date) {
-          req.input(key, sql3.DateTime, value);
-        } else {
-          req.input(key, value);
-        }
-      });
-      req.input("offset", offset);
-      req.input("perPage", perPage);
-      const result = await req.query(dataQuery);
-      const rows = result.recordset;
-      return {
-        evals: rows?.map((row) => transformEvalRow(row)) ?? [],
-        total,
-        page,
-        perPage,
-        hasMore: offset + (rows?.length ?? 0) < total
-      };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "MASTRA_STORAGE_MSSQL_STORE_GET_EVALS_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            agentName: agentName || "all",
-            type: type || "all",
-            page,
-            perPage
-          }
-        },
-        error
-      );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      throw mastraError;
-    }
-  }
-};
+// src/storage/domains/memory/index.ts
 var MemoryMSSQL = class extends MemoryStorage {
   pool;
   schema;
@@ -258,7 +112,7 @@ var MemoryMSSQL = class extends MemoryStorage {
   }
   async getThreadById({ threadId }) {
     try {
-      const sql6 = `SELECT 
+      const sql5 = `SELECT 
         id,
         [resourceId],
         title,
@@ -269,7 +123,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       WHERE id = @threadId`;
       const request = this.pool.request();
       request.input("threadId", threadId);
-      const resultSet = await request.query(sql6);
+      const resultSet = await request.query(sql5);
       const thread = resultSet.recordset[0] || null;
       if (!thread) {
         return null;
@@ -375,12 +229,12 @@ var MemoryMSSQL = class extends MemoryStorage {
       req.input("title", thread.title);
       const metadata = thread.metadata ? JSON.stringify(thread.metadata) : null;
       if (metadata === null) {
-        req.input("metadata", sql3.NVarChar, null);
+        req.input("metadata", sql2.NVarChar, null);
       } else {
         req.input("metadata", metadata);
       }
-      req.input("createdAt", sql3.DateTime2, thread.createdAt);
-      req.input("updatedAt", sql3.DateTime2, thread.updatedAt);
+      req.input("createdAt", sql2.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql2.DateTime2, thread.updatedAt);
       await req.query(mergeSql);
       return thread;
     } catch (error) {
@@ -449,7 +303,7 @@ var MemoryMSSQL = class extends MemoryStorage {
     };
     try {
       const table = getTableName({ indexName: TABLE_THREADS, schemaName: getSchemaName(this.schema) });
-      const sql6 = `UPDATE ${table}
+      const sql5 = `UPDATE ${table}
         SET title = @title,
             metadata = @metadata,
             [updatedAt] = @updatedAt
@@ -460,7 +314,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       req.input("title", title);
       req.input("metadata", JSON.stringify(mergedMetadata));
       req.input("updatedAt", /* @__PURE__ */ new Date());
-      const result = await req.query(sql6);
+      const result = await req.query(sql5);
       let thread = result.recordset && result.recordset[0];
       if (thread && "seq_id" in thread) {
         const { seq_id, ...rest } = thread;
@@ -699,6 +553,20 @@ var MemoryMSSQL = class extends MemoryStorage {
       return [];
     }
   }
+  async listMessages(_args) {
+    throw new Error(
+      `listMessages is not yet implemented by this storage adapter (${this.constructor.name}). This method is currently being rolled out across all storage adapters. Please use getMessages or getMessagesPaginated as an alternative, or wait for the implementation.`
+    );
+  }
+  async listMessagesById({ messageIds }) {
+    return this.getMessagesById({ messageIds, format: "v2" });
+  }
+  async listThreadsByResourceId(args) {
+    const { resourceId, limit, offset, orderBy, sortDirection } = args;
+    const page = Math.floor(offset / limit);
+    const perPage = limit;
+    return this.getThreadsByResourceIdPaginated({ resourceId, page, perPage, orderBy, sortDirection });
+  }
   async getMessagesPaginated(args) {
     const { threadId, resourceId, format, selectBy } = args;
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
@@ -829,7 +697,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             "content",
             typeof message.content === "string" ? message.content : JSON.stringify(message.content)
           );
-          request.input("createdAt", sql3.DateTime2, message.createdAt);
+          request.input("createdAt", sql2.DateTime2, message.createdAt);
           request.input("role", message.role);
           request.input("type", message.type || "v2");
           request.input("resourceId", message.resourceId);
@@ -848,7 +716,7 @@ var MemoryMSSQL = class extends MemoryStorage {
           await request.query(mergeSql);
         }
         const threadReq = transaction.request();
-        threadReq.input("updatedAt", sql3.DateTime2, /* @__PURE__ */ new Date());
+        threadReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
         threadReq.input("id", threadId);
         await threadReq.query(`UPDATE ${tableThreads} SET [updatedAt] = @updatedAt WHERE id = @id`);
         await transaction.commit();
@@ -1550,7 +1418,7 @@ var StoreOperationsMSSQL = class extends StoreOperations {
         const value = record[col];
         const preparedValue = this.prepareValue(value, col, tableName);
         if (preparedValue instanceof Date) {
-          request.input(`param${i}`, sql3.DateTime2, preparedValue);
+          request.input(`param${i}`, sql2.DateTime2, preparedValue);
         } else if (preparedValue === null || preparedValue === void 0) {
           request.input(`param${i}`, this.getMssqlType(tableName, col), null);
         } else {
@@ -1765,7 +1633,7 @@ ${columns}
     try {
       const keyEntries = Object.entries(keys).map(([key, value]) => [parseSqlIdentifier(key, "column name"), value]);
       const conditions = keyEntries.map(([key], i) => `[${key}] = @param${i}`).join(" AND ");
-      const sql6 = `SELECT * FROM ${getTableName({ indexName: tableName, schemaName: getSchemaName(this.schemaName) })} WHERE ${conditions}`;
+      const sql5 = `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);
@@ -1775,7 +1643,7 @@ ${columns}
           request.input(`param${i}`, preparedValue);
         }
       });
-      const resultSet = await request.query(sql6);
+      const resultSet = await request.query(sql5);
       const result = resultSet.recordset[0] || null;
       if (!result) {
         return null;
@@ -1874,23 +1742,23 @@ ${columns}
     const col = TABLE_SCHEMAS[tableName]?.[columnName];
     switch (col?.type) {
       case "text":
-        return sql3.NVarChar;
+        return sql2.NVarChar;
       case "timestamp":
-        return sql3.DateTime2;
+        return sql2.DateTime2;
       case "uuid":
-        return sql3.UniqueIdentifier;
+        return sql2.UniqueIdentifier;
       case "jsonb":
-        return sql3.NVarChar;
+        return sql2.NVarChar;
       case "integer":
-        return sql3.Int;
+        return sql2.Int;
       case "bigint":
-        return sql3.BigInt;
+        return sql2.BigInt;
       case "float":
-        return sql3.Float;
+        return sql2.Float;
       case "boolean":
-        return sql3.Bit;
+        return sql2.Bit;
       default:
-        return sql3.NVarChar;
+        return sql2.NVarChar;
     }
   }
   /**
@@ -2334,11 +2202,6 @@ ${columns}
         table: TABLE_TRACES,
         columns: ["name", "seq_id DESC"]
       },
-      {
-        name: `${schemaPrefix}mastra_evals_agent_name_seqid_idx`,
-        table: TABLE_EVALS,
-        columns: ["agent_name", "seq_id DESC"]
-      },
       {
         name: `${schemaPrefix}mastra_scores_trace_id_span_id_seqid_idx`,
         table: TABLE_SCORERS,
@@ -2403,7 +2266,7 @@ function transformScoreRow(row) {
     metadata: safelyParseJSON(row.metadata),
     output: safelyParseJSON(row.output),
     additionalContext: safelyParseJSON(row.additionalContext),
-    runtimeContext: safelyParseJSON(row.runtimeContext),
+    requestContext: safelyParseJSON(row.requestContext),
     entity: safelyParseJSON(row.entity),
     createdAt: row.createdAt,
     updatedAt: row.updatedAt
@@ -2470,7 +2333,7 @@ var ScoresMSSQL = class extends ScoresStorage {
         input,
         output,
         additionalContext,
-        runtimeContext,
+        requestContext,
         entity,
         ...rest
       } = validatedScore;
@@ -2485,7 +2348,7 @@ var ScoresMSSQL = class extends ScoresStorage {
           analyzeStepResult: analyzeStepResult || null,
           metadata: metadata || null,
           additionalContext: additionalContext || null,
-          runtimeContext: runtimeContext || null,
+          requestContext: requestContext || null,
           entity: entity || null,
           scorer: scorer || null,
           createdAt: (/* @__PURE__ */ new Date()).toISOString(),
@@ -2774,13 +2637,13 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     runId,
     stepId,
     result,
-    runtimeContext
+    requestContext
   }) {
     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);
+      const selectRequest = new sql2.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2799,20 +2662,20 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
           waitingPaths: {},
           status: "pending",
           runId,
-          runtimeContext: {}
+          requestContext: {}
         };
       } 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);
+      snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
+      const upsertReq = new sql2.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());
+      upsertReq.input("createdAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+      upsertReq.input("updatedAt", sql2.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
@@ -2852,7 +2715,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql3.Request(transaction);
+      const selectRequest = new sql2.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2880,11 +2743,11 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         );
       }
       const updatedSnapshot = { ...snapshot, ...opts };
-      const updateRequest = new sql3.Request(transaction);
+      const updateRequest = new sql2.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());
+      updateRequest.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
       await updateRequest.query(
         `UPDATE ${table} SET snapshot = @snapshot, [updatedAt] = @updatedAt WHERE workflow_name = @workflow_name AND run_id = @run_id`
       );
@@ -2923,8 +2786,8 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       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));
+      request.input("createdAt", sql2.DateTime2, new Date(now));
+      request.input("updatedAt", sql2.DateTime2, new Date(now));
       const mergeSql = `MERGE INTO ${table} AS target
         USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
         ON target.workflow_name = src.workflow_name AND target.run_id = src.run_id
@@ -3059,7 +2922,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       const request = this.pool.request();
       Object.entries(paramMap).forEach(([key, value]) => {
         if (value instanceof Date) {
-          request.input(key, sql3.DateTime, value);
+          request.input(key, sql2.DateTime, value);
         } else {
           request.input(key, value);
         }
@@ -3092,6 +2955,9 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       );
     }
   }
+  async listWorkflowRuns(args) {
+    return this.getWorkflowRuns(args);
+  }
 };
 
 // src/storage/index.ts
@@ -3116,7 +2982,7 @@ var MSSQLStore = class extends MastraStorage {
         }
       }
       this.schema = config.schemaName || "dbo";
-      this.pool = "connectionString" in config ? new sql3.ConnectionPool(config.connectionString) : new sql3.ConnectionPool({
+      this.pool = "connectionString" in config ? new sql2.ConnectionPool(config.connectionString) : new sql2.ConnectionPool({
         server: config.server,
         database: config.database,
         user: config.user,
@@ -3124,7 +2990,6 @@ var MSSQLStore = class extends MastraStorage {
         port: config.port,
         options: config.options || { encrypt: true, trustServerCertificate: true }
       });
-      const legacyEvals = new LegacyEvalsMSSQL({ pool: this.pool, schema: this.schema });
       const operations = new StoreOperationsMSSQL({ pool: this.pool, schemaName: this.schema });
       const scores = new ScoresMSSQL({ pool: this.pool, operations, schema: this.schema });
       const workflows = new WorkflowsMSSQL({ pool: this.pool, operations, schema: this.schema });
@@ -3134,7 +2999,6 @@ var MSSQLStore = class extends MastraStorage {
         operations,
         scores,
         workflows,
-        legacyEvals,
         memory,
         observability
       };
@@ -3193,13 +3057,6 @@ var MSSQLStore = class extends MastraStorage {
       indexManagement: true
     };
   }
-  /** @deprecated use getEvals instead */
-  async getEvalsByAgentName(agentName, type) {
-    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
-  }
-  async getEvals(options = {}) {
-    return this.stores.legacyEvals.getEvals(options);
-  }
   async createTable({
     tableName,
     schema
@@ -3300,9 +3157,9 @@ var MSSQLStore = class extends MastraStorage {
     runId,
     stepId,
     result,
-    runtimeContext
+    requestContext
   }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
   }
   async updateWorkflowState({
     workflowName,