@mastra/libsql 0.13.4-alpha.0 → 0.13.5-alpha.0
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/.turbo/turbo-build.log +1 -1
- package/CHANGELOG.md +23 -0
- package/dist/index.cjs +679 -8
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +680 -9
- package/dist/index.js.map +1 -1
- package/dist/storage/domains/memory/index.d.ts.map +1 -1
- package/dist/storage/domains/observability/index.d.ts +34 -0
- package/dist/storage/domains/observability/index.d.ts.map +1 -0
- package/dist/storage/domains/operations/index.d.ts +50 -1
- package/dist/storage/domains/operations/index.d.ts.map +1 -1
- package/dist/storage/domains/utils.d.ts +45 -1
- package/dist/storage/domains/utils.d.ts.map +1 -1
- package/dist/storage/domains/workflows/index.d.ts +27 -2
- package/dist/storage/domains/workflows/index.d.ts.map +1 -1
- package/dist/storage/index.d.ts +42 -2
- package/dist/storage/index.d.ts.map +1 -1
- package/package.json +5 -5
- package/src/storage/domains/memory/index.ts +1 -1
- package/src/storage/domains/observability/index.ts +237 -0
- package/src/storage/domains/operations/index.ts +213 -3
- package/src/storage/domains/utils.ts +207 -2
- package/src/storage/domains/workflows/index.ts +225 -2
- package/src/storage/index.ts +74 -1
|
@@ -1 +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;
|
|
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,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,KAAK,EACV,cAAc,EACd,qBAAqB,EACrB,mBAAmB,EACnB,iBAAiB,EAClB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,aAAa,EAKd,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;IAqDlC;;OAEG;IACU,WAAW,CAAC,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,CAAC,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IACxF,WAAW,CAAC,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAwDvF,eAAe,CAAC,EAC3B,UAAU,EACV,MAAM,GACP,EAAE;QACD,UAAU,EAAE,MAAM,EAAE,CAAC;QACrB,MAAM,EAAE,IAAI,CAAC;KACd,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IACjB,eAAe,CAAC,EAC3B,UAAU,EACV,MAAM,GACP,EAAE;QACD,UAAU,EAAE,MAAM,EAAE,CAAC;QACrB,MAAM,CAAC,EAAE,IAAI,CAAC;KACf,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA2CjB,oBAAoB,CAC/B,IAAI,EAAE,qBAAqB,GAAG;QAC5B,MAAM,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;KACtB,GACA,OAAO,CAAC,cAAc,GAAG;QAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,eAAe,EAAE,CAAA;KAAE,CAAC;IAmG1E,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,SAAS,GAAG,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAC1G,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAAC,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA2F7F,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;IAgC1F;;OAEG;IACU,sBAAsB,CAAC,IAAI,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA4CtG,+BAA+B,CAC1C,IAAI,EAAE;QACJ,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,GAAG,iBAAiB,GACpB,OAAO,CAAC,cAAc,GAAG;QAAE,OAAO,EAAE,iBAAiB,EAAE,CAAA;KAAE,CAAC;IAkEvD,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 { AISpanRecord, AITraceRecord, AITracesPaginatedArg, 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
|
+
createAISpan(span: AISpanRecord): Promise<void>;
|
|
10
|
+
getAITrace(traceId: string): Promise<AITraceRecord | null>;
|
|
11
|
+
updateAISpan({ spanId, traceId, updates, }: {
|
|
12
|
+
spanId: string;
|
|
13
|
+
traceId: string;
|
|
14
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
15
|
+
}): Promise<void>;
|
|
16
|
+
getAITracesPaginated({ filters, pagination, }: AITracesPaginatedArg): Promise<{
|
|
17
|
+
pagination: PaginationInfo;
|
|
18
|
+
spans: AISpanRecord[];
|
|
19
|
+
}>;
|
|
20
|
+
batchCreateAISpans(args: {
|
|
21
|
+
records: AISpanRecord[];
|
|
22
|
+
}): Promise<void>;
|
|
23
|
+
batchUpdateAISpans(args: {
|
|
24
|
+
records: {
|
|
25
|
+
traceId: string;
|
|
26
|
+
spanId: string;
|
|
27
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
28
|
+
}[];
|
|
29
|
+
}): Promise<void>;
|
|
30
|
+
batchDeleteAITraces(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,EAAkB,oBAAoB,EAAkB,MAAM,sBAAsB,CAAC;AAC5F,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC9G,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,YAAY,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB/C,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IA+B1D,YAAY,CAAC,EACjB,MAAM,EACN,OAAO,EACP,OAAO,GACR,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC;KAC5D,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBX,oBAAoB,CAAC,EACzB,OAAO,EACP,UAAU,GACX,EAAE,oBAAoB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAsElF,kBAAkB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,YAAY,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBpE,kBAAkB,CAAC,IAAI,EAAE;QAC7B,OAAO,EAAE;YACP,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC;SAC5D,EAAE,CAAC;KACL,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBX,mBAAmB,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAkBvE"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { Client } from '@libsql/client';
|
|
1
|
+
import type { Client, InValue } from '@libsql/client';
|
|
2
2
|
import { StoreOperations } from '@mastra/core/storage';
|
|
3
3
|
import type { StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
|
|
4
4
|
export declare class StoreOperationsLibSQL extends StoreOperations {
|
|
@@ -35,11 +35,60 @@ export declare class StoreOperationsLibSQL extends StoreOperations {
|
|
|
35
35
|
tableName: TABLE_NAMES;
|
|
36
36
|
keys: Record<string, string>;
|
|
37
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;
|
|
38
62
|
private doBatchInsert;
|
|
39
63
|
batchInsert(args: {
|
|
40
64
|
tableName: TABLE_NAMES;
|
|
41
65
|
records: Record<string, any>[];
|
|
42
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;
|
|
43
92
|
/**
|
|
44
93
|
* Alters table schema to add columns if they don't exist
|
|
45
94
|
* @param tableName Name of the table
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;
|
|
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,EAAkB,MAAM,sBAAsB,CAAC;AAChG,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"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import type { InValue } from '@libsql/client';
|
|
2
2
|
import type { IMastraLogger } from '@mastra/core/logger';
|
|
3
|
-
import type { TABLE_NAMES } from '@mastra/core/storage';
|
|
3
|
+
import type { PaginationArgs, StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
|
|
4
4
|
export declare function createExecuteWriteOperationWithRetry({ logger, maxRetries, initialBackoffMs, }: {
|
|
5
5
|
logger: IMastraLogger;
|
|
6
6
|
maxRetries: number;
|
|
@@ -13,4 +13,48 @@ export declare function prepareStatement({ tableName, record }: {
|
|
|
13
13
|
sql: string;
|
|
14
14
|
args: InValue[];
|
|
15
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 {};
|
|
16
60
|
//# sourceMappingURL=utils.d.ts.map
|
|
@@ -1 +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;
|
|
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"}
|
|
@@ -1,14 +1,39 @@
|
|
|
1
1
|
import type { Client } from '@libsql/client';
|
|
2
|
-
import type { WorkflowRun, WorkflowRuns
|
|
2
|
+
import type { WorkflowRun, WorkflowRuns } from '@mastra/core/storage';
|
|
3
3
|
import { WorkflowsStorage } from '@mastra/core/storage';
|
|
4
|
+
import type { WorkflowRunState, StepResult } from '@mastra/core/workflows';
|
|
4
5
|
import type { StoreOperationsLibSQL } from '../operations/index.js';
|
|
5
6
|
export declare class WorkflowsLibSQL extends WorkflowsStorage {
|
|
6
7
|
operations: StoreOperationsLibSQL;
|
|
7
8
|
client: Client;
|
|
8
|
-
|
|
9
|
+
private readonly maxRetries;
|
|
10
|
+
private readonly initialBackoffMs;
|
|
11
|
+
constructor({ operations, client, maxRetries, initialBackoffMs, }: {
|
|
9
12
|
operations: StoreOperationsLibSQL;
|
|
10
13
|
client: Client;
|
|
14
|
+
maxRetries?: number;
|
|
15
|
+
initialBackoffMs?: number;
|
|
11
16
|
});
|
|
17
|
+
private setupPragmaSettings;
|
|
18
|
+
private executeWithRetry;
|
|
19
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext, }: {
|
|
20
|
+
workflowName: string;
|
|
21
|
+
runId: string;
|
|
22
|
+
stepId: string;
|
|
23
|
+
result: StepResult<any, any, any, any>;
|
|
24
|
+
runtimeContext: 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>;
|
|
12
37
|
persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
|
|
13
38
|
workflowName: string;
|
|
14
39
|
runId: string;
|
|
@@ -1 +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;
|
|
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,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAA2B,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACjF,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;IAqDrD,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,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,gBAAgB,CAAC;KAC5B;IAgBK,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,eAAe,CAAC,EACpB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,KAAK,EACL,MAAM,EACN,UAAU,GACX,GAAE;QACD,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,QAAQ,CAAC,EAAE,IAAI,CAAC;QAChB,MAAM,CAAC,EAAE,IAAI,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,YAAY,CAAC;CA+D/B"}
|
package/dist/storage/index.d.ts
CHANGED
|
@@ -3,9 +3,9 @@ import type { MastraMessageContentV2, MastraMessageV2 } from '@mastra/core/agent
|
|
|
3
3
|
import type { MastraMessageV1, StorageThreadType } from '@mastra/core/memory';
|
|
4
4
|
import type { ScoreRowData, ScoringSource } from '@mastra/core/scores';
|
|
5
5
|
import { MastraStorage } from '@mastra/core/storage';
|
|
6
|
-
import type { EvalRow, PaginationArgs, PaginationInfo, StorageColumn, StoragePagination, StorageGetMessagesArg, StorageResourceType, TABLE_NAMES, WorkflowRun, WorkflowRuns, StorageGetTracesArg, StorageDomains, ThreadSortOptions } from '@mastra/core/storage';
|
|
6
|
+
import type { EvalRow, PaginationArgs, PaginationInfo, StorageColumn, StoragePagination, StorageGetMessagesArg, StorageResourceType, TABLE_NAMES, WorkflowRun, WorkflowRuns, StorageGetTracesArg, StorageDomains, ThreadSortOptions, AISpanRecord, AITraceRecord, AITracesPaginatedArg } from '@mastra/core/storage';
|
|
7
7
|
import type { Trace } from '@mastra/core/telemetry';
|
|
8
|
-
import type { WorkflowRunState } from '@mastra/core/workflows';
|
|
8
|
+
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
9
9
|
export type LibSQLConfig = {
|
|
10
10
|
url: string;
|
|
11
11
|
authToken?: string;
|
|
@@ -37,6 +37,7 @@ export declare class LibSQLStore extends MastraStorage {
|
|
|
37
37
|
hasColumn: boolean;
|
|
38
38
|
createTable: boolean;
|
|
39
39
|
deleteMessages: boolean;
|
|
40
|
+
aiTracing: boolean;
|
|
40
41
|
};
|
|
41
42
|
createTable({ tableName, schema, }: {
|
|
42
43
|
tableName: TABLE_NAMES;
|
|
@@ -193,6 +194,24 @@ export declare class LibSQLStore extends MastraStorage {
|
|
|
193
194
|
/**
|
|
194
195
|
* WORKFLOWS
|
|
195
196
|
*/
|
|
197
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext, }: {
|
|
198
|
+
workflowName: string;
|
|
199
|
+
runId: string;
|
|
200
|
+
stepId: string;
|
|
201
|
+
result: StepResult<any, any, any, any>;
|
|
202
|
+
runtimeContext: Record<string, any>;
|
|
203
|
+
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
204
|
+
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
205
|
+
workflowName: string;
|
|
206
|
+
runId: string;
|
|
207
|
+
opts: {
|
|
208
|
+
status: string;
|
|
209
|
+
result?: StepResult<any, any, any, any>;
|
|
210
|
+
error?: string;
|
|
211
|
+
suspendedPaths?: Record<string, number[]>;
|
|
212
|
+
waitingPaths?: Record<string, number[]>;
|
|
213
|
+
};
|
|
214
|
+
}): Promise<WorkflowRunState | undefined>;
|
|
196
215
|
persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
|
|
197
216
|
workflowName: string;
|
|
198
217
|
runId: string;
|
|
@@ -225,6 +244,27 @@ export declare class LibSQLStore extends MastraStorage {
|
|
|
225
244
|
workingMemory?: string;
|
|
226
245
|
metadata?: Record<string, unknown>;
|
|
227
246
|
}): Promise<StorageResourceType>;
|
|
247
|
+
createAISpan(span: AISpanRecord): Promise<void>;
|
|
248
|
+
updateAISpan(params: {
|
|
249
|
+
spanId: string;
|
|
250
|
+
traceId: string;
|
|
251
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
252
|
+
}): Promise<void>;
|
|
253
|
+
getAITrace(traceId: string): Promise<AITraceRecord | null>;
|
|
254
|
+
getAITracesPaginated(args: AITracesPaginatedArg): Promise<{
|
|
255
|
+
pagination: PaginationInfo;
|
|
256
|
+
spans: AISpanRecord[];
|
|
257
|
+
}>;
|
|
258
|
+
batchCreateAISpans(args: {
|
|
259
|
+
records: AISpanRecord[];
|
|
260
|
+
}): Promise<void>;
|
|
261
|
+
batchUpdateAISpans(args: {
|
|
262
|
+
records: {
|
|
263
|
+
traceId: string;
|
|
264
|
+
spanId: string;
|
|
265
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
266
|
+
}[];
|
|
267
|
+
}): Promise<void>;
|
|
228
268
|
}
|
|
229
269
|
export { LibSQLStore as DefaultStorage };
|
|
230
270
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +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,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,KAAK,EACV,OAAO,EACP,cAAc,EACd,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,qBAAqB,EACrB,mBAAmB,EACnB,WAAW,EACX,WAAW,EACX,YAAY,EACZ,mBAAmB,EACnB,cAAc,EACd,iBAAiB,
|
|
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,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,KAAK,EACV,OAAO,EACP,cAAc,EACd,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,qBAAqB,EACrB,mBAAmB,EACnB,WAAW,EACX,WAAW,EACX,YAAY,EACZ,mBAAmB,EACnB,cAAc,EACd,iBAAiB,EACjB,YAAY,EACZ,aAAa,EACb,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAS3E,MAAM,MAAM,YAAY,GACpB;IACE,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,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;IAwDhC,IAAW,QAAQ;;;;;;;MASlB;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;IAI1F;;OAEG;IACU,sBAAsB,CAAC,IAAI,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IAItG,+BAA+B,CAC1C,IAAI,EAAE;QACJ,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,GAAG,iBAAiB,GACpB,OAAO,CAAC,cAAc,GAAG;QAAE,OAAO,EAAE,iBAAiB,EAAE,CAAA;KAAE,CAAC;IAIvD,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;IAIrE;;OAEG;IACU,WAAW,CAAC,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,CAAC,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IACxF,WAAW,CAAC,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAW9F,eAAe,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAC3G,eAAe,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAWrG,oBAAoB,CAC/B,IAAI,EAAE,qBAAqB,GAAG;QAC5B,MAAM,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;KACtB,GACA,OAAO,CAAC,cAAc,GAAG;QAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,eAAe,EAAE,CAAA;KAAE,CAAC;IAI1E,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAAC,MAAM,CAAC,EAAE,SAAS,GAAG,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAC1G,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAAC,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAO7F,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;IAIzD,uCAAuC;IACjC,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAIlF,QAAQ,CACZ,OAAO,GAAE;QACP,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;KACxB,GAAG,cAAmB,GACtB,OAAO,CAAC,cAAc,GAAG;QAAE,KAAK,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC;IAI3C,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,mBAAmB,CAAC,EACxB,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,gBAAgB,CAAC,EACrB,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,mBAAmB,CAAC,EACxB,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;IAEH;;OAEG;IACG,SAAS,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAItD,kBAAkB,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,MAAM,EAAE,KAAK,EAAE,CAAA;KAAE,CAAC;IAI5F,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/E;;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,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,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,eAAe,CAAC,EACpB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,KAAK,EACL,MAAM,EACN,UAAU,GACX,GAAE;QACD,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,QAAQ,CAAC,EAAE,IAAI,CAAC;QAChB,MAAM,CAAC,EAAE,IAAI,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,YAAY,CAAC;IAIxB,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,YAAY,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/C,YAAY,CAAC,MAAM,EAAE;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC;KAC5D,GAAG,OAAO,CAAC,IAAI,CAAC;IAIX,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IAI1D,oBAAoB,CACxB,IAAI,EAAE,oBAAoB,GACzB,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI3D,kBAAkB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,YAAY,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpE,kBAAkB,CAAC,IAAI,EAAE;QAC7B,OAAO,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAA;SAAE,EAAE,CAAC;KAC5G,GAAG,OAAO,CAAC,IAAI,CAAC;CAGlB;AAED,OAAO,EAAE,WAAW,IAAI,cAAc,EAAE,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/libsql",
|
|
3
|
-
"version": "0.13.
|
|
3
|
+
"version": "0.13.5-alpha.0",
|
|
4
4
|
"description": "Libsql provider for Mastra - includes both vector and db storage capabilities",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -29,10 +29,10 @@
|
|
|
29
29
|
"tsup": "^8.5.0",
|
|
30
30
|
"typescript": "^5.8.3",
|
|
31
31
|
"vitest": "^3.2.4",
|
|
32
|
-
"@internal/lint": "0.0.
|
|
33
|
-
"@internal/storage-test-utils": "0.0.
|
|
34
|
-
"@internal/types-builder": "0.0.
|
|
35
|
-
"@mastra/core": "0.14.
|
|
32
|
+
"@internal/lint": "0.0.31",
|
|
33
|
+
"@internal/storage-test-utils": "0.0.27",
|
|
34
|
+
"@internal/types-builder": "0.0.6",
|
|
35
|
+
"@mastra/core": "0.14.2-alpha.1"
|
|
36
36
|
},
|
|
37
37
|
"peerDependencies": {
|
|
38
38
|
"@mastra/core": ">=0.13.0-0 <0.15.0-0"
|
|
@@ -1,5 +1,4 @@
|
|
|
1
1
|
import type { Client, InValue } from '@libsql/client';
|
|
2
|
-
import { parseSqlIdentifier } from '@mastra/core';
|
|
3
2
|
import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
4
3
|
import { MessageList } from '@mastra/core/agent';
|
|
5
4
|
import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
|
|
@@ -17,6 +16,7 @@ import {
|
|
|
17
16
|
TABLE_RESOURCES,
|
|
18
17
|
TABLE_THREADS,
|
|
19
18
|
} from '@mastra/core/storage';
|
|
19
|
+
import { parseSqlIdentifier } from '@mastra/core/utils';
|
|
20
20
|
import type { StoreOperationsLibSQL } from '../operations';
|
|
21
21
|
|
|
22
22
|
export class MemoryLibSQL extends MemoryStorage {
|
|
@@ -0,0 +1,237 @@
|
|
|
1
|
+
import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
|
|
2
|
+
import { AI_SPAN_SCHEMA, ObservabilityStorage, TABLE_AI_SPANS } from '@mastra/core/storage';
|
|
3
|
+
import type { AISpanRecord, AITraceRecord, AITracesPaginatedArg, PaginationInfo } from '@mastra/core/storage';
|
|
4
|
+
import type { StoreOperationsLibSQL } from '../operations';
|
|
5
|
+
import { buildDateRangeFilter, prepareWhereClause, transformFromSqlRow } from '../utils';
|
|
6
|
+
|
|
7
|
+
export class ObservabilityLibSQL extends ObservabilityStorage {
|
|
8
|
+
private operations: StoreOperationsLibSQL;
|
|
9
|
+
constructor({ operations }: { operations: StoreOperationsLibSQL }) {
|
|
10
|
+
super();
|
|
11
|
+
this.operations = operations;
|
|
12
|
+
}
|
|
13
|
+
|
|
14
|
+
async createAISpan(span: AISpanRecord): Promise<void> {
|
|
15
|
+
try {
|
|
16
|
+
return this.operations.insert({ tableName: TABLE_AI_SPANS, record: span });
|
|
17
|
+
} catch (error) {
|
|
18
|
+
throw new MastraError(
|
|
19
|
+
{
|
|
20
|
+
id: 'LIBSQL_STORE_CREATE_AI_SPAN_FAILED',
|
|
21
|
+
domain: ErrorDomain.STORAGE,
|
|
22
|
+
category: ErrorCategory.USER,
|
|
23
|
+
details: {
|
|
24
|
+
spanId: span.spanId,
|
|
25
|
+
traceId: span.traceId,
|
|
26
|
+
spanType: span.spanType,
|
|
27
|
+
spanName: span.name,
|
|
28
|
+
},
|
|
29
|
+
},
|
|
30
|
+
error,
|
|
31
|
+
);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
async getAITrace(traceId: string): Promise<AITraceRecord | null> {
|
|
36
|
+
try {
|
|
37
|
+
const spans = await this.operations.loadMany<AISpanRecord>({
|
|
38
|
+
tableName: TABLE_AI_SPANS,
|
|
39
|
+
whereClause: { sql: ' WHERE traceId = ?', args: [traceId] },
|
|
40
|
+
orderBy: 'startedAt DESC',
|
|
41
|
+
});
|
|
42
|
+
|
|
43
|
+
if (!spans || spans.length === 0) {
|
|
44
|
+
return null;
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
return {
|
|
48
|
+
traceId,
|
|
49
|
+
spans: spans.map(span => transformFromSqlRow<AISpanRecord>({ tableName: TABLE_AI_SPANS, sqlRow: span })),
|
|
50
|
+
};
|
|
51
|
+
} catch (error) {
|
|
52
|
+
throw new MastraError(
|
|
53
|
+
{
|
|
54
|
+
id: 'LIBSQL_STORE_GET_AI_TRACE_FAILED',
|
|
55
|
+
domain: ErrorDomain.STORAGE,
|
|
56
|
+
category: ErrorCategory.USER,
|
|
57
|
+
details: {
|
|
58
|
+
traceId,
|
|
59
|
+
},
|
|
60
|
+
},
|
|
61
|
+
error,
|
|
62
|
+
);
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
async updateAISpan({
|
|
67
|
+
spanId,
|
|
68
|
+
traceId,
|
|
69
|
+
updates,
|
|
70
|
+
}: {
|
|
71
|
+
spanId: string;
|
|
72
|
+
traceId: string;
|
|
73
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
74
|
+
}): Promise<void> {
|
|
75
|
+
try {
|
|
76
|
+
await this.operations.update({
|
|
77
|
+
tableName: TABLE_AI_SPANS,
|
|
78
|
+
keys: { spanId, traceId },
|
|
79
|
+
data: { ...updates, updatedAt: new Date().toISOString() },
|
|
80
|
+
});
|
|
81
|
+
} catch (error) {
|
|
82
|
+
throw new MastraError(
|
|
83
|
+
{
|
|
84
|
+
id: 'LIBSQL_STORE_UPDATE_AI_SPAN_FAILED',
|
|
85
|
+
domain: ErrorDomain.STORAGE,
|
|
86
|
+
category: ErrorCategory.USER,
|
|
87
|
+
details: {
|
|
88
|
+
spanId,
|
|
89
|
+
traceId,
|
|
90
|
+
},
|
|
91
|
+
},
|
|
92
|
+
error,
|
|
93
|
+
);
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
async getAITracesPaginated({
|
|
98
|
+
filters,
|
|
99
|
+
pagination,
|
|
100
|
+
}: AITracesPaginatedArg): Promise<{ pagination: PaginationInfo; spans: AISpanRecord[] }> {
|
|
101
|
+
const page = pagination?.page ?? 0;
|
|
102
|
+
const perPage = pagination?.perPage ?? 10;
|
|
103
|
+
|
|
104
|
+
const filtersWithDateRange: Record<string, any> = {
|
|
105
|
+
...filters,
|
|
106
|
+
...buildDateRangeFilter(pagination?.dateRange, 'startedAt'),
|
|
107
|
+
};
|
|
108
|
+
const whereClause = prepareWhereClause(filtersWithDateRange, AI_SPAN_SCHEMA);
|
|
109
|
+
const orderBy = 'startedAt DESC';
|
|
110
|
+
|
|
111
|
+
let count = 0;
|
|
112
|
+
try {
|
|
113
|
+
count = await this.operations.loadTotalCount({
|
|
114
|
+
tableName: TABLE_AI_SPANS,
|
|
115
|
+
whereClause: { sql: whereClause.sql, args: whereClause.args },
|
|
116
|
+
});
|
|
117
|
+
} catch (error) {
|
|
118
|
+
throw new MastraError(
|
|
119
|
+
{
|
|
120
|
+
id: 'LIBSQL_STORE_GET_AI_TRACES_PAGINATED_COUNT_FAILED',
|
|
121
|
+
domain: ErrorDomain.STORAGE,
|
|
122
|
+
category: ErrorCategory.USER,
|
|
123
|
+
},
|
|
124
|
+
error,
|
|
125
|
+
);
|
|
126
|
+
}
|
|
127
|
+
|
|
128
|
+
if (count === 0) {
|
|
129
|
+
return {
|
|
130
|
+
pagination: {
|
|
131
|
+
total: 0,
|
|
132
|
+
page,
|
|
133
|
+
perPage,
|
|
134
|
+
hasMore: false,
|
|
135
|
+
},
|
|
136
|
+
spans: [],
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
|
|
140
|
+
try {
|
|
141
|
+
const spans = await this.operations.loadMany<AISpanRecord>({
|
|
142
|
+
tableName: TABLE_AI_SPANS,
|
|
143
|
+
whereClause,
|
|
144
|
+
orderBy,
|
|
145
|
+
offset: page * perPage,
|
|
146
|
+
limit: perPage,
|
|
147
|
+
});
|
|
148
|
+
|
|
149
|
+
return {
|
|
150
|
+
pagination: {
|
|
151
|
+
total: count,
|
|
152
|
+
page,
|
|
153
|
+
perPage,
|
|
154
|
+
hasMore: spans.length === perPage,
|
|
155
|
+
},
|
|
156
|
+
spans: spans.map(span => transformFromSqlRow<AISpanRecord>({ tableName: TABLE_AI_SPANS, sqlRow: span })),
|
|
157
|
+
};
|
|
158
|
+
} catch (error) {
|
|
159
|
+
throw new MastraError(
|
|
160
|
+
{
|
|
161
|
+
id: 'LIBSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED',
|
|
162
|
+
domain: ErrorDomain.STORAGE,
|
|
163
|
+
category: ErrorCategory.USER,
|
|
164
|
+
},
|
|
165
|
+
error,
|
|
166
|
+
);
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
|
|
170
|
+
async batchCreateAISpans(args: { records: AISpanRecord[] }): Promise<void> {
|
|
171
|
+
try {
|
|
172
|
+
return this.operations.batchInsert({
|
|
173
|
+
tableName: TABLE_AI_SPANS,
|
|
174
|
+
records: args.records.map(record => ({
|
|
175
|
+
...record,
|
|
176
|
+
createdAt: new Date().toISOString(),
|
|
177
|
+
updatedAt: new Date().toISOString(),
|
|
178
|
+
})),
|
|
179
|
+
});
|
|
180
|
+
} catch (error) {
|
|
181
|
+
throw new MastraError(
|
|
182
|
+
{
|
|
183
|
+
id: 'LIBSQL_STORE_BATCH_CREATE_AI_SPANS_FAILED',
|
|
184
|
+
domain: ErrorDomain.STORAGE,
|
|
185
|
+
category: ErrorCategory.USER,
|
|
186
|
+
},
|
|
187
|
+
error,
|
|
188
|
+
);
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
async batchUpdateAISpans(args: {
|
|
193
|
+
records: {
|
|
194
|
+
traceId: string;
|
|
195
|
+
spanId: string;
|
|
196
|
+
updates: Partial<Omit<AISpanRecord, 'spanId' | 'traceId'>>;
|
|
197
|
+
}[];
|
|
198
|
+
}): Promise<void> {
|
|
199
|
+
try {
|
|
200
|
+
return this.operations.batchUpdate({
|
|
201
|
+
tableName: TABLE_AI_SPANS,
|
|
202
|
+
updates: args.records.map(record => ({
|
|
203
|
+
keys: { spanId: record.spanId, traceId: record.traceId },
|
|
204
|
+
data: { ...record.updates, updatedAt: new Date().toISOString() },
|
|
205
|
+
})),
|
|
206
|
+
});
|
|
207
|
+
} catch (error) {
|
|
208
|
+
throw new MastraError(
|
|
209
|
+
{
|
|
210
|
+
id: 'LIBSQL_STORE_BATCH_UPDATE_AI_SPANS_FAILED',
|
|
211
|
+
domain: ErrorDomain.STORAGE,
|
|
212
|
+
category: ErrorCategory.USER,
|
|
213
|
+
},
|
|
214
|
+
error,
|
|
215
|
+
);
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
|
|
219
|
+
async batchDeleteAITraces(args: { traceIds: string[] }): Promise<void> {
|
|
220
|
+
try {
|
|
221
|
+
const keys = args.traceIds.map(traceId => ({ traceId }));
|
|
222
|
+
return this.operations.batchDelete({
|
|
223
|
+
tableName: TABLE_AI_SPANS,
|
|
224
|
+
keys,
|
|
225
|
+
});
|
|
226
|
+
} catch (error) {
|
|
227
|
+
throw new MastraError(
|
|
228
|
+
{
|
|
229
|
+
id: 'LIBSQL_STORE_BATCH_DELETE_AI_TRACES_FAILED',
|
|
230
|
+
domain: ErrorDomain.STORAGE,
|
|
231
|
+
category: ErrorCategory.USER,
|
|
232
|
+
},
|
|
233
|
+
error,
|
|
234
|
+
);
|
|
235
|
+
}
|
|
236
|
+
}
|
|
237
|
+
}
|