@mastra/core 1.18.0-alpha.3 → 1.18.0-alpha.4

package/dist/{chunk-W2QYKEBN.js → chunk-ERAZTMF6.js}

@@ -1,7 +1,7 @@
 import { MessageList } from './chunk-SLZ3WO42.js';
-import { TABLE_SCHEMAS, TABLE_SCORERS, listTracesArgsSchema, toTraceSpans, TABLE_WORKFLOW_SNAPSHOT } from './chunk-SLHDZJ4B.js';
-import { deepEqual } from './chunk-AHFHYQS6.js';
-import { EntityType, listMetricsArgsSchema, listLogsArgsSchema, listScoresArgsSchema, listFeedbackArgsSchema } from './chunk-CX5I6VS4.js';
+import { TABLE_SCHEMAS, TABLE_SCORERS, listTracesArgsSchema, toTraceSpans, TABLE_WORKFLOW_SNAPSHOT } from './chunk-RHQHBU6R.js';
+import { deepEqual } from './chunk-LQHNR4L7.js';
+import { EntityType, listMetricsArgsSchema, listLogsArgsSchema, listScoresArgsSchema, listFeedbackArgsSchema } from './chunk-QE7OKYS3.js';
 import { MastraError } from './chunk-FJEVLHJT.js';
 import { MastraBase } from './chunk-WCAFTXGK.js';
 import { jsonSchemaToZod } from '@mastra/schema-compat/json-to-zod';
@@ -1056,7 +1056,9 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
       if (filters.experimentId !== void 0 && m.experimentId !== filters.experimentId) return false;
       if (filters.serviceName !== void 0 && m.serviceName !== filters.serviceName) return false;
       if (filters.environment !== void 0 && m.environment !== filters.environment) return false;
-      if (filters.source !== void 0 && m.source !== filters.source) return false;
+      const metricExecutionSource = m.executionSource ?? m.source ?? null;
+      if (filters.executionSource !== void 0 && metricExecutionSource !== filters.executionSource) return false;
+      if (filters.source !== void 0 && metricExecutionSource !== filters.source) return false;
       if (filters.parentEntityType !== void 0 && m.parentEntityType !== filters.parentEntityType) return false;
       if (filters.parentEntityName !== void 0 && m.parentEntityName !== filters.parentEntityName) return false;
       if (filters.rootEntityType !== void 0 && m.rootEntityType !== filters.rootEntityType) return false;
@@ -1433,7 +1435,9 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     if (filters.rootEntityName !== void 0 && log.rootEntityName !== filters.rootEntityName) return false;
     if (filters.serviceName !== void 0 && log.serviceName !== filters.serviceName) return false;
     if (filters.environment !== void 0 && log.environment !== filters.environment) return false;
-    if (filters.source !== void 0 && log.source !== filters.source) return false;
+    const logExecutionSource = log.executionSource ?? log.source ?? null;
+    if (filters.executionSource !== void 0 && logExecutionSource !== filters.executionSource) return false;
+    if (filters.source !== void 0 && logExecutionSource !== filters.source) return false;
     if (filters.experimentId !== void 0 && log.experimentId !== filters.experimentId) return false;
     if (filters.tags != null && filters.tags.length > 0) {
       if (log.tags == null) return false;
@@ -1447,11 +1451,21 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
   // Scores
   // ============================================================================
   async createScore(args) {
-    this.db.scoreRecords.push(args.score);
+    const scoreSource = args.score.scoreSource ?? args.score.source ?? null;
+    this.db.scoreRecords.push({
+      ...args.score,
+      scoreSource,
+      source: scoreSource
+    });
   }
   async batchCreateScores(args) {
     for (const score of args.scores) {
-      this.db.scoreRecords.push(score);
+      const scoreSource = score.scoreSource ?? score.source ?? null;
+      this.db.scoreRecords.push({
+        ...score,
+        scoreSource,
+        source: scoreSource
+      });
     }
   }
   async listScores(args) {
@@ -1480,22 +1494,57 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     }
     if (filters.traceId !== void 0 && score.traceId !== filters.traceId) return false;
     if (filters.spanId !== void 0 && score.spanId !== filters.spanId) return false;
+    if (filters.entityType !== void 0 && score.entityType !== filters.entityType) return false;
+    if (filters.entityName !== void 0 && score.entityName !== filters.entityName) return false;
+    if (filters.userId !== void 0 && score.userId !== filters.userId) return false;
+    if (filters.organizationId !== void 0 && score.organizationId !== filters.organizationId) return false;
+    if (filters.resourceId !== void 0 && score.resourceId !== filters.resourceId) return false;
+    if (filters.runId !== void 0 && score.runId !== filters.runId) return false;
+    if (filters.sessionId !== void 0 && score.sessionId !== filters.sessionId) return false;
+    if (filters.threadId !== void 0 && score.threadId !== filters.threadId) return false;
+    if (filters.requestId !== void 0 && score.requestId !== filters.requestId) return false;
+    if (filters.parentEntityType !== void 0 && score.parentEntityType !== filters.parentEntityType) return false;
+    if (filters.parentEntityName !== void 0 && score.parentEntityName !== filters.parentEntityName) return false;
+    if (filters.rootEntityType !== void 0 && score.rootEntityType !== filters.rootEntityType) return false;
+    if (filters.rootEntityName !== void 0 && score.rootEntityName !== filters.rootEntityName) return false;
+    if (filters.serviceName !== void 0 && score.serviceName !== filters.serviceName) return false;
+    if (filters.environment !== void 0 && score.environment !== filters.environment) return false;
+    if (filters.executionSource !== void 0 && score.executionSource !== filters.executionSource) return false;
     if (filters.scorerId !== void 0) {
       const names = Array.isArray(filters.scorerId) ? filters.scorerId : [filters.scorerId];
       if (!names.includes(score.scorerId)) return false;
     }
+    const scoreSource = score.scoreSource ?? score.source ?? null;
+    if (filters.scoreSource !== void 0 && scoreSource !== filters.scoreSource) return false;
+    if (filters.source !== void 0 && scoreSource !== filters.source) return false;
     if (filters.experimentId !== void 0 && score.experimentId !== filters.experimentId) return false;
+    if (filters.tags != null && filters.tags.length > 0) {
+      if (score.tags == null) return false;
+      for (const tag of filters.tags) {
+        if (!score.tags.includes(tag)) return false;
+      }
+    }
     return true;
   }
   // ============================================================================
   // Feedback
   // ============================================================================
   async createFeedback(args) {
-    this.db.feedbackRecords.push(args.feedback);
+    this.db.feedbackRecords.push({
+      ...args.feedback,
+      feedbackSource: args.feedback.feedbackSource ?? args.feedback.source ?? "",
+      source: args.feedback.feedbackSource ?? args.feedback.source ?? "",
+      feedbackUserId: args.feedback.feedbackUserId ?? args.feedback.userId ?? (typeof args.feedback.metadata?.userId === "string" ? args.feedback.metadata.userId : null)
+    });
   }
   async batchCreateFeedback(args) {
     for (const fb of args.feedbacks) {
-      this.db.feedbackRecords.push(fb);
+      this.db.feedbackRecords.push({
+        ...fb,
+        feedbackSource: fb.feedbackSource ?? fb.source ?? "",
+        source: fb.feedbackSource ?? fb.source ?? "",
+        feedbackUserId: fb.feedbackUserId ?? fb.userId ?? (typeof fb.metadata?.userId === "string" ? fb.metadata.userId : null)
+      });
     }
   }
   async listFeedback(args) {
@@ -1520,13 +1569,37 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     }
     if (filters.traceId !== void 0 && fb.traceId !== filters.traceId) return false;
     if (filters.spanId !== void 0 && fb.spanId !== filters.spanId) return false;
+    if (filters.entityType !== void 0 && fb.entityType !== filters.entityType) return false;
+    if (filters.entityName !== void 0 && fb.entityName !== filters.entityName) return false;
+    if (filters.userId !== void 0 && fb.userId !== filters.userId) return false;
+    if (filters.organizationId !== void 0 && fb.organizationId !== filters.organizationId) return false;
+    if (filters.resourceId !== void 0 && fb.resourceId !== filters.resourceId) return false;
+    if (filters.runId !== void 0 && fb.runId !== filters.runId) return false;
+    if (filters.sessionId !== void 0 && fb.sessionId !== filters.sessionId) return false;
+    if (filters.threadId !== void 0 && fb.threadId !== filters.threadId) return false;
+    if (filters.requestId !== void 0 && fb.requestId !== filters.requestId) return false;
+    if (filters.parentEntityType !== void 0 && fb.parentEntityType !== filters.parentEntityType) return false;
+    if (filters.parentEntityName !== void 0 && fb.parentEntityName !== filters.parentEntityName) return false;
+    if (filters.rootEntityType !== void 0 && fb.rootEntityType !== filters.rootEntityType) return false;
+    if (filters.rootEntityName !== void 0 && fb.rootEntityName !== filters.rootEntityName) return false;
+    if (filters.serviceName !== void 0 && fb.serviceName !== filters.serviceName) return false;
+    if (filters.environment !== void 0 && fb.environment !== filters.environment) return false;
+    if (filters.executionSource !== void 0 && fb.executionSource !== filters.executionSource) return false;
     if (filters.feedbackType !== void 0) {
       const types = Array.isArray(filters.feedbackType) ? filters.feedbackType : [filters.feedbackType];
       if (!types.includes(fb.feedbackType)) return false;
     }
-    if (filters.source !== void 0 && fb.source !== filters.source) return false;
+    const feedbackSource = fb.feedbackSource ?? fb.source ?? "";
+    if (filters.feedbackSource !== void 0 && feedbackSource !== filters.feedbackSource) return false;
+    if (filters.source !== void 0 && feedbackSource !== filters.source) return false;
     if (filters.experimentId !== void 0 && fb.experimentId !== filters.experimentId) return false;
-    if (filters.userId !== void 0 && fb.userId !== filters.userId) return false;
+    if (filters.feedbackUserId !== void 0 && fb.feedbackUserId !== filters.feedbackUserId) return false;
+    if (filters.tags != null && filters.tags.length > 0) {
+      if (fb.tags == null) return false;
+      for (const tag of filters.tags) {
+        if (!fb.tags.includes(tag)) return false;
+      }
+    }
     return true;
   }
 };
@@ -1564,8 +1637,6 @@ function serializeSpanAttributes(span) {
 }
 function buildCorrelationRecordFields(context) {
   return {
-    traceId: context?.traceId ?? null,
-    spanId: context?.spanId ?? null,
     tags: context?.tags ?? null,
     entityType: context?.entityType ?? null,
     entityId: context?.entityId ?? null,
@@ -1584,7 +1655,7 @@ function buildCorrelationRecordFields(context) {
     threadId: context?.threadId ?? null,
     requestId: context?.requestId ?? null,
     environment: context?.environment ?? null,
-    source: context?.source ?? null,
+    executionSource: context?.source ?? null,
     serviceName: context?.serviceName ?? null,
     experimentId: context?.experimentId ?? null
   };
@@ -1616,7 +1687,7 @@ function buildLegacyLogMetadataCorrelationFields(metadata) {
     rootEntityType: toEntityType(getStringOrNull(metadata?.root_type) ?? void 0),
     rootEntityName: getStringOrNull(metadata?.root_name),
     environment: getStringOrNull(metadata?.environment),
-    source: getStringOrNull(metadata?.source),
+    executionSource: getStringOrNull(metadata?.source),
     serviceName: getStringOrNull(metadata?.service_name)
   };
 }
@@ -1686,6 +1757,8 @@ function buildMetricRecord(event) {
     name: m.name,
     value: m.value,
     labels,
+    traceId: m.traceId ?? m.correlationContext?.traceId ?? null,
+    spanId: m.spanId ?? m.correlationContext?.spanId ?? null,
     ...correlationFields,
     scope: null,
     entityType: correlationFields.entityType ?? legacyCorrelationFields.entityType ?? null,
@@ -1711,8 +1784,8 @@ function buildLogRecord(event) {
     message: l.message,
     data: l.data ?? null,
     ...correlationFields,
-    traceId: correlationFields.traceId ?? l.traceId ?? null,
-    spanId: correlationFields.spanId ?? l.spanId ?? null,
+    traceId: l.traceId ?? l.correlationContext?.traceId ?? null,
+    spanId: l.spanId ?? l.correlationContext?.spanId ?? null,
     tags: correlationFields.tags ?? l.tags ?? null,
     entityType: correlationFields.entityType ?? legacyCorrelationFields.entityType ?? null,
     entityName: correlationFields.entityName ?? legacyCorrelationFields.entityName ?? null,
@@ -1721,7 +1794,7 @@ function buildLogRecord(event) {
     rootEntityType: correlationFields.rootEntityType ?? legacyCorrelationFields.rootEntityType ?? null,
     rootEntityName: correlationFields.rootEntityName ?? legacyCorrelationFields.rootEntityName ?? null,
     environment: correlationFields.environment ?? legacyCorrelationFields.environment ?? null,
-    source: correlationFields.source ?? legacyCorrelationFields.source ?? null,
+    executionSource: correlationFields.executionSource ?? legacyCorrelationFields.executionSource ?? null,
     serviceName: correlationFields.serviceName ?? legacyCorrelationFields.serviceName ?? null,
     scope: null,
     metadata: l.metadata ?? null
@@ -1729,33 +1802,41 @@ function buildLogRecord(event) {
 }
 function buildScoreRecord(event) {
   const s = event.score;
+  const correlationFields = buildCorrelationRecordFields(s.correlationContext);
   return {
     timestamp: s.timestamp,
-    traceId: s.traceId,
-    spanId: s.spanId ?? null,
+    traceId: s.traceId ?? s.correlationContext?.traceId ?? null,
+    spanId: s.spanId ?? s.correlationContext?.spanId ?? null,
     scorerId: s.scorerId,
     scorerVersion: s.scorerVersion ?? null,
-    source: s.source ?? null,
+    scoreSource: s.scoreSource ?? s.source ?? null,
+    source: s.scoreSource ?? s.source ?? null,
     score: s.score,
     reason: s.reason ?? null,
-    experimentId: s.experimentId ?? null,
+    ...correlationFields,
+    experimentId: correlationFields.experimentId ?? s.experimentId ?? null,
+    scope: null,
     scoreTraceId: s.scoreTraceId ?? null,
     metadata: s.metadata ?? null
   };
 }
 function buildFeedbackRecord(event) {
   const fb = event.feedback;
-  const userId = typeof fb.metadata?.userId === "string" ? fb.metadata.userId : null;
+  const correlationFields = buildCorrelationRecordFields(fb.correlationContext);
   return {
     timestamp: fb.timestamp,
-    traceId: fb.traceId,
-    spanId: fb.spanId ?? null,
-    source: fb.source,
+    traceId: fb.traceId ?? fb.correlationContext?.traceId ?? null,
+    spanId: fb.spanId ?? fb.correlationContext?.spanId ?? null,
+    feedbackSource: fb.feedbackSource ?? fb.source ?? "",
+    source: fb.feedbackSource ?? fb.source ?? "",
     feedbackType: fb.feedbackType,
     value: fb.value,
     comment: fb.comment ?? null,
-    experimentId: fb.experimentId ?? null,
-    userId,
+    ...correlationFields,
+    experimentId: correlationFields.experimentId ?? fb.experimentId ?? null,
+    feedbackUserId: fb.feedbackUserId ?? fb.userId ?? (typeof fb.metadata?.userId === "string" ? fb.metadata.userId : null),
+    scope: null,
+    sourceId: fb.sourceId ?? null,
     metadata: fb.metadata ?? null
   };
 }
@@ -2473,6 +2554,7 @@ var DatasetsInMemory = class extends DatasetsStorage {
       requestContextSchema: input.requestContextSchema,
       targetType: input.targetType,
       targetIds: input.targetIds,
+      scorerIds: input.scorerIds ?? null,
       version: 0,
       createdAt: now,
       updatedAt: now
@@ -2500,6 +2582,7 @@ var DatasetsInMemory = class extends DatasetsStorage {
       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,
+      scorerIds: args.scorerIds !== void 0 ? args.scorerIds : existing.scorerIds,
       updatedAt: /* @__PURE__ */ new Date()
     };
     this.db.datasets.set(args.id, updated);
@@ -8123,5 +8206,5 @@ var StoreOperationsInMemory = class extends StoreOperations {
 };
 
 export { AgentsStorage, BlobStore, DatasetsInMemory, DatasetsStorage, EDITOR_DOMAINS, ExperimentsInMemory, ExperimentsStorage, FilesystemAgentsStorage, FilesystemDB, FilesystemMCPClientsStorage, FilesystemMCPServersStorage, FilesystemPromptBlocksStorage, FilesystemScorerDefinitionsStorage, FilesystemSkillsStorage, FilesystemStore, FilesystemVersionedHelpers, FilesystemWorkspacesStorage, GitHistory, InMemoryAgentsStorage, InMemoryBlobStore, InMemoryDB, InMemoryMCPClientsStorage, InMemoryMCPServersStorage, InMemoryMemory, InMemoryPromptBlocksStorage, InMemoryScorerDefinitionsStorage, InMemorySkillsStorage, InMemoryStore, InMemoryWorkspacesStorage, MCPClientsStorage, MCPServersStorage, MastraCompositeStore, MastraStorage, MemoryStorage, MockStore, ObservabilityInMemory, ObservabilityStorage, PromptBlocksStorage, SchemaUpdateValidationError, SchemaValidationError, SchemaValidator, ScorerDefinitionsStorage, ScoresInMemory, ScoresStorage, SkillsStorage, StorageDomain, StoreOperations, StoreOperationsInMemory, VersionedStorageDomain, WorkflowsInMemory, WorkflowsStorage, WorkspacesStorage, buildCreateSpanRecord, buildFeedbackRecord, buildLogRecord, buildMetricRecord, buildScoreRecord, buildUpdateSpanRecord, calculatePagination, createPendingMarker, createStorageErrorId, createStoreErrorId, createValidator, createVectorErrorId, ensureDate, filterByDateRange, getDefaultValue, getObjectOrNull, getSchemaValidator, getSqlType, getStringOrNull, jsonValueEquals, normalizePerPage, safelyParseJSON, serializeDate, serializeSpanAttributes, toEntityType, transformRow, transformScoreRow };
-//# sourceMappingURL=chunk-W2QYKEBN.js.map
-//# sourceMappingURL=chunk-W2QYKEBN.js.map
+//# sourceMappingURL=chunk-ERAZTMF6.js.map
+//# sourceMappingURL=chunk-ERAZTMF6.js.map