@mastra/lance 0.0.0-toolOptionTypes-20250917085558 → 0.0.0-top-level-fix-20251211111608
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 +711 -3
- package/README.md +61 -4
- package/dist/index.cjs +661 -738
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +662 -739
- package/dist/index.js.map +1 -1
- package/dist/storage/domains/memory/index.d.ts +15 -39
- package/dist/storage/domains/memory/index.d.ts.map +1 -1
- package/dist/storage/domains/operations/index.d.ts.map +1 -1
- package/dist/storage/domains/scores/index.d.ts +21 -5
- package/dist/storage/domains/scores/index.d.ts.map +1 -1
- package/dist/storage/domains/utils.d.ts.map +1 -1
- package/dist/storage/domains/workflows/index.d.ts +8 -11
- package/dist/storage/domains/workflows/index.d.ts.map +1 -1
- package/dist/storage/index.d.ts +68 -88
- package/dist/storage/index.d.ts.map +1 -1
- package/dist/vector/filter.d.ts +5 -5
- package/dist/vector/index.d.ts +6 -3
- package/dist/vector/index.d.ts.map +1 -1
- package/package.json +15 -10
- package/dist/storage/domains/legacy-evals/index.d.ts +0 -25
- package/dist/storage/domains/legacy-evals/index.d.ts.map +0 -1
- package/dist/storage/domains/traces/index.d.ts +0 -34
- package/dist/storage/domains/traces/index.d.ts.map +0 -1
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
import type { Connection } from '@lancedb/lancedb';
|
|
2
2
|
import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
3
|
-
import type {
|
|
3
|
+
import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
|
|
4
4
|
import { MemoryStorage } from '@mastra/core/storage';
|
|
5
|
-
import type {
|
|
5
|
+
import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput } from '@mastra/core/storage';
|
|
6
6
|
import type { StoreOperationsLance } from '../operations/index.js';
|
|
7
7
|
export declare class StoreMemoryLance extends MemoryStorage {
|
|
8
8
|
private client;
|
|
@@ -11,12 +11,10 @@ export declare class StoreMemoryLance extends MemoryStorage {
|
|
|
11
11
|
client: Connection;
|
|
12
12
|
operations: StoreOperationsLance;
|
|
13
13
|
});
|
|
14
|
+
private escapeSql;
|
|
14
15
|
getThreadById({ threadId }: {
|
|
15
16
|
threadId: string;
|
|
16
17
|
}): Promise<StorageThreadType | null>;
|
|
17
|
-
getThreadsByResourceId({ resourceId }: {
|
|
18
|
-
resourceId: string;
|
|
19
|
-
}): Promise<StorageThreadType[]>;
|
|
20
18
|
/**
|
|
21
19
|
* Saves a thread to the database. This function doesn't overwrite existing threads.
|
|
22
20
|
* @param thread - The thread to save
|
|
@@ -34,35 +32,18 @@ export declare class StoreMemoryLance extends MemoryStorage {
|
|
|
34
32
|
threadId: string;
|
|
35
33
|
}): Promise<void>;
|
|
36
34
|
private normalizeMessage;
|
|
37
|
-
|
|
38
|
-
format?: 'v1';
|
|
39
|
-
}): Promise<MastraMessageV1[]>;
|
|
40
|
-
getMessages(args: StorageGetMessagesArg & {
|
|
41
|
-
format: 'v2';
|
|
42
|
-
}): Promise<MastraMessageV2[]>;
|
|
43
|
-
getMessagesById({ messageIds, format, }: {
|
|
44
|
-
messageIds: string[];
|
|
45
|
-
format: 'v1';
|
|
46
|
-
}): Promise<MastraMessageV1[]>;
|
|
47
|
-
getMessagesById({ messageIds, format, }: {
|
|
35
|
+
listMessagesById({ messageIds }: {
|
|
48
36
|
messageIds: string[];
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
format?: undefined | 'v1';
|
|
54
|
-
}): Promise<MastraMessageV1[]>;
|
|
37
|
+
}): Promise<{
|
|
38
|
+
messages: MastraDBMessage[];
|
|
39
|
+
}>;
|
|
40
|
+
listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
|
|
55
41
|
saveMessages(args: {
|
|
56
|
-
messages:
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
getThreadsByResourceIdPaginated(args: {
|
|
60
|
-
resourceId: string;
|
|
61
|
-
page?: number;
|
|
62
|
-
perPage?: number;
|
|
63
|
-
}): Promise<PaginationInfo & {
|
|
64
|
-
threads: StorageThreadType[];
|
|
42
|
+
messages: MastraDBMessage[];
|
|
43
|
+
}): Promise<{
|
|
44
|
+
messages: MastraDBMessage[];
|
|
65
45
|
}>;
|
|
46
|
+
listThreadsByResourceId(args: StorageListThreadsByResourceIdInput): Promise<StorageListThreadsByResourceIdOutput>;
|
|
66
47
|
/**
|
|
67
48
|
* Processes messages to include context messages based on withPreviousMessages and withNextMessages
|
|
68
49
|
* @param records - The sorted array of records to process
|
|
@@ -70,24 +51,19 @@ export declare class StoreMemoryLance extends MemoryStorage {
|
|
|
70
51
|
* @returns The processed array with context messages included
|
|
71
52
|
*/
|
|
72
53
|
private processMessagesWithContext;
|
|
73
|
-
getMessagesPaginated(args: StorageGetMessagesArg & {
|
|
74
|
-
format?: 'v1' | 'v2';
|
|
75
|
-
}): Promise<PaginationInfo & {
|
|
76
|
-
messages: MastraMessageV1[] | MastraMessageV2[];
|
|
77
|
-
}>;
|
|
78
54
|
/**
|
|
79
|
-
* Parse message data from LanceDB record format to
|
|
55
|
+
* Parse message data from LanceDB record format to MastraDBMessage format
|
|
80
56
|
*/
|
|
81
57
|
private parseMessageData;
|
|
82
58
|
updateMessages(args: {
|
|
83
|
-
messages: Partial<Omit<
|
|
59
|
+
messages: Partial<Omit<MastraDBMessage, 'createdAt'>> & {
|
|
84
60
|
id: string;
|
|
85
61
|
content?: {
|
|
86
62
|
metadata?: MastraMessageContentV2['metadata'];
|
|
87
63
|
content?: MastraMessageContentV2['content'];
|
|
88
64
|
};
|
|
89
65
|
}[];
|
|
90
|
-
}): Promise<
|
|
66
|
+
}): Promise<MastraDBMessage[]>;
|
|
91
67
|
getResourceById({ resourceId }: {
|
|
92
68
|
resourceId: string;
|
|
93
69
|
}): Promise<StorageResourceType | null>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EAEL,aAAa,EAMd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAG1D,qBAAa,gBAAiB,SAAQ,aAAa;IACjD,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,UAAU,CAAuB;gBAC7B,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE;QAAE,MAAM,EAAE,UAAU,CAAC;QAAC,UAAU,EAAE,oBAAoB,CAAA;KAAE;IAO5F,OAAO,CAAC,SAAS;IAIX,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAyB1F;;;;OAIG;IACG,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAmBjF,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;IA6DxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBrE,OAAO,CAAC,gBAAgB;IAkBX,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAiCpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAqMvF,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IA8DtF,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IAwEhD;;;;;OAKG;IACH,OAAO,CAAC,0BAA0B;IAuElC;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAoBlB,cAAc,CAAC,IAAI,EAAE;QACzB,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GACnD;YACE,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,EAAE,CAAC;KACP,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAgGxB,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAiG5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAyB3F,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;CA+DjC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,EAAwB,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAKvE,qBAAa,oBAAqB,SAAQ,eAAe;IACvD,MAAM,EAAE,UAAU,CAAC;gBACP,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,UAAU,CAAA;KAAE;IAK9C,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAkBxD,SAAS,CAAC,SAAS,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAM7E,OAAO,CAAC,eAAe;IAyCjB,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;IA2CX,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCnE,UAAU,CAAC,EACf,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwEX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAuCpE,MAAM,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwDhG,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA+DzG,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,GAAG,CAAC;CA8ErG"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { Connection } from '@lancedb/lancedb';
|
|
2
|
-
import type { ScoreRowData, ScoringSource } from '@mastra/core/
|
|
2
|
+
import type { SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
3
3
|
import { ScoresStorage } from '@mastra/core/storage';
|
|
4
4
|
import type { PaginationInfo, StoragePagination } from '@mastra/core/storage';
|
|
5
5
|
export declare class StoreScoresLance extends ScoresStorage {
|
|
@@ -7,13 +7,21 @@ export declare class StoreScoresLance extends ScoresStorage {
|
|
|
7
7
|
constructor({ client }: {
|
|
8
8
|
client: Connection;
|
|
9
9
|
});
|
|
10
|
-
saveScore(score:
|
|
10
|
+
saveScore(score: SaveScorePayload): Promise<{
|
|
11
11
|
score: ScoreRowData;
|
|
12
12
|
}>;
|
|
13
13
|
getScoreById({ id }: {
|
|
14
14
|
id: string;
|
|
15
15
|
}): Promise<ScoreRowData | null>;
|
|
16
|
-
|
|
16
|
+
/**
|
|
17
|
+
* LanceDB-specific score row transformation.
|
|
18
|
+
*
|
|
19
|
+
* Note: This implementation does NOT use coreTransformScoreRow because:
|
|
20
|
+
* 1. LanceDB stores schema information in the table itself (requires async fetch)
|
|
21
|
+
* 2. Uses processResultWithTypeConversion utility for LanceDB-specific type handling
|
|
22
|
+
*/
|
|
23
|
+
private transformScoreRow;
|
|
24
|
+
listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
|
|
17
25
|
scorerId: string;
|
|
18
26
|
pagination: StoragePagination;
|
|
19
27
|
entityId?: string;
|
|
@@ -23,14 +31,14 @@ export declare class StoreScoresLance extends ScoresStorage {
|
|
|
23
31
|
pagination: PaginationInfo;
|
|
24
32
|
scores: ScoreRowData[];
|
|
25
33
|
}>;
|
|
26
|
-
|
|
34
|
+
listScoresByRunId({ runId, pagination, }: {
|
|
27
35
|
runId: string;
|
|
28
36
|
pagination: StoragePagination;
|
|
29
37
|
}): Promise<{
|
|
30
38
|
pagination: PaginationInfo;
|
|
31
39
|
scores: ScoreRowData[];
|
|
32
40
|
}>;
|
|
33
|
-
|
|
41
|
+
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
34
42
|
pagination: StoragePagination;
|
|
35
43
|
entityId: string;
|
|
36
44
|
entityType: string;
|
|
@@ -38,5 +46,13 @@ export declare class StoreScoresLance extends ScoresStorage {
|
|
|
38
46
|
pagination: PaginationInfo;
|
|
39
47
|
scores: ScoreRowData[];
|
|
40
48
|
}>;
|
|
49
|
+
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
50
|
+
traceId: string;
|
|
51
|
+
spanId: string;
|
|
52
|
+
pagination: StoragePagination;
|
|
53
|
+
}): Promise<{
|
|
54
|
+
pagination: PaginationInfo;
|
|
55
|
+
scores: ScoreRowData[];
|
|
56
|
+
}>;
|
|
41
57
|
}
|
|
42
58
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAA6B,MAAM,oBAAoB,CAAC;AAEnH,OAAO,EAEL,aAAa,EAId,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAG9E,qBAAa,gBAAiB,SAAQ,aAAa;IACjD,OAAO,CAAC,MAAM,CAAa;gBACf,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,UAAU,CAAA;KAAE;IAKxC,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAoEpE,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAwBxE;;;;;;OAMG;YACW,iBAAiB;IAUzB,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;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAqE7D,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;IAiD7D,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;IAoD7D,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;CAgDpE"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAa,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG1E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,wBAAgB,cAAc,CAAC,SAAS,EAAE,WAAW,GAAG,MAAM,EAAE,
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAa,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG1E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,wBAAgB,cAAc,CAAC,SAAS,EAAE,WAAW,GAAG,MAAM,EAAE,CAO/D;AAED,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,WAAW,EAAE,UAAU,GAAG,IAAI,CA4BzF;AAED,wBAAgB,+BAA+B,CAC7C,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EACtD,WAAW,EAAE,UAAU,GACtB,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAqD7C;AAED,wBAAsB,cAAc,CAAC,EACnC,SAAS,EACT,MAAM,GACP,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,MAAM,EAAE,UAAU,CAAC;CACpB,GAAG,OAAO,CAAC,UAAU,CAAC,CA6CtB"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import type { Connection } from '@lancedb/lancedb';
|
|
2
|
-
import type {
|
|
2
|
+
import type { StorageListWorkflowRunsInput, WorkflowRuns } from '@mastra/core/storage';
|
|
3
3
|
import { WorkflowsStorage } from '@mastra/core/storage';
|
|
4
|
+
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
4
5
|
export declare class StoreWorkflowsLance extends WorkflowsStorage {
|
|
5
6
|
client: Connection;
|
|
6
7
|
constructor({ client }: {
|
|
@@ -11,7 +12,7 @@ export declare class StoreWorkflowsLance extends WorkflowsStorage {
|
|
|
11
12
|
runId: string;
|
|
12
13
|
stepId: string;
|
|
13
14
|
result: StepResult<any, any, any, any>;
|
|
14
|
-
|
|
15
|
+
requestContext: Record<string, any>;
|
|
15
16
|
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
16
17
|
updateWorkflowState({}: {
|
|
17
18
|
workflowName: string;
|
|
@@ -44,14 +45,10 @@ export declare class StoreWorkflowsLance extends WorkflowsStorage {
|
|
|
44
45
|
createdAt: Date;
|
|
45
46
|
updatedAt: Date;
|
|
46
47
|
} | null>;
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
toDate?: Date;
|
|
53
|
-
limit?: number;
|
|
54
|
-
offset?: number;
|
|
55
|
-
}): Promise<WorkflowRuns>;
|
|
48
|
+
deleteWorkflowRunById({ runId, workflowName }: {
|
|
49
|
+
runId: string;
|
|
50
|
+
workflowName: string;
|
|
51
|
+
}): Promise<void>;
|
|
52
|
+
listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
|
|
56
53
|
}
|
|
57
54
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAe,4BAA4B,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpG,OAAO,EAKL,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAuB3E,qBAAa,mBAAoB,SAAQ,gBAAgB;IACvD,MAAM,EAAE,UAAU,CAAC;gBACP,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,UAAU,CAAA;KAAE;IAK9C,qBAAqB,CACnB,EAMC,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,GACA,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAG1D,mBAAmB,CACjB,EAIC,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,GACA,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAIlC,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;IA4CX,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;IAmB9B,kBAAkB,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QAChF,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,GAAG,CAAC;QACd,SAAS,EAAE,IAAI,CAAC;QAChB,SAAS,EAAE,IAAI,CAAC;KACjB,GAAG,IAAI,CAAC;IAyBH,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBtG,gBAAgB,CAAC,IAAI,CAAC,EAAE,4BAA4B,GAAG,OAAO,CAAC,YAAY,CAAC;CAkFnF"}
|
package/dist/storage/index.d.ts
CHANGED
|
@@ -1,37 +1,66 @@
|
|
|
1
1
|
import type { ConnectionOptions } from '@lancedb/lancedb';
|
|
2
2
|
import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
3
|
-
import type {
|
|
4
|
-
import type {
|
|
3
|
+
import type { SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
4
|
+
import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
|
|
5
5
|
import { MastraStorage } from '@mastra/core/storage';
|
|
6
|
-
import type { TABLE_NAMES, PaginationInfo,
|
|
7
|
-
import type { Trace } from '@mastra/core/telemetry';
|
|
6
|
+
import type { TABLE_NAMES, PaginationInfo, StorageColumn, WorkflowRuns, StoragePagination, StorageDomains, StorageResourceType, StorageListWorkflowRunsInput } from '@mastra/core/storage';
|
|
8
7
|
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
8
|
+
export interface LanceStorageOptions {
|
|
9
|
+
/**
|
|
10
|
+
* When true, automatic initialization (table creation/migrations) is disabled.
|
|
11
|
+
* This is useful for CI/CD pipelines where you want to:
|
|
12
|
+
* 1. Run migrations explicitly during deployment (not at runtime)
|
|
13
|
+
* 2. Use different credentials for schema changes vs runtime operations
|
|
14
|
+
*
|
|
15
|
+
* When disableInit is true:
|
|
16
|
+
* - The storage will not automatically create/alter tables on first use
|
|
17
|
+
* - You must call `storage.init()` explicitly in your CI/CD scripts
|
|
18
|
+
*
|
|
19
|
+
* @example
|
|
20
|
+
* // In CI/CD script:
|
|
21
|
+
* const storage = await LanceStorage.create('id', 'name', '/path/to/db', undefined, { disableInit: false });
|
|
22
|
+
* await storage.init(); // Explicitly run migrations
|
|
23
|
+
*
|
|
24
|
+
* // In runtime application:
|
|
25
|
+
* const storage = await LanceStorage.create('id', 'name', '/path/to/db', undefined, { disableInit: true });
|
|
26
|
+
* // No auto-init, tables must already exist
|
|
27
|
+
*/
|
|
28
|
+
disableInit?: boolean;
|
|
29
|
+
}
|
|
9
30
|
export declare class LanceStorage extends MastraStorage {
|
|
10
31
|
stores: StorageDomains;
|
|
11
32
|
private lanceClient;
|
|
12
33
|
/**
|
|
13
34
|
* Creates a new instance of LanceStorage
|
|
35
|
+
* @param id The unique identifier for this storage instance
|
|
36
|
+
* @param name The name for this storage instance
|
|
14
37
|
* @param uri The URI to connect to LanceDB
|
|
15
|
-
* @param
|
|
38
|
+
* @param connectionOptions connection options for LanceDB
|
|
39
|
+
* @param storageOptions storage options including disableInit
|
|
16
40
|
*
|
|
17
41
|
* Usage:
|
|
18
42
|
*
|
|
19
43
|
* Connect to a local database
|
|
20
44
|
* ```ts
|
|
21
|
-
* const store = await LanceStorage.create('/path/to/db');
|
|
45
|
+
* const store = await LanceStorage.create('my-storage-id', 'MyStorage', '/path/to/db');
|
|
22
46
|
* ```
|
|
23
47
|
*
|
|
24
48
|
* Connect to a LanceDB cloud database
|
|
25
49
|
* ```ts
|
|
26
|
-
* const store = await LanceStorage.create('db://host:port');
|
|
50
|
+
* const store = await LanceStorage.create('my-storage-id', 'MyStorage', 'db://host:port');
|
|
27
51
|
* ```
|
|
28
52
|
*
|
|
29
53
|
* Connect to a cloud database
|
|
30
54
|
* ```ts
|
|
31
|
-
* const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
|
|
55
|
+
* const store = await LanceStorage.create('my-storage-id', 'MyStorage', 's3://bucket/db', { storageOptions: { timeout: '60s' } });
|
|
56
|
+
* ```
|
|
57
|
+
*
|
|
58
|
+
* Disable auto-init for runtime (after CI/CD has run migrations)
|
|
59
|
+
* ```ts
|
|
60
|
+
* const store = await LanceStorage.create('my-storage-id', 'MyStorage', '/path/to/db', undefined, { disableInit: true });
|
|
32
61
|
* ```
|
|
33
62
|
*/
|
|
34
|
-
static create(name: string, uri: string,
|
|
63
|
+
static create(id: string, name: string, uri: string, connectionOptions?: ConnectionOptions, storageOptions?: LanceStorageOptions): Promise<LanceStorage>;
|
|
35
64
|
/**
|
|
36
65
|
* @internal
|
|
37
66
|
* Private constructor to enforce using the create factory method
|
|
@@ -67,9 +96,6 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
67
96
|
getThreadById({ threadId }: {
|
|
68
97
|
threadId: string;
|
|
69
98
|
}): Promise<StorageThreadType | null>;
|
|
70
|
-
getThreadsByResourceId({ resourceId }: {
|
|
71
|
-
resourceId: string;
|
|
72
|
-
}): Promise<StorageThreadType[]>;
|
|
73
99
|
/**
|
|
74
100
|
* Saves a thread to the database. This function doesn't overwrite existing threads.
|
|
75
101
|
* @param thread - The thread to save
|
|
@@ -92,6 +118,7 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
92
118
|
hasColumn: boolean;
|
|
93
119
|
createTable: boolean;
|
|
94
120
|
deleteMessages: boolean;
|
|
121
|
+
listScoresBySpan: boolean;
|
|
95
122
|
};
|
|
96
123
|
getResourceById({ resourceId }: {
|
|
97
124
|
resourceId: string;
|
|
@@ -111,85 +138,26 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
111
138
|
* @returns The processed array with context messages included
|
|
112
139
|
*/
|
|
113
140
|
private processMessagesWithContext;
|
|
114
|
-
|
|
115
|
-
format?: 'v1';
|
|
116
|
-
}): Promise<MastraMessageV1[]>;
|
|
117
|
-
getMessages(args: StorageGetMessagesArg & {
|
|
118
|
-
format: 'v2';
|
|
119
|
-
}): Promise<MastraMessageV2[]>;
|
|
120
|
-
getMessagesById({ messageIds, format }: {
|
|
141
|
+
listMessagesById({ messageIds }: {
|
|
121
142
|
messageIds: string[];
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
getMessagesById({ messageIds, format }: {
|
|
125
|
-
messageIds: string[];
|
|
126
|
-
format?: 'v2';
|
|
127
|
-
}): Promise<MastraMessageV2[]>;
|
|
128
|
-
saveMessages(args: {
|
|
129
|
-
messages: MastraMessageV1[];
|
|
130
|
-
format?: undefined | 'v1';
|
|
131
|
-
}): Promise<MastraMessageV1[]>;
|
|
132
|
-
saveMessages(args: {
|
|
133
|
-
messages: MastraMessageV2[];
|
|
134
|
-
format: 'v2';
|
|
135
|
-
}): Promise<MastraMessageV2[]>;
|
|
136
|
-
getThreadsByResourceIdPaginated(args: {
|
|
137
|
-
resourceId: string;
|
|
138
|
-
page: number;
|
|
139
|
-
perPage: number;
|
|
140
|
-
}): Promise<PaginationInfo & {
|
|
141
|
-
threads: StorageThreadType[];
|
|
143
|
+
}): Promise<{
|
|
144
|
+
messages: MastraDBMessage[];
|
|
142
145
|
}>;
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
}): Promise<
|
|
146
|
-
messages:
|
|
146
|
+
saveMessages(args: {
|
|
147
|
+
messages: MastraDBMessage[];
|
|
148
|
+
}): Promise<{
|
|
149
|
+
messages: MastraDBMessage[];
|
|
147
150
|
}>;
|
|
148
151
|
updateMessages(_args: {
|
|
149
|
-
messages: Partial<Omit<
|
|
152
|
+
messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
|
|
150
153
|
id: string;
|
|
151
154
|
content?: {
|
|
152
155
|
metadata?: MastraMessageContentV2['metadata'];
|
|
153
156
|
content?: MastraMessageContentV2['content'];
|
|
154
157
|
};
|
|
155
|
-
}[];
|
|
156
|
-
}): Promise<
|
|
157
|
-
|
|
158
|
-
traceId: string;
|
|
159
|
-
}): Promise<TraceType>;
|
|
160
|
-
getTraces(args: {
|
|
161
|
-
name?: string;
|
|
162
|
-
scope?: string;
|
|
163
|
-
page: number;
|
|
164
|
-
perPage: number;
|
|
165
|
-
attributes?: Record<string, string>;
|
|
166
|
-
}): Promise<Trace[]>;
|
|
167
|
-
getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
|
|
168
|
-
traces: Trace[];
|
|
169
|
-
}>;
|
|
170
|
-
getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
|
|
171
|
-
getEvals(options: {
|
|
172
|
-
agentName?: string;
|
|
173
|
-
type?: 'test' | 'live';
|
|
174
|
-
page?: number;
|
|
175
|
-
perPage?: number;
|
|
176
|
-
fromDate?: Date;
|
|
177
|
-
toDate?: Date;
|
|
178
|
-
dateRange?: {
|
|
179
|
-
start?: Date;
|
|
180
|
-
end?: Date;
|
|
181
|
-
};
|
|
182
|
-
}): Promise<PaginationInfo & {
|
|
183
|
-
evals: EvalRow[];
|
|
184
|
-
}>;
|
|
185
|
-
getWorkflowRuns(args?: {
|
|
186
|
-
namespace?: string;
|
|
187
|
-
workflowName?: string;
|
|
188
|
-
fromDate?: Date;
|
|
189
|
-
toDate?: Date;
|
|
190
|
-
limit?: number;
|
|
191
|
-
offset?: number;
|
|
192
|
-
}): Promise<WorkflowRuns>;
|
|
158
|
+
})[];
|
|
159
|
+
}): Promise<MastraDBMessage[]>;
|
|
160
|
+
listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
|
|
193
161
|
getWorkflowRunById(args: {
|
|
194
162
|
runId: string;
|
|
195
163
|
workflowName?: string;
|
|
@@ -200,12 +168,16 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
200
168
|
createdAt: Date;
|
|
201
169
|
updatedAt: Date;
|
|
202
170
|
} | null>;
|
|
203
|
-
|
|
171
|
+
deleteWorkflowRunById({ runId, workflowName }: {
|
|
172
|
+
runId: string;
|
|
173
|
+
workflowName: string;
|
|
174
|
+
}): Promise<void>;
|
|
175
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
|
|
204
176
|
workflowName: string;
|
|
205
177
|
runId: string;
|
|
206
178
|
stepId: string;
|
|
207
179
|
result: StepResult<any, any, any, any>;
|
|
208
|
-
|
|
180
|
+
requestContext: Record<string, any>;
|
|
209
181
|
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
210
182
|
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
211
183
|
workflowName: string;
|
|
@@ -231,7 +203,7 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
231
203
|
getScoreById({ id: _id }: {
|
|
232
204
|
id: string;
|
|
233
205
|
}): Promise<ScoreRowData | null>;
|
|
234
|
-
|
|
206
|
+
listScoresByScorerId({ scorerId, source, entityId, entityType, pagination, }: {
|
|
235
207
|
scorerId: string;
|
|
236
208
|
pagination: StoragePagination;
|
|
237
209
|
source?: ScoringSource;
|
|
@@ -241,17 +213,17 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
241
213
|
pagination: PaginationInfo;
|
|
242
214
|
scores: ScoreRowData[];
|
|
243
215
|
}>;
|
|
244
|
-
saveScore(
|
|
216
|
+
saveScore(score: SaveScorePayload): Promise<{
|
|
245
217
|
score: ScoreRowData;
|
|
246
218
|
}>;
|
|
247
|
-
|
|
219
|
+
listScoresByRunId({ runId, pagination, }: {
|
|
248
220
|
runId: string;
|
|
249
221
|
pagination: StoragePagination;
|
|
250
222
|
}): Promise<{
|
|
251
223
|
pagination: PaginationInfo;
|
|
252
224
|
scores: ScoreRowData[];
|
|
253
225
|
}>;
|
|
254
|
-
|
|
226
|
+
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
255
227
|
pagination: StoragePagination;
|
|
256
228
|
entityId: string;
|
|
257
229
|
entityType: string;
|
|
@@ -259,5 +231,13 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
259
231
|
pagination: PaginationInfo;
|
|
260
232
|
scores: ScoreRowData[];
|
|
261
233
|
}>;
|
|
234
|
+
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
235
|
+
traceId: string;
|
|
236
|
+
spanId: string;
|
|
237
|
+
pagination: StoragePagination;
|
|
238
|
+
}): Promise<{
|
|
239
|
+
pagination: PaginationInfo;
|
|
240
|
+
scores: ScoreRowData[];
|
|
241
|
+
}>;
|
|
262
242
|
}
|
|
263
243
|
//# 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,EAAc,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAE,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAc,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAwB,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,KAAK,EACV,WAAW,EACX,cAAc,EACd,aAAa,EACb,YAAY,EACZ,iBAAiB,EACjB,cAAc,EACd,mBAAmB,EACnB,4BAA4B,EAC7B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAM3E,MAAM,WAAW,mBAAmB;IAClC;;;;;;;;;;;;;;;;;;OAkBG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,qBAAa,YAAa,SAAQ,aAAa;IAC7C,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,CAAC,WAAW,CAAc;IACjC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;WACiB,MAAM,CACxB,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,EACZ,GAAG,EAAE,MAAM,EACX,iBAAiB,CAAC,EAAE,iBAAiB,EACrC,cAAc,CAAC,EAAE,mBAAmB,GACnC,OAAO,CAAC,YAAY,CAAC;IA0BxB;;;OAGG;IACH,OAAO;IAYD,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;IAIX,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAInE,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,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;IAIrG,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;IAI9G,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,GAAG,CAAC;IAI9F,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAI1F;;;;OAIG;IACG,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,IAAW,QAAQ;;;;;;;MASlB;IAEK,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;IAIhC;;;;;OAKG;IACH,OAAO,CAAC,0BAA0B;IAuE5B,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;IAI7F,cAAc,CAAC,KAAK,EAAE;QAC1B,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,gBAAgB,CAAC,IAAI,CAAC,EAAE,4BAA4B,GAAG,OAAO,CAAC,YAAY,CAAC;IAI5E,kBAAkB,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QAChF,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,GAAG,CAAC;QACd,SAAS,EAAE,IAAI,CAAC;QAChB,SAAS,EAAE,IAAI,CAAC;KACjB,GAAG,IAAI,CAAC;IAIH,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAItG,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,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAIvE,oBAAoB,CAAC,EACzB,QAAQ,EACR,MAAM,EACN,QAAQ,EACR,UAAU,EACV,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,MAAM,CAAC,EAAE,aAAa,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAI7D,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAIpE,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;IAI7D,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;CAGpE"}
|
package/dist/vector/filter.d.ts
CHANGED
|
@@ -29,12 +29,12 @@ export declare class LanceFilterTranslator extends BaseFilterTranslator<LanceVec
|
|
|
29
29
|
*/
|
|
30
30
|
protected getSupportedOperators(): {
|
|
31
31
|
custom: string[];
|
|
32
|
-
logical: import("@mastra/core/vector
|
|
33
|
-
basic: import("@mastra/core/vector
|
|
34
|
-
numeric: import("@mastra/core/vector
|
|
35
|
-
array: import("@mastra/core/vector
|
|
32
|
+
logical: import("@mastra/core/vector").LogicalOperator[];
|
|
33
|
+
basic: import("@mastra/core/vector").BasicOperator[];
|
|
34
|
+
numeric: import("@mastra/core/vector").NumericOperator[];
|
|
35
|
+
array: import("@mastra/core/vector").ArrayOperator[];
|
|
36
36
|
element: "$exists"[];
|
|
37
|
-
regex: import("@mastra/core/vector
|
|
37
|
+
regex: import("@mastra/core/vector").RegexOperator[];
|
|
38
38
|
};
|
|
39
39
|
}
|
|
40
40
|
export {};
|
package/dist/vector/index.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { ConnectionOptions, CreateTableOptions, Table, TableLike } from '@lancedb/lancedb';
|
|
2
|
-
import type { CreateIndexParams, DeleteIndexParams, DeleteVectorParams, DescribeIndexParams, IndexStats, QueryResult, QueryVectorParams, UpdateVectorParams, UpsertVectorParams } from '@mastra/core/vector';
|
|
2
|
+
import type { CreateIndexParams, DeleteIndexParams, DeleteVectorParams, DescribeIndexParams, IndexStats, QueryResult, QueryVectorParams, UpdateVectorParams, UpsertVectorParams, DeleteVectorsParams } from '@mastra/core/vector';
|
|
3
3
|
import { MastraVector } from '@mastra/core/vector';
|
|
4
4
|
import type { LanceVectorFilter } from './filter.js';
|
|
5
5
|
import type { IndexConfig } from './types.js';
|
|
@@ -43,7 +43,9 @@ export declare class LanceVectorStore extends MastraVector<LanceVectorFilter> {
|
|
|
43
43
|
* const store = await LanceVectorStore.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
|
|
44
44
|
* ```
|
|
45
45
|
*/
|
|
46
|
-
static create(uri: string, options?: ConnectionOptions
|
|
46
|
+
static create(uri: string, options?: ConnectionOptions & {
|
|
47
|
+
id: string;
|
|
48
|
+
}): Promise<LanceVectorStore>;
|
|
47
49
|
/**
|
|
48
50
|
* @internal
|
|
49
51
|
* Private constructor to enforce using the create factory method
|
|
@@ -73,13 +75,14 @@ export declare class LanceVectorStore extends MastraVector<LanceVectorFilter> {
|
|
|
73
75
|
*/
|
|
74
76
|
deleteAllTables(): Promise<void>;
|
|
75
77
|
deleteTable(tableName: string): Promise<void>;
|
|
76
|
-
updateVector(
|
|
78
|
+
updateVector(params: UpdateVectorParams<LanceVectorFilter>): Promise<void>;
|
|
77
79
|
deleteVector({ indexName, id }: DeleteVectorParams): Promise<void>;
|
|
78
80
|
/**
|
|
79
81
|
* Converts a flattened object with keys using underscore notation back to a nested object.
|
|
80
82
|
* Example: { name: 'test', details_text: 'test' } → { name: 'test', details: { text: 'test' } }
|
|
81
83
|
*/
|
|
82
84
|
private unflattenObject;
|
|
85
|
+
deleteVectors({ indexName, filter, ids }: DeleteVectorsParams<LanceVectorFilter>): Promise<void>;
|
|
83
86
|
}
|
|
84
87
|
export {};
|
|
85
88
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAc,iBAAiB,EAAE,kBAAkB,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAc,iBAAiB,EAAE,kBAAkB,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAI5G,OAAO,KAAK,EACV,iBAAiB,EACjB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,UAAU,EACV,WAAW,EACX,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACpB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAElD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAE3C,UAAU,sBAAuB,SAAQ,iBAAiB;IACxD,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,UAAU,gBAAiB,SAAQ,WAAW;IAC5C,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,UAAU,uBAAwB,SAAQ,kBAAkB;IAC1D,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,UAAU,sBAAuB,SAAQ,iBAAiB,CAAC,iBAAiB,CAAC;IAC3E,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,qBAAa,gBAAiB,SAAQ,YAAY,CAAC,iBAAiB,CAAC;IACnE,OAAO,CAAC,WAAW,CAAc;IAEjC;;;;;;;;;;;;;;;;;;;;;OAqBG;WACiB,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAkBhH;;;OAGG;IACH,OAAO;IAIP,KAAK;IAMC,KAAK,CAAC,EACV,SAAS,EACT,WAAW,EACX,MAAM,EACN,aAAqB,EACrB,IAAS,EACT,OAAY,EACZ,iBAAyB,GAC1B,EAAE,sBAAsB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAoGlD,OAAO,CAAC,gBAAgB;IA0ClB,MAAM,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,QAAa,EAAE,GAAQ,EAAE,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA4EzG;;;OAGG;IACH,OAAO,CAAC,aAAa;IAYf,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,SAAS,EAC3C,OAAO,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC,GACpC,OAAO,CAAC,KAAK,CAAC;IA+BX,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAwB/B,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IA2BrD;;OAEG;IACG,WAAW,CAAC,EAChB,SAAS,EACT,SAAS,EACT,SAAS,EACT,MAAiB,EACjB,WAAgB,GACjB,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;IAkFnC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAkChC,aAAa,CAAC,EAAE,SAAS,EAAE,EAAE,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC;IAkEtE,WAAW,CAAC,EAAE,SAAS,EAAE,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAgDlE;;OAEG;IACG,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAyBhC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA2B7C,YAAY,CAAC,MAAM,EAAE,kBAAkB,CAAC,iBAAiB,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAsL1E,YAAY,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuExE;;;OAGG;IACH,OAAO,CAAC,eAAe;IAiCjB,aAAa,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,mBAAmB,CAAC,iBAAiB,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;CA6HvG"}
|