@mastra/pg 1.15.0-alpha.0 → 1.15.0-alpha.1
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 +153 -0
- package/dist/docs/SKILL.md +1 -1
- package/dist/docs/assets/SOURCE_MAP.json +1 -1
- package/dist/index.cjs +183 -51
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +183 -51
- package/dist/index.js.map +1 -1
- package/dist/storage/domains/datasets/index.d.ts +6 -7
- package/dist/storage/domains/datasets/index.d.ts.map +1 -1
- package/dist/storage/domains/experiments/index.d.ts +9 -5
- package/dist/storage/domains/experiments/index.d.ts.map +1 -1
- package/dist/storage/domains/scorer-definitions/index.d.ts.map +1 -1
- package/dist/storage/domains/scores/index.d.ts +9 -5
- package/dist/storage/domains/scores/index.d.ts.map +1 -1
- package/dist/storage/domains/utils.d.ts +20 -1
- package/dist/storage/domains/utils.d.ts.map +1 -1
- package/dist/storage/test-utils.d.ts.map +1 -1
- package/package.json +4 -4
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { DatasetsStorage } from '@mastra/core/storage';
|
|
2
|
-
import type { DatasetRecord, DatasetItem, DatasetItemRow, DatasetVersion, CreateDatasetInput, UpdateDatasetInput, AddDatasetItemInput, UpdateDatasetItemInput, ListDatasetsInput, ListDatasetsOutput, ListDatasetItemsInput, ListDatasetItemsOutput, ListDatasetVersionsInput, ListDatasetVersionsOutput, BatchInsertItemsInput, BatchDeleteItemsInput, CreateIndexOptions } from '@mastra/core/storage';
|
|
2
|
+
import type { DatasetRecord, DatasetItem, DatasetItemRow, DatasetVersion, CreateDatasetInput, UpdateDatasetInput, AddDatasetItemInput, UpdateDatasetItemInput, DeleteDatasetItemInput, ListDatasetsInput, ListDatasetsOutput, ListDatasetItemsInput, ListDatasetItemsOutput, ListDatasetVersionsInput, ListDatasetVersionsOutput, BatchInsertItemsInput, BatchDeleteItemsInput, CreateIndexOptions, DatasetTenancyFilters } from '@mastra/core/storage';
|
|
3
3
|
import type { PgDomainConfig } from '../../db/index.js';
|
|
4
4
|
export declare class DatasetsPG extends DatasetsStorage {
|
|
5
5
|
#private;
|
|
@@ -15,20 +15,19 @@ export declare class DatasetsPG extends DatasetsStorage {
|
|
|
15
15
|
private transformItemRowFull;
|
|
16
16
|
private transformDatasetVersionRow;
|
|
17
17
|
createDataset(input: CreateDatasetInput): Promise<DatasetRecord>;
|
|
18
|
-
getDatasetById({ id }: {
|
|
18
|
+
getDatasetById({ id, filters, }: {
|
|
19
19
|
id: string;
|
|
20
|
+
filters?: DatasetTenancyFilters;
|
|
20
21
|
}): Promise<DatasetRecord | null>;
|
|
21
22
|
protected _doUpdateDataset(args: UpdateDatasetInput): Promise<DatasetRecord>;
|
|
22
|
-
deleteDataset({ id }: {
|
|
23
|
+
deleteDataset({ id, filters }: {
|
|
23
24
|
id: string;
|
|
25
|
+
filters?: DatasetTenancyFilters;
|
|
24
26
|
}): Promise<void>;
|
|
25
27
|
listDatasets(args: ListDatasetsInput): Promise<ListDatasetsOutput>;
|
|
26
28
|
protected _doAddItem(args: AddDatasetItemInput): Promise<DatasetItem>;
|
|
27
29
|
protected _doUpdateItem(args: UpdateDatasetItemInput): Promise<DatasetItem>;
|
|
28
|
-
protected _doDeleteItem({ id, datasetId }:
|
|
29
|
-
id: string;
|
|
30
|
-
datasetId: string;
|
|
31
|
-
}): Promise<void>;
|
|
30
|
+
protected _doDeleteItem({ id, datasetId }: DeleteDatasetItemInput): Promise<void>;
|
|
32
31
|
protected _doBatchInsertItems(input: BatchInsertItemsInput): Promise<DatasetItem[]>;
|
|
33
32
|
protected _doBatchDeleteItems(input: BatchDeleteItemsInput): Promise<void>;
|
|
34
33
|
getItemById(args: {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/datasets/index.ts"],"names":[],"mappings":"AACA,OAAO,EAYL,eAAe,EAKhB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,cAAc,EACd,cAAc,EACd,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EACjB,kBAAkB,EAClB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,qBAAqB,EACrB,qBAAqB,EACrB,kBAAkB,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/datasets/index.ts"],"names":[],"mappings":"AACA,OAAO,EAYL,eAAe,EAKhB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,cAAc,EACd,cAAc,EACd,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,iBAAiB,EACjB,kBAAkB,EAClB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,qBAAqB,EACrB,qBAAqB,EACrB,kBAAkB,EAElB,qBAAqB,EACtB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAQ/C,qBAAa,UAAW,SAAQ,eAAe;;IAM7C,MAAM,CAAC,QAAQ,CAAC,cAAc,kFAA0E;gBAE5F,MAAM,EAAE,cAAc;IASlC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAgB5C,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAsC3B,0BAA0B,IAAI,kBAAkB,EAAE;IA2C5C,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAWrC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAa1C,OAAO,CAAC,mBAAmB;IAuB3B,OAAO,CAAC,gBAAgB;IAmBxB,OAAO,CAAC,oBAAoB;IAqB5B,OAAO,CAAC,0BAA0B;IAW5B,aAAa,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,aAAa,CAAC;IA4DhE,cAAc,CAAC,EACnB,EAAE,EACF,OAAO,GACR,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,CAAC,EAAE,qBAAqB,CAAC;KACjC,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;cAmBjB,gBAAgB,CAAC,IAAI,EAAE,kBAAkB,GAAG,OAAO,CAAC,aAAa,CAAC;IAqG5E,aAAa,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,qBAAqB,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAiE9F,YAAY,CAAC,IAAI,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;cAsFxD,UAAU,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAAC,WAAW,CAAC;cAoF3D,aAAa,CAAC,IAAI,EAAE,sBAAsB,GAAG,OAAO,CAAC,WAAW,CAAC;cAuHjE,aAAa,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;cAkFvE,mBAAmB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;cAqGzE,mBAAmB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8F1E,WAAW,CAAC,IAAI,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA8BvF,iBAAiB,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAoBzG,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAoBzD,SAAS,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAwFvE,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAwBjF,mBAAmB,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IA+CxF,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;CAK3C"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { ExperimentsStorage } from '@mastra/core/storage';
|
|
2
|
-
import type { Experiment, ExperimentResult, ExperimentReviewCounts, CreateExperimentInput, UpdateExperimentInput, AddExperimentResultInput, UpdateExperimentResultInput, ListExperimentsInput, ListExperimentsOutput, ListExperimentResultsInput, ListExperimentResultsOutput, CreateIndexOptions, PruneOptions, PruneResult, RetentionTablesDescriptor, TableRetentionPolicy } from '@mastra/core/storage';
|
|
2
|
+
import type { Experiment, ExperimentResult, ExperimentReviewCounts, ExperimentTenancyFilters, CreateExperimentInput, UpdateExperimentInput, AddExperimentResultInput, UpdateExperimentResultInput, ListExperimentsInput, ListExperimentsOutput, ListExperimentResultsInput, ListExperimentResultsOutput, CreateIndexOptions, PruneOptions, PruneResult, RetentionTablesDescriptor, TableRetentionPolicy } from '@mastra/core/storage';
|
|
3
3
|
import type { PgDomainConfig } from '../../db/index.js';
|
|
4
4
|
export declare class ExperimentsPG extends ExperimentsStorage {
|
|
5
5
|
#private;
|
|
@@ -44,21 +44,25 @@ export declare class ExperimentsPG extends ExperimentsStorage {
|
|
|
44
44
|
private transformExperimentResultRow;
|
|
45
45
|
createExperiment(input: CreateExperimentInput): Promise<Experiment>;
|
|
46
46
|
updateExperiment(input: UpdateExperimentInput): Promise<Experiment>;
|
|
47
|
-
getExperimentById({ id }: {
|
|
47
|
+
getExperimentById({ id, filters, }: {
|
|
48
48
|
id: string;
|
|
49
|
+
filters?: ExperimentTenancyFilters;
|
|
49
50
|
}): Promise<Experiment | null>;
|
|
50
51
|
listExperiments(args: ListExperimentsInput): Promise<ListExperimentsOutput>;
|
|
51
|
-
deleteExperiment({ id }: {
|
|
52
|
+
deleteExperiment({ id, filters }: {
|
|
52
53
|
id: string;
|
|
54
|
+
filters?: ExperimentTenancyFilters;
|
|
53
55
|
}): Promise<void>;
|
|
54
56
|
addExperimentResult(input: AddExperimentResultInput): Promise<ExperimentResult>;
|
|
55
57
|
updateExperimentResult(input: UpdateExperimentResultInput): Promise<ExperimentResult>;
|
|
56
|
-
getExperimentResultById({ id }: {
|
|
58
|
+
getExperimentResultById({ id, filters, }: {
|
|
57
59
|
id: string;
|
|
60
|
+
filters?: ExperimentTenancyFilters;
|
|
58
61
|
}): Promise<ExperimentResult | null>;
|
|
59
62
|
listExperimentResults(args: ListExperimentResultsInput): Promise<ListExperimentResultsOutput>;
|
|
60
|
-
deleteExperimentResults({ experimentId }: {
|
|
63
|
+
deleteExperimentResults({ experimentId, filters, }: {
|
|
61
64
|
experimentId: string;
|
|
65
|
+
filters?: ExperimentTenancyFilters;
|
|
62
66
|
}): Promise<void>;
|
|
63
67
|
getReviewSummary(): Promise<ExperimentReviewCounts[]>;
|
|
64
68
|
dangerouslyClearAll(): Promise<void>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/experiments/index.ts"],"names":[],"mappings":"AACA,OAAO,EAOL,kBAAkB,EAKnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,UAAU,EACV,gBAAgB,EAChB,sBAAsB,EACtB,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,2BAA2B,EAC3B,oBAAoB,EACpB,qBAAqB,EACrB,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,EAElB,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAM/C,qBAAa,aAAc,SAAQ,kBAAkB;;IAMnD,MAAM,CAAC,QAAQ,CAAC,cAAc,+DAA0D;IAExF;;;;;;;OAOG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAEjE;gBAEU,MAAM,EAAE,cAAc;IASlC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAe5C,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB3B;;;;;;;;OAQG;YACW,sBAAsB;IAgBpC;;;;;;;;;OASG;IACG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IA6C3G,0BAA0B,IAAI,kBAAkB,EAAE;IAwB5C,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAWrC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAa1C,OAAO,CAAC,sBAAsB;IAyB9B,OAAO,CAAC,4BAA4B;IAyB9B,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC;IAkEnE,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC;IAiFnE,iBAAiB,CAAC,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/experiments/index.ts"],"names":[],"mappings":"AACA,OAAO,EAOL,kBAAkB,EAKnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,UAAU,EACV,gBAAgB,EAChB,sBAAsB,EACtB,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,2BAA2B,EAC3B,oBAAoB,EACpB,qBAAqB,EACrB,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,EAElB,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAM/C,qBAAa,aAAc,SAAQ,kBAAkB;;IAMnD,MAAM,CAAC,QAAQ,CAAC,cAAc,+DAA0D;IAExF;;;;;;;OAOG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAEjE;gBAEU,MAAM,EAAE,cAAc;IASlC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAe5C,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB3B;;;;;;;;OAQG;YACW,sBAAsB;IAgBpC;;;;;;;;;OASG;IACG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IA6C3G,0BAA0B,IAAI,kBAAkB,EAAE;IAwB5C,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAWrC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAa1C,OAAO,CAAC,sBAAsB;IAyB9B,OAAO,CAAC,4BAA4B;IAyB9B,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC;IAkEnE,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC;IAiFnE,iBAAiB,CAAC,EACtB,EAAE,EACF,OAAO,GACR,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,CAAC,EAAE,wBAAwB,CAAC;KACpC,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAmBxB,eAAe,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAqF3E,gBAAgB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,wBAAwB,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkCpG,mBAAmB,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA8D/E,sBAAsB,CAAC,KAAK,EAAE,2BAA2B,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAgErF,uBAAuB,CAAC,EAC5B,EAAE,EACF,OAAO,GACR,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,CAAC,EAAE,wBAAwB,CAAC;KACpC,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAmB9B,qBAAqB,CAAC,IAAI,EAAE,0BAA0B,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAuE7F,uBAAuB,CAAC,EAC5B,YAAY,EACZ,OAAO,GACR,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,wBAAwB,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAkCX,gBAAgB,IAAI,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAmCrD,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;CAI3C"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scorer-definitions/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,wBAAwB,EAOzB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,2BAA2B,EAC3B,kCAAkC,EAClC,kCAAkC,EAClC,iCAAiC,EACjC,kCAAkC,EAClC,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,uBAAuB,EACvB,kCAAkC,EAClC,iCAAiC,EACjC,kCAAkC,EACnC,MAAM,iDAAiD,CAAC;AAGzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAc/C,qBAAa,mBAAoB,SAAQ,wBAAwB;;IAM/D,MAAM,CAAC,QAAQ,CAAC,cAAc,8EAAyE;gBAE3F,MAAM,EAAE,cAAc;IAWlC;;;OAGG;IACH,MAAM,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,GAAG,kBAAkB,EAAE;IAWtE;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAyBlD,0BAA0B,IAAI,kBAAkB,EAAE;IAK5C,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAarC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scorer-definitions/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,wBAAwB,EAOzB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,2BAA2B,EAC3B,kCAAkC,EAClC,kCAAkC,EAClC,iCAAiC,EACjC,kCAAkC,EAClC,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,uBAAuB,EACvB,kCAAkC,EAClC,iCAAiC,EACjC,kCAAkC,EACnC,MAAM,iDAAiD,CAAC;AAGzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAc/C,qBAAa,mBAAoB,SAAQ,wBAAwB;;IAM/D,MAAM,CAAC,QAAQ,CAAC,cAAc,8EAAyE;gBAE3F,MAAM,EAAE,cAAc;IAWlC;;;OAGG;IACH,MAAM,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,GAAG,kBAAkB,EAAE;IAWtE;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAyBlD,0BAA0B,IAAI,kBAAkB,EAAE;IAK5C,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAarC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAmBrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAapC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IASpC,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,2BAA2B,GAAG,IAAI,CAAC;IAwBhE,MAAM,CAAC,KAAK,EAAE;QAAE,gBAAgB,EAAE,kCAAkC,CAAA;KAAE,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAsF7G,MAAM,CAAC,KAAK,EAAE,kCAAkC,GAAG,OAAO,CAAC,2BAA2B,CAAC;IAiFvF,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBjC,IAAI,CAAC,IAAI,CAAC,EAAE,iCAAiC,GAAG,OAAO,CAAC,kCAAkC,CAAC;IAuH3F,aAAa,CAAC,KAAK,EAAE,kCAAkC,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAqD1F,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,GAAG,IAAI,CAAC;IA2B/D,kBAAkB,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,GAAG,IAAI,CAAC;IA8B9G,gBAAgB,CAAC,kBAAkB,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,GAAG,IAAI,CAAC;IA8BrF,YAAY,CAAC,KAAK,EAAE,iCAAiC,GAAG,OAAO,CAAC,kCAAkC,CAAC;IAgFnG,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBxC,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBzD,aAAa,CAAC,kBAAkB,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IA6BhE,OAAO,CAAC,cAAc;IActB,OAAO,CAAC,eAAe;CAkBxB"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { ListScoresResponse, SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
2
|
-
import type {
|
|
2
|
+
import type { CreateIndexOptions, PruneOptions, PruneResult, RetentionTablesDescriptor, ScoreTenancyFilters, StoragePagination, TableRetentionPolicy } from '@mastra/core/storage';
|
|
3
3
|
import { ScoresStorage } from '@mastra/core/storage';
|
|
4
4
|
import type { PgDomainConfig } from '../../db/index.js';
|
|
5
5
|
export declare class ScoresPG extends ScoresStorage {
|
|
@@ -51,29 +51,33 @@ export declare class ScoresPG extends ScoresStorage {
|
|
|
51
51
|
getScoreById({ id }: {
|
|
52
52
|
id: string;
|
|
53
53
|
}): Promise<ScoreRowData | null>;
|
|
54
|
-
listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
|
|
54
|
+
listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, filters, }: {
|
|
55
55
|
scorerId: string;
|
|
56
56
|
pagination: StoragePagination;
|
|
57
57
|
entityId?: string;
|
|
58
58
|
entityType?: string;
|
|
59
59
|
source?: ScoringSource;
|
|
60
|
+
filters?: ScoreTenancyFilters;
|
|
60
61
|
}): Promise<ListScoresResponse>;
|
|
61
62
|
saveScore(score: SaveScorePayload): Promise<{
|
|
62
63
|
score: ScoreRowData;
|
|
63
64
|
}>;
|
|
64
|
-
listScoresByRunId({ runId, pagination, }: {
|
|
65
|
+
listScoresByRunId({ runId, pagination, filters, }: {
|
|
65
66
|
runId: string;
|
|
66
67
|
pagination: StoragePagination;
|
|
68
|
+
filters?: ScoreTenancyFilters;
|
|
67
69
|
}): Promise<ListScoresResponse>;
|
|
68
|
-
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
70
|
+
listScoresByEntityId({ entityId, entityType, pagination, filters, }: {
|
|
69
71
|
pagination: StoragePagination;
|
|
70
72
|
entityId: string;
|
|
71
73
|
entityType: string;
|
|
74
|
+
filters?: ScoreTenancyFilters;
|
|
72
75
|
}): Promise<ListScoresResponse>;
|
|
73
|
-
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
76
|
+
listScoresBySpan({ traceId, spanId, pagination, filters, }: {
|
|
74
77
|
traceId: string;
|
|
75
78
|
spanId: string;
|
|
76
79
|
pagination: StoragePagination;
|
|
80
|
+
filters?: ScoreTenancyFilters;
|
|
77
81
|
}): Promise<ListScoresResponse>;
|
|
78
82
|
}
|
|
79
83
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAE5G,OAAO,KAAK,EACV,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAE5G,OAAO,KAAK,EACV,kBAAkB,EAClB,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,mBAAmB,EACnB,iBAAiB,EAEjB,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAIL,aAAa,EAId,MAAM,sBAAsB,CAAC;AAG9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AA+C/C,qBAAa,QAAS,SAAQ,aAAa;;IAMzC,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,8BAA4B;IAE1D;;;OAGG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAEjE;gBAEU,MAAM,EAAE,cAAc;IAU5B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAY3B;;;;;;;;OAQG;YACW,sBAAsB;IAgBpC;;;OAGG;IACH,MAAM,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,GAAG,kBAAkB,EAAE;IAUtE;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAuBlD;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAKlD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C,iFAAiF;IAC3E,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAUrG,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAoBlE,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,UAAU,EACV,MAAM,EACN,OAAO,GACR,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;QACvB,OAAO,CAAC,EAAE,mBAAmB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAwEzB,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAuEpE,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,EACV,OAAO,GACR,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;QAC9B,OAAO,CAAC,EAAE,mBAAmB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsDzB,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,UAAU,EACV,OAAO,GACR,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,mBAAmB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAuDzB,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,EACV,OAAO,GACR,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;QAC9B,OAAO,CAAC,EAAE,mBAAmB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA+ChC"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
|
|
1
|
+
import type { DatasetTenancyFilters, ExperimentTenancyFilters, StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
|
|
2
2
|
export declare function getSchemaName(schema?: string): string | undefined;
|
|
3
3
|
export declare function getTableName({ indexName, schemaName }: {
|
|
4
4
|
indexName: string;
|
|
@@ -36,4 +36,23 @@ export declare function transformFromSqlRow<T>({ tableName, sqlRow, }: {
|
|
|
36
36
|
tableName: TABLE_NAMES;
|
|
37
37
|
sqlRow: Record<string, any>;
|
|
38
38
|
}): T;
|
|
39
|
+
/**
|
|
40
|
+
* Tenancy scope shape shared by domains that carry `organizationId` / `projectId`
|
|
41
|
+
* columns (datasets, experiments, ...). Aliases the core tenancy filter types so
|
|
42
|
+
* this helper stays in sync if the core shape ever changes.
|
|
43
|
+
*/
|
|
44
|
+
export type TenancyScope = DatasetTenancyFilters | ExperimentTenancyFilters;
|
|
45
|
+
/**
|
|
46
|
+
* Build additional `AND "col" = $N` conditions for a tenancy read-scope filter.
|
|
47
|
+
* Uses double-quoted PG identifiers and continues numbering from `startIndex`.
|
|
48
|
+
* Returns empty arrays and unchanged `nextIndex` when no filters apply.
|
|
49
|
+
*
|
|
50
|
+
* Shared across domains (datasets, experiments, ...) so tenancy predicates stay
|
|
51
|
+
* consistent and cross-tenant reads/deletes do not leak.
|
|
52
|
+
*/
|
|
53
|
+
export declare function tenancyWhere(filters: TenancyScope | undefined, startIndex: number): {
|
|
54
|
+
conditions: string[];
|
|
55
|
+
params: any[];
|
|
56
|
+
nextIndex: number;
|
|
57
|
+
};
|
|
39
58
|
//# sourceMappingURL=utils.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAIxH,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,sBAE5C;AAED,wBAAgB,YAAY,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,UAKjG;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC5B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,GACtC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,GAAG,EAAE,CAAA;CAAE,CA6B9B;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,GAAG,CASvE;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,EACrC,SAAS,EACT,MAAM,GACP,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC7B,GAAG,CAAC,CA+BJ;AAED;;;;GAIG;AACH,MAAM,MAAM,YAAY,GAAG,qBAAqB,GAAG,wBAAwB,CAAC;AAE5E;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAC1B,OAAO,EAAE,YAAY,GAAG,SAAS,EACjC,UAAU,EAAE,MAAM,GACjB;IAAE,UAAU,EAAE,MAAM,EAAE,CAAC;IAAC,MAAM,EAAE,GAAG,EAAE,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAa5D"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"test-utils.d.ts","sourceRoot":"","sources":["../../src/storage/test-utils.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAK5D,eAAO,MAAM,WAAW,EAAE,mBAOF,CAAC;AAEzB,eAAO,MAAM,gBAAgB,QAA0K,CAAC;AAExM,wBAAgB,OAAO,
|
|
1
|
+
{"version":3,"file":"test-utils.d.ts","sourceRoot":"","sources":["../../src/storage/test-utils.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAK5D,eAAO,MAAM,WAAW,EAAE,mBAOF,CAAC;AAEzB,eAAO,MAAM,gBAAgB,QAA0K,CAAC;AAExM,wBAAgB,OAAO,SA2xCtB"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/pg",
|
|
3
|
-
"version": "1.15.0-alpha.
|
|
3
|
+
"version": "1.15.0-alpha.1",
|
|
4
4
|
"description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -35,10 +35,10 @@
|
|
|
35
35
|
"tsx": "^4.22.4",
|
|
36
36
|
"typescript": "^6.0.3",
|
|
37
37
|
"vitest": "4.1.8",
|
|
38
|
-
"@internal/lint": "0.0.110",
|
|
39
38
|
"@internal/storage-test-utils": "0.0.106",
|
|
40
|
-
"@
|
|
41
|
-
"@
|
|
39
|
+
"@mastra/core": "1.49.0-alpha.5",
|
|
40
|
+
"@internal/lint": "0.0.110",
|
|
41
|
+
"@internal/types-builder": "0.0.85"
|
|
42
42
|
},
|
|
43
43
|
"peerDependencies": {
|
|
44
44
|
"@mastra/core": ">=1.42.1-0 <2.0.0-0"
|