@mastra/dynamodb 0.10.4-alpha.0 → 0.10.4-alpha.1

package/dist/_tsup-dts-rollup.d.cts

@@ -1,17 +1,20 @@
 import type { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { Entity } from 'electrodb';
 import type { EvalRow } from '@mastra/core/storage';
-import type { MastraMessageV1 } from '@mastra/core';
-import type { MastraMessageV2 } from '@mastra/core';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
+import type { PaginationInfo } from '@mastra/core/storage';
 import { Service } from 'electrodb';
-import type { StorageColumn } from '@mastra/core';
+import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
-import type { StorageThreadType } from '@mastra/core';
+import type { StorageGetTracesArg } from '@mastra/core/storage';
+import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_NAMES } from '@mastra/core/storage';
+import type { Trace } from '@mastra/core/telemetry';
 import type { WorkflowRun } from '@mastra/core/storage';
 import type { WorkflowRuns } from '@mastra/core/storage';
-import type { WorkflowRunState } from '@mastra/core';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 
 export declare const baseAttributes: {
     readonly createdAt: {
@@ -175,6 +178,19 @@ declare class DynamoDBStore extends MastraStorage {
     private formatWorkflowRun;
     private getEntityNameForTable;
     getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    getThreadsByResourceIdPaginated(_args: {
+        resourceId: string;
+        page?: number;
+        perPage?: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
+    getMessagesPaginated(_args: StorageGetMessagesArg): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     /**
      * Closes the DynamoDB client connection and cleans up resources.
      * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.

package/dist/_tsup-dts-rollup.d.ts

@@ -1,17 +1,20 @@
 import type { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { Entity } from 'electrodb';
 import type { EvalRow } from '@mastra/core/storage';
-import type { MastraMessageV1 } from '@mastra/core';
-import type { MastraMessageV2 } from '@mastra/core';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
+import type { PaginationInfo } from '@mastra/core/storage';
 import { Service } from 'electrodb';
-import type { StorageColumn } from '@mastra/core';
+import type { StorageColumn } from '@mastra/core/storage';
 import type { StorageGetMessagesArg } from '@mastra/core/storage';
-import type { StorageThreadType } from '@mastra/core';
+import type { StorageGetTracesArg } from '@mastra/core/storage';
+import type { StorageThreadType } from '@mastra/core/memory';
 import type { TABLE_NAMES } from '@mastra/core/storage';
+import type { Trace } from '@mastra/core/telemetry';
 import type { WorkflowRun } from '@mastra/core/storage';
 import type { WorkflowRuns } from '@mastra/core/storage';
-import type { WorkflowRunState } from '@mastra/core';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 
 export declare const baseAttributes: {
     readonly createdAt: {
@@ -175,6 +178,19 @@ declare class DynamoDBStore extends MastraStorage {
     private formatWorkflowRun;
     private getEntityNameForTable;
     getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & {
+        traces: Trace[];
+    }>;
+    getThreadsByResourceIdPaginated(_args: {
+        resourceId: string;
+        page?: number;
+        perPage?: number;
+    }): Promise<PaginationInfo & {
+        threads: StorageThreadType[];
+    }>;
+    getMessagesPaginated(_args: StorageGetMessagesArg): Promise<PaginationInfo & {
+        messages: MastraMessageV1[] | MastraMessageV2[];
+    }>;
     /**
      * Closes the DynamoDB client connection and cleans up resources.
      * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.

package/dist/index.cjs

@@ -1078,7 +1078,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
         updatedAt: now,
         resourceId
       };
-      await this.service.entities.workflowSnapshot.create(data).go();
+      await this.service.entities.workflowSnapshot.upsert(data).go();
     } catch (error) {
       this.logger.error("Failed to persist workflow snapshot", { workflowName, runId, error });
       throw error;
@@ -1293,6 +1293,15 @@ var DynamoDBStore = class extends storage.MastraStorage {
       throw error;
     }
   }
+  async getTracesPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
+  async getThreadsByResourceIdPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
+  async getMessagesPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
   /**
    * Closes the DynamoDB client connection and cleans up resources.
    * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.

package/dist/index.js

@@ -1076,7 +1076,7 @@ var DynamoDBStore = class extends MastraStorage {
         updatedAt: now,
         resourceId
       };
-      await this.service.entities.workflowSnapshot.create(data).go();
+      await this.service.entities.workflowSnapshot.upsert(data).go();
     } catch (error) {
       this.logger.error("Failed to persist workflow snapshot", { workflowName, runId, error });
       throw error;
@@ -1291,6 +1291,15 @@ var DynamoDBStore = class extends MastraStorage {
       throw error;
     }
   }
+  async getTracesPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
+  async getThreadsByResourceIdPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
+  async getMessagesPaginated(_args) {
+    throw new Error("Method not implemented.");
+  }
   /**
    * Closes the DynamoDB client connection and cleans up resources.
    * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dynamodb",
-  "version": "0.10.4-alpha.0",
+  "version": "0.10.4-alpha.1",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -42,7 +42,7 @@
     "vitest": "^3.2.2",
     "@internal/lint": "0.0.10",
     "@internal/storage-test-utils": "0.0.6",
-    "@mastra/core": "0.10.4-alpha.1"
+    "@mastra/core": "0.10.4-alpha.3"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/storage/index.test.ts

@@ -901,6 +901,53 @@ describe('DynamoDBStore Integration Tests', () => {
       expect(loadedSnapshot?.context).toEqual(snapshot.context);
     });
 
+    test('should allow updating an existing workflow snapshot', async () => {
+      const wfName = 'update-test-wf';
+      const runId = `run-${randomUUID()}`;
+
+      // Create initial snapshot
+      const { snapshot: initialSnapshot } = sampleWorkflowSnapshot(wfName, runId);
+
+      await expect(
+        store.persistWorkflowSnapshot({
+          workflowName: wfName,
+          runId: runId,
+          snapshot: initialSnapshot,
+        }),
+      ).resolves.not.toThrow();
+
+      // Create updated snapshot with different data
+      const updatedSnapshot: WorkflowRunState = {
+        ...initialSnapshot,
+        value: { currentState: 'completed' },
+        context: {
+          step1: { status: 'success', output: { data: 'updated-test' } },
+          step2: { status: 'success', output: { data: 'new-step' } },
+          input: { source: 'updated-test' },
+        } as unknown as WorkflowRunState['context'],
+        timestamp: Date.now(),
+      };
+
+      // This should succeed (update existing snapshot)
+      await expect(
+        store.persistWorkflowSnapshot({
+          workflowName: wfName,
+          runId: runId,
+          snapshot: updatedSnapshot,
+        }),
+      ).resolves.not.toThrow();
+
+      // Verify the snapshot was updated
+      const loadedSnapshot = await store.loadWorkflowSnapshot({
+        workflowName: wfName,
+        runId: runId,
+      });
+
+      expect(loadedSnapshot?.runId).toEqual(updatedSnapshot.runId);
+      expect(loadedSnapshot?.value).toEqual(updatedSnapshot.value);
+      expect(loadedSnapshot?.context).toEqual(updatedSnapshot.context);
+    });
+
     test('getWorkflowRunById should retrieve correct run', async () => {
       const wfName = 'get-by-id-wf';
       const runId1 = `run-${randomUUID()}`;

package/src/storage/index.ts

@@ -1,13 +1,8 @@
 import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
-import type {
-  StorageThreadType,
-  WorkflowRunState,
-  MastraMessageV1,
-  MastraMessageV2,
-  StorageColumn,
-} from '@mastra/core';
 import { MessageList } from '@mastra/core/agent';
+import type { StorageThreadType, MastraMessageV2, MastraMessageV1 } from '@mastra/core/memory';
+
 import {
   MastraStorage,
   TABLE_THREADS,
@@ -16,7 +11,18 @@ import {
   TABLE_EVALS,
   TABLE_TRACES,
 } from '@mastra/core/storage';
-import type { EvalRow, StorageGetMessagesArg, WorkflowRun, WorkflowRuns, TABLE_NAMES } from '@mastra/core/storage';
+import type {
+  EvalRow,
+  StorageGetMessagesArg,
+  WorkflowRun,
+  WorkflowRuns,
+  TABLE_NAMES,
+  StorageGetTracesArg,
+  PaginationInfo,
+  StorageColumn,
+} from '@mastra/core/storage';
+import type { Trace } from '@mastra/core/telemetry';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 import type { Service } from 'electrodb';
 import { getElectroDbService } from '../entities';
 
@@ -771,8 +777,8 @@ export class DynamoDBStore extends MastraStorage {
         updatedAt: now,
         resourceId,
       };
-      // Pass the data including 'entity'
-      await this.service.entities.workflowSnapshot.create(data).go();
+      // Use upsert instead of create to handle both create and update cases
+      await this.service.entities.workflowSnapshot.upsert(data).go();
     } catch (error) {
       this.logger.error('Failed to persist workflow snapshot', { workflowName, runId, error });
       throw error;
@@ -1077,6 +1083,24 @@ export class DynamoDBStore extends MastraStorage {
     }
   }
 
+  async getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & { traces: Trace[] }> {
+    throw new Error('Method not implemented.');
+  }
+
+  async getThreadsByResourceIdPaginated(_args: {
+    resourceId: string;
+    page?: number;
+    perPage?: number;
+  }): Promise<PaginationInfo & { threads: StorageThreadType[] }> {
+    throw new Error('Method not implemented.');
+  }
+
+  async getMessagesPaginated(
+    _args: StorageGetMessagesArg,
+  ): Promise<PaginationInfo & { messages: MastraMessageV1[] | MastraMessageV2[] }> {
+    throw new Error('Method not implemented.');
+  }
+
   /**
    * Closes the DynamoDB client connection and cleans up resources.
    * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.