@mastra/core 0.18.1-alpha.0 → 0.19.0

package/dist/storage/index.cjs

@@ -1,9 +1,9 @@
 'use strict';
 
-var chunkO6PQYYT6_cjs = require('../chunk-O6PQYYT6.cjs');
-var chunkXUAWHRUO_cjs = require('../chunk-XUAWHRUO.cjs');
-var chunk6X4FZVGT_cjs = require('../chunk-6X4FZVGT.cjs');
-var chunkC73WLCY3_cjs = require('../chunk-C73WLCY3.cjs');
+var chunkHQPFM22N_cjs = require('../chunk-HQPFM22N.cjs');
+var chunk4R2TBRS7_cjs = require('../chunk-4R2TBRS7.cjs');
+var chunkCQ73JM2N_cjs = require('../chunk-CQ73JM2N.cjs');
+var chunkS7UXT3V4_cjs = require('../chunk-S7UXT3V4.cjs');
 var chunk4L3P3PCP_cjs = require('../chunk-4L3P3PCP.cjs');
 
 // src/storage/domains/legacy-evals/base.ts
@@ -311,7 +311,7 @@ var InMemoryMemory = class extends MemoryStorage {
   }) {
     this.logger.debug(`MockStore: getMessagesById called`);
     const rawMessages = messageIds.map((id) => this.collection.messages.get(id)).filter((message) => !!message);
-    const list = new chunk6X4FZVGT_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
+    const list = new chunkCQ73JM2N_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
     if (format === "v1") return list.get.all.v1();
     return list.get.all.v2();
   }
@@ -341,7 +341,7 @@ var InMemoryMemory = class extends MemoryStorage {
       };
       this.collection.messages.set(key, storageMessage);
     }
-    const list = new chunk6X4FZVGT_cjs.MessageList().add(messages, "memory");
+    const list = new chunkCQ73JM2N_cjs.MessageList().add(messages, "memory");
     if (format === `v2`) return list.get.all.v2();
     return list.get.all.v1();
   }
@@ -655,7 +655,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Creates a single AI span record in the storage provider.
    */
   createAISpan(_span) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_CREATE_AI_SPAN_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -666,7 +666,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Updates a single AI span with partial data. Primarily used for realtime trace creation.
    */
   updateAISpan(_params) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_UPDATE_AI_SPAN_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -677,7 +677,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Retrieves a single AI trace with all its associated spans.
    */
   getAITrace(_traceId) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_GET_AI_TRACE_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -688,7 +688,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Retrieves a paginated list of AI traces with optional filtering.
    */
   getAITracesPaginated(_args) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_GET_AI_TRACES_PAGINATED_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -699,7 +699,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Creates multiple AI spans in a single batch.
    */
   batchCreateAISpans(_args) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_BATCH_CREATE_AI_SPAN_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -710,7 +710,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Updates multiple AI spans in a single batch.
    */
   batchUpdateAISpans(_args) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_BATCH_UPDATE_AI_SPAN_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -721,7 +721,7 @@ var ObservabilityStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
    * Deletes multiple AI traces and all their associated spans in a single batch operation.
    */
   batchDeleteAITraces(_args) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "OBSERVABILITY_STORAGE_BATCH_DELETE_AI_SPAN_NOT_IMPLEMENTED",
       domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
       category: "SYSTEM" /* SYSTEM */,
@@ -748,11 +748,19 @@ 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) {
-      throw new chunkC73WLCY3_cjs.MastraError({
+      throw new chunkS7UXT3V4_cjs.MastraError({
         id: "OBSERVABILITY_SPAN_ID_REQUIRED",
         domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
         category: "SYSTEM" /* SYSTEM */,
@@ -760,7 +768,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
       });
     }
     if (!record.traceId) {
-      throw new chunkC73WLCY3_cjs.MastraError({
+      throw new chunkS7UXT3V4_cjs.MastraError({
         id: "OBSERVABILITY_TRACE_ID_REQUIRED",
         domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
         category: "SYSTEM" /* SYSTEM */,
@@ -829,45 +837,28 @@ 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 chunkC73WLCY3_cjs.MastraError({
+      throw new chunkS7UXT3V4_cjs.MastraError({
         id: "OBSERVABILITY_UPDATE_AI_SPAN_NOT_FOUND",
         domain: "MASTRA_OBSERVABILITY" /* MASTRA_OBSERVABILITY */,
         category: "SYSTEM" /* SYSTEM */,
         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 chunkC73WLCY3_cjs.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));
       }
     }
   }
@@ -926,7 +917,7 @@ var StoreOperations = class extends chunk4L3P3PCP_cjs.MastraBase {
    * @throws {MastraError} if not supported by the storage adapter
    */
   async createIndex(_options) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "MASTRA_STORAGE_CREATE_INDEX_NOT_SUPPORTED",
       domain: "STORAGE" /* STORAGE */,
       category: "SYSTEM" /* SYSTEM */,
@@ -938,7 +929,7 @@ var StoreOperations = class extends chunk4L3P3PCP_cjs.MastraBase {
    * @throws {MastraError} if not supported by the storage adapter
    */
   async dropIndex(_indexName) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "MASTRA_STORAGE_DROP_INDEX_NOT_SUPPORTED",
       domain: "STORAGE" /* STORAGE */,
       category: "SYSTEM" /* SYSTEM */,
@@ -950,7 +941,7 @@ var StoreOperations = class extends chunk4L3P3PCP_cjs.MastraBase {
    * @throws {MastraError} if not supported by the storage adapter
    */
   async listIndexes(_tableName) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "MASTRA_STORAGE_LIST_INDEXES_NOT_SUPPORTED",
       domain: "STORAGE" /* STORAGE */,
       category: "SYSTEM" /* SYSTEM */,
@@ -962,7 +953,7 @@ var StoreOperations = class extends chunk4L3P3PCP_cjs.MastraBase {
    * @throws {MastraError} if not supported by the storage adapter
    */
   async describeIndex(_indexName) {
-    throw new chunkC73WLCY3_cjs.MastraError({
+    throw new chunkS7UXT3V4_cjs.MastraError({
       id: "MASTRA_STORAGE_DESCRIBE_INDEX_NOT_SUPPORTED",
       domain: "STORAGE" /* STORAGE */,
       category: "SYSTEM" /* SYSTEM */,
@@ -993,7 +984,7 @@ var StoreOperationsInMemory = class extends StoreOperations {
   async insert({ tableName, record }) {
     const table = this.data[tableName];
     let key = record.id;
-    if ([chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT, chunkXUAWHRUO_cjs.TABLE_EVALS].includes(tableName) && !record.id && record.run_id) {
+    if ([chunk4R2TBRS7_cjs.TABLE_WORKFLOW_SNAPSHOT, chunk4R2TBRS7_cjs.TABLE_EVALS].includes(tableName) && !record.id && record.run_id) {
       key = record.workflow_name ? `${record.workflow_name}-${record.run_id}` : record.run_id;
       record.id = key;
     } else if (!record.id) {
@@ -1006,7 +997,7 @@ var StoreOperationsInMemory = class extends StoreOperations {
     const table = this.data[tableName];
     for (const record of records) {
       let key = record.id;
-      if ([chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT, chunkXUAWHRUO_cjs.TABLE_EVALS].includes(tableName) && !record.id && record.run_id) {
+      if ([chunk4R2TBRS7_cjs.TABLE_WORKFLOW_SNAPSHOT, chunk4R2TBRS7_cjs.TABLE_EVALS].includes(tableName) && !record.id && record.run_id) {
         key = record.run_id;
         record.id = key;
       } else if (!record.id) {
@@ -1057,6 +1048,18 @@ var ScoresStorage = class extends chunk4L3P3PCP_cjs.MastraBase {
       name: "SCORES"
     });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination: _pagination
+  }) {
+    throw new chunkS7UXT3V4_cjs.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
@@ -1223,7 +1245,7 @@ var TracesInMemory = class extends TracesStorage {
   async batchTraceInsert({ records }) {
     this.logger.debug("Batch inserting traces", { count: records.length });
     await this.operations.batchInsert({
-      tableName: chunkXUAWHRUO_cjs.TABLE_TRACES,
+      tableName: chunk4R2TBRS7_cjs.TABLE_TRACES,
       records
     });
   }
@@ -1344,7 +1366,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
       updatedAt: /* @__PURE__ */ new Date()
     };
     await this.operations.insert({
-      tableName: chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT,
+      tableName: chunk4R2TBRS7_cjs.TABLE_WORKFLOW_SNAPSHOT,
       record: data
     });
   }
@@ -1354,7 +1376,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
   }) {
     this.logger.debug("Loading workflow snapshot", { workflowName, runId });
     const d = await this.operations.load({
-      tableName: chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT,
+      tableName: chunk4R2TBRS7_cjs.TABLE_WORKFLOW_SNAPSHOT,
       keys: { workflow_name: workflowName, run_id: runId }
     });
     return d ? JSON.parse(JSON.stringify(d.snapshot)) : null;
@@ -1418,7 +1440,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
 };
 
 // src/storage/mock.ts
-var InMemoryStore = class extends chunkO6PQYYT6_cjs.MastraStorage {
+var InMemoryStore = class extends chunkHQPFM22N_cjs.MastraStorage {
   stores;
   constructor() {
     super({ name: "InMemoryStorage" });
@@ -1469,7 +1491,8 @@ var InMemoryStore = class extends chunkO6PQYYT6_cjs.MastraStorage {
       createTable: false,
       deleteMessages: true,
       aiTracing: true,
-      indexManagement: false
+      indexManagement: false,
+      getScoresBySpan: true
     };
   }
   async persistWorkflowSnapshot({
@@ -1644,6 +1667,13 @@ var InMemoryStore = class extends chunkO6PQYYT6_cjs.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);
   }
@@ -1703,63 +1733,63 @@ function safelyParseJSON(input) {
 
 Object.defineProperty(exports, "MastraStorage", {
   enumerable: true,
-  get: function () { return chunkO6PQYYT6_cjs.MastraStorage; }
+  get: function () { return chunkHQPFM22N_cjs.MastraStorage; }
 });
 Object.defineProperty(exports, "ensureDate", {
   enumerable: true,
-  get: function () { return chunkO6PQYYT6_cjs.ensureDate; }
+  get: function () { return chunkHQPFM22N_cjs.ensureDate; }
 });
 Object.defineProperty(exports, "resolveMessageLimit", {
   enumerable: true,
-  get: function () { return chunkO6PQYYT6_cjs.resolveMessageLimit; }
+  get: function () { return chunkHQPFM22N_cjs.resolveMessageLimit; }
 });
 Object.defineProperty(exports, "serializeDate", {
   enumerable: true,
-  get: function () { return chunkO6PQYYT6_cjs.serializeDate; }
+  get: function () { return chunkHQPFM22N_cjs.serializeDate; }
 });
 Object.defineProperty(exports, "AI_SPAN_SCHEMA", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.AI_SPAN_SCHEMA; }
+  get: function () { return chunk4R2TBRS7_cjs.AI_SPAN_SCHEMA; }
 });
 Object.defineProperty(exports, "SCORERS_SCHEMA", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.SCORERS_SCHEMA; }
+  get: function () { return chunk4R2TBRS7_cjs.SCORERS_SCHEMA; }
 });
 Object.defineProperty(exports, "TABLE_AI_SPANS", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_AI_SPANS; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_AI_SPANS; }
 });
 Object.defineProperty(exports, "TABLE_EVALS", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_EVALS; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_EVALS; }
 });
 Object.defineProperty(exports, "TABLE_MESSAGES", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_MESSAGES; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_MESSAGES; }
 });
 Object.defineProperty(exports, "TABLE_RESOURCES", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_RESOURCES; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_RESOURCES; }
 });
 Object.defineProperty(exports, "TABLE_SCHEMAS", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_SCHEMAS; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_SCHEMAS; }
 });
 Object.defineProperty(exports, "TABLE_SCORERS", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_SCORERS; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_SCORERS; }
 });
 Object.defineProperty(exports, "TABLE_THREADS", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_THREADS; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_THREADS; }
 });
 Object.defineProperty(exports, "TABLE_TRACES", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_TRACES; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_TRACES; }
 });
 Object.defineProperty(exports, "TABLE_WORKFLOW_SNAPSHOT", {
   enumerable: true,
-  get: function () { return chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT; }
+  get: function () { return chunk4R2TBRS7_cjs.TABLE_WORKFLOW_SNAPSHOT; }
 });
 exports.InMemoryMemory = InMemoryMemory;
 exports.InMemoryStore = InMemoryStore;