@mastra/lance 0.0.0-scorers-ui-refactored-20250916094952 → 0.0.0-scorers-logs-20251208101616
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 +703 -3
- package/README.md +61 -4
- package/dist/index.cjs +641 -738
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +642 -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 +4 -11
- package/dist/storage/domains/workflows/index.d.ts.map +1 -1
- package/dist/storage/index.d.ts +64 -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,6 @@ export declare class StoreWorkflowsLance extends WorkflowsStorage {
|
|
|
44
45
|
createdAt: Date;
|
|
45
46
|
updatedAt: Date;
|
|
46
47
|
} | null>;
|
|
47
|
-
|
|
48
|
-
namespace?: string;
|
|
49
|
-
resourceId?: string;
|
|
50
|
-
workflowName?: string;
|
|
51
|
-
fromDate?: Date;
|
|
52
|
-
toDate?: Date;
|
|
53
|
-
limit?: number;
|
|
54
|
-
offset?: number;
|
|
55
|
-
}): Promise<WorkflowRuns>;
|
|
48
|
+
listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
|
|
56
49
|
}
|
|
57
50
|
//# 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,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 }: {
|
|
121
|
-
messageIds: string[];
|
|
122
|
-
format: 'v1';
|
|
123
|
-
}): Promise<MastraMessageV1[]>;
|
|
124
|
-
getMessagesById({ messageIds, format }: {
|
|
141
|
+
listMessagesById({ messageIds }: {
|
|
125
142
|
messageIds: string[];
|
|
126
|
-
|
|
127
|
-
|
|
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,12 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
200
168
|
createdAt: Date;
|
|
201
169
|
updatedAt: Date;
|
|
202
170
|
} | null>;
|
|
203
|
-
updateWorkflowResults({ workflowName, runId, stepId, result,
|
|
171
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
|
|
204
172
|
workflowName: string;
|
|
205
173
|
runId: string;
|
|
206
174
|
stepId: string;
|
|
207
175
|
result: StepResult<any, any, any, any>;
|
|
208
|
-
|
|
176
|
+
requestContext: Record<string, any>;
|
|
209
177
|
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
210
178
|
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
211
179
|
workflowName: string;
|
|
@@ -231,7 +199,7 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
231
199
|
getScoreById({ id: _id }: {
|
|
232
200
|
id: string;
|
|
233
201
|
}): Promise<ScoreRowData | null>;
|
|
234
|
-
|
|
202
|
+
listScoresByScorerId({ scorerId, source, entityId, entityType, pagination, }: {
|
|
235
203
|
scorerId: string;
|
|
236
204
|
pagination: StoragePagination;
|
|
237
205
|
source?: ScoringSource;
|
|
@@ -241,17 +209,17 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
241
209
|
pagination: PaginationInfo;
|
|
242
210
|
scores: ScoreRowData[];
|
|
243
211
|
}>;
|
|
244
|
-
saveScore(
|
|
212
|
+
saveScore(score: SaveScorePayload): Promise<{
|
|
245
213
|
score: ScoreRowData;
|
|
246
214
|
}>;
|
|
247
|
-
|
|
215
|
+
listScoresByRunId({ runId, pagination, }: {
|
|
248
216
|
runId: string;
|
|
249
217
|
pagination: StoragePagination;
|
|
250
218
|
}): Promise<{
|
|
251
219
|
pagination: PaginationInfo;
|
|
252
220
|
scores: ScoreRowData[];
|
|
253
221
|
}>;
|
|
254
|
-
|
|
222
|
+
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
255
223
|
pagination: StoragePagination;
|
|
256
224
|
entityId: string;
|
|
257
225
|
entityType: string;
|
|
@@ -259,5 +227,13 @@ export declare class LanceStorage extends MastraStorage {
|
|
|
259
227
|
pagination: PaginationInfo;
|
|
260
228
|
scores: ScoreRowData[];
|
|
261
229
|
}>;
|
|
230
|
+
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
231
|
+
traceId: string;
|
|
232
|
+
spanId: string;
|
|
233
|
+
pagination: StoragePagination;
|
|
234
|
+
}): Promise<{
|
|
235
|
+
pagination: PaginationInfo;
|
|
236
|
+
scores: ScoreRowData[];
|
|
237
|
+
}>;
|
|
262
238
|
}
|
|
263
239
|
//# 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,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"}
|