@mastra/dynamodb 0.0.0-scorers-api-v2-20250801171841 → 0.0.0-scorers-logs-20251208093427

package/src/entities/eval.ts

@@ -1,102 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const evalEntity = new Entity({
-  model: {
-    entity: 'eval',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    input: {
-      type: 'string',
-      required: true,
-    },
-    output: {
-      type: 'string',
-      required: true,
-    },
-    result: {
-      type: 'string', // JSON stringified
-      required: true,
-      // Stringify object on set
-      set: (value?: any) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get
-      get: (value?: string) => {
-        if (value) {
-          return JSON.parse(value);
-        }
-        return value;
-      },
-    },
-    agent_name: {
-      type: 'string',
-      required: true,
-    },
-    metric_name: {
-      type: 'string',
-      required: true,
-    },
-    instructions: {
-      type: 'string',
-      required: true,
-    },
-    test_info: {
-      type: 'string', // JSON stringified
-      required: false,
-      // Stringify object on set
-      set: (value?: any) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get
-      get: (value?: string) => {
-        return value;
-      },
-    },
-    global_run_id: {
-      type: 'string',
-      required: true,
-    },
-    run_id: {
-      type: 'string',
-      required: true,
-    },
-    created_at: {
-      type: 'string',
-      required: true,
-      // Initialize with current timestamp if not provided
-      default: () => new Date().toISOString(),
-      // Convert Date to ISO string on set
-      set: (value?: Date | string) => {
-        if (value instanceof Date) {
-          return value.toISOString();
-        }
-        return value || new Date().toISOString();
-      },
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'run_id'] },
-      sk: { field: 'sk', composite: [] },
-    },
-    byAgent: {
-      index: 'gsi1',
-      pk: { field: 'gsi1pk', composite: ['entity', 'agent_name'] },
-      sk: { field: 'gsi1sk', composite: ['created_at'] },
-    },
-  },
-});

package/src/entities/index.ts

@@ -1,27 +0,0 @@
-import type { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
-import { Service } from 'electrodb';
-import { evalEntity } from './eval';
-import { messageEntity } from './message';
-import { resourceEntity } from './resource';
-import { scoreEntity } from './score';
-import { threadEntity } from './thread';
-import { traceEntity } from './trace';
-import { workflowSnapshotEntity } from './workflow-snapshot';
-
-export function getElectroDbService(client: DynamoDBDocumentClient, tableName: string) {
-  return new Service(
-    {
-      thread: threadEntity,
-      message: messageEntity,
-      eval: evalEntity,
-      trace: traceEntity,
-      workflow_snapshot: workflowSnapshotEntity,
-      resource: resourceEntity,
-      score: scoreEntity,
-    },
-    {
-      client,
-      table: tableName,
-    },
-  );
-}

package/src/entities/message.ts

@@ -1,143 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const messageEntity = new Entity({
-  model: {
-    entity: 'message',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    id: {
-      type: 'string',
-      required: true,
-    },
-    threadId: {
-      type: 'string',
-      required: true,
-    },
-    content: {
-      type: 'string',
-      required: true,
-      // Stringify content object on set if it's not already a string
-      set: (value?: string | void) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get ONLY if it looks like JSON
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            // Attempt to parse only if it might be JSON (e.g., starts with { or [)
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            // Ignore parse error, return original string
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    role: {
-      type: 'string',
-      required: true,
-    },
-    type: {
-      type: 'string',
-      default: 'text',
-    },
-    resourceId: {
-      type: 'string',
-      required: false,
-    },
-    toolCallIds: {
-      type: 'string',
-      required: false,
-      set: (value?: string[] | string) => {
-        if (Array.isArray(value)) {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to array on get
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            return JSON.parse(value);
-          } catch {
-            // Return raw value on error, consistent with 'content' field
-            return value;
-          }
-        }
-        // If value was not a string, or if it was an empty string, return it as is.
-        return value;
-      },
-    },
-    toolCallArgs: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown>[] | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            return JSON.parse(value);
-          } catch {
-            // Return raw value on error, consistent with 'content' field
-            return value;
-          }
-        }
-        // If value was not a string, or if it was an empty string, return it as is.
-        return value;
-      },
-    },
-    toolNames: {
-      type: 'string',
-      required: false,
-      set: (value?: string[] | string) => {
-        if (Array.isArray(value)) {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to array on get
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            return JSON.parse(value);
-          } catch {
-            // Return raw value on error, consistent with 'content' field
-            return value;
-          }
-        }
-        // If value was not a string, or if it was an empty string, return it as is.
-        return value;
-      },
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'id'] },
-      sk: { field: 'sk', composite: ['entity'] },
-    },
-    byThread: {
-      index: 'gsi1',
-      pk: { field: 'gsi1pk', composite: ['entity', 'threadId'] },
-      sk: { field: 'gsi1sk', composite: ['createdAt'] },
-    },
-  },
-});

package/src/entities/resource.ts

@@ -1,57 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const resourceEntity = new Entity({
-  model: {
-    entity: 'resource',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    id: {
-      type: 'string',
-      required: true,
-    },
-    workingMemory: {
-      type: 'string',
-      required: false,
-    },
-    metadata: {
-      type: 'string',
-      required: false,
-      // Stringify content object on set if it's not already a string
-      set: (value?: string | void) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get ONLY if it looks like JSON
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            // Attempt to parse only if it might be JSON (e.g., starts with { or [)
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            // Ignore parse error, return original string
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'id'] },
-      sk: { field: 'sk', composite: ['entity'] },
-    },
-  },
-});

package/src/entities/score.ts

@@ -1,317 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const scoreEntity = new Entity({
-  model: {
-    entity: 'score',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    id: {
-      type: 'string',
-      required: true,
-    },
-    scorerId: {
-      type: 'string',
-      required: true,
-    },
-    traceId: {
-      type: 'string',
-      required: false,
-    },
-    runId: {
-      type: 'string',
-      required: true,
-    },
-    scorer: {
-      type: 'string',
-      required: true,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    extractStepResult: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    preprocessStepResult: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    analyzeStepResult: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    score: {
-      type: 'number',
-      required: true,
-    },
-    reason: {
-      type: 'string',
-      required: false,
-    },
-    extractPrompt: {
-      type: 'string',
-      required: false,
-    },
-    analyzePrompt: {
-      type: 'string',
-      required: false,
-    },
-
-    // Deprecated in favor of generateReasonPrompt
-    reasonPrompt: {
-      type: 'string',
-      required: false,
-    },
-    generateScorePrompt: {
-      type: 'string',
-      required: false,
-    },
-    generateReasonPrompt: {
-      type: 'string',
-      required: false,
-    },
-    input: {
-      type: 'string',
-      required: true,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    output: {
-      type: 'string',
-      required: true,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    additionalContext: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    runtimeContext: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    entityType: {
-      type: 'string',
-      required: false,
-    },
-    entityData: {
-      type: 'string',
-      required: false,
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-    entityId: {
-      type: 'string',
-      required: false,
-    },
-    source: {
-      type: 'string',
-      required: true,
-    },
-    resourceId: {
-      type: 'string',
-      required: false,
-    },
-    threadId: {
-      type: 'string',
-      required: false,
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'id'] },
-      sk: { field: 'sk', composite: ['entity'] },
-    },
-    byScorer: {
-      index: 'gsi1',
-      pk: { field: 'gsi1pk', composite: ['entity', 'scorerId'] },
-      sk: { field: 'gsi1sk', composite: ['createdAt'] },
-    },
-    byRun: {
-      index: 'gsi2',
-      pk: { field: 'gsi2pk', composite: ['entity', 'runId'] },
-      sk: { field: 'gsi2sk', composite: ['createdAt'] },
-    },
-    byTrace: {
-      index: 'gsi3',
-      pk: { field: 'gsi3pk', composite: ['entity', 'traceId'] },
-      sk: { field: 'gsi3sk', composite: ['createdAt'] },
-    },
-    byEntityData: {
-      index: 'gsi4',
-      pk: { field: 'gsi4pk', composite: ['entity', 'entityId'] },
-      sk: { field: 'gsi4sk', composite: ['createdAt'] },
-    },
-    byResource: {
-      index: 'gsi5',
-      pk: { field: 'gsi5pk', composite: ['entity', 'resourceId'] },
-      sk: { field: 'gsi5sk', composite: ['createdAt'] },
-    },
-    byThread: {
-      index: 'gsi6',
-      pk: { field: 'gsi6pk', composite: ['entity', 'threadId'] },
-      sk: { field: 'gsi6sk', composite: ['createdAt'] },
-    },
-  },
-});

package/src/entities/thread.ts

@@ -1,66 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const threadEntity = new Entity({
-  model: {
-    entity: 'thread',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    id: {
-      type: 'string',
-      required: true,
-    },
-    resourceId: {
-      type: 'string',
-      required: true,
-    },
-    title: {
-      type: 'string',
-      required: true,
-    },
-    metadata: {
-      type: 'string',
-      required: false,
-      // Stringify metadata object on set if it's not already a string
-      set: (value?: Record<string, unknown> | string) => {
-        if (value && typeof value !== 'string') {
-          return JSON.stringify(value);
-        }
-        return value;
-      },
-      // Parse JSON string to object on get
-      get: (value?: string) => {
-        if (value && typeof value === 'string') {
-          try {
-            // Attempt to parse only if it might be JSON (e.g., starts with { or [)
-            if (value.startsWith('{') || value.startsWith('[')) {
-              return JSON.parse(value);
-            }
-          } catch {
-            // Ignore parse error, return original string
-            return value;
-          }
-        }
-        return value;
-      },
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'id'] },
-      sk: { field: 'sk', composite: ['id'] },
-    },
-    byResource: {
-      index: 'gsi1',
-      pk: { field: 'gsi1pk', composite: ['entity', 'resourceId'] },
-      sk: { field: 'gsi1sk', composite: ['createdAt'] },
-    },
-  },
-});