@mastra/mssql 0.0.0-main-test-20251105183450 → 0.0.0-main-test-05-11-2025-2-20251105223343

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_AI_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_SCORERS, TABLE_SPANS, ScoresStorage, normalizePerPage, calculatePagination, WorkflowsStorage, MemoryStorage, TABLE_RESOURCES, ObservabilityStorage, safelyParseJSON } from '@mastra/core/storage';
 import sql2 from 'mssql';
 import { MessageList } from '@mastra/core/agent';
 import { parseSqlIdentifier } from '@mastra/core/utils';
@@ -982,11 +982,11 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         endedAt
         // Note: createdAt/updatedAt will be set by default values
       };
-      return this.operations.insert({ tableName: TABLE_AI_SPANS, record });
+      return this.operations.insert({ tableName: TABLE_SPANS, record });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_CREATE_AI_SPAN_FAILED",
+          id: "MSSQL_STORE_CREATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1000,10 +1000,10 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITrace(traceId) {
+  async getTrace(traceId) {
     try {
       const tableName = getTableName({
-        indexName: TABLE_AI_SPANS,
+        indexName: TABLE_SPANS,
         schemaName: getSchemaName(this.schema)
       });
       const request = this.pool.request();
@@ -1024,7 +1024,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         traceId,
         spans: result.recordset.map(
           (span) => transformFromSqlRow({
-            tableName: TABLE_AI_SPANS,
+            tableName: TABLE_SPANS,
             sqlRow: span
           })
         )
@@ -1032,7 +1032,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_AI_TRACE_FAILED",
+          id: "MSSQL_STORE_GET_TRACE_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1057,14 +1057,14 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         data.startedAt = data.startedAt.toISOString();
       }
       await this.operations.update({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         keys: { spanId, traceId },
         data
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_UPDATE_AI_SPAN_FAILED",
+          id: "MSSQL_STORE_UPDATE_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1076,7 +1076,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async getAITracesPaginated({
+  async getTracesPaginated({
     filters,
     pagination
   }) {
@@ -1101,7 +1101,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         name = `agent run: '${entityId}'`;
       } else {
         const error = new MastraError({
-          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1120,7 +1120,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       params[entityParam] = name;
     }
     const tableName = getTableName({
-      indexName: TABLE_AI_SPANS,
+      indexName: TABLE_SPANS,
       schemaName: getSchemaName(this.schema)
     });
     try {
@@ -1154,7 +1154,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
       const spans = dataResult.recordset.map(
         (row) => transformFromSqlRow({
-          tableName: TABLE_AI_SPANS,
+          tableName: TABLE_SPANS,
           sqlRow: row
         })
       );
@@ -1170,7 +1170,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          id: "MSSQL_STORE_GET_TRACES_PAGINATED_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER
         },
@@ -1184,7 +1184,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     }
     try {
       await this.operations.batchInsert({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         records: args.records.map((span) => ({
           ...span,
           startedAt: span.startedAt instanceof Date ? span.startedAt.toISOString() : span.startedAt,
@@ -1194,7 +1194,7 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_CREATE_AI_SPANS_FAILED",
+          id: "MSSQL_STORE_BATCH_CREATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1224,13 +1224,13 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
         };
       });
       await this.operations.batchUpdate({
-        tableName: TABLE_AI_SPANS,
+        tableName: TABLE_SPANS,
         updates
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_UPDATE_AI_SPANS_FAILED",
+          id: "MSSQL_STORE_BATCH_UPDATE_SPANS_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -1241,20 +1241,20 @@ var ObservabilityMSSQL = class extends ObservabilityStorage {
       );
     }
   }
-  async batchDeleteAITraces(args) {
+  async batchDeleteTraces(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,
+        tableName: TABLE_SPANS,
         keys
       });
     } catch (error) {
       throw new MastraError(
         {
-          id: "MSSQL_STORE_BATCH_DELETE_AI_TRACES_FAILED",
+          id: "MSSQL_STORE_BATCH_DELETE_TRACES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
@@ -2161,22 +2161,22 @@ ${columns}
       // Spans indexes for optimal trace querying
       {
         name: `${schemaPrefix}mastra_ai_spans_traceid_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["traceId", "startedAt DESC"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_parentspanid_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["parentSpanId", "startedAt DESC"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_name_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["name"]
       },
       {
         name: `${schemaPrefix}mastra_ai_spans_spantype_startedat_idx`,
-        table: TABLE_AI_SPANS,
+        table: TABLE_SPANS,
         columns: ["spanType", "startedAt DESC"]
       }
     ];
@@ -3192,11 +3192,11 @@ var MSSQLStore = class extends MastraStorage {
   }) {
     return this.getObservabilityStore().updateSpan({ spanId, traceId, updates });
   }
-  async getAITrace(traceId) {
-    return this.getObservabilityStore().getAITrace(traceId);
+  async getTrace(traceId) {
+    return this.getObservabilityStore().getTrace(traceId);
   }
-  async getAITracesPaginated(args) {
-    return this.getObservabilityStore().getAITracesPaginated(args);
+  async getTracesPaginated(args) {
+    return this.getObservabilityStore().getTracesPaginated(args);
   }
   async batchCreateSpans(args) {
     return this.getObservabilityStore().batchCreateSpans(args);
@@ -3204,8 +3204,8 @@ var MSSQLStore = class extends MastraStorage {
   async batchUpdateSpans(args) {
     return this.getObservabilityStore().batchUpdateSpans(args);
   }
-  async batchDeleteAITraces(args) {
-    return this.getObservabilityStore().batchDeleteAITraces(args);
+  async batchDeleteTraces(args) {
+    return this.getObservabilityStore().batchDeleteTraces(args);
   }
   /**
    * Scorers