@mastra/core 1.15.0 → 1.16.0-alpha.0

package/dist/{chunk-VSGRSX3N.cjs → chunk-XAFCD3W2.cjs}

@@ -1,9 +1,9 @@
 'use strict';
 
 var chunkO6O5MD7Q_cjs = require('./chunk-O6O5MD7Q.cjs');
-var chunkZHWQKPUV_cjs = require('./chunk-ZHWQKPUV.cjs');
-var chunkYVFFNB2F_cjs = require('./chunk-YVFFNB2F.cjs');
-var chunkJUL4UHWQ_cjs = require('./chunk-JUL4UHWQ.cjs');
+var chunk57NZ23ZE_cjs = require('./chunk-57NZ23ZE.cjs');
+var chunk542IIQM4_cjs = require('./chunk-542IIQM4.cjs');
+var chunk4SE5WPZS_cjs = require('./chunk-4SE5WPZS.cjs');
 var chunk4U7ZLI36_cjs = require('./chunk-4U7ZLI36.cjs');
 var chunkRO47SMI7_cjs = require('./chunk-RO47SMI7.cjs');
 var jsonToZod = require('@mastra/schema-compat/json-to-zod');
@@ -539,7 +539,7 @@ function safelyParseJSON(input) {
 }
 function transformRow(row, tableName, options = {}) {
   const { preferredTimestampFields = {}, convertTimestamps = false, nullValuePattern, fieldMappings = {} } = options;
-  const tableSchema = chunkZHWQKPUV_cjs.TABLE_SCHEMAS[tableName];
+  const tableSchema = chunk57NZ23ZE_cjs.TABLE_SCHEMAS[tableName];
   const result = {};
   for (const [key, columnSchema] of Object.entries(tableSchema)) {
     const sourceKey = fieldMappings[key] ?? key;
@@ -570,7 +570,7 @@ function transformRow(row, tableName, options = {}) {
   return result;
 }
 function transformScoreRow(row, options = {}) {
-  return transformRow(row, chunkZHWQKPUV_cjs.TABLE_SCORERS, options);
+  return transformRow(row, chunk57NZ23ZE_cjs.TABLE_SCORERS, options);
 }
 function toUpperSnakeCase(str) {
   return str.replace(/([a-z])([A-Z])/g, "$1_$2").replace(/([A-Z])([A-Z][a-z])/g, "$1_$2").toUpperCase().replace(/[^A-Z0-9]+/g, "_").replace(/^_+|_+$/g, "");
@@ -822,7 +822,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     };
   }
   async listTraces(args) {
-    const { filters, pagination, orderBy } = chunkZHWQKPUV_cjs.listTracesArgsSchema.parse(args);
+    const { filters, pagination, orderBy } = chunk57NZ23ZE_cjs.listTracesArgsSchema.parse(args);
     const matchingRootSpans = [];
     for (const [, traceEntry] of this.db.traces) {
       if (!traceEntry.rootSpan) continue;
@@ -851,7 +851,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     const end = start + perPage;
     const paged = matchingRootSpans.slice(start, end);
     return {
-      spans: chunkZHWQKPUV_cjs.toTraceSpans(paged),
+      spans: chunk57NZ23ZE_cjs.toTraceSpans(paged),
       pagination: { total, page, perPage, hasMore: end < total }
     };
   }
@@ -1237,7 +1237,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     return { values };
   }
   async getEntityTypes(_args) {
-    const validTypes = new Set(Object.values(chunkJUL4UHWQ_cjs.EntityType));
+    const validTypes = new Set(Object.values(chunk4SE5WPZS_cjs.EntityType));
     const typeSet = /* @__PURE__ */ new Set();
     for (const [, traceEntry] of this.db.traces) {
       for (const span of Object.values(traceEntry.spans)) {
@@ -1299,7 +1299,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     }
   }
   async listLogs(args) {
-    const { filters, pagination, orderBy } = chunkJUL4UHWQ_cjs.listLogsArgsSchema.parse(args);
+    const { filters, pagination, orderBy } = chunk4SE5WPZS_cjs.listLogsArgsSchema.parse(args);
     let matching = this.db.logRecords.filter((log) => this.logMatchesFilters(log, filters));
     const dir = orderBy.direction === "DESC" ? -1 : 1;
     matching.sort((a, b) => dir * (a.timestamp.getTime() - b.timestamp.getTime()));
@@ -1354,7 +1354,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     }
   }
   async listScores(args) {
-    const { filters, pagination, orderBy } = chunkJUL4UHWQ_cjs.listScoresArgsSchema.parse(args);
+    const { filters, pagination, orderBy } = chunk4SE5WPZS_cjs.listScoresArgsSchema.parse(args);
     let matching = this.db.scoreRecords.filter((score) => this.scoreMatchesFilters(score, filters));
     const dir = orderBy.direction === "DESC" ? -1 : 1;
     if (orderBy.field === "score") {
@@ -1398,7 +1398,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     }
   }
   async listFeedback(args) {
-    const { filters, pagination, orderBy } = chunkJUL4UHWQ_cjs.listFeedbackArgsSchema.parse(args);
+    const { filters, pagination, orderBy } = chunk4SE5WPZS_cjs.listFeedbackArgsSchema.parse(args);
     let matching = this.db.feedbackRecords.filter((fb) => this.feedbackMatchesFilters(fb, filters));
     const dir = orderBy.direction === "DESC" ? -1 : 1;
     matching.sort((a, b) => dir * (a.timestamp.getTime() - b.timestamp.getTime()));
@@ -1431,7 +1431,7 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
 };
 
 // src/storage/domains/observability/record-builders.ts
-var entityTypeValues = new Set(Object.values(chunkJUL4UHWQ_cjs.EntityType));
+var entityTypeValues = new Set(Object.values(chunk4SE5WPZS_cjs.EntityType));
 function toEntityType(value) {
   if (value && entityTypeValues.has(value)) {
     return value;
@@ -1836,7 +1836,7 @@ var InMemoryAgentsStorage = class extends AgentsStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       agents = agents.filter((agent) => {
         if (!agent.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(agent.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(agent.metadata[key], value));
       });
     }
     const sortedAgents = this.sortAgents(agents, field, direction);
@@ -2283,6 +2283,7 @@ function toDatasetItem(row) {
     groundTruth: row.groundTruth,
     requestContext: row.requestContext,
     metadata: row.metadata,
+    source: row.source,
     createdAt: row.createdAt,
     updatedAt: row.updatedAt
   };
@@ -2318,6 +2319,8 @@ var DatasetsInMemory = class extends DatasetsStorage {
       inputSchema: input.inputSchema,
       groundTruthSchema: input.groundTruthSchema,
       requestContextSchema: input.requestContextSchema,
+      targetType: input.targetType,
+      targetIds: input.targetIds,
       version: 0,
       createdAt: now,
       updatedAt: now
@@ -2342,6 +2345,9 @@ var DatasetsInMemory = class extends DatasetsStorage {
       inputSchema: args.inputSchema !== void 0 ? args.inputSchema : existing.inputSchema,
       groundTruthSchema: args.groundTruthSchema !== void 0 ? args.groundTruthSchema : existing.groundTruthSchema,
       requestContextSchema: args.requestContextSchema !== void 0 ? args.requestContextSchema : existing.requestContextSchema,
+      tags: args.tags !== void 0 ? args.tags : existing.tags,
+      targetType: args.targetType !== void 0 ? args.targetType : existing.targetType,
+      targetIds: args.targetIds !== void 0 ? args.targetIds : existing.targetIds,
       updatedAt: /* @__PURE__ */ new Date()
     };
     this.db.datasets.set(args.id, updated);
@@ -2402,6 +2408,7 @@ var DatasetsInMemory = class extends DatasetsStorage {
       groundTruth: args.groundTruth,
       requestContext: args.requestContext,
       metadata: args.metadata,
+      source: args.source,
       createdAt: now,
       updatedAt: now
     };
@@ -2439,6 +2446,7 @@ var DatasetsInMemory = class extends DatasetsStorage {
       groundTruth: args.groundTruth ?? currentRow.groundTruth,
       requestContext: args.requestContext ?? currentRow.requestContext,
       metadata: args.metadata ?? currentRow.metadata,
+      source: args.source ?? currentRow.source,
       createdAt: currentRow.createdAt,
       updatedAt: now
     };
@@ -2604,6 +2612,7 @@ var DatasetsInMemory = class extends DatasetsStorage {
         groundTruth: itemInput.groundTruth,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
+        source: itemInput.source,
         createdAt: now,
         updatedAt: now
       };
@@ -2762,11 +2771,29 @@ var ExperimentsInMemory = class extends ExperimentsStorage {
       completedAt: input.completedAt,
       retryCount: input.retryCount,
       traceId: input.traceId ?? null,
+      status: input.status ?? null,
+      tags: input.tags ?? null,
       createdAt: now
     };
     this.db.experimentResults.set(result.id, result);
     return result;
   }
+  async updateExperimentResult(input) {
+    const existing = this.db.experimentResults.get(input.id);
+    if (!existing) {
+      throw new Error(`Experiment result not found: ${input.id}`);
+    }
+    if (input.experimentId && existing.experimentId !== input.experimentId) {
+      throw new Error(`Experiment result ${input.id} does not belong to experiment ${input.experimentId}`);
+    }
+    const updated = {
+      ...existing,
+      status: input.status !== void 0 ? input.status : existing.status,
+      tags: input.tags !== void 0 ? input.tags : existing.tags
+    };
+    this.db.experimentResults.set(input.id, updated);
+    return updated;
+  }
   async getExperimentResultById(args) {
     return this.db.experimentResults.get(args.id) ?? null;
   }
@@ -2984,7 +3011,7 @@ var InMemoryMCPClientsStorage = class extends MCPClientsStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       configs = configs.filter((config) => {
         if (!config.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(config.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(config.metadata[key], value));
       });
     }
     const sortedConfigs = this.sortConfigs(configs, field, direction);
@@ -3252,7 +3279,7 @@ var InMemoryMCPServersStorage = class extends MCPServersStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       configs = configs.filter((config) => {
         if (!config.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(config.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(config.metadata[key], value));
       });
     }
     const sortedConfigs = this.sortConfigs(configs, field, direction);
@@ -4717,7 +4744,7 @@ var InMemoryPromptBlocksStorage = class extends PromptBlocksStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       blocks = blocks.filter((block) => {
         if (!block.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(block.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(block.metadata[key], value));
       });
     }
     const sortedBlocks = this.sortBlocks(blocks, field, direction);
@@ -4983,7 +5010,7 @@ var InMemoryScorerDefinitionsStorage = class extends ScorerDefinitionsStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       scorers = scorers.filter((scorer) => {
         if (!scorer.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(scorer.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(scorer.metadata[key], value));
       });
     }
     const sortedScorers = this.sortScorers(scorers, field, direction);
@@ -6031,7 +6058,7 @@ var InMemoryWorkspacesStorage = class extends WorkspacesStorage {
     if (metadata && Object.keys(metadata).length > 0) {
       configs = configs.filter((config) => {
         if (!config.metadata) return false;
-        return Object.entries(metadata).every(([key, value]) => chunkYVFFNB2F_cjs.deepEqual(config.metadata[key], value));
+        return Object.entries(metadata).every(([key, value]) => chunk542IIQM4_cjs.deepEqual(config.metadata[key], value));
       });
     }
     const sortedConfigs = this.sortConfigs(configs, field, direction);
@@ -7871,7 +7898,7 @@ var StoreOperationsInMemory = class extends StoreOperations {
   async insert({ tableName, record }) {
     const table = this.data[tableName];
     let key = record.id;
-    if ([chunkZHWQKPUV_cjs.TABLE_WORKFLOW_SNAPSHOT].includes(tableName) && !record.id && record.run_id) {
+    if ([chunk57NZ23ZE_cjs.TABLE_WORKFLOW_SNAPSHOT].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) {
@@ -7884,7 +7911,7 @@ var StoreOperationsInMemory = class extends StoreOperations {
     const table = this.data[tableName];
     for (const record of records) {
       let key = record.id;
-      if ([chunkZHWQKPUV_cjs.TABLE_WORKFLOW_SNAPSHOT].includes(tableName) && !record.id && record.run_id) {
+      if ([chunk57NZ23ZE_cjs.TABLE_WORKFLOW_SNAPSHOT].includes(tableName) && !record.id && record.run_id) {
         key = record.run_id;
         record.id = key;
       } else if (!record.id) {
@@ -8006,5 +8033,5 @@ exports.serializeSpanAttributes = serializeSpanAttributes;
 exports.toEntityType = toEntityType;
 exports.transformRow = transformRow;
 exports.transformScoreRow = transformScoreRow;
-//# sourceMappingURL=chunk-VSGRSX3N.cjs.map
-//# sourceMappingURL=chunk-VSGRSX3N.cjs.map
+//# sourceMappingURL=chunk-XAFCD3W2.cjs.map
+//# sourceMappingURL=chunk-XAFCD3W2.cjs.map