@mastra/libsql 0.0.0-1.x-tester-20251106055847

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.
@@ -0,0 +1,64 @@
1
+ import type { Client } from '@libsql/client';
2
+ import type { MastraMessageContentV2 } from '@mastra/core/agent';
3
+ import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
4
+ import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput } from '@mastra/core/storage';
5
+ import { MemoryStorage } from '@mastra/core/storage';
6
+ import type { StoreOperationsLibSQL } from '../operations/index.js';
7
+ export declare class MemoryLibSQL extends MemoryStorage {
8
+ private client;
9
+ private operations;
10
+ constructor({ client, operations }: {
11
+ client: Client;
12
+ operations: StoreOperationsLibSQL;
13
+ });
14
+ private parseRow;
15
+ private _getIncludedMessages;
16
+ listMessagesById({ messageIds }: {
17
+ messageIds: string[];
18
+ }): Promise<{
19
+ messages: MastraDBMessage[];
20
+ }>;
21
+ listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
22
+ saveMessages({ messages }: {
23
+ messages: MastraDBMessage[];
24
+ }): Promise<{
25
+ messages: MastraDBMessage[];
26
+ }>;
27
+ updateMessages({ messages, }: {
28
+ messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
29
+ id: string;
30
+ content?: {
31
+ metadata?: MastraMessageContentV2['metadata'];
32
+ content?: MastraMessageContentV2['content'];
33
+ };
34
+ })[];
35
+ }): Promise<MastraDBMessage[]>;
36
+ deleteMessages(messageIds: string[]): Promise<void>;
37
+ getResourceById({ resourceId }: {
38
+ resourceId: string;
39
+ }): Promise<StorageResourceType | null>;
40
+ saveResource({ resource }: {
41
+ resource: StorageResourceType;
42
+ }): Promise<StorageResourceType>;
43
+ updateResource({ resourceId, workingMemory, metadata, }: {
44
+ resourceId: string;
45
+ workingMemory?: string;
46
+ metadata?: Record<string, unknown>;
47
+ }): Promise<StorageResourceType>;
48
+ getThreadById({ threadId }: {
49
+ threadId: string;
50
+ }): Promise<StorageThreadType | null>;
51
+ listThreadsByResourceId(args: StorageListThreadsByResourceIdInput): Promise<StorageListThreadsByResourceIdOutput>;
52
+ saveThread({ thread }: {
53
+ thread: StorageThreadType;
54
+ }): Promise<StorageThreadType>;
55
+ updateThread({ id, title, metadata, }: {
56
+ id: string;
57
+ title: string;
58
+ metadata: Record<string, unknown>;
59
+ }): Promise<StorageThreadType>;
60
+ deleteThread({ threadId }: {
61
+ threadId: string;
62
+ }): Promise<void>;
63
+ }
64
+ //# 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":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAGjE,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,aAAa,EAMd,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAE3D,qBAAa,YAAa,SAAQ,aAAa;IAC7C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAwB;gBAC9B,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,qBAAqB,CAAA;KAAE;IAMzF,OAAO,CAAC,QAAQ;YAmBF,oBAAoB;IAsDrB,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAmCpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IA8JvF,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAoFrG,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;gBAAE,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAAC,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAA;aAAE,CAAC;SAC1G,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAmGxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoEnD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAuB5F,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;IAmD1B,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAgC7E,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IAqF1C,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA2BjF,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;IA6CxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAwBtE"}
@@ -0,0 +1,34 @@
1
+ import { ObservabilityStorage } from '@mastra/core/storage';
2
+ import type { SpanRecord, CreateSpanRecord, UpdateSpanRecord, TraceRecord, TracesPaginatedArg, PaginationInfo } from '@mastra/core/storage';
3
+ import type { StoreOperationsLibSQL } from '../operations/index.js';
4
+ export declare class ObservabilityLibSQL extends ObservabilityStorage {
5
+ private operations;
6
+ constructor({ operations }: {
7
+ operations: StoreOperationsLibSQL;
8
+ });
9
+ createSpan(span: CreateSpanRecord): Promise<void>;
10
+ getTrace(traceId: string): Promise<TraceRecord | null>;
11
+ updateSpan({ spanId, traceId, updates, }: {
12
+ spanId: string;
13
+ traceId: string;
14
+ updates: Partial<UpdateSpanRecord>;
15
+ }): Promise<void>;
16
+ getTracesPaginated({ filters, pagination, }: TracesPaginatedArg): Promise<{
17
+ pagination: PaginationInfo;
18
+ spans: SpanRecord[];
19
+ }>;
20
+ batchCreateSpans(args: {
21
+ records: CreateSpanRecord[];
22
+ }): Promise<void>;
23
+ batchUpdateSpans(args: {
24
+ records: {
25
+ traceId: string;
26
+ spanId: string;
27
+ updates: Partial<UpdateSpanRecord>;
28
+ }[];
29
+ }): Promise<void>;
30
+ batchDeleteTraces(args: {
31
+ traceIds: string[];
32
+ }): Promise<void>;
33
+ }
34
+ //# 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,EAAe,oBAAoB,EAAe,MAAM,sBAAsB,CAAC;AACtF,OAAO,KAAK,EACV,UAAU,EACV,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EACX,kBAAkB,EAClB,cAAc,EACf,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAG3D,qBAAa,mBAAoB,SAAQ,oBAAoB;IAC3D,OAAO,CAAC,UAAU,CAAwB;gBAC9B,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,qBAAqB,CAAA;KAAE;IAK3D,UAAU,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IA4BjD,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA+BtD,UAAU,CAAC,EACf,MAAM,EACN,OAAO,EACP,OAAO,GACR,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBX,kBAAkB,CAAC,EACvB,OAAO,EACP,UAAU,GACX,EAAE,kBAAkB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IA4G9E,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,gBAAgB,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBtE,gBAAgB,CAAC,IAAI,EAAE;QAC3B,OAAO,EAAE;YACP,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;SACpC,EAAE,CAAC;KACL,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBX,iBAAiB,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAkBrE"}
@@ -0,0 +1,110 @@
1
+ import type { Client, InValue } from '@libsql/client';
2
+ import { StoreOperations } from '@mastra/core/storage';
3
+ import type { StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
4
+ export declare class StoreOperationsLibSQL extends StoreOperations {
5
+ private client;
6
+ /**
7
+ * Maximum number of retries for write operations if an SQLITE_BUSY error occurs.
8
+ * @default 5
9
+ */
10
+ maxRetries: number;
11
+ /**
12
+ * Initial backoff time in milliseconds for retrying write operations on SQLITE_BUSY.
13
+ * The backoff time will double with each retry (exponential backoff).
14
+ * @default 100
15
+ */
16
+ initialBackoffMs: number;
17
+ constructor({ client, maxRetries, initialBackoffMs, }: {
18
+ client: Client;
19
+ maxRetries?: number;
20
+ initialBackoffMs?: number;
21
+ });
22
+ hasColumn(table: string, column: string): Promise<boolean>;
23
+ private getCreateTableSQL;
24
+ createTable({ tableName, schema, }: {
25
+ tableName: TABLE_NAMES;
26
+ schema: Record<string, StorageColumn>;
27
+ }): Promise<void>;
28
+ protected getSqlType(type: StorageColumn['type']): string;
29
+ private doInsert;
30
+ insert(args: {
31
+ tableName: TABLE_NAMES;
32
+ record: Record<string, any>;
33
+ }): Promise<void>;
34
+ load<R>({ tableName, keys }: {
35
+ tableName: TABLE_NAMES;
36
+ keys: Record<string, string>;
37
+ }): Promise<R | null>;
38
+ loadMany<R>({ tableName, whereClause, orderBy, offset, limit, args, }: {
39
+ tableName: TABLE_NAMES;
40
+ whereClause?: {
41
+ sql: string;
42
+ args: InValue[];
43
+ };
44
+ orderBy?: string;
45
+ offset?: number;
46
+ limit?: number;
47
+ args?: any[];
48
+ }): Promise<R[]>;
49
+ loadTotalCount({ tableName, whereClause, }: {
50
+ tableName: TABLE_NAMES;
51
+ whereClause?: {
52
+ sql: string;
53
+ args: InValue[];
54
+ };
55
+ }): Promise<number>;
56
+ update(args: {
57
+ tableName: TABLE_NAMES;
58
+ keys: Record<string, any>;
59
+ data: Record<string, any>;
60
+ }): Promise<void>;
61
+ private executeUpdate;
62
+ private doBatchInsert;
63
+ batchInsert(args: {
64
+ tableName: TABLE_NAMES;
65
+ records: Record<string, any>[];
66
+ }): Promise<void>;
67
+ /**
68
+ * Public batch update method with retry logic
69
+ */
70
+ batchUpdate(args: {
71
+ tableName: TABLE_NAMES;
72
+ updates: Array<{
73
+ keys: Record<string, any>;
74
+ data: Record<string, any>;
75
+ }>;
76
+ }): Promise<void>;
77
+ /**
78
+ * Updates multiple records in batch. Each record can be updated based on single or composite keys.
79
+ */
80
+ private executeBatchUpdate;
81
+ /**
82
+ * Public batch delete method with retry logic
83
+ */
84
+ batchDelete({ tableName, keys }: {
85
+ tableName: TABLE_NAMES;
86
+ keys: Array<Record<string, any>>;
87
+ }): Promise<void>;
88
+ /**
89
+ * Deletes multiple records in batch. Each record can be deleted based on single or composite keys.
90
+ */
91
+ private executeBatchDelete;
92
+ /**
93
+ * Alters table schema to add columns if they don't exist
94
+ * @param tableName Name of the table
95
+ * @param schema Schema of the table
96
+ * @param ifNotExists Array of column names to add if they don't exist
97
+ */
98
+ alterTable({ tableName, schema, ifNotExists, }: {
99
+ tableName: TABLE_NAMES;
100
+ schema: Record<string, StorageColumn>;
101
+ ifNotExists: string[];
102
+ }): Promise<void>;
103
+ clearTable({ tableName }: {
104
+ tableName: TABLE_NAMES;
105
+ }): Promise<void>;
106
+ dropTable({ tableName }: {
107
+ tableName: TABLE_NAMES;
108
+ }): Promise<void>;
109
+ }
110
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAEtD,OAAO,EAA2B,eAAe,EAAe,MAAM,sBAAsB,CAAC;AAC7F,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AASvE,qBAAa,qBAAsB,SAAQ,eAAe;IACxD,OAAO,CAAC,MAAM,CAAS;IACvB;;;OAGG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;;;OAIG;IACH,gBAAgB,EAAE,MAAM,CAAC;gBAEb,EACV,MAAM,EACN,UAAU,EACV,gBAAgB,GACjB,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B;IAQK,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAOhE,OAAO,CAAC,iBAAiB;IAmCnB,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;IAoBjB,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;YAW3C,QAAQ;IAef,MAAM,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IASrF,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;IAkCzG,QAAQ,CAAC,CAAC,EAAE,EAChB,SAAS,EACT,WAAW,EACX,OAAO,EACP,MAAM,EACN,KAAK,EACL,IAAI,GACL,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,WAAW,CAAC,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,OAAO,EAAE,CAAA;SAAE,CAAC;QAC/C,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;KACd,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IA6BV,cAAc,CAAC,EACnB,SAAS,EACT,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,WAAW,CAAC,EAAE;YAAE,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,OAAO,EAAE,CAAA;SAAE,CAAC;KAChD,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBZ,MAAM,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;YAStG,aAAa;YAYb,aAAa;IAYpB,WAAW,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyBnG;;OAEG;IACI,WAAW,CAAC,IAAI,EAAE;QACvB,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;IAyBjB;;OAEG;YACW,kBAAkB;IAuBhC;;OAEG;IACI,WAAW,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyBpH;;OAEG;YACW,kBAAkB;IAmBhC;;;;;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,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBpE,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAkB1E"}
@@ -0,0 +1,54 @@
1
+ import type { Client } from '@libsql/client';
2
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/evals';
3
+ import { ScoresStorage } from '@mastra/core/storage';
4
+ import type { PaginationInfo, StoragePagination } from '@mastra/core/storage';
5
+ import type { StoreOperationsLibSQL } from '../operations/index.js';
6
+ export declare class ScoresLibSQL extends ScoresStorage {
7
+ private operations;
8
+ private client;
9
+ constructor({ client, operations }: {
10
+ client: Client;
11
+ operations: StoreOperationsLibSQL;
12
+ });
13
+ listScoresByRunId({ runId, pagination, }: {
14
+ runId: string;
15
+ pagination: StoragePagination;
16
+ }): Promise<{
17
+ pagination: PaginationInfo;
18
+ scores: ScoreRowData[];
19
+ }>;
20
+ listScoresByScorerId({ scorerId, entityId, entityType, source, pagination, }: {
21
+ scorerId: string;
22
+ entityId?: string;
23
+ entityType?: string;
24
+ source?: ScoringSource;
25
+ pagination: StoragePagination;
26
+ }): Promise<{
27
+ pagination: PaginationInfo;
28
+ scores: ScoreRowData[];
29
+ }>;
30
+ private transformScoreRow;
31
+ getScoreById({ id }: {
32
+ id: string;
33
+ }): Promise<ScoreRowData | null>;
34
+ saveScore(score: Omit<ScoreRowData, 'id' | 'createdAt' | 'updatedAt'>): Promise<{
35
+ score: ScoreRowData;
36
+ }>;
37
+ listScoresByEntityId({ entityId, entityType, pagination, }: {
38
+ pagination: StoragePagination;
39
+ entityId: string;
40
+ entityType: string;
41
+ }): Promise<{
42
+ pagination: PaginationInfo;
43
+ scores: ScoreRowData[];
44
+ }>;
45
+ listScoresBySpan({ traceId, spanId, pagination, }: {
46
+ traceId: string;
47
+ spanId: string;
48
+ pagination: StoragePagination;
49
+ }): Promise<{
50
+ pagination: PaginationInfo;
51
+ scores: ScoreRowData[];
52
+ }>;
53
+ }
54
+ //# 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":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AAGtD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAA6B,MAAM,oBAAoB,CAAC;AACjG,OAAO,EAEL,aAAa,EAId,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAE3D,qBAAa,YAAa,SAAQ,aAAa;IAC7C,OAAO,CAAC,UAAU,CAAwB;IAC1C,OAAO,CAAC,MAAM,CAAS;gBACX,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,qBAAqB,CAAA;KAAE;IAMnF,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAwD7D,oBAAoB,CAAC,EACzB,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,MAAM,EACN,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;QACvB,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAiFnE,OAAO,CAAC,iBAAiB;IA0CnB,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAQlE,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAiDxG,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;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAwD7D,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;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;CA2CpE"}
@@ -0,0 +1,60 @@
1
+ import type { InValue } from '@libsql/client';
2
+ import type { IMastraLogger } from '@mastra/core/logger';
3
+ import type { PaginationArgs, StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
4
+ export declare function createExecuteWriteOperationWithRetry({ logger, maxRetries, initialBackoffMs, }: {
5
+ logger: IMastraLogger;
6
+ maxRetries: number;
7
+ initialBackoffMs: number;
8
+ }): <T>(operationFn: () => Promise<T>, operationDescription: string) => Promise<T>;
9
+ export declare function prepareStatement({ tableName, record }: {
10
+ tableName: TABLE_NAMES;
11
+ record: Record<string, any>;
12
+ }): {
13
+ sql: string;
14
+ args: InValue[];
15
+ };
16
+ export declare function prepareUpdateStatement({ tableName, updates, keys, }: {
17
+ tableName: TABLE_NAMES;
18
+ updates: Record<string, any>;
19
+ keys: Record<string, any>;
20
+ }): {
21
+ sql: string;
22
+ args: InValue[];
23
+ };
24
+ export declare function transformToSqlValue(value: any): InValue;
25
+ export declare function prepareDeleteStatement({ tableName, keys }: {
26
+ tableName: TABLE_NAMES;
27
+ keys: Record<string, any>;
28
+ }): {
29
+ sql: string;
30
+ args: InValue[];
31
+ };
32
+ type WhereValue = InValue | {
33
+ startAt?: InValue;
34
+ endAt?: InValue;
35
+ };
36
+ export declare function prepareWhereClause(filters: Record<string, WhereValue>, schema: Record<string, StorageColumn>): {
37
+ sql: string;
38
+ args: InValue[];
39
+ };
40
+ type DateRangeFilter = {
41
+ startAt?: string;
42
+ endAt?: string;
43
+ };
44
+ /**
45
+ * Converts pagination date range to where clause date range format
46
+ * @param dateRange - The date range from pagination
47
+ * @param columnName - The timestamp column to filter on (defaults to 'createdAt')
48
+ * @returns Object with the date range filter, or empty object if no date range
49
+ */
50
+ export declare function buildDateRangeFilter(dateRange?: PaginationArgs['dateRange'], columnName?: string): Record<string, DateRangeFilter>;
51
+ /**
52
+ * Transforms SQL row data back to a typed object format
53
+ * Reverses the transformations done in prepareStatement
54
+ */
55
+ export declare function transformFromSqlRow<T>({ tableName, sqlRow, }: {
56
+ tableName: TABLE_NAMES;
57
+ sqlRow: Record<string, any>;
58
+ }): T;
59
+ export {};
60
+ //# 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,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEzD,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGvF,wBAAgB,oCAAoC,CAAC,EACnD,MAAM,EACN,UAAU,EACV,gBAAgB,GACjB,EAAE;IACD,MAAM,EAAE,aAAa,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;CAC1B,IACsD,CAAC,EACpD,aAAa,MAAM,OAAO,CAAC,CAAC,CAAC,EAC7B,sBAAsB,MAAM,KAC3B,OAAO,CAAC,CAAC,CAAC,CAyBd;AAED,wBAAgB,gBAAgB,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;IAAE,SAAS,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;CAAE,GAAG;IAChH,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,OAAO,EAAE,CAAC;CACjB,CAmBA;AAED,wBAAgB,sBAAsB,CAAC,EACrC,SAAS,EACT,OAAO,EACP,IAAI,GACL,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC3B,GAAG;IACF,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,OAAO,EAAE,CAAC;CACjB,CAeA;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,CAQvD;AAED,wBAAgB,sBAAsB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;IAAE,SAAS,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;CAAE,GAAG;IAClH,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,OAAO,EAAE,CAAC;CACjB,CAQA;AAED,KAAK,UAAU,GAAG,OAAO,GAAG;IAAE,OAAO,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAEnE,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,EACnC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,GACpC;IACD,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,OAAO,EAAE,CAAC;CACjB,CAqBA;AA+CD,KAAK,eAAe,GAAG;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAClC,SAAS,CAAC,EAAE,cAAc,CAAC,WAAW,CAAC,EACvC,UAAU,GAAE,MAAoB,GAC/B,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAgBjC;AAED;;;GAGG;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,CAiCJ"}
@@ -0,0 +1,53 @@
1
+ import type { Client } from '@libsql/client';
2
+ import type { WorkflowRun, WorkflowRuns, StorageListWorkflowRunsInput } from '@mastra/core/storage';
3
+ import { WorkflowsStorage } from '@mastra/core/storage';
4
+ import type { WorkflowRunState, StepResult } from '@mastra/core/workflows';
5
+ import type { StoreOperationsLibSQL } from '../operations/index.js';
6
+ export declare class WorkflowsLibSQL extends WorkflowsStorage {
7
+ operations: StoreOperationsLibSQL;
8
+ client: Client;
9
+ private readonly maxRetries;
10
+ private readonly initialBackoffMs;
11
+ constructor({ operations, client, maxRetries, initialBackoffMs, }: {
12
+ operations: StoreOperationsLibSQL;
13
+ client: Client;
14
+ maxRetries?: number;
15
+ initialBackoffMs?: number;
16
+ });
17
+ private setupPragmaSettings;
18
+ private executeWithRetry;
19
+ updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
20
+ workflowName: string;
21
+ runId: string;
22
+ stepId: string;
23
+ result: StepResult<any, any, any, any>;
24
+ requestContext: Record<string, any>;
25
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
26
+ updateWorkflowState({ workflowName, runId, opts, }: {
27
+ workflowName: string;
28
+ runId: string;
29
+ opts: {
30
+ status: string;
31
+ result?: StepResult<any, any, any, any>;
32
+ error?: string;
33
+ suspendedPaths?: Record<string, number[]>;
34
+ waitingPaths?: Record<string, number[]>;
35
+ };
36
+ }): Promise<WorkflowRunState | undefined>;
37
+ persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
38
+ workflowName: string;
39
+ runId: string;
40
+ resourceId?: string;
41
+ snapshot: WorkflowRunState;
42
+ }): Promise<void>;
43
+ loadWorkflowSnapshot({ workflowName, runId, }: {
44
+ workflowName: string;
45
+ runId: string;
46
+ }): Promise<WorkflowRunState | null>;
47
+ getWorkflowRunById({ runId, workflowName, }: {
48
+ runId: string;
49
+ workflowName?: string;
50
+ }): Promise<WorkflowRun | null>;
51
+ listWorkflowRuns({ workflowName, fromDate, toDate, page, perPage, resourceId, }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
52
+ }
53
+ //# 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":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AAEtD,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,4BAA4B,EAAE,MAAM,sBAAsB,CAAC;AACpG,OAAO,EAA6C,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACnG,OAAO,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAC3E,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAsB3D,qBAAa,eAAgB,SAAQ,gBAAgB;IACnD,UAAU,EAAE,qBAAqB,CAAC;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;gBAE9B,EACV,UAAU,EACV,MAAM,EACN,UAAc,EACd,gBAAsB,GACvB,EAAE;QACD,UAAU,EAAE,qBAAqB,CAAC;QAClC,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B;YAca,mBAAmB;YA0BnB,gBAAgB;IAmDxB,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;IAsDrD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM,CAAC;YACf,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACxC,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;SACzC,CAAC;KACH,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IA6CnC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;KAC5B;IAiBK,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;IAU9B,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;IAuCzB,gBAAgB,CAAC,EACrB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,IAAI,EACJ,OAAO,EACP,UAAU,GACX,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;CAkE7D"}
@@ -0,0 +1,220 @@
1
+ import type { Client } from '@libsql/client';
2
+ import type { MastraMessageContentV2, MastraDBMessage } from '@mastra/core/agent';
3
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/evals';
4
+ import type { StorageThreadType } from '@mastra/core/memory';
5
+ import { MastraStorage } from '@mastra/core/storage';
6
+ import type { PaginationInfo, StorageColumn, StoragePagination, StorageResourceType, TABLE_NAMES, WorkflowRun, WorkflowRuns, StorageDomains, SpanRecord, TraceRecord, TracesPaginatedArg, StorageListWorkflowRunsInput } from '@mastra/core/storage';
7
+ import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
8
+ export type LibSQLConfig = {
9
+ id: string;
10
+ url: string;
11
+ authToken?: string;
12
+ /**
13
+ * Maximum number of retries for write operations if an SQLITE_BUSY error occurs.
14
+ * @default 5
15
+ */
16
+ maxRetries?: number;
17
+ /**
18
+ * Initial backoff time in milliseconds for retrying write operations on SQLITE_BUSY.
19
+ * The backoff time will double with each retry (exponential backoff).
20
+ * @default 100
21
+ */
22
+ initialBackoffMs?: number;
23
+ } | {
24
+ id: string;
25
+ client: Client;
26
+ maxRetries?: number;
27
+ initialBackoffMs?: number;
28
+ };
29
+ export declare class LibSQLStore extends MastraStorage {
30
+ private client;
31
+ private readonly maxRetries;
32
+ private readonly initialBackoffMs;
33
+ stores: StorageDomains;
34
+ constructor(config: LibSQLConfig);
35
+ get supports(): {
36
+ selectByIncludeResourceScope: boolean;
37
+ resourceWorkingMemory: boolean;
38
+ hasColumn: boolean;
39
+ createTable: boolean;
40
+ deleteMessages: boolean;
41
+ observabilityInstance: boolean;
42
+ listScoresBySpan: boolean;
43
+ };
44
+ createTable({ tableName, schema, }: {
45
+ tableName: TABLE_NAMES;
46
+ schema: Record<string, StorageColumn>;
47
+ }): Promise<void>;
48
+ /**
49
+ * Alters table schema to add columns if they don't exist
50
+ * @param tableName Name of the table
51
+ * @param schema Schema of the table
52
+ * @param ifNotExists Array of column names to add if they don't exist
53
+ */
54
+ alterTable({ tableName, schema, ifNotExists, }: {
55
+ tableName: TABLE_NAMES;
56
+ schema: Record<string, StorageColumn>;
57
+ ifNotExists: string[];
58
+ }): Promise<void>;
59
+ clearTable({ tableName }: {
60
+ tableName: TABLE_NAMES;
61
+ }): Promise<void>;
62
+ dropTable({ tableName }: {
63
+ tableName: TABLE_NAMES;
64
+ }): Promise<void>;
65
+ insert(args: {
66
+ tableName: TABLE_NAMES;
67
+ record: Record<string, any>;
68
+ }): Promise<void>;
69
+ batchInsert(args: {
70
+ tableName: TABLE_NAMES;
71
+ records: Record<string, any>[];
72
+ }): Promise<void>;
73
+ load<R>({ tableName, keys }: {
74
+ tableName: TABLE_NAMES;
75
+ keys: Record<string, string>;
76
+ }): Promise<R | null>;
77
+ getThreadById({ threadId }: {
78
+ threadId: string;
79
+ }): Promise<StorageThreadType | null>;
80
+ saveThread({ thread }: {
81
+ thread: StorageThreadType;
82
+ }): Promise<StorageThreadType>;
83
+ updateThread({ id, title, metadata, }: {
84
+ id: string;
85
+ title: string;
86
+ metadata: Record<string, unknown>;
87
+ }): Promise<StorageThreadType>;
88
+ deleteThread({ threadId }: {
89
+ threadId: string;
90
+ }): Promise<void>;
91
+ listMessagesById({ messageIds }: {
92
+ messageIds: string[];
93
+ }): Promise<{
94
+ messages: MastraDBMessage[];
95
+ }>;
96
+ saveMessages(args: {
97
+ messages: MastraDBMessage[];
98
+ }): Promise<{
99
+ messages: MastraDBMessage[];
100
+ }>;
101
+ updateMessages({ messages, }: {
102
+ messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
103
+ id: string;
104
+ content?: {
105
+ metadata?: MastraMessageContentV2['metadata'];
106
+ content?: MastraMessageContentV2['content'];
107
+ };
108
+ })[];
109
+ }): Promise<MastraDBMessage[]>;
110
+ deleteMessages(messageIds: string[]): Promise<void>;
111
+ getScoreById({ id }: {
112
+ id: string;
113
+ }): Promise<ScoreRowData | null>;
114
+ saveScore(score: Omit<ScoreRowData, 'id' | 'createdAt' | 'updatedAt'>): Promise<{
115
+ score: ScoreRowData;
116
+ }>;
117
+ listScoresByScorerId({ scorerId, entityId, entityType, source, pagination, }: {
118
+ scorerId: string;
119
+ entityId?: string;
120
+ entityType?: string;
121
+ source?: ScoringSource;
122
+ pagination: StoragePagination;
123
+ }): Promise<{
124
+ pagination: PaginationInfo;
125
+ scores: ScoreRowData[];
126
+ }>;
127
+ listScoresByRunId({ runId, pagination, }: {
128
+ runId: string;
129
+ pagination: StoragePagination;
130
+ }): Promise<{
131
+ pagination: PaginationInfo;
132
+ scores: ScoreRowData[];
133
+ }>;
134
+ listScoresByEntityId({ entityId, entityType, pagination, }: {
135
+ pagination: StoragePagination;
136
+ entityId: string;
137
+ entityType: string;
138
+ }): Promise<{
139
+ pagination: PaginationInfo;
140
+ scores: ScoreRowData[];
141
+ }>;
142
+ /**
143
+ * WORKFLOWS
144
+ */
145
+ updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
146
+ workflowName: string;
147
+ runId: string;
148
+ stepId: string;
149
+ result: StepResult<any, any, any, any>;
150
+ requestContext: Record<string, any>;
151
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
152
+ updateWorkflowState({ workflowName, runId, opts, }: {
153
+ workflowName: string;
154
+ runId: string;
155
+ opts: {
156
+ status: string;
157
+ result?: StepResult<any, any, any, any>;
158
+ error?: string;
159
+ suspendedPaths?: Record<string, number[]>;
160
+ waitingPaths?: Record<string, number[]>;
161
+ };
162
+ }): Promise<WorkflowRunState | undefined>;
163
+ persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
164
+ workflowName: string;
165
+ runId: string;
166
+ resourceId?: string;
167
+ snapshot: WorkflowRunState;
168
+ }): Promise<void>;
169
+ loadWorkflowSnapshot({ workflowName, runId, }: {
170
+ workflowName: string;
171
+ runId: string;
172
+ }): Promise<WorkflowRunState | null>;
173
+ listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, }?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
174
+ getWorkflowRunById({ runId, workflowName, }: {
175
+ runId: string;
176
+ workflowName?: string;
177
+ }): Promise<WorkflowRun | null>;
178
+ getResourceById({ resourceId }: {
179
+ resourceId: string;
180
+ }): Promise<StorageResourceType | null>;
181
+ saveResource({ resource }: {
182
+ resource: StorageResourceType;
183
+ }): Promise<StorageResourceType>;
184
+ updateResource({ resourceId, workingMemory, metadata, }: {
185
+ resourceId: string;
186
+ workingMemory?: string;
187
+ metadata?: Record<string, unknown>;
188
+ }): Promise<StorageResourceType>;
189
+ createSpan(span: SpanRecord): Promise<void>;
190
+ updateSpan(params: {
191
+ spanId: string;
192
+ traceId: string;
193
+ updates: Partial<Omit<SpanRecord, 'spanId' | 'traceId'>>;
194
+ }): Promise<void>;
195
+ getTrace(traceId: string): Promise<TraceRecord | null>;
196
+ getTracesPaginated(args: TracesPaginatedArg): Promise<{
197
+ pagination: PaginationInfo;
198
+ spans: SpanRecord[];
199
+ }>;
200
+ listScoresBySpan({ traceId, spanId, pagination, }: {
201
+ traceId: string;
202
+ spanId: string;
203
+ pagination: StoragePagination;
204
+ }): Promise<{
205
+ pagination: PaginationInfo;
206
+ scores: ScoreRowData[];
207
+ }>;
208
+ batchCreateSpans(args: {
209
+ records: SpanRecord[];
210
+ }): Promise<void>;
211
+ batchUpdateSpans(args: {
212
+ records: {
213
+ traceId: string;
214
+ spanId: string;
215
+ updates: Partial<Omit<SpanRecord, 'spanId' | 'traceId'>>;
216
+ }[];
217
+ }): Promise<void>;
218
+ }
219
+ export { LibSQLStore as DefaultStorage };
220
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAClF,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,KAAK,EACV,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,WAAW,EACX,WAAW,EACX,YAAY,EACZ,cAAc,EACd,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,4BAA4B,EAC7B,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAO3E,MAAM,MAAM,YAAY,GACpB;IACE,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B,GACD;IACE,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B,CAAC;AAEN,qBAAa,WAAY,SAAQ,aAAa;IAC5C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAE1C,MAAM,EAAE,cAAc,CAAC;gBAEX,MAAM,EAAE,YAAY;IAuDhC,IAAW,QAAQ;;;;;;;;MAUlB;IAEK,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;IAIjB;;;;;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;IAIX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpE,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlE,MAAM,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpF,WAAW,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7F,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;IAIzG,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAIpF,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAIjF,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;IAIxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/D,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAIpG,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAK7F,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;gBAAE,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAAC,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAA;aAAE,CAAC;SAC1G,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAIxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAInD,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAIlE,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAIxG,oBAAoB,CAAC,EACzB,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,MAAM,EACN,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;QACvB,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI7D,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI7D,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;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAInE;;OAEG;IAEG,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;IAIrD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE;YACJ,MAAM,EAAE,MAAM,CAAC;YACf,MAAM,CAAC,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACxC,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;SACzC,CAAC;KACH,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAInC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,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;IAI9B,gBAAgB,CAAC,EACrB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,OAAO,EACP,IAAI,EACJ,UAAU,GACX,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;IAItD,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;IAIzB,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAI5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAI3F,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;IAI1B,UAAU,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAI3C,UAAU,CAAC,MAAM,EAAE;QACvB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC;KAC1D,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAItD,kBAAkB,CAAC,IAAI,EAAE,kBAAkB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IAI1G,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;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI7D,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,UAAU,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIhE,gBAAgB,CAAC,IAAI,EAAE;QAC3B,OAAO,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAA;SAAE,EAAE,CAAC;KAC1G,GAAG,OAAO,CAAC,IAAI,CAAC;CAGlB;AAED,OAAO,EAAE,WAAW,IAAI,cAAc,EAAE,CAAC"}