@mastra/dynamodb 0.10.1-alpha.1 → 0.10.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/_tsup-dts-rollup.d.cts +1120 -0
- package/dist/_tsup-dts-rollup.d.ts +1120 -0
- package/dist/index.cjs +7 -4
- package/dist/index.d.cts +2 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.js +7 -4
- package/package.json +18 -5
- package/src/storage/index.test.ts +3 -0
- package/src/storage/index.ts +4 -5
package/dist/index.cjs
CHANGED
|
@@ -560,9 +560,12 @@ function getElectroDbService(client, tableName) {
|
|
|
560
560
|
|
|
561
561
|
// src/storage/index.ts
|
|
562
562
|
var DynamoDBStore = class extends storage.MastraStorage {
|
|
563
|
+
tableName;
|
|
564
|
+
client;
|
|
565
|
+
service;
|
|
566
|
+
hasInitialized = null;
|
|
563
567
|
constructor({ name, config }) {
|
|
564
568
|
super({ name });
|
|
565
|
-
this.hasInitialized = null;
|
|
566
569
|
if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
|
|
567
570
|
throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
|
|
568
571
|
}
|
|
@@ -906,7 +909,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
|
|
|
906
909
|
results2.data.map((data) => this.parseMessageData(data)),
|
|
907
910
|
"memory"
|
|
908
911
|
);
|
|
909
|
-
if (format === `v2`) return list2.get.all.
|
|
912
|
+
if (format === `v2`) return list2.get.all.v2();
|
|
910
913
|
return list2.get.all.v1();
|
|
911
914
|
}
|
|
912
915
|
const results = await query.go();
|
|
@@ -914,7 +917,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
|
|
|
914
917
|
results.data.map((data) => this.parseMessageData(data)),
|
|
915
918
|
"memory"
|
|
916
919
|
);
|
|
917
|
-
if (format === `v2`) return list.get.all.
|
|
920
|
+
if (format === `v2`) return list.get.all.v2();
|
|
918
921
|
return list.get.all.v1();
|
|
919
922
|
} catch (error) {
|
|
920
923
|
this.logger.error("Failed to get messages", { threadId, error });
|
|
@@ -965,7 +968,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
|
|
|
965
968
|
}
|
|
966
969
|
const list = new agent.MessageList().add(messages, "memory");
|
|
967
970
|
if (format === `v1`) return list.get.all.v1();
|
|
968
|
-
return list.get.all.
|
|
971
|
+
return list.get.all.v2();
|
|
969
972
|
} catch (error) {
|
|
970
973
|
this.logger.error("Failed to save messages", { error });
|
|
971
974
|
throw error;
|
package/dist/index.d.cts
ADDED
package/dist/index.d.ts
ADDED
package/dist/index.js
CHANGED
|
@@ -558,9 +558,12 @@ function getElectroDbService(client, tableName) {
|
|
|
558
558
|
|
|
559
559
|
// src/storage/index.ts
|
|
560
560
|
var DynamoDBStore = class extends MastraStorage {
|
|
561
|
+
tableName;
|
|
562
|
+
client;
|
|
563
|
+
service;
|
|
564
|
+
hasInitialized = null;
|
|
561
565
|
constructor({ name, config }) {
|
|
562
566
|
super({ name });
|
|
563
|
-
this.hasInitialized = null;
|
|
564
567
|
if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
|
|
565
568
|
throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
|
|
566
569
|
}
|
|
@@ -904,7 +907,7 @@ var DynamoDBStore = class extends MastraStorage {
|
|
|
904
907
|
results2.data.map((data) => this.parseMessageData(data)),
|
|
905
908
|
"memory"
|
|
906
909
|
);
|
|
907
|
-
if (format === `v2`) return list2.get.all.
|
|
910
|
+
if (format === `v2`) return list2.get.all.v2();
|
|
908
911
|
return list2.get.all.v1();
|
|
909
912
|
}
|
|
910
913
|
const results = await query.go();
|
|
@@ -912,7 +915,7 @@ var DynamoDBStore = class extends MastraStorage {
|
|
|
912
915
|
results.data.map((data) => this.parseMessageData(data)),
|
|
913
916
|
"memory"
|
|
914
917
|
);
|
|
915
|
-
if (format === `v2`) return list.get.all.
|
|
918
|
+
if (format === `v2`) return list.get.all.v2();
|
|
916
919
|
return list.get.all.v1();
|
|
917
920
|
} catch (error) {
|
|
918
921
|
this.logger.error("Failed to get messages", { threadId, error });
|
|
@@ -963,7 +966,7 @@ var DynamoDBStore = class extends MastraStorage {
|
|
|
963
966
|
}
|
|
964
967
|
const list = new MessageList().add(messages, "memory");
|
|
965
968
|
if (format === `v1`) return list.get.all.v1();
|
|
966
|
-
return list.get.all.
|
|
969
|
+
return list.get.all.v2();
|
|
967
970
|
} catch (error) {
|
|
968
971
|
this.logger.error("Failed to save messages", { error });
|
|
969
972
|
throw error;
|
package/package.json
CHANGED
|
@@ -1,10 +1,23 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/dynamodb",
|
|
3
|
-
"version": "0.10.1
|
|
3
|
+
"version": "0.10.1",
|
|
4
4
|
"description": "DynamoDB storage adapter for Mastra",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
7
7
|
"types": "dist/index.d.ts",
|
|
8
|
+
"exports": {
|
|
9
|
+
".": {
|
|
10
|
+
"import": {
|
|
11
|
+
"types": "./dist/index.d.ts",
|
|
12
|
+
"default": "./dist/index.js"
|
|
13
|
+
},
|
|
14
|
+
"require": {
|
|
15
|
+
"types": "./dist/index.d.cts",
|
|
16
|
+
"default": "./dist/index.cjs"
|
|
17
|
+
}
|
|
18
|
+
},
|
|
19
|
+
"./package.json": "./package.json"
|
|
20
|
+
},
|
|
8
21
|
"files": [
|
|
9
22
|
"dist",
|
|
10
23
|
"src"
|
|
@@ -15,7 +28,7 @@
|
|
|
15
28
|
"electrodb": "^3.4.1"
|
|
16
29
|
},
|
|
17
30
|
"peerDependencies": {
|
|
18
|
-
"@mastra/core": "^0.10.
|
|
31
|
+
"@mastra/core": "^0.10.2-alpha.0"
|
|
19
32
|
},
|
|
20
33
|
"devDependencies": {
|
|
21
34
|
"@microsoft/api-extractor": "^7.52.1",
|
|
@@ -27,11 +40,11 @@
|
|
|
27
40
|
"tsup": "^8.4.0",
|
|
28
41
|
"typescript": "^5.8.2",
|
|
29
42
|
"vitest": "^3.0.9",
|
|
30
|
-
"@internal/lint": "0.0.
|
|
31
|
-
"@mastra/core": "0.10.2
|
|
43
|
+
"@internal/lint": "0.0.8",
|
|
44
|
+
"@mastra/core": "0.10.2"
|
|
32
45
|
},
|
|
33
46
|
"scripts": {
|
|
34
|
-
"build": "tsup src/index.ts --format esm,cjs --clean --treeshake=smallest --splitting",
|
|
47
|
+
"build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",
|
|
35
48
|
"dev": "tsup --watch",
|
|
36
49
|
"clean": "rm -rf dist",
|
|
37
50
|
"lint": "eslint .",
|
|
@@ -320,6 +320,7 @@ describe('DynamoDBStore Integration Tests', () => {
|
|
|
320
320
|
input: { source: 'test' },
|
|
321
321
|
step1: { status: 'success', output: { data: 'test' } },
|
|
322
322
|
} as unknown as WorkflowRunState['context'],
|
|
323
|
+
serializedStepGraph: [],
|
|
323
324
|
activePaths: [{ stepPath: ['test'], stepId: 'step1', status: 'success' }],
|
|
324
325
|
suspendedPaths: { test: [1] },
|
|
325
326
|
runId: 'test-run-large', // Use unique runId
|
|
@@ -486,6 +487,7 @@ describe('DynamoDBStore Integration Tests', () => {
|
|
|
486
487
|
step1: { status: 'success', output: { data: 'test' } },
|
|
487
488
|
input: { source: 'test' },
|
|
488
489
|
} as unknown as WorkflowRunState['context'],
|
|
490
|
+
serializedStepGraph: [],
|
|
489
491
|
activePaths: [{ stepPath: ['test'], stepId: 'step1', status: 'success' }],
|
|
490
492
|
suspendedPaths: { test: [1] },
|
|
491
493
|
runId: 'mixed-run',
|
|
@@ -734,6 +736,7 @@ describe('DynamoDBStore Integration Tests', () => {
|
|
|
734
736
|
step1: { status: 'success', output: { data: 'test' } },
|
|
735
737
|
input: { source: 'test' },
|
|
736
738
|
} as unknown as WorkflowRunState['context'],
|
|
739
|
+
serializedStepGraph: [],
|
|
737
740
|
activePaths: [],
|
|
738
741
|
suspendedPaths: {},
|
|
739
742
|
runId: runId,
|
package/src/storage/index.ts
CHANGED
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
|
|
2
2
|
import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
|
|
3
|
-
import type { StorageThreadType, WorkflowRunState, MastraMessageV1 } from '@mastra/core';
|
|
4
|
-
import type { MastraMessageV2 } from '@mastra/core/agent';
|
|
3
|
+
import type { StorageThreadType, WorkflowRunState, MastraMessageV1, MastraMessageV2 } from '@mastra/core';
|
|
5
4
|
import { MessageList } from '@mastra/core/agent';
|
|
6
5
|
import {
|
|
7
6
|
MastraStorage,
|
|
@@ -542,7 +541,7 @@ export class DynamoDBStore extends MastraStorage {
|
|
|
542
541
|
results.data.map((data: any) => this.parseMessageData(data)),
|
|
543
542
|
'memory',
|
|
544
543
|
);
|
|
545
|
-
if (format === `v2`) return list.get.all.
|
|
544
|
+
if (format === `v2`) return list.get.all.v2();
|
|
546
545
|
return list.get.all.v1();
|
|
547
546
|
}
|
|
548
547
|
|
|
@@ -553,7 +552,7 @@ export class DynamoDBStore extends MastraStorage {
|
|
|
553
552
|
results.data.map((data: any) => this.parseMessageData(data)),
|
|
554
553
|
'memory',
|
|
555
554
|
);
|
|
556
|
-
if (format === `v2`) return list.get.all.
|
|
555
|
+
if (format === `v2`) return list.get.all.v2();
|
|
557
556
|
return list.get.all.v1();
|
|
558
557
|
} catch (error) {
|
|
559
558
|
this.logger.error('Failed to get messages', { threadId, error });
|
|
@@ -618,7 +617,7 @@ export class DynamoDBStore extends MastraStorage {
|
|
|
618
617
|
|
|
619
618
|
const list = new MessageList().add(messages, 'memory');
|
|
620
619
|
if (format === `v1`) return list.get.all.v1();
|
|
621
|
-
return list.get.all.
|
|
620
|
+
return list.get.all.v2();
|
|
622
621
|
} catch (error) {
|
|
623
622
|
this.logger.error('Failed to save messages', { error });
|
|
624
623
|
throw error;
|