@mastra/clickhouse 1.0.0-beta.5 → 1.0.0-beta.7
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 +238 -0
- package/README.md +3 -1
- package/dist/index.cjs +963 -383
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.ts +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +964 -384
- package/dist/index.js.map +1 -1
- package/dist/storage/{domains/operations → db}/index.d.ts +34 -3
- package/dist/storage/db/index.d.ts.map +1 -0
- package/dist/storage/db/utils.d.ts.map +1 -0
- package/dist/storage/domains/memory/index.d.ts +6 -6
- package/dist/storage/domains/memory/index.d.ts.map +1 -1
- package/dist/storage/domains/observability/index.d.ts +28 -0
- package/dist/storage/domains/observability/index.d.ts.map +1 -0
- package/dist/storage/domains/scores/index.d.ts +11 -24
- package/dist/storage/domains/scores/index.d.ts.map +1 -1
- package/dist/storage/domains/workflows/index.d.ts +12 -17
- package/dist/storage/domains/workflows/index.d.ts.map +1 -1
- package/dist/storage/index.d.ts +78 -173
- package/dist/storage/index.d.ts.map +1 -1
- package/package.json +3 -4
- package/dist/storage/domains/operations/index.d.ts.map +0 -1
- package/dist/storage/domains/utils.d.ts.map +0 -1
- /package/dist/storage/{domains → db}/utils.d.ts +0 -0
|
@@ -1,8 +1,39 @@
|
|
|
1
1
|
import type { ClickHouseClient } from '@clickhouse/client';
|
|
2
|
-
import {
|
|
2
|
+
import { MastraBase } from '@mastra/core/base';
|
|
3
3
|
import type { StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
|
|
4
|
-
import type { ClickhouseConfig } from '
|
|
5
|
-
|
|
4
|
+
import type { ClickhouseConfig } from './utils.js';
|
|
5
|
+
/**
|
|
6
|
+
* Configuration for standalone domain usage.
|
|
7
|
+
* Accepts either:
|
|
8
|
+
* 1. An existing ClickHouse client with optional ttl config
|
|
9
|
+
* 2. Config to create a new client internally
|
|
10
|
+
*/
|
|
11
|
+
export type ClickhouseDomainConfig = ClickhouseDomainClientConfig | ClickhouseDomainRestConfig;
|
|
12
|
+
/**
|
|
13
|
+
* Pass an existing ClickHouse client
|
|
14
|
+
*/
|
|
15
|
+
export interface ClickhouseDomainClientConfig {
|
|
16
|
+
client: ClickHouseClient;
|
|
17
|
+
ttl?: ClickhouseConfig['ttl'];
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Pass config to create a new ClickHouse client internally
|
|
21
|
+
*/
|
|
22
|
+
export interface ClickhouseDomainRestConfig {
|
|
23
|
+
url: string;
|
|
24
|
+
username: string;
|
|
25
|
+
password: string;
|
|
26
|
+
ttl?: ClickhouseConfig['ttl'];
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Resolves ClickhouseDomainConfig to a ClickHouse client and ttl config.
|
|
30
|
+
* Handles creating a new client if config is provided.
|
|
31
|
+
*/
|
|
32
|
+
export declare function resolveClickhouseConfig(config: ClickhouseDomainConfig): {
|
|
33
|
+
client: ClickHouseClient;
|
|
34
|
+
ttl?: ClickhouseConfig['ttl'];
|
|
35
|
+
};
|
|
36
|
+
export declare class ClickhouseDB extends MastraBase {
|
|
6
37
|
protected ttl: ClickhouseConfig['ttl'];
|
|
7
38
|
protected client: ClickHouseClient;
|
|
8
39
|
constructor({ client, ttl }: {
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/storage/db/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAU/C,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAGhD;;;;;GAKG;AACH,MAAM,MAAM,sBAAsB,GAAG,4BAA4B,GAAG,0BAA0B,CAAC;AAE/F;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C,MAAM,EAAE,gBAAgB,CAAC;IACzB,GAAG,CAAC,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;CAC/B;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,sBAAsB,GAAG;IACvE,MAAM,EAAE,gBAAgB,CAAC;IACzB,GAAG,CAAC,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;CAC/B,CAoBA;AAED,qBAAa,YAAa,SAAQ,UAAU;IAC1C,SAAS,CAAC,GAAG,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACvC,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBACvB,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAAE,MAAM,EAAE,gBAAgB,CAAC;QAAC,GAAG,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAA;KAAE;IAQjF,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAShE,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAoBnD,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;IAmFX,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;IA0CX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyBpE,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAMnE,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;IAqCrG,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;IAmC9G,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;CA6DhH"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/db/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAYtF,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAUrD,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,CAS9D,CAAC;AAEF,MAAM,MAAM,YAAY,GACpB,YAAY,GACZ,aAAa,GACb,aAAa,GACb,QAAQ,GACR,QAAQ,GACR,MAAM,GACN,KAAK,GACL,MAAM,GACN,OAAO,GACP,SAAS,GACT,MAAM,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG;IAC7B,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE;SACH,QAAQ,IAAI,WAAW,CAAC,CAAC,EAAE;YAC1B,GAAG,CAAC,EAAE;gBAAE,QAAQ,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,YAAY,CAAC;gBAAC,MAAM,CAAC,EAAE,MAAM,CAAA;aAAE,CAAC;YAChE,OAAO,CAAC,EAAE,OAAO,CAAC;iBACf,SAAS,IAAI,MAAM,CAAC,OAAO,aAAa,CAAC,CAAC,QAAQ,CAAC,GAAG;oBACrD,QAAQ,EAAE,MAAM,CAAC;oBACjB,IAAI,EAAE,YAAY,CAAC;oBACnB,MAAM,CAAC,EAAE,MAAM,CAAC;iBACjB;aACF,CAAC,CAAC;SACJ;KACF,CAAC;CACH,CAAC;AAQF,wBAAgB,YAAY,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,CAiC3C;AAED,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAEjD"}
|
|
@@ -3,14 +3,14 @@ import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
|
3
3
|
import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
|
|
4
4
|
import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput } from '@mastra/core/storage';
|
|
5
5
|
import { MemoryStorage } from '@mastra/core/storage';
|
|
6
|
-
import type {
|
|
6
|
+
import type { ClickhouseDomainConfig } from '../../db/index.js';
|
|
7
7
|
export declare class MemoryStorageClickhouse extends MemoryStorage {
|
|
8
|
+
#private;
|
|
8
9
|
protected client: ClickHouseClient;
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
});
|
|
10
|
+
constructor(config: ClickhouseDomainConfig);
|
|
11
|
+
init(): Promise<void>;
|
|
12
|
+
dangerouslyClearAll(): Promise<void>;
|
|
13
|
+
deleteMessages(messageIds: string[]): Promise<void>;
|
|
14
14
|
listMessagesById({ messageIds }: {
|
|
15
15
|
messageIds: string[];
|
|
16
16
|
}): Promise<{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,aAAa,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,aAAa,EAOd,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAiCvD,qBAAa,uBAAwB,SAAQ,aAAa;;IACxD,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBAEvB,MAAM,EAAE,sBAAsB;IAOpC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAYrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAMpC,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyC5C,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IA2DpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAuUvF,YAAY,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAsL7F,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAkDpF,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAqCjF,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;IAuDxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCxD,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IA+G1C,cAAc,CAAC,IAAI,EAAE;QACzB,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,QAAQ,CAAC,EAAE,MAAM,CAAC;YAClB,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;IAqWxB,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IA6C5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAmC3F,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;CAuEjC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import type { ClickHouseClient } from '@clickhouse/client';
|
|
2
|
+
import { ObservabilityStorage } from '@mastra/core/storage';
|
|
3
|
+
import type { SpanRecord, ListTracesArgs, PaginationInfo, TracingStorageStrategy, UpdateSpanArgs, BatchDeleteTracesArgs, BatchUpdateSpansArgs, BatchCreateSpansArgs, CreateSpanArgs, GetSpanArgs, GetSpanResponse, GetRootSpanArgs, GetRootSpanResponse, GetTraceArgs, GetTraceResponse } from '@mastra/core/storage';
|
|
4
|
+
import type { ClickhouseDomainConfig } from '../../db/index.js';
|
|
5
|
+
export declare class ObservabilityStorageClickhouse extends ObservabilityStorage {
|
|
6
|
+
#private;
|
|
7
|
+
protected client: ClickHouseClient;
|
|
8
|
+
constructor(config: ClickhouseDomainConfig);
|
|
9
|
+
init(): Promise<void>;
|
|
10
|
+
dangerouslyClearAll(): Promise<void>;
|
|
11
|
+
get tracingStrategy(): {
|
|
12
|
+
preferred: TracingStorageStrategy;
|
|
13
|
+
supported: TracingStorageStrategy[];
|
|
14
|
+
};
|
|
15
|
+
createSpan(args: CreateSpanArgs): Promise<void>;
|
|
16
|
+
getSpan(args: GetSpanArgs): Promise<GetSpanResponse | null>;
|
|
17
|
+
getRootSpan(args: GetRootSpanArgs): Promise<GetRootSpanResponse | null>;
|
|
18
|
+
getTrace(args: GetTraceArgs): Promise<GetTraceResponse | null>;
|
|
19
|
+
updateSpan(args: UpdateSpanArgs): Promise<void>;
|
|
20
|
+
listTraces(args: ListTracesArgs): Promise<{
|
|
21
|
+
pagination: PaginationInfo;
|
|
22
|
+
spans: SpanRecord[];
|
|
23
|
+
}>;
|
|
24
|
+
batchCreateSpans(args: BatchCreateSpansArgs): Promise<void>;
|
|
25
|
+
batchUpdateSpans(args: BatchUpdateSpansArgs): Promise<void>;
|
|
26
|
+
batchDeleteTraces(args: BatchDeleteTracesArgs): Promise<void>;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/observability/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EAGL,oBAAoB,EAIrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,UAAU,EACV,cAAc,EACd,cAAc,EACd,sBAAsB,EACtB,cAAc,EACd,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,WAAW,EACX,eAAe,EACf,eAAe,EACf,mBAAmB,EACnB,YAAY,EACZ,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAGvD,qBAAa,8BAA+B,SAAQ,oBAAoB;;IACtE,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBAGvB,MAAM,EAAE,sBAAsB;IAOpC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAIrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C,IAAoB,eAAe,IAAI;QACrC,SAAS,EAAE,sBAAsB,CAAC;QAClC,SAAS,EAAE,sBAAsB,EAAE,CAAC;KACrC,CAQA;IAEK,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IA+B/C,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IA6C3D,WAAW,CAAC,IAAI,EAAE,eAAe,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IA6CvE,QAAQ,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IA2C9D,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IA0D/C,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IAqQ9F,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IA0B3D,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuD3D,iBAAiB,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CAmBpE"}
|
|
@@ -1,15 +1,14 @@
|
|
|
1
1
|
import type { ClickHouseClient } from '@clickhouse/client';
|
|
2
|
-
import type { SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
2
|
+
import type { ListScoresResponse, SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
3
3
|
import { ScoresStorage } from '@mastra/core/storage';
|
|
4
|
-
import type {
|
|
5
|
-
import type {
|
|
4
|
+
import type { StoragePagination } from '@mastra/core/storage';
|
|
5
|
+
import type { ClickhouseDomainConfig } from '../../db/index.js';
|
|
6
6
|
export declare class ScoresStorageClickhouse extends ScoresStorage {
|
|
7
|
+
#private;
|
|
7
8
|
protected client: ClickHouseClient;
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
operations: StoreOperationsClickhouse;
|
|
12
|
-
});
|
|
9
|
+
constructor(config: ClickhouseDomainConfig);
|
|
10
|
+
init(): Promise<void>;
|
|
11
|
+
dangerouslyClearAll(): Promise<void>;
|
|
13
12
|
/**
|
|
14
13
|
* ClickHouse-specific score row transformation.
|
|
15
14
|
* Converts timestamps to Date objects and filters out '_null_' values.
|
|
@@ -24,35 +23,23 @@ export declare class ScoresStorageClickhouse extends ScoresStorage {
|
|
|
24
23
|
listScoresByRunId({ runId, pagination, }: {
|
|
25
24
|
runId: string;
|
|
26
25
|
pagination: StoragePagination;
|
|
27
|
-
}): Promise<
|
|
28
|
-
pagination: PaginationInfo;
|
|
29
|
-
scores: ScoreRowData[];
|
|
30
|
-
}>;
|
|
26
|
+
}): Promise<ListScoresResponse>;
|
|
31
27
|
listScoresByScorerId({ scorerId, entityId, entityType, source, pagination, }: {
|
|
32
28
|
scorerId: string;
|
|
33
29
|
pagination: StoragePagination;
|
|
34
30
|
entityId?: string;
|
|
35
31
|
entityType?: string;
|
|
36
32
|
source?: ScoringSource;
|
|
37
|
-
}): Promise<
|
|
38
|
-
pagination: PaginationInfo;
|
|
39
|
-
scores: ScoreRowData[];
|
|
40
|
-
}>;
|
|
33
|
+
}): Promise<ListScoresResponse>;
|
|
41
34
|
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
42
35
|
pagination: StoragePagination;
|
|
43
36
|
entityId: string;
|
|
44
37
|
entityType: string;
|
|
45
|
-
}): Promise<
|
|
46
|
-
pagination: PaginationInfo;
|
|
47
|
-
scores: ScoreRowData[];
|
|
48
|
-
}>;
|
|
38
|
+
}): Promise<ListScoresResponse>;
|
|
49
39
|
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
50
40
|
traceId: string;
|
|
51
41
|
spanId: string;
|
|
52
42
|
pagination: StoragePagination;
|
|
53
|
-
}): Promise<
|
|
54
|
-
pagination: PaginationInfo;
|
|
55
|
-
scores: ScoreRowData[];
|
|
56
|
-
}>;
|
|
43
|
+
}): Promise<ListScoresResponse>;
|
|
57
44
|
}
|
|
58
45
|
//# 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,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAG3D,OAAO,KAAK,EAAE,gBAAgB,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,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAG3D,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAC5G,OAAO,EAEL,aAAa,EAOd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAE9D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAEvD,qBAAa,uBAAwB,SAAQ,aAAa;;IACxD,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBAEvB,MAAM,EAAE,sBAAsB;IAOpC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAIrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAOnB,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAmClE,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAmEpE,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA0EzB,oBAAoB,CAAC,EACzB,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,MAAM,EACN,UAAU,GACX,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,kBAAkB,CAAC;IA6FzB,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,kBAAkB,CAAC;IA2EzB,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,kBAAkB,CAAC;CA6EhC"}
|
|
@@ -1,38 +1,33 @@
|
|
|
1
1
|
import type { ClickHouseClient } from '@clickhouse/client';
|
|
2
2
|
import { WorkflowsStorage } from '@mastra/core/storage';
|
|
3
|
-
import type { WorkflowRun, WorkflowRuns, StorageListWorkflowRunsInput } from '@mastra/core/storage';
|
|
3
|
+
import type { WorkflowRun, WorkflowRuns, StorageListWorkflowRunsInput, UpdateWorkflowStateOptions } from '@mastra/core/storage';
|
|
4
4
|
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
5
|
-
import type {
|
|
5
|
+
import type { ClickhouseDomainConfig } from '../../db/index.js';
|
|
6
6
|
export declare class WorkflowsStorageClickhouse extends WorkflowsStorage {
|
|
7
|
+
#private;
|
|
7
8
|
protected client: ClickHouseClient;
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
});
|
|
13
|
-
updateWorkflowResults({}: {
|
|
9
|
+
constructor(config: ClickhouseDomainConfig);
|
|
10
|
+
init(): Promise<void>;
|
|
11
|
+
dangerouslyClearAll(): Promise<void>;
|
|
12
|
+
updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
|
|
14
13
|
workflowName: string;
|
|
15
14
|
runId: string;
|
|
16
15
|
stepId: string;
|
|
17
16
|
result: StepResult<any, any, any, any>;
|
|
18
17
|
requestContext: Record<string, any>;
|
|
19
18
|
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
20
|
-
updateWorkflowState({}: {
|
|
19
|
+
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
21
20
|
workflowName: string;
|
|
22
21
|
runId: string;
|
|
23
|
-
opts:
|
|
24
|
-
status: string;
|
|
25
|
-
result?: StepResult<any, any, any, any>;
|
|
26
|
-
error?: string;
|
|
27
|
-
suspendedPaths?: Record<string, number[]>;
|
|
28
|
-
waitingPaths?: Record<string, number[]>;
|
|
29
|
-
};
|
|
22
|
+
opts: UpdateWorkflowStateOptions;
|
|
30
23
|
}): Promise<WorkflowRunState | undefined>;
|
|
31
|
-
persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
|
|
24
|
+
persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, }: {
|
|
32
25
|
workflowName: string;
|
|
33
26
|
runId: string;
|
|
34
27
|
resourceId?: string;
|
|
35
28
|
snapshot: WorkflowRunState;
|
|
29
|
+
createdAt?: Date;
|
|
30
|
+
updatedAt?: Date;
|
|
36
31
|
}): Promise<void>;
|
|
37
32
|
loadWorkflowSnapshot({ workflowName, runId, }: {
|
|
38
33
|
workflowName: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EAKL,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,4BAA4B,EAC5B,0BAA0B,EAC3B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE3E,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAGvD,qBAAa,0BAA2B,SAAQ,gBAAgB;;IAC9D,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBAEvB,MAAM,EAAE,sBAAsB;IAOpC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAWrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC,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;IAgCrD,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,0BAA0B,CAAC;KAClC,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IA0BnC,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,EACR,SAAS,EACT,SAAS,GACV,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;QAC3B,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,SAAS,CAAC,EAAE,IAAI,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC;IAgDX,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;IA4BpC,OAAO,CAAC,gBAAgB;IAqBlB,gBAAgB,CAAC,EACrB,YAAY,EACZ,QAAQ,EACR,MAAM,EACN,IAAI,EACJ,OAAO,EACP,UAAU,EACV,MAAM,GACP,GAAE,4BAAiC,GAAG,OAAO,CAAC,YAAY,CAAC;IA+FtD,kBAAkB,CAAC,EACvB,KAAK,EACL,YAAY,GACb,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAqDzB,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAuB7G"}
|
package/dist/storage/index.d.ts
CHANGED
|
@@ -1,32 +1,33 @@
|
|
|
1
1
|
import type { ClickHouseClient } from '@clickhouse/client';
|
|
2
|
-
import type { MastraMessageContentV2 } from '@mastra/core/agent';
|
|
3
|
-
import type { SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
|
|
4
|
-
import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
|
|
5
2
|
import { MastraStorage } from '@mastra/core/storage';
|
|
6
|
-
import type {
|
|
7
|
-
import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
|
|
3
|
+
import type { TABLE_NAMES, StorageDomains, TABLE_SCHEMAS, StorageSupports } from '@mastra/core/storage';
|
|
8
4
|
type IntervalUnit = 'NANOSECOND' | 'MICROSECOND' | 'MILLISECOND' | 'SECOND' | 'MINUTE' | 'HOUR' | 'DAY' | 'WEEK' | 'MONTH' | 'QUARTER' | 'YEAR';
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
5
|
+
type ClickhouseTtlConfig = {
|
|
6
|
+
[TableKey in TABLE_NAMES]?: {
|
|
7
|
+
row?: {
|
|
8
|
+
interval: number;
|
|
9
|
+
unit: IntervalUnit;
|
|
10
|
+
ttlKey?: string;
|
|
11
|
+
};
|
|
12
|
+
columns?: Partial<{
|
|
13
|
+
[ColumnKey in keyof (typeof TABLE_SCHEMAS)[TableKey]]: {
|
|
17
14
|
interval: number;
|
|
18
15
|
unit: IntervalUnit;
|
|
19
16
|
ttlKey?: string;
|
|
20
17
|
};
|
|
21
|
-
|
|
22
|
-
[ColumnKey in keyof (typeof TABLE_SCHEMAS)[TableKey]]: {
|
|
23
|
-
interval: number;
|
|
24
|
-
unit: IntervalUnit;
|
|
25
|
-
ttlKey?: string;
|
|
26
|
-
};
|
|
27
|
-
}>;
|
|
28
|
-
};
|
|
18
|
+
}>;
|
|
29
19
|
};
|
|
20
|
+
};
|
|
21
|
+
/**
|
|
22
|
+
* ClickHouse configuration type.
|
|
23
|
+
*
|
|
24
|
+
* Accepts either:
|
|
25
|
+
* - A pre-configured ClickHouse client: `{ id, client, ttl? }`
|
|
26
|
+
* - URL/credentials config: `{ id, url, username, password, ttl? }`
|
|
27
|
+
*/
|
|
28
|
+
export type ClickhouseConfig = {
|
|
29
|
+
id: string;
|
|
30
|
+
ttl?: ClickhouseTtlConfig;
|
|
30
31
|
/**
|
|
31
32
|
* When true, automatic initialization (table creation/migrations) is disabled.
|
|
32
33
|
* This is useful for CI/CD pipelines where you want to:
|
|
@@ -47,169 +48,73 @@ export type ClickhouseConfig = {
|
|
|
47
48
|
* // No auto-init, tables must already exist
|
|
48
49
|
*/
|
|
49
50
|
disableInit?: boolean;
|
|
50
|
-
}
|
|
51
|
+
} & ({
|
|
52
|
+
/**
|
|
53
|
+
* Pre-configured ClickHouse client.
|
|
54
|
+
* Use this when you need to configure the client before initialization,
|
|
55
|
+
* e.g., to set custom connection settings or interceptors.
|
|
56
|
+
*
|
|
57
|
+
* @example
|
|
58
|
+
* ```typescript
|
|
59
|
+
* import { createClient } from '@clickhouse/client';
|
|
60
|
+
*
|
|
61
|
+
* const client = createClient({
|
|
62
|
+
* url: 'http://localhost:8123',
|
|
63
|
+
* username: 'default',
|
|
64
|
+
* password: '',
|
|
65
|
+
* // Custom settings
|
|
66
|
+
* request_timeout: 60000,
|
|
67
|
+
* });
|
|
68
|
+
*
|
|
69
|
+
* const store = new ClickhouseStore({ id: 'my-store', client });
|
|
70
|
+
* ```
|
|
71
|
+
*/
|
|
72
|
+
client: ClickHouseClient;
|
|
73
|
+
} | {
|
|
74
|
+
url: string;
|
|
75
|
+
username: string;
|
|
76
|
+
password: string;
|
|
77
|
+
});
|
|
78
|
+
/**
|
|
79
|
+
* ClickHouse storage adapter for Mastra.
|
|
80
|
+
*
|
|
81
|
+
* Access domain-specific storage via `getStore()`:
|
|
82
|
+
*
|
|
83
|
+
* @example
|
|
84
|
+
* ```typescript
|
|
85
|
+
* const storage = new ClickhouseStore({ id: 'my-store', url: '...', username: '...', password: '...' });
|
|
86
|
+
*
|
|
87
|
+
* // Access memory domain
|
|
88
|
+
* const memory = await storage.getStore('memory');
|
|
89
|
+
* await memory?.saveThread({ thread });
|
|
90
|
+
*
|
|
91
|
+
* // Access workflows domain
|
|
92
|
+
* const workflows = await storage.getStore('workflows');
|
|
93
|
+
* await workflows?.persistWorkflowSnapshot({ workflowName, runId, snapshot });
|
|
94
|
+
*
|
|
95
|
+
* // Access observability domain
|
|
96
|
+
* const observability = await storage.getStore('observability');
|
|
97
|
+
* await observability?.createSpan(span);
|
|
98
|
+
* ```
|
|
99
|
+
*/
|
|
51
100
|
export declare class ClickhouseStore extends MastraStorage {
|
|
52
101
|
protected db: ClickHouseClient;
|
|
53
102
|
protected ttl: ClickhouseConfig['ttl'];
|
|
54
103
|
stores: StorageDomains;
|
|
55
104
|
constructor(config: ClickhouseConfig);
|
|
56
|
-
get supports():
|
|
57
|
-
selectByIncludeResourceScope: boolean;
|
|
58
|
-
resourceWorkingMemory: boolean;
|
|
59
|
-
hasColumn: boolean;
|
|
60
|
-
createTable: boolean;
|
|
61
|
-
deleteMessages: boolean;
|
|
62
|
-
listScoresBySpan: boolean;
|
|
63
|
-
};
|
|
64
|
-
batchInsert({ tableName, records }: {
|
|
65
|
-
tableName: TABLE_NAMES;
|
|
66
|
-
records: Record<string, any>[];
|
|
67
|
-
}): Promise<void>;
|
|
105
|
+
get supports(): StorageSupports;
|
|
68
106
|
optimizeTable({ tableName }: {
|
|
69
107
|
tableName: TABLE_NAMES;
|
|
70
108
|
}): Promise<void>;
|
|
71
109
|
materializeTtl({ tableName }: {
|
|
72
110
|
tableName: TABLE_NAMES;
|
|
73
111
|
}): Promise<void>;
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
}): Promise<void>;
|
|
81
|
-
alterTable({ tableName, schema, ifNotExists, }: {
|
|
82
|
-
tableName: TABLE_NAMES;
|
|
83
|
-
schema: Record<string, StorageColumn>;
|
|
84
|
-
ifNotExists: string[];
|
|
85
|
-
}): Promise<void>;
|
|
86
|
-
clearTable({ tableName }: {
|
|
87
|
-
tableName: TABLE_NAMES;
|
|
88
|
-
}): Promise<void>;
|
|
89
|
-
insert({ tableName, record }: {
|
|
90
|
-
tableName: TABLE_NAMES;
|
|
91
|
-
record: Record<string, any>;
|
|
92
|
-
}): Promise<void>;
|
|
93
|
-
load<R>({ tableName, keys }: {
|
|
94
|
-
tableName: TABLE_NAMES;
|
|
95
|
-
keys: Record<string, string>;
|
|
96
|
-
}): Promise<R | null>;
|
|
97
|
-
updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, }: {
|
|
98
|
-
workflowName: string;
|
|
99
|
-
runId: string;
|
|
100
|
-
stepId: string;
|
|
101
|
-
result: StepResult<any, any, any, any>;
|
|
102
|
-
requestContext: Record<string, any>;
|
|
103
|
-
}): Promise<Record<string, StepResult<any, any, any, any>>>;
|
|
104
|
-
updateWorkflowState({ workflowName, runId, opts, }: {
|
|
105
|
-
workflowName: string;
|
|
106
|
-
runId: string;
|
|
107
|
-
opts: {
|
|
108
|
-
status: string;
|
|
109
|
-
result?: StepResult<any, any, any, any>;
|
|
110
|
-
error?: string;
|
|
111
|
-
suspendedPaths?: Record<string, number[]>;
|
|
112
|
-
waitingPaths?: Record<string, number[]>;
|
|
113
|
-
};
|
|
114
|
-
}): Promise<WorkflowRunState | undefined>;
|
|
115
|
-
persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, }: {
|
|
116
|
-
workflowName: string;
|
|
117
|
-
runId: string;
|
|
118
|
-
resourceId?: string;
|
|
119
|
-
snapshot: WorkflowRunState;
|
|
120
|
-
}): Promise<void>;
|
|
121
|
-
loadWorkflowSnapshot({ workflowName, runId, }: {
|
|
122
|
-
workflowName: string;
|
|
123
|
-
runId: string;
|
|
124
|
-
}): Promise<WorkflowRunState | null>;
|
|
125
|
-
listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<WorkflowRuns>;
|
|
126
|
-
getWorkflowRunById({ runId, workflowName, }: {
|
|
127
|
-
runId: string;
|
|
128
|
-
workflowName?: string;
|
|
129
|
-
}): Promise<WorkflowRun | null>;
|
|
130
|
-
deleteWorkflowRunById({ runId, workflowName }: {
|
|
131
|
-
runId: string;
|
|
132
|
-
workflowName: string;
|
|
133
|
-
}): Promise<void>;
|
|
134
|
-
getThreadById({ threadId }: {
|
|
135
|
-
threadId: string;
|
|
136
|
-
}): Promise<StorageThreadType | null>;
|
|
137
|
-
saveThread({ thread }: {
|
|
138
|
-
thread: StorageThreadType;
|
|
139
|
-
}): Promise<StorageThreadType>;
|
|
140
|
-
updateThread({ id, title, metadata, }: {
|
|
141
|
-
id: string;
|
|
142
|
-
title: string;
|
|
143
|
-
metadata: Record<string, unknown>;
|
|
144
|
-
}): Promise<StorageThreadType>;
|
|
145
|
-
deleteThread({ threadId }: {
|
|
146
|
-
threadId: string;
|
|
147
|
-
}): Promise<void>;
|
|
148
|
-
saveMessages(args: {
|
|
149
|
-
messages: MastraDBMessage[];
|
|
150
|
-
}): Promise<{
|
|
151
|
-
messages: MastraDBMessage[];
|
|
152
|
-
}>;
|
|
153
|
-
updateMessages(args: {
|
|
154
|
-
messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
|
|
155
|
-
id: string;
|
|
156
|
-
threadId?: string;
|
|
157
|
-
content?: {
|
|
158
|
-
metadata?: MastraMessageContentV2['metadata'];
|
|
159
|
-
content?: MastraMessageContentV2['content'];
|
|
160
|
-
};
|
|
161
|
-
})[];
|
|
162
|
-
}): Promise<MastraDBMessage[]>;
|
|
163
|
-
getResourceById({ resourceId }: {
|
|
164
|
-
resourceId: string;
|
|
165
|
-
}): Promise<StorageResourceType | null>;
|
|
166
|
-
saveResource({ resource }: {
|
|
167
|
-
resource: StorageResourceType;
|
|
168
|
-
}): Promise<StorageResourceType>;
|
|
169
|
-
updateResource({ resourceId, workingMemory, metadata, }: {
|
|
170
|
-
resourceId: string;
|
|
171
|
-
workingMemory?: string;
|
|
172
|
-
metadata?: Record<string, unknown>;
|
|
173
|
-
}): Promise<StorageResourceType>;
|
|
174
|
-
getScoreById({ id }: {
|
|
175
|
-
id: string;
|
|
176
|
-
}): Promise<ScoreRowData | null>;
|
|
177
|
-
saveScore(score: SaveScorePayload): Promise<{
|
|
178
|
-
score: ScoreRowData;
|
|
179
|
-
}>;
|
|
180
|
-
listScoresByRunId({ runId, pagination, }: {
|
|
181
|
-
runId: string;
|
|
182
|
-
pagination: StoragePagination;
|
|
183
|
-
}): Promise<{
|
|
184
|
-
pagination: PaginationInfo;
|
|
185
|
-
scores: ScoreRowData[];
|
|
186
|
-
}>;
|
|
187
|
-
listScoresByEntityId({ entityId, entityType, pagination, }: {
|
|
188
|
-
pagination: StoragePagination;
|
|
189
|
-
entityId: string;
|
|
190
|
-
entityType: string;
|
|
191
|
-
}): Promise<{
|
|
192
|
-
pagination: PaginationInfo;
|
|
193
|
-
scores: ScoreRowData[];
|
|
194
|
-
}>;
|
|
195
|
-
listScoresByScorerId({ scorerId, pagination, entityId, entityType, source, }: {
|
|
196
|
-
scorerId: string;
|
|
197
|
-
pagination: StoragePagination;
|
|
198
|
-
entityId?: string;
|
|
199
|
-
entityType?: string;
|
|
200
|
-
source?: ScoringSource;
|
|
201
|
-
}): Promise<{
|
|
202
|
-
pagination: PaginationInfo;
|
|
203
|
-
scores: ScoreRowData[];
|
|
204
|
-
}>;
|
|
205
|
-
listScoresBySpan({ traceId, spanId, pagination, }: {
|
|
206
|
-
traceId: string;
|
|
207
|
-
spanId: string;
|
|
208
|
-
pagination: StoragePagination;
|
|
209
|
-
}): Promise<{
|
|
210
|
-
pagination: PaginationInfo;
|
|
211
|
-
scores: ScoreRowData[];
|
|
212
|
-
}>;
|
|
112
|
+
/**
|
|
113
|
+
* Closes the ClickHouse client connection.
|
|
114
|
+
*
|
|
115
|
+
* This will close the ClickHouse client, including pre-configured clients.
|
|
116
|
+
* The store assumes ownership of all clients and manages their lifecycle.
|
|
117
|
+
*/
|
|
213
118
|
close(): Promise<void>;
|
|
214
119
|
}
|
|
215
120
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAG3D,OAAO,EAAwB,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAMxG,KAAK,YAAY,GACb,YAAY,GACZ,aAAa,GACb,aAAa,GACb,QAAQ,GACR,QAAQ,GACR,MAAM,GACN,KAAK,GACL,MAAM,GACN,OAAO,GACP,SAAS,GACT,MAAM,CAAC;AAEX,KAAK,mBAAmB,GAAG;KACxB,QAAQ,IAAI,WAAW,CAAC,CAAC,EAAE;QAC1B,GAAG,CAAC,EAAE;YAAE,QAAQ,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,YAAY,CAAC;YAAC,MAAM,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC;QAChE,OAAO,CAAC,EAAE,OAAO,CAAC;aACf,SAAS,IAAI,MAAM,CAAC,OAAO,aAAa,CAAC,CAAC,QAAQ,CAAC,GAAG;gBACrD,QAAQ,EAAE,MAAM,CAAC;gBACjB,IAAI,EAAE,YAAY,CAAC;gBACnB,MAAM,CAAC,EAAE,MAAM,CAAC;aACjB;SACF,CAAC,CAAC;KACJ;CACF,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,CAAC,EAAE,mBAAmB,CAAC;IAC1B;;;;;;;;;;;;;;;;;;OAkBG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,GAAG,CACA;IACE;;;;;;;;;;;;;;;;;;;OAmBG;IACH,MAAM,EAAE,gBAAgB,CAAC;CAC1B,GACD;IACE,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB,CACJ,CAAC;AASF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,eAAgB,SAAQ,aAAa;IAChD,SAAS,CAAC,EAAE,EAAE,gBAAgB,CAAC;IAC/B,SAAS,CAAC,GAAG,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAM;IAE5C,MAAM,EAAE,cAAc,CAAC;gBAEX,MAAM,EAAE,gBAAgB;IAiDpC,IAAI,QAAQ,IAAI,eAAe,CAY9B;IAEK,aAAa,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBvE,cAAc,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkB9E;;;;;OAKG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7B"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/clickhouse",
|
|
3
|
-
"version": "1.0.0-beta.
|
|
3
|
+
"version": "1.0.0-beta.7",
|
|
4
4
|
"description": "Clickhouse provider for Mastra - includes db storage capabilities",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -23,7 +23,6 @@
|
|
|
23
23
|
"@clickhouse/client": "^1.12.1"
|
|
24
24
|
},
|
|
25
25
|
"devDependencies": {
|
|
26
|
-
"@microsoft/api-extractor": "^7.52.8",
|
|
27
26
|
"@types/node": "22.13.17",
|
|
28
27
|
"@vitest/coverage-v8": "4.0.12",
|
|
29
28
|
"@vitest/ui": "4.0.12",
|
|
@@ -32,9 +31,9 @@
|
|
|
32
31
|
"typescript": "^5.8.3",
|
|
33
32
|
"vitest": "4.0.12",
|
|
34
33
|
"@internal/lint": "0.0.53",
|
|
35
|
-
"@internal/storage-test-utils": "0.0.49",
|
|
36
34
|
"@internal/types-builder": "0.0.28",
|
|
37
|
-
"@mastra/core": "1.0.0-beta.
|
|
35
|
+
"@mastra/core": "1.0.0-beta.15",
|
|
36
|
+
"@internal/storage-test-utils": "0.0.49"
|
|
38
37
|
},
|
|
39
38
|
"peerDependencies": {
|
|
40
39
|
"@mastra/core": ">=1.0.0-0 <2.0.0-0"
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/operations/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EAAwB,eAAe,EAA0C,MAAM,sBAAsB,CAAC;AACrH,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAGjD,qBAAa,yBAA0B,SAAQ,eAAe;IAC5D,SAAS,CAAC,GAAG,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACvC,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBACvB,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAAE,MAAM,EAAE,gBAAgB,CAAC;QAAC,GAAG,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAA;KAAE;IAMjF,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAShE,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAgBnD,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;IA8DX,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;IA0CX,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAyBpE,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAMnE,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;IAoCrG,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;IAmC9G,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;CA6DhH"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAYtF,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAUrD,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,CAS9D,CAAC;AAEF,MAAM,MAAM,YAAY,GACpB,YAAY,GACZ,aAAa,GACb,aAAa,GACb,QAAQ,GACR,QAAQ,GACR,MAAM,GACN,KAAK,GACL,MAAM,GACN,OAAO,GACP,SAAS,GACT,MAAM,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG;IAC7B,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE;SACH,QAAQ,IAAI,WAAW,CAAC,CAAC,EAAE;YAC1B,GAAG,CAAC,EAAE;gBAAE,QAAQ,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,YAAY,CAAC;gBAAC,MAAM,CAAC,EAAE,MAAM,CAAA;aAAE,CAAC;YAChE,OAAO,CAAC,EAAE,OAAO,CAAC;iBACf,SAAS,IAAI,MAAM,CAAC,OAAO,aAAa,CAAC,CAAC,QAAQ,CAAC,GAAG;oBACrD,QAAQ,EAAE,MAAM,CAAC;oBACjB,IAAI,EAAE,YAAY,CAAC;oBACnB,MAAM,CAAC,EAAE,MAAM,CAAC;iBACjB;aACF,CAAC,CAAC;SACJ;KACF,CAAC;CACH,CAAC;AAEF,wBAAgB,YAAY,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,CAkB3C;AAED,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAEjD"}
|
|
File without changes
|