@mastra/dsql 0.0.0 → 1.0.0-alpha.3
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 +27 -0
- package/LICENSE.md +30 -0
- package/dist/index.cjs +4476 -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 +4467 -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 +17 -17
|
@@ -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
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/dsql",
|
|
3
|
-
"version": "0.0.
|
|
3
|
+
"version": "1.0.0-alpha.3",
|
|
4
4
|
"description": "Amazon Aurora DSQL storage provider for Mastra",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -18,14 +18,6 @@
|
|
|
18
18
|
},
|
|
19
19
|
"./package.json": "./package.json"
|
|
20
20
|
},
|
|
21
|
-
"scripts": {
|
|
22
|
-
"build": "tsup --silent --config tsup.config.ts",
|
|
23
|
-
"build:watch": "pnpm build --watch",
|
|
24
|
-
"test": "vitest run",
|
|
25
|
-
"test:perf": "NODE_OPTIONS='--max-old-space-size=16384' vitest run -c vitest.perf.config.ts",
|
|
26
|
-
"test:watch": "vitest watch",
|
|
27
|
-
"lint": "eslint ."
|
|
28
|
-
},
|
|
29
21
|
"license": "Apache-2.0",
|
|
30
22
|
"dependencies": {
|
|
31
23
|
"@aws-sdk/types": "^3.731.0",
|
|
@@ -33,19 +25,19 @@
|
|
|
33
25
|
"pg": "^8.14.0"
|
|
34
26
|
},
|
|
35
27
|
"devDependencies": {
|
|
36
|
-
"@internal/lint": "workspace:*",
|
|
37
|
-
"@internal/storage-test-utils": "workspace:*",
|
|
38
|
-
"@internal/types-builder": "workspace:*",
|
|
39
|
-
"@mastra/core": "workspace:*",
|
|
40
28
|
"@microsoft/api-extractor": "^7.52.8",
|
|
41
29
|
"@types/node": "22.13.17",
|
|
42
30
|
"@types/pg": "^8.11.14",
|
|
43
|
-
"@vitest/coverage-v8": "
|
|
44
|
-
"@vitest/ui": "
|
|
31
|
+
"@vitest/coverage-v8": "4.1.5",
|
|
32
|
+
"@vitest/ui": "4.1.5",
|
|
45
33
|
"eslint": "^9.37.0",
|
|
46
34
|
"tsup": "^8.5.0",
|
|
47
35
|
"typescript": "^5.8.3",
|
|
48
|
-
"vitest": "
|
|
36
|
+
"vitest": "4.1.5",
|
|
37
|
+
"@internal/storage-test-utils": "0.0.88",
|
|
38
|
+
"@internal/lint": "0.0.92",
|
|
39
|
+
"@internal/types-builder": "0.0.67",
|
|
40
|
+
"@mastra/core": "1.33.0-alpha.13"
|
|
49
41
|
},
|
|
50
42
|
"peerDependencies": {
|
|
51
43
|
"@mastra/core": ">=1.0.0-0 <2.0.0-0"
|
|
@@ -65,5 +57,13 @@
|
|
|
65
57
|
},
|
|
66
58
|
"engines": {
|
|
67
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
68
|
}
|
|
69
|
-
}
|
|
69
|
+
}
|