@mastra/core 0.18.1-alpha.0 → 0.19.0

package/dist/storage/index.js

@@ -1,9 +1,9 @@
-import { MastraStorage } from '../chunk-ITD7IHN5.js';
-export { MastraStorage, ensureDate, resolveMessageLimit, serializeDate } from '../chunk-ITD7IHN5.js';
-import { TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS, TABLE_TRACES } from '../chunk-R5GESRGB.js';
-export { AI_SPAN_SCHEMA, SCORERS_SCHEMA, TABLE_AI_SPANS, TABLE_EVALS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT } from '../chunk-R5GESRGB.js';
-import { MessageList } from '../chunk-245ANI3P.js';
-import { MastraError } from '../chunk-MCOVMKIS.js';
+import { MastraStorage } from '../chunk-2DGWSOVQ.js';
+export { MastraStorage, ensureDate, resolveMessageLimit, serializeDate } from '../chunk-2DGWSOVQ.js';
+import { TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS, TABLE_TRACES } from '../chunk-SJMKDSRF.js';
+export { AI_SPAN_SCHEMA, SCORERS_SCHEMA, TABLE_AI_SPANS, TABLE_EVALS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT } from '../chunk-SJMKDSRF.js';
+import { MessageList } from '../chunk-BUY5X7DT.js';
+import { MastraError } from '../chunk-WHFNGECY.js';
 import { MastraBase } from '../chunk-BMVFEBPE.js';
 
 // src/storage/domains/legacy-evals/base.ts
@@ -748,7 +748,15 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
   async createAISpan(span) {
     this.validateCreateAISpan(span);
     const id = this.generateId(span);
-    this.collection.set(id, span);
+    const record = span;
+    record.createdAt = /* @__PURE__ */ new Date();
+    record.updatedAt = record.createdAt;
+    this.collection.set(id, record);
+  }
+  async batchCreateAISpans(args) {
+    for (const record of args.records) {
+      await this.createAISpan(record);
+    }
   }
   validateCreateAISpan(record) {
     if (!record.spanId) {
@@ -829,7 +837,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     return spans.slice(start, end);
   }
   async updateAISpan(params) {
-    const id = this.generateId({ traceId: params.traceId, spanId: params.spanId });
+    const id = this.generateId(params);
     const span = this.collection.get(id);
     if (!span) {
       throw new MastraError({
@@ -839,35 +847,18 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
         text: "Span not found for update"
       });
     }
-    this.collection.set(id, { ...span, ...params.updates });
-  }
-  async batchCreateAISpans(args) {
-    for (const record of args.records) {
-      this.validateCreateAISpan(record);
-      const id = this.generateId({ traceId: record.traceId, spanId: record.spanId });
-      this.collection.set(id, record);
-    }
+    this.collection.set(id, { ...span, ...params.updates, updatedAt: /* @__PURE__ */ new Date() });
   }
   async batchUpdateAISpans(args) {
     for (const record of args.records) {
-      const id = this.generateId({ traceId: record.traceId, spanId: record.spanId });
-      const span = this.collection.get(id);
-      if (!span) {
-        throw new MastraError({
-          id: "OBSERVABILITY_BATCH_UPDATE_AI_SPAN_NOT_FOUND",
-          domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
-          category: "SYSTEM" /* SYSTEM */,
-          text: "Span not found for batch update"
-        });
-      }
-      this.collection.set(id, { ...span, ...record.updates });
+      await this.updateAISpan(record);
     }
   }
   async batchDeleteAITraces(args) {
     for (const traceId of args.traceIds) {
       const spans = Array.from(this.collection.values()).filter((span) => span.traceId === traceId);
       for (const span of spans) {
-        this.collection.delete(this.generateId({ traceId: span.traceId, spanId: span.spanId }));
+        this.collection.delete(this.generateId(span));
       }
     }
   }
@@ -1057,6 +1048,18 @@ var ScoresStorage = class extends MastraBase {
       name: "SCORES"
     });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination: _pagination
+  }) {
+    throw new MastraError({
+      id: "SCORES_STORAGE_GET_SCORES_BY_SPAN_NOT_IMPLEMENTED",
+      domain: "STORAGE" /* STORAGE */,
+      category: "SYSTEM" /* SYSTEM */,
+      details: { traceId, spanId }
+    });
+  }
 };
 
 // src/storage/domains/scores/inmemory.ts
@@ -1138,6 +1141,25 @@ var ScoresInMemory = class extends ScoresStorage {
       }
     };
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    const scores = Array.from(this.scores.values()).filter(
+      (score) => score.traceId === traceId && score.spanId === spanId
+    );
+    scores.sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime());
+    return {
+      scores: scores.slice(pagination.page * pagination.perPage, (pagination.page + 1) * pagination.perPage),
+      pagination: {
+        total: scores.length,
+        page: pagination.page,
+        perPage: pagination.perPage,
+        hasMore: scores.length > (pagination.page + 1) * pagination.perPage
+      }
+    };
+  }
 };
 
 // src/storage/domains/traces/base.ts
@@ -1469,7 +1491,8 @@ var InMemoryStore = class extends MastraStorage {
       createTable: false,
       deleteMessages: true,
       aiTracing: true,
-      indexManagement: false
+      indexManagement: false,
+      getScoresBySpan: true
     };
   }
   async persistWorkflowSnapshot({
@@ -1644,6 +1667,13 @@ var InMemoryStore = class extends MastraStorage {
   }) {
     return this.stores.scores.getScoresByEntityId({ entityId, entityType, pagination });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
   async getEvals(options) {
     return this.stores.legacyEvals.getEvals(options);
   }