@mastra/lance 0.0.0-http-transporter-20250702160118 → 0.0.0-issue-7087-20250909202326

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.
Files changed (46) hide show
  1. package/CHANGELOG.md +370 -3
  2. package/LICENSE.md +11 -42
  3. package/README.md +3 -3
  4. package/dist/index.cjs +1932 -848
  5. package/dist/index.cjs.map +1 -0
  6. package/dist/index.d.ts +3 -2
  7. package/dist/index.d.ts.map +1 -0
  8. package/dist/index.js +1933 -849
  9. package/dist/index.js.map +1 -0
  10. package/dist/storage/domains/legacy-evals/index.d.ts +25 -0
  11. package/dist/storage/domains/legacy-evals/index.d.ts.map +1 -0
  12. package/dist/storage/domains/memory/index.d.ts +103 -0
  13. package/dist/storage/domains/memory/index.d.ts.map +1 -0
  14. package/dist/storage/domains/operations/index.d.ts +40 -0
  15. package/dist/storage/domains/operations/index.d.ts.map +1 -0
  16. package/dist/storage/domains/scores/index.d.ts +42 -0
  17. package/dist/storage/domains/scores/index.d.ts.map +1 -0
  18. package/dist/storage/domains/traces/index.d.ts +34 -0
  19. package/dist/storage/domains/traces/index.d.ts.map +1 -0
  20. package/dist/storage/domains/utils.d.ts +10 -0
  21. package/dist/storage/domains/utils.d.ts.map +1 -0
  22. package/dist/storage/domains/workflows/index.d.ts +56 -0
  23. package/dist/storage/domains/workflows/index.d.ts.map +1 -0
  24. package/dist/storage/index.d.ts +262 -0
  25. package/dist/storage/index.d.ts.map +1 -0
  26. package/dist/vector/filter.d.ts +41 -0
  27. package/dist/vector/filter.d.ts.map +1 -0
  28. package/dist/vector/index.d.ts +85 -0
  29. package/dist/vector/index.d.ts.map +1 -0
  30. package/dist/vector/types.d.ts +15 -0
  31. package/dist/vector/types.d.ts.map +1 -0
  32. package/package.json +24 -10
  33. package/dist/_tsup-dts-rollup.d.cts +0 -409
  34. package/dist/_tsup-dts-rollup.d.ts +0 -409
  35. package/dist/index.d.cts +0 -2
  36. package/eslint.config.js +0 -6
  37. package/src/index.ts +0 -2
  38. package/src/storage/index.test.ts +0 -1336
  39. package/src/storage/index.ts +0 -1447
  40. package/src/vector/filter.test.ts +0 -295
  41. package/src/vector/filter.ts +0 -443
  42. package/src/vector/index.test.ts +0 -1493
  43. package/src/vector/index.ts +0 -941
  44. package/src/vector/types.ts +0 -16
  45. package/tsconfig.json +0 -5
  46. package/vitest.config.ts +0 -11
@@ -0,0 +1,25 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import { LegacyEvalsStorage } from '@mastra/core/storage';
3
+ import type { EvalRow, PaginationInfo } from '@mastra/core/storage';
4
+ export declare class StoreLegacyEvalsLance extends LegacyEvalsStorage {
5
+ private client;
6
+ constructor({ client }: {
7
+ client: Connection;
8
+ });
9
+ getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
10
+ getEvals(options: {
11
+ agentName?: string;
12
+ type?: 'test' | 'live';
13
+ page?: number;
14
+ perPage?: number;
15
+ fromDate?: Date;
16
+ toDate?: Date;
17
+ dateRange?: {
18
+ start?: Date;
19
+ end?: Date;
20
+ };
21
+ }): Promise<PaginationInfo & {
22
+ evals: EvalRow[];
23
+ }>;
24
+ }
25
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/legacy-evals/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,EAAE,kBAAkB,EAAe,MAAM,sBAAsB,CAAC;AACvE,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEpE,qBAAa,qBAAsB,SAAQ,kBAAkB;IAC3D,OAAO,CAAC,MAAM,CAAa;gBACf,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,UAAU,CAAA;KAAE;IAKxC,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IA4ClF,QAAQ,CAAC,OAAO,EAAE;QACtB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;QACvB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,IAAI,CAAC;QAChB,MAAM,CAAC,EAAE,IAAI,CAAC;QACd,SAAS,CAAC,EAAE;YAAE,KAAK,CAAC,EAAE,IAAI,CAAC;YAAC,GAAG,CAAC,EAAE,IAAI,CAAA;SAAE,CAAC;KAC1C,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,KAAK,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC;CA2FnD"}
@@ -0,0 +1,103 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import type { MastraMessageContentV2 } from '@mastra/core/agent';
3
+ import type { MastraMessageV1, MastraMessageV2, StorageThreadType } from '@mastra/core/memory';
4
+ import { MemoryStorage } from '@mastra/core/storage';
5
+ import type { PaginationInfo, StorageGetMessagesArg, StorageResourceType } from '@mastra/core/storage';
6
+ import type { StoreOperationsLance } from '../operations/index.js';
7
+ export declare class StoreMemoryLance extends MemoryStorage {
8
+ private client;
9
+ private operations;
10
+ constructor({ client, operations }: {
11
+ client: Connection;
12
+ operations: StoreOperationsLance;
13
+ });
14
+ getThreadById({ threadId }: {
15
+ threadId: string;
16
+ }): Promise<StorageThreadType | null>;
17
+ getThreadsByResourceId({ resourceId }: {
18
+ resourceId: string;
19
+ }): Promise<StorageThreadType[]>;
20
+ /**
21
+ * Saves a thread to the database. This function doesn't overwrite existing threads.
22
+ * @param thread - The thread to save
23
+ * @returns The saved thread
24
+ */
25
+ saveThread({ thread }: {
26
+ thread: StorageThreadType;
27
+ }): Promise<StorageThreadType>;
28
+ updateThread({ id, title, metadata, }: {
29
+ id: string;
30
+ title: string;
31
+ metadata: Record<string, unknown>;
32
+ }): Promise<StorageThreadType>;
33
+ deleteThread({ threadId }: {
34
+ threadId: string;
35
+ }): Promise<void>;
36
+ private normalizeMessage;
37
+ getMessages(args: StorageGetMessagesArg & {
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, }: {
48
+ messageIds: string[];
49
+ format?: 'v2';
50
+ }): Promise<MastraMessageV2[]>;
51
+ saveMessages(args: {
52
+ messages: MastraMessageV1[];
53
+ format?: undefined | 'v1';
54
+ }): Promise<MastraMessageV1[]>;
55
+ saveMessages(args: {
56
+ messages: MastraMessageV2[];
57
+ format: 'v2';
58
+ }): Promise<MastraMessageV2[]>;
59
+ getThreadsByResourceIdPaginated(args: {
60
+ resourceId: string;
61
+ page?: number;
62
+ perPage?: number;
63
+ }): Promise<PaginationInfo & {
64
+ threads: StorageThreadType[];
65
+ }>;
66
+ /**
67
+ * Processes messages to include context messages based on withPreviousMessages and withNextMessages
68
+ * @param records - The sorted array of records to process
69
+ * @param include - The array of include specifications with context parameters
70
+ * @returns The processed array with context messages included
71
+ */
72
+ private processMessagesWithContext;
73
+ getMessagesPaginated(args: StorageGetMessagesArg & {
74
+ format?: 'v1' | 'v2';
75
+ }): Promise<PaginationInfo & {
76
+ messages: MastraMessageV1[] | MastraMessageV2[];
77
+ }>;
78
+ /**
79
+ * Parse message data from LanceDB record format to MastraMessageV2 format
80
+ */
81
+ private parseMessageData;
82
+ updateMessages(args: {
83
+ messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
84
+ id: string;
85
+ content?: {
86
+ metadata?: MastraMessageContentV2['metadata'];
87
+ content?: MastraMessageContentV2['content'];
88
+ };
89
+ }[];
90
+ }): Promise<MastraMessageV2[]>;
91
+ getResourceById({ resourceId }: {
92
+ resourceId: string;
93
+ }): Promise<StorageResourceType | null>;
94
+ saveResource({ resource }: {
95
+ resource: StorageResourceType;
96
+ }): Promise<StorageResourceType>;
97
+ updateResource({ resourceId, workingMemory, metadata, }: {
98
+ resourceId: string;
99
+ workingMemory?: string;
100
+ metadata?: Record<string, unknown>;
101
+ }): Promise<StorageResourceType>;
102
+ }
103
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAE,eAAe,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EACL,aAAa,EAKd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,cAAc,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AACvG,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;IAMtF,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAyBpF,sBAAsB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IAuBlG;;;;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,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;IA6EvF,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;IAsCxB,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;IAkE7F,+BAA+B,CAAC,IAAI,EAAE;QAC1C,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,OAAO,EAAE,iBAAiB,EAAE,CAAA;KAAE,CAAC;IA2C9D;;;;;OAKG;IACH,OAAO,CAAC,0BAA0B;IAuE5B,oBAAoB,CACxB,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,CAAC,EAAE,IAAI,GAAG,IAAI,CAAA;KAAE,GACrD,OAAO,CAAC,cAAc,GAAG;QAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,eAAe,EAAE,CAAA;KAAE,CAAC;IAqKhF;;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"}
@@ -0,0 +1,40 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import { StoreOperations } from '@mastra/core/storage';
3
+ import type { TABLE_NAMES, StorageColumn } from '@mastra/core/storage';
4
+ export declare class StoreOperationsLance extends StoreOperations {
5
+ client: Connection;
6
+ constructor({ client }: {
7
+ client: Connection;
8
+ });
9
+ protected getDefaultValue(type: StorageColumn['type']): string;
10
+ hasColumn(tableName: TABLE_NAMES, columnName: string): Promise<boolean>;
11
+ private translateSchema;
12
+ createTable({ tableName, schema, }: {
13
+ tableName: TABLE_NAMES;
14
+ schema: Record<string, StorageColumn>;
15
+ }): Promise<void>;
16
+ dropTable({ tableName }: {
17
+ tableName: TABLE_NAMES;
18
+ }): Promise<void>;
19
+ alterTable({ tableName, schema, ifNotExists, }: {
20
+ tableName: string;
21
+ schema: Record<string, StorageColumn>;
22
+ ifNotExists: string[];
23
+ }): Promise<void>;
24
+ clearTable({ tableName }: {
25
+ tableName: TABLE_NAMES;
26
+ }): Promise<void>;
27
+ insert({ tableName, record }: {
28
+ tableName: string;
29
+ record: Record<string, any>;
30
+ }): Promise<void>;
31
+ batchInsert({ tableName, records }: {
32
+ tableName: string;
33
+ records: Record<string, any>[];
34
+ }): Promise<void>;
35
+ load({ tableName, keys }: {
36
+ tableName: TABLE_NAMES;
37
+ keys: Record<string, any>;
38
+ }): Promise<any>;
39
+ }
40
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,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;IAyDhG,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;IAiEzG,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"}
@@ -0,0 +1,42 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/scores';
3
+ import { ScoresStorage } from '@mastra/core/storage';
4
+ import type { PaginationInfo, StoragePagination } from '@mastra/core/storage';
5
+ export declare class StoreScoresLance extends ScoresStorage {
6
+ private client;
7
+ constructor({ client }: {
8
+ client: Connection;
9
+ });
10
+ saveScore(score: ScoreRowData): Promise<{
11
+ score: ScoreRowData;
12
+ }>;
13
+ getScoreById({ id }: {
14
+ id: string;
15
+ }): Promise<ScoreRowData | null>;
16
+ getScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
17
+ scorerId: string;
18
+ pagination: StoragePagination;
19
+ entityId?: string;
20
+ entityType?: string;
21
+ source?: ScoringSource;
22
+ }): Promise<{
23
+ pagination: PaginationInfo;
24
+ scores: ScoreRowData[];
25
+ }>;
26
+ getScoresByRunId({ runId, pagination, }: {
27
+ runId: string;
28
+ pagination: StoragePagination;
29
+ }): Promise<{
30
+ pagination: PaginationInfo;
31
+ scores: ScoreRowData[];
32
+ }>;
33
+ getScoresByEntityId({ entityId, entityType, pagination, }: {
34
+ pagination: StoragePagination;
35
+ entityId: string;
36
+ entityType: string;
37
+ }): Promise<{
38
+ pagination: PaginationInfo;
39
+ scores: ScoreRowData[];
40
+ }>;
41
+ }
42
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,aAAa,EAAiB,MAAM,sBAAsB,CAAC;AACpE,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,YAAY,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IA0ChE,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAyBlE,mBAAmB,CAAC,EACxB,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;IAwD7D,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;IAqC7D,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;CA0CpE"}
@@ -0,0 +1,34 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import type { TraceType } from '@mastra/core/memory';
3
+ import { TracesStorage } from '@mastra/core/storage';
4
+ import type { PaginationInfo, StorageGetTracesPaginatedArg } from '@mastra/core/storage';
5
+ import type { Trace } from '@mastra/core/telemetry';
6
+ import type { StoreOperationsLance } from '../operations/index.js';
7
+ export declare class StoreTracesLance extends TracesStorage {
8
+ private client;
9
+ private operations;
10
+ constructor({ client, operations }: {
11
+ client: Connection;
12
+ operations: StoreOperationsLance;
13
+ });
14
+ saveTrace({ trace }: {
15
+ trace: TraceType;
16
+ }): Promise<TraceType>;
17
+ getTraceById({ traceId }: {
18
+ traceId: string;
19
+ }): Promise<TraceType>;
20
+ getTraces({ name, scope, page, perPage, attributes, }: {
21
+ name?: string;
22
+ scope?: string;
23
+ page: number;
24
+ perPage: number;
25
+ attributes?: Record<string, string>;
26
+ }): Promise<Trace[]>;
27
+ getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
28
+ traces: Trace[];
29
+ }>;
30
+ batchTraceInsert({ records }: {
31
+ records: Record<string, any>[];
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/traces/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAgB,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,KAAK,EAAE,cAAc,EAAE,4BAA4B,EAAE,MAAM,sBAAsB,CAAC;AACzF,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAE1D,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;IAMtF,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,SAAS,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,CAAC;IAyB9D,YAAY,CAAC,EAAE,OAAO,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,CAAC;IAkBlE,SAAS,CAAC,EACd,IAAI,EACJ,KAAK,EACL,IAAQ,EACR,OAAY,EACZ,UAAU,GACX,EAAE;QACD,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACrC,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAoDd,kBAAkB,CAAC,IAAI,EAAE,4BAA4B,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,MAAM,EAAE,KAAK,EAAE,CAAA;KAAE,CAAC;IAgFrG,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAOvF"}
@@ -0,0 +1,10 @@
1
+ import type { Connection, SchemaLike } from '@lancedb/lancedb';
2
+ import type { TABLE_NAMES } from '@mastra/core/storage';
3
+ export declare function getPrimaryKeys(tableName: TABLE_NAMES): string[];
4
+ export declare function validateKeyTypes(keys: Record<string, any>, tableSchema: SchemaLike): void;
5
+ export declare function processResultWithTypeConversion(rawResult: Record<string, any> | Record<string, any>[], tableSchema: SchemaLike): Record<string, any> | Record<string, any>[];
6
+ export declare function getTableSchema({ tableName, client, }: {
7
+ tableName: TABLE_NAMES;
8
+ client: Connection;
9
+ }): Promise<SchemaLike>;
10
+ //# sourceMappingURL=utils.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,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,CAS/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,CAuD7C;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"}
@@ -0,0 +1,56 @@
1
+ import type { Connection } from '@lancedb/lancedb';
2
+ import type { StepResult, WorkflowRunState, WorkflowRuns } from '@mastra/core';
3
+ import { WorkflowsStorage } from '@mastra/core/storage';
4
+ export declare class StoreWorkflowsLance extends WorkflowsStorage {
5
+ client: Connection;
6
+ constructor({ client }: {
7
+ client: Connection;
8
+ });
9
+ updateWorkflowResults({}: {
10
+ workflowName: string;
11
+ runId: string;
12
+ stepId: string;
13
+ result: StepResult<any, any, any, any>;
14
+ runtimeContext: Record<string, any>;
15
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
16
+ updateWorkflowState({}: {
17
+ workflowName: string;
18
+ runId: string;
19
+ opts: {
20
+ status: string;
21
+ result?: StepResult<any, any, any, any>;
22
+ error?: string;
23
+ suspendedPaths?: Record<string, number[]>;
24
+ waitingPaths?: Record<string, number[]>;
25
+ };
26
+ }): Promise<WorkflowRunState | undefined>;
27
+ persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
28
+ workflowName: string;
29
+ runId: string;
30
+ snapshot: WorkflowRunState;
31
+ }): Promise<void>;
32
+ loadWorkflowSnapshot({ workflowName, runId, }: {
33
+ workflowName: string;
34
+ runId: string;
35
+ }): Promise<WorkflowRunState | null>;
36
+ getWorkflowRunById(args: {
37
+ runId: string;
38
+ workflowName?: string;
39
+ }): Promise<{
40
+ workflowName: string;
41
+ runId: string;
42
+ snapshot: any;
43
+ createdAt: Date;
44
+ updatedAt: Date;
45
+ } | null>;
46
+ getWorkflowRuns(args?: {
47
+ namespace?: string;
48
+ resourceId?: string;
49
+ workflowName?: string;
50
+ fromDate?: Date;
51
+ toDate?: Date;
52
+ limit?: number;
53
+ offset?: number;
54
+ }): Promise<WorkflowRuns>;
55
+ }
56
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAG/E,OAAO,EAAuC,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAuB7F,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,QAAQ,GACT,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,gBAAgB,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAyCX,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,eAAe,CAAC,IAAI,CAAC,EAAE;QAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,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;KACjB,GAAG,OAAO,CAAC,YAAY,CAAC;CA4D1B"}
@@ -0,0 +1,262 @@
1
+ import type { ConnectionOptions } from '@lancedb/lancedb';
2
+ import type { MastraMessageContentV2 } from '@mastra/core/agent';
3
+ import type { MastraMessageV1, MastraMessageV2, StorageThreadType, TraceType } from '@mastra/core/memory';
4
+ import type { ScoreRowData, ScoringSource } from '@mastra/core/scores';
5
+ import { MastraStorage } from '@mastra/core/storage';
6
+ import type { TABLE_NAMES, PaginationInfo, StorageGetMessagesArg, StorageColumn, EvalRow, WorkflowRuns, StoragePagination, StorageDomains, StorageGetTracesPaginatedArg, StorageResourceType } from '@mastra/core/storage';
7
+ import type { Trace } from '@mastra/core/telemetry';
8
+ import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
9
+ export declare class LanceStorage extends MastraStorage {
10
+ stores: StorageDomains;
11
+ private lanceClient;
12
+ /**
13
+ * Creates a new instance of LanceStorage
14
+ * @param uri The URI to connect to LanceDB
15
+ * @param options connection options
16
+ *
17
+ * Usage:
18
+ *
19
+ * Connect to a local database
20
+ * ```ts
21
+ * const store = await LanceStorage.create('/path/to/db');
22
+ * ```
23
+ *
24
+ * Connect to a LanceDB cloud database
25
+ * ```ts
26
+ * const store = await LanceStorage.create('db://host:port');
27
+ * ```
28
+ *
29
+ * Connect to a cloud database
30
+ * ```ts
31
+ * const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
32
+ * ```
33
+ */
34
+ static create(name: string, uri: string, options?: ConnectionOptions): Promise<LanceStorage>;
35
+ /**
36
+ * @internal
37
+ * Private constructor to enforce using the create factory method
38
+ */
39
+ private constructor();
40
+ createTable({ tableName, schema, }: {
41
+ tableName: TABLE_NAMES;
42
+ schema: Record<string, StorageColumn>;
43
+ }): Promise<void>;
44
+ dropTable({ tableName }: {
45
+ tableName: TABLE_NAMES;
46
+ }): Promise<void>;
47
+ alterTable({ tableName, schema, ifNotExists, }: {
48
+ tableName: TABLE_NAMES;
49
+ schema: Record<string, StorageColumn>;
50
+ ifNotExists: string[];
51
+ }): Promise<void>;
52
+ clearTable({ tableName }: {
53
+ tableName: TABLE_NAMES;
54
+ }): Promise<void>;
55
+ insert({ tableName, record }: {
56
+ tableName: TABLE_NAMES;
57
+ record: Record<string, any>;
58
+ }): Promise<void>;
59
+ batchInsert({ tableName, records }: {
60
+ tableName: TABLE_NAMES;
61
+ records: Record<string, any>[];
62
+ }): Promise<void>;
63
+ load({ tableName, keys }: {
64
+ tableName: TABLE_NAMES;
65
+ keys: Record<string, any>;
66
+ }): Promise<any>;
67
+ getThreadById({ threadId }: {
68
+ threadId: string;
69
+ }): Promise<StorageThreadType | null>;
70
+ getThreadsByResourceId({ resourceId }: {
71
+ resourceId: string;
72
+ }): Promise<StorageThreadType[]>;
73
+ /**
74
+ * Saves a thread to the database. This function doesn't overwrite existing threads.
75
+ * @param thread - The thread to save
76
+ * @returns The saved thread
77
+ */
78
+ saveThread({ thread }: {
79
+ thread: StorageThreadType;
80
+ }): Promise<StorageThreadType>;
81
+ updateThread({ id, title, metadata, }: {
82
+ id: string;
83
+ title: string;
84
+ metadata: Record<string, unknown>;
85
+ }): Promise<StorageThreadType>;
86
+ deleteThread({ threadId }: {
87
+ threadId: string;
88
+ }): Promise<void>;
89
+ get supports(): {
90
+ selectByIncludeResourceScope: boolean;
91
+ resourceWorkingMemory: boolean;
92
+ hasColumn: boolean;
93
+ createTable: boolean;
94
+ deleteMessages: boolean;
95
+ };
96
+ getResourceById({ resourceId }: {
97
+ resourceId: string;
98
+ }): Promise<StorageResourceType | null>;
99
+ saveResource({ resource }: {
100
+ resource: StorageResourceType;
101
+ }): Promise<StorageResourceType>;
102
+ updateResource({ resourceId, workingMemory, metadata, }: {
103
+ resourceId: string;
104
+ workingMemory?: string;
105
+ metadata?: Record<string, unknown>;
106
+ }): Promise<StorageResourceType>;
107
+ /**
108
+ * Processes messages to include context messages based on withPreviousMessages and withNextMessages
109
+ * @param records - The sorted array of records to process
110
+ * @param include - The array of include specifications with context parameters
111
+ * @returns The processed array with context messages included
112
+ */
113
+ private processMessagesWithContext;
114
+ getMessages(args: StorageGetMessagesArg & {
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 }: {
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[];
142
+ }>;
143
+ getMessagesPaginated(args: StorageGetMessagesArg & {
144
+ format?: 'v1' | 'v2';
145
+ }): Promise<PaginationInfo & {
146
+ messages: MastraMessageV1[] | MastraMessageV2[];
147
+ }>;
148
+ updateMessages(_args: {
149
+ messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
150
+ id: string;
151
+ content?: {
152
+ metadata?: MastraMessageContentV2['metadata'];
153
+ content?: MastraMessageContentV2['content'];
154
+ };
155
+ }[];
156
+ }): Promise<MastraMessageV2[]>;
157
+ getTraceById(args: {
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>;
193
+ getWorkflowRunById(args: {
194
+ runId: string;
195
+ workflowName?: string;
196
+ }): Promise<{
197
+ workflowName: string;
198
+ runId: string;
199
+ snapshot: any;
200
+ createdAt: Date;
201
+ updatedAt: Date;
202
+ } | null>;
203
+ updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext, }: {
204
+ workflowName: string;
205
+ runId: string;
206
+ stepId: string;
207
+ result: StepResult<any, any, any, any>;
208
+ runtimeContext: Record<string, any>;
209
+ }): Promise<Record<string, StepResult<any, any, any, any>>>;
210
+ updateWorkflowState({ workflowName, runId, opts, }: {
211
+ workflowName: string;
212
+ runId: string;
213
+ opts: {
214
+ status: string;
215
+ result?: StepResult<any, any, any, any>;
216
+ error?: string;
217
+ suspendedPaths?: Record<string, number[]>;
218
+ waitingPaths?: Record<string, number[]>;
219
+ };
220
+ }): Promise<WorkflowRunState | undefined>;
221
+ persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
222
+ workflowName: string;
223
+ runId: string;
224
+ snapshot: WorkflowRunState;
225
+ }): Promise<void>;
226
+ loadWorkflowSnapshot({ workflowName, runId, }: {
227
+ workflowName: string;
228
+ runId: string;
229
+ }): Promise<WorkflowRunState | null>;
230
+ getScoreById({ id: _id }: {
231
+ id: string;
232
+ }): Promise<ScoreRowData | null>;
233
+ getScoresByScorerId({ scorerId, source, entityId, entityType, pagination, }: {
234
+ scorerId: string;
235
+ pagination: StoragePagination;
236
+ source?: ScoringSource;
237
+ entityId?: string;
238
+ entityType?: string;
239
+ }): Promise<{
240
+ pagination: PaginationInfo;
241
+ scores: ScoreRowData[];
242
+ }>;
243
+ saveScore(_score: ScoreRowData): Promise<{
244
+ score: ScoreRowData;
245
+ }>;
246
+ getScoresByRunId({ runId, pagination, }: {
247
+ runId: string;
248
+ pagination: StoragePagination;
249
+ }): Promise<{
250
+ pagination: PaginationInfo;
251
+ scores: ScoreRowData[];
252
+ }>;
253
+ getScoresByEntityId({ entityId, entityType, pagination, }: {
254
+ pagination: StoragePagination;
255
+ entityId: string;
256
+ entityType: string;
257
+ }): Promise<{
258
+ pagination: PaginationInfo;
259
+ scores: ScoreRowData[];
260
+ }>;
261
+ }
262
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAc,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAE,eAAe,EAAE,eAAe,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAC1G,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,KAAK,EACV,WAAW,EACX,cAAc,EACd,qBAAqB,EACrB,aAAa,EACb,OAAO,EACP,YAAY,EACZ,iBAAiB,EACjB,cAAc,EACd,4BAA4B,EAC5B,mBAAmB,EACpB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAQ3E,qBAAa,YAAa,SAAQ,aAAa;IAC7C,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,CAAC,WAAW,CAAc;IACjC;;;;;;;;;;;;;;;;;;;;;OAqBG;WACiB,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC;IA4BzG;;;OAGG;IACH,OAAO;IAcD,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;IAIpF,sBAAsB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IAIlG;;;;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;;;;;;MAQlB;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;IAuErB,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;IAW5G,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,+BAA+B,CAAC,IAAI,EAAE;QAC1C,UAAU,EAAE,MAAM,CAAC;QACnB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,OAAO,EAAE,iBAAiB,EAAE,CAAA;KAAE,CAAC;IAIxD,oBAAoB,CACxB,IAAI,EAAE,qBAAqB,GAAG;QAAE,MAAM,CAAC,EAAE,IAAI,GAAG,IAAI,CAAA;KAAE,GACrD,OAAO,CAAC,cAAc,GAAG;QAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,eAAe,EAAE,CAAA;KAAE,CAAC;IAI1E,cAAc,CAAC,KAAK,EAAE;QAC1B,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;IAIxB,YAAY,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,CAAC;IAI3D,SAAS,CAAC,IAAI,EAAE;QACpB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACrC,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAId,kBAAkB,CAAC,IAAI,EAAE,4BAA4B,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,MAAM,EAAE,KAAK,EAAE,CAAA;KAAE,CAAC;IAIrG,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAIlF,QAAQ,CAAC,OAAO,EAAE;QACtB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;QACvB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,IAAI,CAAC;QAChB,MAAM,CAAC,EAAE,IAAI,CAAC;QACd,SAAS,CAAC,EAAE;YAAE,KAAK,CAAC,EAAE,IAAI,CAAC;YAAC,GAAG,CAAC,EAAE,IAAI,CAAA;SAAE,CAAC;KAC1C,GAAG,OAAO,CAAC,cAAc,GAAG;QAAE,KAAK,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC;IAI5C,eAAe,CAAC,IAAI,CAAC,EAAE;QAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,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;KACjB,GAAG,OAAO,CAAC,YAAY,CAAC;IAInB,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,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,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAIvE,mBAAmB,CAAC,EACxB,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,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAIjE,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;CAGpE"}