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

package/dist/index.js

@@ -1,6 +1,6 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraStorage, createStorageErrorId, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
-import sql2 from 'mssql';
+import sql3 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
 import { parseSqlIdentifier } from '@mastra/core/utils';
 import { randomUUID } from 'crypto';
@@ -227,7 +227,7 @@ var MemoryMSSQL = class extends MemoryStorage {
       dataRequest.input("resourceId", resourceId);
       dataRequest.input("offset", offset);
       if (limitValue > 2147483647) {
-        dataRequest.input("perPage", sql2.BigInt, limitValue);
+        dataRequest.input("perPage", sql3.BigInt, limitValue);
       } else {
         dataRequest.input("perPage", limitValue);
       }
@@ -291,12 +291,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", sql2.NVarChar, null);
+        req.input("metadata", sql3.NVarChar, null);
       } else {
         req.input("metadata", metadata);
       }
-      req.input("createdAt", sql2.DateTime2, thread.createdAt);
-      req.input("updatedAt", sql2.DateTime2, thread.updatedAt);
+      req.input("createdAt", sql3.DateTime2, thread.createdAt);
+      req.input("updatedAt", sql3.DateTime2, thread.updatedAt);
       await req.query(mergeSql);
       return thread;
     } catch (error) {
@@ -592,7 +592,7 @@ var MemoryMSSQL = class extends MemoryStorage {
           return result2.recordset || [];
         }
         request.input("offset", offset);
-        request.input("limit", perPage > 2147483647 ? sql2.BigInt : sql2.Int, perPage);
+        request.input("limit", perPage > 2147483647 ? sql3.BigInt : sql3.Int, perPage);
         const result = await request.query(
           `${baseQuery}${actualWhereClause} ${orderByStatement} OFFSET @offset ROWS FETCH NEXT @limit ROWS ONLY`
         );
@@ -717,7 +717,7 @@ var MemoryMSSQL = class extends MemoryStorage {
             "content",
             typeof message.content === "string" ? message.content : JSON.stringify(message.content)
           );
-          request.input("createdAt", sql2.DateTime2, message.createdAt);
+          request.input("createdAt", sql3.DateTime2, message.createdAt);
           request.input("role", message.role);
           request.input("type", message.type || "v2");
           request.input("resourceId", message.resourceId);
@@ -736,7 +736,7 @@ var MemoryMSSQL = class extends MemoryStorage {
           await request.query(mergeSql);
         }
         const threadReq = transaction.request();
-        threadReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+        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();
@@ -1437,7 +1437,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}`, sql2.DateTime2, preparedValue);
+          request.input(`param${i}`, sql3.DateTime2, preparedValue);
         } else if (preparedValue === null || preparedValue === void 0) {
           request.input(`param${i}`, this.getMssqlType(tableName, col), null);
         } else {
@@ -1775,23 +1775,23 @@ ${columns}
     const col = TABLE_SCHEMAS[tableName]?.[columnName];
     switch (col?.type) {
       case "text":
-        return sql2.NVarChar;
+        return sql3.NVarChar;
       case "timestamp":
-        return sql2.DateTime2;
+        return sql3.DateTime2;
       case "uuid":
-        return sql2.UniqueIdentifier;
+        return sql3.UniqueIdentifier;
       case "jsonb":
-        return sql2.NVarChar;
+        return sql3.NVarChar;
       case "integer":
-        return sql2.Int;
+        return sql3.Int;
       case "bigint":
-        return sql2.BigInt;
+        return sql3.BigInt;
       case "float":
-        return sql2.Float;
+        return sql3.Float;
       case "boolean":
-        return sql2.Bit;
+        return sql3.Bit;
       default:
-        return sql2.NVarChar;
+        return sql3.NVarChar;
     }
   }
   /**
@@ -2340,13 +2340,21 @@ var ScoresMSSQL = class extends ScoresStorage {
         {
           id: createStorageErrorId("MSSQL", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.USER,
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
     try {
       const scoreId = randomUUID();
+      const now = /* @__PURE__ */ new Date();
       const {
         scorer,
         preprocessStepResult,
@@ -2373,12 +2381,11 @@ var ScoresMSSQL = class extends ScoresStorage {
           requestContext: requestContext || null,
           entity: entity || null,
           scorer: scorer || null,
-          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+          createdAt: now.toISOString(),
+          updatedAt: now.toISOString()
         }
       });
-      const scoreFromDb = await this.getScoreById({ id: scoreId });
-      return { score: scoreFromDb };
+      return { score: { ...validatedScore, id: scoreId, createdAt: now, updatedAt: now } };
     } catch (error) {
       throw new MastraError(
         {
@@ -2684,7 +2691,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql2.Request(transaction);
+      const selectRequest = new sql3.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2712,12 +2719,12 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       }
       snapshot.context[stepId] = result;
       snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
-      const upsertReq = new sql2.Request(transaction);
+      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", sql2.DateTime2, /* @__PURE__ */ new Date());
-      upsertReq.input("updatedAt", sql2.DateTime2, /* @__PURE__ */ new Date());
+      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
@@ -2757,7 +2764,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
     const transaction = this.pool.transaction();
     try {
       await transaction.begin();
-      const selectRequest = new sql2.Request(transaction);
+      const selectRequest = new sql3.Request(transaction);
       selectRequest.input("workflow_name", workflowName);
       selectRequest.input("run_id", runId);
       const existingSnapshotResult = await selectRequest.query(
@@ -2785,11 +2792,11 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
         );
       }
       const updatedSnapshot = { ...snapshot, ...opts };
-      const updateRequest = new sql2.Request(transaction);
+      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", sql2.DateTime2, /* @__PURE__ */ new Date());
+      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`
       );
@@ -2829,8 +2836,8 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       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));
+      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
@@ -2927,6 +2934,35 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       );
     }
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    const table = getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName(this.schema) });
+    const transaction = this.pool.transaction();
+    try {
+      await transaction.begin();
+      const deleteRequest = new sql3.Request(transaction);
+      deleteRequest.input("workflow_name", workflowName);
+      deleteRequest.input("run_id", runId);
+      await deleteRequest.query(`DELETE FROM ${table} WHERE workflow_name = @workflow_name AND run_id = @run_id`);
+      await transaction.commit();
+    } catch (error) {
+      try {
+        await transaction.rollback();
+      } catch {
+      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("MSSQL", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            runId,
+            workflowName
+          }
+        },
+        error
+      );
+    }
+  }
   async listWorkflowRuns({
     workflowName,
     fromDate,
@@ -2970,7 +3006,7 @@ var WorkflowsMSSQL = class extends WorkflowsStorage {
       const request = this.pool.request();
       Object.entries(paramMap).forEach(([key, value]) => {
         if (value instanceof Date) {
-          request.input(key, sql2.DateTime, value);
+          request.input(key, sql3.DateTime, value);
         } else {
           request.input(key, value);
         }
@@ -3033,7 +3069,7 @@ var MSSQLStore = class extends MastraStorage {
         }
       }
       this.schema = config.schemaName || "dbo";
-      this.pool = "connectionString" in config ? new sql2.ConnectionPool(config.connectionString) : new sql2.ConnectionPool({
+      this.pool = "connectionString" in config ? new sql3.ConnectionPool(config.connectionString) : new sql3.ConnectionPool({
         server: config.server,
         database: config.database,
         user: config.user,
@@ -3224,6 +3260,9 @@ var MSSQLStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.getWorkflowRunById({ runId, workflowName });
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async close() {
     await this.pool.close();
   }
@@ -3302,8 +3341,8 @@ var MSSQLStore = class extends MastraStorage {
       source: _source
     });
   }
-  async saveScore(_score) {
-    return this.stores.scores.saveScore(_score);
+  async saveScore(score) {
+    return this.stores.scores.saveScore(score);
   }
   async listScoresByRunId({
     runId: _runId,