@mastra/dynamodb 0.14.3 → 0.14.6-alpha.0

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'] },
-    },
-  },
-});

package/src/entities/trace.ts

@@ -1,129 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const traceEntity = new Entity({
-  model: {
-    entity: 'trace',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    id: {
-      type: 'string',
-      required: true,
-    },
-    parentSpanId: {
-      type: 'string',
-      required: false,
-    },
-    name: {
-      type: 'string',
-      required: true,
-    },
-    traceId: {
-      type: 'string',
-      required: true,
-    },
-    scope: {
-      type: 'string',
-      required: true,
-    },
-    kind: {
-      type: 'number',
-      required: true,
-    },
-    attributes: {
-      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 ? JSON.parse(value) : value;
-      },
-    },
-    status: {
-      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;
-      },
-    },
-    events: {
-      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;
-      },
-    },
-    links: {
-      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;
-      },
-    },
-    other: {
-      type: 'string',
-      required: false,
-    },
-    startTime: {
-      type: 'number',
-      required: true,
-    },
-    endTime: {
-      type: 'number',
-      required: true,
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'id'] },
-      sk: { field: 'sk', composite: [] },
-    },
-    byName: {
-      index: 'gsi1',
-      pk: { field: 'gsi1pk', composite: ['entity', 'name'] },
-      sk: { field: 'gsi1sk', composite: ['startTime'] },
-    },
-    byScope: {
-      index: 'gsi2',
-      pk: { field: 'gsi2pk', composite: ['entity', 'scope'] },
-      sk: { field: 'gsi2sk', composite: ['startTime'] },
-    },
-  },
-});

package/src/entities/utils.ts

@@ -1,51 +0,0 @@
-export const baseAttributes = {
-  createdAt: {
-    type: 'string',
-    required: true,
-    readOnly: true,
-    // Convert Date to ISO string on set
-    set: (value?: Date | string) => {
-      if (value instanceof Date) {
-        return value.toISOString();
-      }
-      return value || new Date().toISOString();
-    },
-    // Initialize with current timestamp if not provided
-    default: () => new Date().toISOString(),
-  },
-  updatedAt: {
-    type: 'string',
-    required: true,
-    // Convert Date to ISO string on set
-    set: (value?: Date | string) => {
-      if (value instanceof Date) {
-        return value.toISOString();
-      }
-      return value || new Date().toISOString();
-    },
-    // Always use current timestamp when creating/updating
-    default: () => new Date().toISOString(),
-  },
-  metadata: {
-    type: 'string', // JSON stringified
-    // Stringify objects on set
-    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) {
-        try {
-          return JSON.parse(value);
-        } catch {
-          // If parsing fails, return the original string
-          return value;
-        }
-      }
-      return value;
-    },
-  },
-} as const;

package/src/entities/workflow-snapshot.ts

@@ -1,56 +0,0 @@
-import { Entity } from 'electrodb';
-import { baseAttributes } from './utils';
-
-export const workflowSnapshotEntity = new Entity({
-  model: {
-    entity: 'workflow_snapshot',
-    version: '1',
-    service: 'mastra',
-  },
-  attributes: {
-    entity: {
-      type: 'string',
-      required: true,
-    },
-    ...baseAttributes,
-    workflow_name: {
-      type: 'string',
-      required: true,
-    },
-    run_id: {
-      type: 'string',
-      required: true,
-    },
-    snapshot: {
-      type: 'string', // JSON stringified
-      required: true,
-      // Stringify snapshot 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 ? JSON.parse(value) : value;
-      },
-    },
-    resourceId: {
-      type: 'string',
-      required: false,
-    },
-  },
-  indexes: {
-    primary: {
-      pk: { field: 'pk', composite: ['entity', 'workflow_name'] },
-      sk: { field: 'sk', composite: ['run_id'] },
-    },
-    // GSI to allow querying by run_id efficiently without knowing the workflow_name
-    gsi2: {
-      index: 'gsi2',
-      pk: { field: 'gsi2pk', composite: ['entity', 'run_id'] },
-      sk: { field: 'gsi2sk', composite: ['workflow_name'] },
-    },
-  },
-});

package/src/index.ts

@@ -1 +0,0 @@
-export * from './storage';

package/src/storage/docker-compose.yml

@@ -1,16 +0,0 @@
-version: '3.8'
-
-services:
-  dynamodb-local:
-    image: amazon/dynamodb-local:latest
-    # Use host network mode for simpler connection from tests,
-    # or define ports if running in bridge mode.
-    # network_mode: host
-    ports:
-      - '8000:8000' # Map container port 8000 to host port 8000
-    command: ['-jar', 'DynamoDBLocal.jar', '-sharedDb']
-    volumes:
-      - dynamodb_data:/home/dynamodblocal/data
-
-volumes:
-  dynamodb_data: {}