@mastra/dsql 0.0.0-ag-example-20260516005230
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/CHANGELOG.md +60 -0
- package/LICENSE.md +30 -0
- package/README.md +357 -0
- package/dist/index.cjs +4478 -0
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.ts +4 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +4469 -0
- package/dist/index.js.map +1 -0
- package/dist/shared/batch.d.ts +55 -0
- package/dist/shared/batch.d.ts.map +1 -0
- package/dist/shared/config.d.ts +129 -0
- package/dist/shared/config.d.ts.map +1 -0
- package/dist/shared/retry.d.ts +207 -0
- package/dist/shared/retry.d.ts.map +1 -0
- package/dist/storage/client.d.ts +91 -0
- package/dist/storage/client.d.ts.map +1 -0
- package/dist/storage/db/index.d.ts +179 -0
- package/dist/storage/db/index.d.ts.map +1 -0
- package/dist/storage/domains/agents/index.d.ts +47 -0
- package/dist/storage/domains/agents/index.d.ts.map +1 -0
- package/dist/storage/domains/memory/index.d.ts +80 -0
- package/dist/storage/domains/memory/index.d.ts.map +1 -0
- package/dist/storage/domains/observability/index.d.ts +37 -0
- package/dist/storage/domains/observability/index.d.ts.map +1 -0
- package/dist/storage/domains/scores/index.d.ts +53 -0
- package/dist/storage/domains/scores/index.d.ts.map +1 -0
- package/dist/storage/domains/utils.d.ts +14 -0
- package/dist/storage/domains/utils.d.ts.map +1 -0
- package/dist/storage/domains/workflows/index.d.ts +61 -0
- package/dist/storage/domains/workflows/index.d.ts.map +1 -0
- package/dist/storage/index.d.ts +46 -0
- package/dist/storage/index.d.ts.map +1 -0
- package/dist/storage/performance-indexes/performance-test.d.ts +47 -0
- package/dist/storage/performance-indexes/performance-test.d.ts.map +1 -0
- package/dist/storage/test-utils.d.ts +14 -0
- package/dist/storage/test-utils.d.ts.map +1 -0
- package/package.json +69 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/storage/db/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAU/C,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,kBAAkB,EAClB,SAAS,EACT,iBAAiB,EAClB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAG1B,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAI1C,YAAY,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAE1C;;;;;GAKG;AACH,MAAM,MAAM,gBAAgB,GAAG,sBAAsB,GAAG,oBAAoB,GAAG,oBAAoB,CAAC;AAEpG;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,mCAAmC;IACnC,MAAM,EAAE,QAAQ,CAAC;IACjB,2BAA2B;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,uCAAuC;IACvC,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,qEAAqE;IACrE,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,6BAA6B;IAC7B,IAAI,EAAE,IAAI,CAAC;IACX,kDAAkD;IAClD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,wDAAwD;IACxD,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,kDAAkD;IAClD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,wDAAwD;IACxD,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,GAAG;IACF,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AACF;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,gBAAgB,GAAG;IAC3D,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,CAiCA;AAaD;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAOD,qBAAa,MAAO,SAAQ,UAAU;IAC7B,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;gBAExB,MAAM,EAAE,oBAAoB;IAWlC,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAWhE;;;OAGG;IACH,OAAO,CAAC,sBAAsB;IAc9B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAY5B;;;;OAIG;IACH,OAAO,CAAC,YAAY;YA2BN,WAAW;IA6DzB;;;;OAIG;IACH,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAczD,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAaxD,MAAM,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmCrG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA8BpE,WAAW,CAAC,EAChB,SAAS,EACT,MAAM,GACP,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;KACvC,GAAG,OAAO,CAAC,IAAI,CAAC;IAqGjB;;;;;OAKG;YACW,sBAAsB;IAQpC;;;OAGG;YACW,iBAAiB;IAqD/B;;;;;OAKG;IACG,UAAU,CAAC,EACf,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuCX,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAuCzG,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqD9G,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBzE;;;OAGG;YACW,cAAc;IAwBtB,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAoFvD,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwC3C,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAiFrD,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAuE5D,MAAM,CAAC,EACX,SAAS,EACT,IAAI,EACJ,IAAI,GACL,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC3B,GAAG,OAAO,CAAC,IAAI,CAAC;IA4DX,WAAW,CAAC,EAChB,SAAS,EACT,OAAO,GACR,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,CAAC;YACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAC3B,CAAC,CAAC;KACJ,GAAG,OAAO,CAAC,IAAI,CAAC;IAyEX,WAAW,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA0D9G;;OAEG;IACG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAG3E"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { AgentsStorage } from '@mastra/core/storage';
|
|
2
|
+
import type { StorageAgentType, StorageCreateAgentInput, StorageUpdateAgentInput, StorageListAgentsInput, StorageListAgentsOutput, CreateIndexOptions } from '@mastra/core/storage';
|
|
3
|
+
import type { AgentVersion, CreateVersionInput, ListVersionsInput, ListVersionsOutput } from '@mastra/core/storage/domains/agents';
|
|
4
|
+
import type { DsqlDomainConfig } from '../../db/index.js';
|
|
5
|
+
export declare class AgentsDSQL extends AgentsStorage {
|
|
6
|
+
#private;
|
|
7
|
+
/** Tables managed by this domain */
|
|
8
|
+
static readonly MANAGED_TABLES: readonly ["mastra_agents", "mastra_agent_versions"];
|
|
9
|
+
constructor(config: DsqlDomainConfig);
|
|
10
|
+
/**
|
|
11
|
+
* Returns default index definitions for the agents domain tables.
|
|
12
|
+
* Currently no default indexes are defined for agents.
|
|
13
|
+
*/
|
|
14
|
+
getDefaultIndexDefinitions(): CreateIndexOptions[];
|
|
15
|
+
/**
|
|
16
|
+
* Creates default indexes for optimal query performance.
|
|
17
|
+
* Currently no default indexes are defined for agents.
|
|
18
|
+
*/
|
|
19
|
+
createDefaultIndexes(): Promise<void>;
|
|
20
|
+
init(): Promise<void>;
|
|
21
|
+
/**
|
|
22
|
+
* Creates custom user-defined indexes for this domain's tables.
|
|
23
|
+
*/
|
|
24
|
+
createCustomIndexes(): Promise<void>;
|
|
25
|
+
dangerouslyClearAll(): Promise<void>;
|
|
26
|
+
private parseJson;
|
|
27
|
+
private parseRow;
|
|
28
|
+
getById(id: string): Promise<StorageAgentType | null>;
|
|
29
|
+
create(input: {
|
|
30
|
+
agent: StorageCreateAgentInput;
|
|
31
|
+
}): Promise<StorageAgentType>;
|
|
32
|
+
update(input: StorageUpdateAgentInput): Promise<StorageAgentType>;
|
|
33
|
+
delete(id: string): Promise<void>;
|
|
34
|
+
list(args?: StorageListAgentsInput): Promise<StorageListAgentsOutput>;
|
|
35
|
+
createVersion(input: CreateVersionInput): Promise<AgentVersion>;
|
|
36
|
+
getVersion(id: string): Promise<AgentVersion | null>;
|
|
37
|
+
getVersionByNumber(agentId: string, versionNumber: number): Promise<AgentVersion | null>;
|
|
38
|
+
getLatestVersion(agentId: string): Promise<AgentVersion | null>;
|
|
39
|
+
listVersions(input: ListVersionsInput): Promise<ListVersionsOutput>;
|
|
40
|
+
deleteVersion(id: string): Promise<void>;
|
|
41
|
+
deleteVersionsByParentId(agentId: string): Promise<void>;
|
|
42
|
+
countVersions(agentId: string): Promise<number>;
|
|
43
|
+
private serializeInstructions;
|
|
44
|
+
private deserializeInstructions;
|
|
45
|
+
private parseVersionRow;
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/agents/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EAOd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,gBAAgB,EAChB,uBAAuB,EACvB,uBAAuB,EACvB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAEnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EACnB,MAAM,qCAAqC,CAAC;AAG7C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAGjD,qBAAa,UAAW,SAAQ,aAAa;;IAM3C,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,sDAAiD;gBAEnE,MAAM,EAAE,gBAAgB;IAUpC;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAIlD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAO3B;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAK1C,OAAO,CAAC,SAAS;IA2BjB,OAAO,CAAC,QAAQ;IAYV,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAyBrD,MAAM,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,uBAAuB,CAAA;KAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA6E5E,MAAM,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAqFjE,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBjC,IAAI,CAAC,IAAI,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAgGrE,aAAa,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,YAAY,CAAC;IAiE/D,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAwBpD,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IA2BxF,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IA2B/D,YAAY,CAAC,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAyDnE,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBxC,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBxD,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAyBrD,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,uBAAuB;IAW/B,OAAO,CAAC,eAAe;CA4BxB"}
|
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
2
|
+
import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
|
|
3
|
+
import { MemoryStorage } from '@mastra/core/storage';
|
|
4
|
+
import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsInput, StorageListThreadsOutput, CreateIndexOptions } from '@mastra/core/storage';
|
|
5
|
+
import type { DsqlDomainConfig } from '../../db/index.js';
|
|
6
|
+
export declare class MemoryDSQL extends MemoryStorage {
|
|
7
|
+
#private;
|
|
8
|
+
/** Tables managed by this domain */
|
|
9
|
+
static readonly MANAGED_TABLES: readonly ["mastra_threads", "mastra_messages", "mastra_resources"];
|
|
10
|
+
constructor(config: DsqlDomainConfig);
|
|
11
|
+
init(): Promise<void>;
|
|
12
|
+
/**
|
|
13
|
+
* Returns default index definitions for the memory domain tables.
|
|
14
|
+
* Note: Aurora DSQL does not support ASC/DESC in index columns.
|
|
15
|
+
*/
|
|
16
|
+
getDefaultIndexDefinitions(): CreateIndexOptions[];
|
|
17
|
+
/**
|
|
18
|
+
* Creates default indexes for optimal query performance.
|
|
19
|
+
*/
|
|
20
|
+
createDefaultIndexes(): Promise<void>;
|
|
21
|
+
/**
|
|
22
|
+
* Creates custom user-defined indexes for this domain's tables.
|
|
23
|
+
*/
|
|
24
|
+
createCustomIndexes(): Promise<void>;
|
|
25
|
+
dangerouslyClearAll(): Promise<void>;
|
|
26
|
+
/**
|
|
27
|
+
* Normalizes message row from database by applying createdAtZ fallback
|
|
28
|
+
*/
|
|
29
|
+
private normalizeMessageRow;
|
|
30
|
+
getThreadById({ threadId }: {
|
|
31
|
+
threadId: string;
|
|
32
|
+
}): Promise<StorageThreadType | null>;
|
|
33
|
+
listThreads(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
|
|
34
|
+
saveThread({ thread }: {
|
|
35
|
+
thread: StorageThreadType;
|
|
36
|
+
}): Promise<StorageThreadType>;
|
|
37
|
+
updateThread({ id, title, metadata, }: {
|
|
38
|
+
id: string;
|
|
39
|
+
title: string;
|
|
40
|
+
metadata: Record<string, unknown>;
|
|
41
|
+
}): Promise<StorageThreadType>;
|
|
42
|
+
deleteThread({ threadId }: {
|
|
43
|
+
threadId: string;
|
|
44
|
+
}): Promise<void>;
|
|
45
|
+
private _getIncludedMessages;
|
|
46
|
+
private parseRow;
|
|
47
|
+
listMessagesById({ messageIds }: {
|
|
48
|
+
messageIds: string[];
|
|
49
|
+
}): Promise<{
|
|
50
|
+
messages: MastraDBMessage[];
|
|
51
|
+
}>;
|
|
52
|
+
listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
|
|
53
|
+
saveMessages({ messages }: {
|
|
54
|
+
messages: MastraDBMessage[];
|
|
55
|
+
}): Promise<{
|
|
56
|
+
messages: MastraDBMessage[];
|
|
57
|
+
}>;
|
|
58
|
+
updateMessages({ messages, }: {
|
|
59
|
+
messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
|
|
60
|
+
id: string;
|
|
61
|
+
content?: {
|
|
62
|
+
metadata?: MastraMessageContentV2['metadata'];
|
|
63
|
+
content?: MastraMessageContentV2['content'];
|
|
64
|
+
};
|
|
65
|
+
})[];
|
|
66
|
+
}): Promise<MastraDBMessage[]>;
|
|
67
|
+
deleteMessages(messageIds: string[]): Promise<void>;
|
|
68
|
+
getResourceById({ resourceId }: {
|
|
69
|
+
resourceId: string;
|
|
70
|
+
}): Promise<StorageResourceType | null>;
|
|
71
|
+
saveResource({ resource }: {
|
|
72
|
+
resource: StorageResourceType;
|
|
73
|
+
}): Promise<StorageResourceType>;
|
|
74
|
+
updateResource({ resourceId, workingMemory, metadata, }: {
|
|
75
|
+
resourceId: string;
|
|
76
|
+
workingMemory?: string;
|
|
77
|
+
metadata?: Record<string, unknown>;
|
|
78
|
+
}): Promise<StorageResourceType>;
|
|
79
|
+
}
|
|
80
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EACL,aAAa,EAQd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAwBjD,qBAAa,UAAW,SAAQ,aAAa;;IAM3C,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,qEAA6D;gBAE/E,MAAM,EAAE,gBAAgB;IAU9B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAgBlD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAM1C;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAYrB,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAoC7E,WAAW,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAuHpF,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA0DjF,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA4ExB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;YAkCvD,oBAAoB;IA4DlC,OAAO,CAAC,QAAQ;IAmBH,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAoCpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IA+JvF,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAsHrG,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBACR,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC;aAC7C,CAAC;SACH,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA2IxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmDnD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAgC5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAY3F,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC;CAkFjC"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { ObservabilityStorage } from '@mastra/core/storage';
|
|
2
|
+
import type { TracingStorageStrategy, ListTracesArgs, ListTracesResponse, UpdateSpanArgs, BatchDeleteTracesArgs, BatchUpdateSpansArgs, BatchCreateSpansArgs, CreateSpanArgs, GetSpanArgs, GetSpanResponse, GetRootSpanArgs, GetRootSpanResponse, GetTraceArgs, GetTraceResponse, CreateIndexOptions } from '@mastra/core/storage';
|
|
3
|
+
import type { DsqlDomainConfig } from '../../db/index.js';
|
|
4
|
+
export declare class ObservabilityDSQL extends ObservabilityStorage {
|
|
5
|
+
#private;
|
|
6
|
+
/** Tables managed by this domain */
|
|
7
|
+
static readonly MANAGED_TABLES: readonly ["mastra_ai_spans"];
|
|
8
|
+
constructor(config: DsqlDomainConfig);
|
|
9
|
+
init(): Promise<void>;
|
|
10
|
+
/**
|
|
11
|
+
* Returns default index definitions for the observability domain tables.
|
|
12
|
+
*/
|
|
13
|
+
getDefaultIndexDefinitions(): CreateIndexOptions[];
|
|
14
|
+
/**
|
|
15
|
+
* Creates default indexes for optimal query performance.
|
|
16
|
+
*/
|
|
17
|
+
createDefaultIndexes(): Promise<void>;
|
|
18
|
+
/**
|
|
19
|
+
* Creates custom user-defined indexes for this domain's tables.
|
|
20
|
+
*/
|
|
21
|
+
createCustomIndexes(): Promise<void>;
|
|
22
|
+
dangerouslyClearAll(): Promise<void>;
|
|
23
|
+
get tracingStrategy(): {
|
|
24
|
+
preferred: TracingStorageStrategy;
|
|
25
|
+
supported: TracingStorageStrategy[];
|
|
26
|
+
};
|
|
27
|
+
createSpan(args: CreateSpanArgs): Promise<void>;
|
|
28
|
+
getSpan(args: GetSpanArgs): Promise<GetSpanResponse | null>;
|
|
29
|
+
getRootSpan(args: GetRootSpanArgs): Promise<GetRootSpanResponse | null>;
|
|
30
|
+
getTrace(args: GetTraceArgs): Promise<GetTraceResponse | null>;
|
|
31
|
+
updateSpan(args: UpdateSpanArgs): Promise<void>;
|
|
32
|
+
listTraces(args: ListTracesArgs): Promise<ListTracesResponse>;
|
|
33
|
+
batchCreateSpans(args: BatchCreateSpansArgs): Promise<void>;
|
|
34
|
+
batchUpdateSpans(args: BatchUpdateSpansArgs): Promise<void>;
|
|
35
|
+
batchDeleteTraces(args: BatchDeleteTracesArgs): Promise<void>;
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/observability/index.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,oBAAoB,EAKrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAEV,sBAAsB,EACtB,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,WAAW,EACX,eAAe,EACf,eAAe,EACf,mBAAmB,EACnB,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAI9B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAGjD,qBAAa,iBAAkB,SAAQ,oBAAoB;;IAMzD,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,+BAA0B;gBAE5C,MAAM,EAAE,gBAAgB;IAU9B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IA2ClD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C,IAAoB,eAAe,IAAI;QACrC,SAAS,EAAE,sBAAsB,CAAC;QAClC,SAAS,EAAE,sBAAsB,EAAE,CAAC;KACrC,CAKA;IAEK,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAqC/C,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IA+C3D,WAAW,CAAC,IAAI,EAAE,eAAe,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IA+CvE,QAAQ,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAqD9D,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAsC/C,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAiO7D,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAmC3D,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqC3D,iBAAiB,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CAkCpE"}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import type { ListScoresResponse, SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
2
|
+
import type { StoragePagination, CreateIndexOptions } from '@mastra/core/storage';
|
|
3
|
+
import { ScoresStorage } from '@mastra/core/storage';
|
|
4
|
+
import type { DsqlDomainConfig } from '../../db/index.js';
|
|
5
|
+
export declare class ScoresDSQL extends ScoresStorage {
|
|
6
|
+
#private;
|
|
7
|
+
/** Tables managed by this domain */
|
|
8
|
+
static readonly MANAGED_TABLES: readonly ["mastra_scorers"];
|
|
9
|
+
constructor(config: DsqlDomainConfig);
|
|
10
|
+
init(): Promise<void>;
|
|
11
|
+
/**
|
|
12
|
+
* Returns default index definitions for the scores domain tables.
|
|
13
|
+
* Note: Aurora DSQL does not support ASC/DESC in index columns.
|
|
14
|
+
*/
|
|
15
|
+
getDefaultIndexDefinitions(): CreateIndexOptions[];
|
|
16
|
+
/**
|
|
17
|
+
* Creates default indexes for optimal query performance.
|
|
18
|
+
*/
|
|
19
|
+
createDefaultIndexes(): Promise<void>;
|
|
20
|
+
/**
|
|
21
|
+
* Creates custom user-defined indexes for this domain's tables.
|
|
22
|
+
*/
|
|
23
|
+
createCustomIndexes(): Promise<void>;
|
|
24
|
+
dangerouslyClearAll(): Promise<void>;
|
|
25
|
+
getScoreById({ id }: {
|
|
26
|
+
id: string;
|
|
27
|
+
}): Promise<ScoreRowData | null>;
|
|
28
|
+
listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
|
|
29
|
+
scorerId: string;
|
|
30
|
+
pagination: StoragePagination;
|
|
31
|
+
entityId?: string;
|
|
32
|
+
entityType?: string;
|
|
33
|
+
source?: ScoringSource;
|
|
34
|
+
}): Promise<ListScoresResponse>;
|
|
35
|
+
saveScore(score: SaveScorePayload): Promise<{
|
|
36
|
+
score: ScoreRowData;
|
|
37
|
+
}>;
|
|
38
|
+
listScoresByRunId({ runId, pagination, }: {
|
|
39
|
+
runId: string;
|
|
40
|
+
pagination: StoragePagination;
|
|
41
|
+
}): Promise<ListScoresResponse>;
|
|
42
|
+
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
43
|
+
pagination: StoragePagination;
|
|
44
|
+
entityId: string;
|
|
45
|
+
entityType: string;
|
|
46
|
+
}): Promise<ListScoresResponse>;
|
|
47
|
+
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
48
|
+
traceId: string;
|
|
49
|
+
spanId: string;
|
|
50
|
+
pagination: StoragePagination;
|
|
51
|
+
}): Promise<ListScoresResponse>;
|
|
52
|
+
}
|
|
53
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAE5G,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAClF,OAAO,EAIL,aAAa,EAId,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAgBjD,qBAAa,UAAW,SAAQ,aAAa;;IAM3C,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,8BAA4B;gBAE9C,MAAM,EAAE,gBAAgB;IAU9B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAWlD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAoBlE,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,UAAU,EACV,MAAM,GACP,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsEzB,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAuEpE,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkDzB,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,UAAU,GACX,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkDzB,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,GACX,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA0ChC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { TABLE_NAMES } from '@mastra/core/storage';
|
|
2
|
+
export declare function getSchemaName(schema?: string): string | undefined;
|
|
3
|
+
export declare function getTableName({ indexName, schemaName }: {
|
|
4
|
+
indexName: string;
|
|
5
|
+
schemaName?: string;
|
|
6
|
+
}): string;
|
|
7
|
+
/**
|
|
8
|
+
* Transform SQL row to record format, handling JSON columns
|
|
9
|
+
*/
|
|
10
|
+
export declare function transformFromSqlRow<T>({ tableName, sqlRow, }: {
|
|
11
|
+
tableName: TABLE_NAMES;
|
|
12
|
+
sqlRow: Record<string, any>;
|
|
13
|
+
}): T;
|
|
14
|
+
//# sourceMappingURL=utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAIxD,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,sBAE5C;AAED,wBAAgB,YAAY,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,UAKjG;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,EACrC,SAAS,EACT,MAAM,GACP,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC7B,GAAG,CAAC,CA8BJ"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { WorkflowsStorage } from '@mastra/core/storage';
|
|
2
|
+
import type { StorageListWorkflowRunsInput, UpdateWorkflowStateOptions, WorkflowRun, WorkflowRuns, CreateIndexOptions } from '@mastra/core/storage';
|
|
3
|
+
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
4
|
+
import type { DsqlDomainConfig } from '../../db/index.js';
|
|
5
|
+
export declare class WorkflowsDSQL extends WorkflowsStorage {
|
|
6
|
+
#private;
|
|
7
|
+
/** Tables managed by this domain */
|
|
8
|
+
static readonly MANAGED_TABLES: readonly ["mastra_workflow_snapshot"];
|
|
9
|
+
constructor(config: DsqlDomainConfig);
|
|
10
|
+
supportsConcurrentUpdates(): boolean;
|
|
11
|
+
/**
|
|
12
|
+
* Returns default index definitions for the workflows domain tables.
|
|
13
|
+
* Currently no default indexes are defined for workflows.
|
|
14
|
+
*/
|
|
15
|
+
getDefaultIndexDefinitions(): CreateIndexOptions[];
|
|
16
|
+
/**
|
|
17
|
+
* Creates default indexes for optimal query performance.
|
|
18
|
+
* Currently no default indexes are defined for workflows.
|
|
19
|
+
*/
|
|
20
|
+
createDefaultIndexes(): Promise<void>;
|
|
21
|
+
init(): Promise<void>;
|
|
22
|
+
/**
|
|
23
|
+
* Creates custom user-defined indexes for this domain's tables.
|
|
24
|
+
*/
|
|
25
|
+
createCustomIndexes(): Promise<void>;
|
|
26
|
+
dangerouslyClearAll(): Promise<void>;
|
|
27
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
|
|
28
|
+
workflowName: string;
|
|
29
|
+
runId: string;
|
|
30
|
+
stepId: string;
|
|
31
|
+
result: StepResult<any, any, any, any>;
|
|
32
|
+
requestContext: Record<string, any>;
|
|
33
|
+
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
34
|
+
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
35
|
+
workflowName: string;
|
|
36
|
+
runId: string;
|
|
37
|
+
opts: UpdateWorkflowStateOptions;
|
|
38
|
+
}): Promise<WorkflowRunState | undefined>;
|
|
39
|
+
persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, }: {
|
|
40
|
+
workflowName: string;
|
|
41
|
+
runId: string;
|
|
42
|
+
resourceId?: string;
|
|
43
|
+
snapshot: WorkflowRunState;
|
|
44
|
+
createdAt?: Date;
|
|
45
|
+
updatedAt?: Date;
|
|
46
|
+
}): Promise<void>;
|
|
47
|
+
loadWorkflowSnapshot({ workflowName, runId, }: {
|
|
48
|
+
workflowName: string;
|
|
49
|
+
runId: string;
|
|
50
|
+
}): Promise<WorkflowRunState | null>;
|
|
51
|
+
getWorkflowRunById({ runId, workflowName, }: {
|
|
52
|
+
runId: string;
|
|
53
|
+
workflowName?: string;
|
|
54
|
+
}): Promise<WorkflowRun | null>;
|
|
55
|
+
deleteWorkflowRunById({ runId, workflowName }: {
|
|
56
|
+
runId: string;
|
|
57
|
+
workflowName: string;
|
|
58
|
+
}): Promise<void>;
|
|
59
|
+
listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, status, }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AACA,OAAO,EAIL,gBAAgB,EAEjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,4BAA4B,EAC5B,0BAA0B,EAC1B,WAAW,EACX,YAAY,EACZ,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAG3E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAsBjD,qBAAa,aAAc,SAAQ,gBAAgB;;IAMjD,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,wCAAsC;gBAExD,MAAM,EAAE,gBAAgB;IAUpC,yBAAyB,IAAI,OAAO;IAIpC;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAIlD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAW3B;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC,qBAAqB,CAAC,EAC1B,YAAY,EACZ,KAAK,EACL,MAAM,EACN,MAAM,EACN,cAAc,GACf,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACvC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACrC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IA8ErD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,0BAA0B,CAAC;KAClC,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IA8DnC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,EACR,SAAS,EACT,SAAS,GACV,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;QAC3B,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,SAAS,CAAC,EAAE,IAAI,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+BX,oBAAoB,CAAC,EACzB,YAAY,EACZ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAoB9B,kBAAkB,CAAC,EACvB,KAAK,EACL,YAAY,GACb,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAmDzB,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBtG,gBAAgB,CAAC,EACrB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,OAAO,EACP,IAAI,EACJ,UAAU,EACV,MAAM,GACP,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;CAqF7D"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import { MastraStorage } from '@mastra/core/storage';
|
|
2
|
+
import type { StorageDomains } from '@mastra/core/storage';
|
|
3
|
+
import { Pool } from 'pg';
|
|
4
|
+
import type { DSQLStoreConfig } from '../shared/config.js';
|
|
5
|
+
import type { DbClient } from './client.js';
|
|
6
|
+
import { AgentsDSQL } from './domains/agents/index.js';
|
|
7
|
+
import { MemoryDSQL } from './domains/memory/index.js';
|
|
8
|
+
import { ObservabilityDSQL } from './domains/observability/index.js';
|
|
9
|
+
import { ScoresDSQL } from './domains/scores/index.js';
|
|
10
|
+
import { WorkflowsDSQL } from './domains/workflows/index.js';
|
|
11
|
+
export { AgentsDSQL, MemoryDSQL, ObservabilityDSQL, ScoresDSQL, WorkflowsDSQL };
|
|
12
|
+
export { PoolAdapter } from './client.js';
|
|
13
|
+
export type { DbClient, TxClient, QueryValues, Pool, PoolClient, QueryResult } from './client.js';
|
|
14
|
+
export type { DsqlDomainConfig, DsqlDomainClientConfig, DsqlDomainPoolConfig, DsqlDomainRestConfig } from './db/index.js';
|
|
15
|
+
export declare class DSQLStore extends MastraStorage {
|
|
16
|
+
#private;
|
|
17
|
+
private schema;
|
|
18
|
+
private isInitialized;
|
|
19
|
+
stores: StorageDomains;
|
|
20
|
+
constructor(config: DSQLStoreConfig);
|
|
21
|
+
/**
|
|
22
|
+
* Creates a connection pool with AuroraDSQLClient for IAM authentication.
|
|
23
|
+
*/
|
|
24
|
+
private createPool;
|
|
25
|
+
init(): Promise<void>;
|
|
26
|
+
/**
|
|
27
|
+
* Database client for executing queries.
|
|
28
|
+
*
|
|
29
|
+
* @example
|
|
30
|
+
* ```typescript
|
|
31
|
+
* const rows = await store.db.any('SELECT * FROM users WHERE active = $1', [true]);
|
|
32
|
+
* const user = await store.db.one('SELECT * FROM users WHERE id = $1', [userId]);
|
|
33
|
+
* ```
|
|
34
|
+
*/
|
|
35
|
+
get db(): DbClient;
|
|
36
|
+
/**
|
|
37
|
+
* The underlying pg.Pool for direct database access or ORM integration.
|
|
38
|
+
*/
|
|
39
|
+
get pool(): Pool;
|
|
40
|
+
/**
|
|
41
|
+
* Closes the connection pool if it was created by this store.
|
|
42
|
+
* If a pool was passed in via config, it will not be closed.
|
|
43
|
+
*/
|
|
44
|
+
close(): Promise<void>;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAwB,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAE1B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAExD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEzC,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,iBAAiB,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAC/F,YAAY,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,MAAM,CAAC;AAEjH,qBAAa,SAAU,SAAQ,aAAa;;IAI1C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,aAAa,CAAkB;IAEvC,MAAM,EAAE,cAAc,CAAC;gBAEX,MAAM,EAAE,eAAe;IA2CnC;;OAEG;IACH,OAAO,CAAC,UAAU;IA2BZ,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAqB3B;;;;;;;;OAQG;IACH,IAAW,EAAE,IAAI,QAAQ,CAExB;IAED;;OAEG;IACH,IAAW,IAAI,IAAI,IAAI,CAEtB;IAED;;;OAGG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAM7B"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Performance testing script to demonstrate the impact of database indexes
|
|
3
|
+
*
|
|
4
|
+
* This script can be used to measure query performance before and after
|
|
5
|
+
* index creation to validate the performance improvements.
|
|
6
|
+
*/
|
|
7
|
+
interface PerformanceTestConfig {
|
|
8
|
+
testDataSize: number;
|
|
9
|
+
iterations: number;
|
|
10
|
+
}
|
|
11
|
+
interface PerformanceResult {
|
|
12
|
+
operation: string;
|
|
13
|
+
avgTimeMs: number;
|
|
14
|
+
minTimeMs: number;
|
|
15
|
+
maxTimeMs: number;
|
|
16
|
+
iterations: number;
|
|
17
|
+
scenario: 'without_indexes' | 'with_indexes';
|
|
18
|
+
}
|
|
19
|
+
interface PerformanceComparison {
|
|
20
|
+
operation: string;
|
|
21
|
+
withoutIndexes: PerformanceResult;
|
|
22
|
+
withIndexes: PerformanceResult;
|
|
23
|
+
improvementFactor: number;
|
|
24
|
+
improvementPercentage: number;
|
|
25
|
+
}
|
|
26
|
+
export declare class DSQLPerformanceTest {
|
|
27
|
+
private store;
|
|
28
|
+
private memory;
|
|
29
|
+
private dbOps;
|
|
30
|
+
private config;
|
|
31
|
+
constructor(config: PerformanceTestConfig);
|
|
32
|
+
init(): Promise<void>;
|
|
33
|
+
cleanup(): Promise<void>;
|
|
34
|
+
resetDatabase(): Promise<void>;
|
|
35
|
+
dropPerformanceIndexes(): Promise<void>;
|
|
36
|
+
createDefaultIndexes(): Promise<void>;
|
|
37
|
+
seedTestData(): Promise<void>;
|
|
38
|
+
measureOperation(name: string, operation: () => Promise<any>, scenario: 'without_indexes' | 'with_indexes'): Promise<PerformanceResult>;
|
|
39
|
+
runPerformanceTests(scenario: 'without_indexes' | 'with_indexes'): Promise<PerformanceResult[]>;
|
|
40
|
+
runComparisonTest(): Promise<PerformanceComparison[]>;
|
|
41
|
+
analyzeCurrentQueries(): Promise<void>;
|
|
42
|
+
printComparison(comparisons: PerformanceComparison[]): void;
|
|
43
|
+
printResults(results: PerformanceResult[]): void;
|
|
44
|
+
checkIndexes(): Promise<void>;
|
|
45
|
+
}
|
|
46
|
+
export {};
|
|
47
|
+
//# sourceMappingURL=performance-test.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"performance-test.d.ts","sourceRoot":"","sources":["../../../src/storage/performance-indexes/performance-test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH,UAAU,qBAAqB;IAC7B,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,UAAU,iBAAiB;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,iBAAiB,GAAG,cAAc,CAAC;CAC9C;AAED,UAAU,qBAAqB;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,EAAE,iBAAiB,CAAC;IAClC,WAAW,EAAE,iBAAiB,CAAC;IAC/B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,KAAK,CAAY;IACzB,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,MAAM,CAAwB;gBAE1B,MAAM,EAAE,qBAAqB;IAOnC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAKrB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAmDxB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAiB9B,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IA2BvC,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAoM7B,gBAAgB,CACpB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,OAAO,CAAC,GAAG,CAAC,EAC7B,QAAQ,EAAE,iBAAiB,GAAG,cAAc,GAC3C,OAAO,CAAC,iBAAiB,CAAC;IA6BvB,mBAAmB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,cAAc,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA+B/F,iBAAiB,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAoCrD,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IA+B5C,eAAe,CAAC,WAAW,EAAE,qBAAqB,EAAE,GAAG,IAAI;IAwB3D,YAAY,CAAC,OAAO,EAAE,iBAAiB,EAAE,GAAG,IAAI;IAiB1C,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CAkBpC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { Pool } from 'pg';
|
|
2
|
+
import type { HostConfig } from '../shared/config.js';
|
|
3
|
+
export declare const TEST_CONFIG: HostConfig;
|
|
4
|
+
/**
|
|
5
|
+
* Check if DSQL tests can run (requires real Aurora DSQL cluster)
|
|
6
|
+
*/
|
|
7
|
+
export declare function canRunDSQLTests(): boolean;
|
|
8
|
+
/**
|
|
9
|
+
* Creates a test pool with IAM authentication for DSQL.
|
|
10
|
+
* Returns a pre-configured pg.Pool using AuroraDSQLClient.
|
|
11
|
+
*/
|
|
12
|
+
export declare function createTestPool(): Pool;
|
|
13
|
+
export declare function dsqlTests(): void;
|
|
14
|
+
//# sourceMappingURL=test-utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"test-utils.d.ts","sourceRoot":"","sources":["../../src/storage/test-utils.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAG1B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAInD,eAAO,MAAM,WAAW,EAAE,UAMzB,CAAC;AAEF;;GAEG;AACH,wBAAgB,eAAe,IAAI,OAAO,CAEzC;AAED;;;GAGG;AACH,wBAAgB,cAAc,IAAI,IAAI,CAgBrC;AAED,wBAAgB,SAAS,SAqZxB"}
|
package/package.json
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@mastra/dsql",
|
|
3
|
+
"version": "0.0.0-ag-example-20260516005230",
|
|
4
|
+
"description": "Amazon Aurora DSQL storage provider for Mastra",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"main": "dist/index.js",
|
|
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.ts",
|
|
16
|
+
"default": "./dist/index.cjs"
|
|
17
|
+
}
|
|
18
|
+
},
|
|
19
|
+
"./package.json": "./package.json"
|
|
20
|
+
},
|
|
21
|
+
"license": "Apache-2.0",
|
|
22
|
+
"dependencies": {
|
|
23
|
+
"@aws-sdk/types": "^3.731.0",
|
|
24
|
+
"@aws/aurora-dsql-node-postgres-connector": "^0.1.1",
|
|
25
|
+
"pg": "^8.14.0"
|
|
26
|
+
},
|
|
27
|
+
"devDependencies": {
|
|
28
|
+
"@microsoft/api-extractor": "^7.52.8",
|
|
29
|
+
"@types/node": "22.13.17",
|
|
30
|
+
"@types/pg": "^8.11.14",
|
|
31
|
+
"@vitest/coverage-v8": "4.1.5",
|
|
32
|
+
"@vitest/ui": "4.1.5",
|
|
33
|
+
"eslint": "^9.37.0",
|
|
34
|
+
"tsup": "^8.5.0",
|
|
35
|
+
"typescript": "^5.8.3",
|
|
36
|
+
"vitest": "4.1.5",
|
|
37
|
+
"@internal/lint": "0.0.0-ag-example-20260516005230",
|
|
38
|
+
"@internal/types-builder": "0.0.0-ag-example-20260516005230",
|
|
39
|
+
"@mastra/core": "0.0.0-ag-example-20260516005230",
|
|
40
|
+
"@internal/storage-test-utils": "0.0.91"
|
|
41
|
+
},
|
|
42
|
+
"peerDependencies": {
|
|
43
|
+
"@mastra/core": "0.0.0-ag-example-20260516005230"
|
|
44
|
+
},
|
|
45
|
+
"files": [
|
|
46
|
+
"dist",
|
|
47
|
+
"CHANGELOG.md"
|
|
48
|
+
],
|
|
49
|
+
"homepage": "https://mastra.ai",
|
|
50
|
+
"repository": {
|
|
51
|
+
"type": "git",
|
|
52
|
+
"url": "git+https://github.com/mastra-ai/mastra.git",
|
|
53
|
+
"directory": "stores/dsql"
|
|
54
|
+
},
|
|
55
|
+
"bugs": {
|
|
56
|
+
"url": "https://github.com/mastra-ai/mastra/issues"
|
|
57
|
+
},
|
|
58
|
+
"engines": {
|
|
59
|
+
"node": ">=22.13.0"
|
|
60
|
+
},
|
|
61
|
+
"scripts": {
|
|
62
|
+
"build": "tsup --silent --config tsup.config.ts",
|
|
63
|
+
"build:watch": "pnpm build --watch",
|
|
64
|
+
"test": "vitest run",
|
|
65
|
+
"test:perf": "NODE_OPTIONS='--max-old-space-size=16384' vitest run -c vitest.perf.config.ts",
|
|
66
|
+
"test:watch": "vitest watch",
|
|
67
|
+
"lint": "eslint ."
|
|
68
|
+
}
|
|
69
|
+
}
|