@mastra/mssql 0.0.0-stream-vnext-usage-20250908171242 → 0.0.0-suspendRuntimeContextTypeFix-20250930142630

package/dist/index.js

@@ -3,6 +3,7 @@ import { MastraStorage, LegacyEvalsStorage, StoreOperations, TABLE_WORKFLOW_SNAP
 import sql2 from 'mssql';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MessageList } from '@mastra/core/agent';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // src/storage/index.ts
 function getSchemaName(schema) {
@@ -1473,6 +1474,19 @@ var ScoresMSSQL = class extends ScoresStorage {
     }
   }
   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 = crypto.randomUUID();
       const {
@@ -1486,7 +1500,7 @@ var ScoresMSSQL = class extends ScoresStorage {
         runtimeContext,
         entity,
         ...rest
-      } = score;
+      } = validatedScore;
       await this.operations.insert({
         tableName: TABLE_SCORERS,
         record: {
@@ -1671,6 +1685,60 @@ var ScoresMSSQL = class extends ScoresStorage {
       );
     }
   }
+  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 TracesMSSQL = class extends TracesStorage {
   pool;
@@ -1890,6 +1958,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
@@ -1898,6 +1967,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       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", sql2.DateTime2, new Date(now));
       request.input("updatedAt", sql2.DateTime2, new Date(now));
@@ -1905,10 +1975,11 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         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, snapshot, [createdAt], [updatedAt])
-          VALUES (@workflow_name, @run_id, @snapshot, @createdAt, @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(
@@ -2157,7 +2228,8 @@ var MSSQLStore = class extends MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: true,
       createTable: true,
-      deleteMessages: true
+      deleteMessages: true,
+      getScoresBySpan: true
     };
   }
   /** @deprecated use getEvals instead */
@@ -2293,9 +2365,10 @@ var MSSQLStore = class extends MastraStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot({
     workflowName,
@@ -2354,6 +2427,13 @@ var MSSQLStore = class extends MastraStorage {
       pagination: _pagination
     });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination: _pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination: _pagination });
+  }
 };
 
 export { MSSQLStore };