@mastra/clickhouse 0.12.3 → 0.13.0-alpha.2

package/dist/_tsup-dts-rollup.d.ts

@@ -1,479 +0,0 @@
-import type { ClickHouseClient } from '@clickhouse/client';
-import type { EvalRow } from '@mastra/core/storage';
-import { LegacyEvalsStorage } from '@mastra/core/storage';
-import type { MastraMessageContentV2 } from '@mastra/core/agent';
-import type { MastraMessageV1 } from '@mastra/core/memory';
-import type { MastraMessageV2 } from '@mastra/core/memory';
-import { MastraStorage } from '@mastra/core/storage';
-import { MemoryStorage } from '@mastra/core/storage';
-import type { PaginationArgs } from '@mastra/core/storage';
-import type { PaginationInfo } from '@mastra/core/storage';
-import type { ScoreRowData } from '@mastra/core/scores';
-import { ScoresStorage } from '@mastra/core/storage';
-import type { StorageColumn } from '@mastra/core/storage';
-import type { StorageDomains } from '@mastra/core/storage';
-import type { StorageGetMessagesArg } from '@mastra/core/storage';
-import type { StorageGetTracesArg } from '@mastra/core/storage';
-import type { StorageGetTracesPaginatedArg } from '@mastra/core/storage';
-import type { StoragePagination } from '@mastra/core/storage';
-import type { StorageResourceType } from '@mastra/core/storage';
-import type { StorageThreadType } from '@mastra/core/memory';
-import { StoreOperations } from '@mastra/core/storage';
-import type { TABLE_NAMES } from '@mastra/core/storage';
-import type { TABLE_SCHEMAS } from '@mastra/core/storage';
-import type { Trace } from '@mastra/core/telemetry';
-import { TracesStorage } from '@mastra/core/storage';
-import type { WorkflowRun } from '@mastra/core/storage';
-import type { WorkflowRuns } from '@mastra/core/storage';
-import type { WorkflowRunState } from '@mastra/core/workflows';
-import { WorkflowsStorage } from '@mastra/core/storage';
-
-declare type ClickhouseConfig = {
-    url: string;
-    username: string;
-    password: string;
-    ttl?: {
-        [TableKey in TABLE_NAMES]?: {
-            row?: {
-                interval: number;
-                unit: IntervalUnit_2;
-                ttlKey?: string;
-            };
-            columns?: Partial<{
-                [ColumnKey in keyof (typeof TABLE_SCHEMAS)[TableKey]]: {
-                    interval: number;
-                    unit: IntervalUnit_2;
-                    ttlKey?: string;
-                };
-            }>;
-        };
-    };
-};
-export { ClickhouseConfig }
-export { ClickhouseConfig as ClickhouseConfig_alias_1 }
-
-export declare type ClickhouseConfig_alias_2 = {
-    url: string;
-    username: string;
-    password: string;
-    ttl?: {
-        [TableKey in TABLE_NAMES]?: {
-            row?: {
-                interval: number;
-                unit: IntervalUnit;
-                ttlKey?: string;
-            };
-            columns?: Partial<{
-                [ColumnKey in keyof (typeof TABLE_SCHEMAS)[TableKey]]: {
-                    interval: number;
-                    unit: IntervalUnit;
-                    ttlKey?: string;
-                };
-            }>;
-        };
-    };
-};
-
-declare class ClickhouseStore extends MastraStorage {
-    protected db: ClickHouseClient;
-    protected ttl: ClickhouseConfig['ttl'];
-    stores: StorageDomains;
-    constructor(config: ClickhouseConfig);
-    get supports(): {
-        selectByIncludeResourceScope: boolean;
-        resourceWorkingMemory: boolean;
-        hasColumn: boolean;
-        createTable: boolean;
-        deleteMessages: boolean;
-    };
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getEvals(options: {
-        agentName?: string;
-        type?: 'test' | 'live';
-    } & PaginationArgs): Promise<PaginationInfo & {
-        evals: EvalRow[];
-    }>;
-    batchInsert({ tableName, records }: {
-        tableName: TABLE_NAMES;
-        records: Record<string, any>[];
-    }): Promise<void>;
-    optimizeTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    materializeTtl({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    createTable({ tableName, schema, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-    }): Promise<void>;
-    dropTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    alterTable({ tableName, schema, ifNotExists, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-        ifNotExists: string[];
-    }): Promise<void>;
-    clearTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    insert({ tableName, record }: {
-        tableName: TABLE_NAMES;
-        record: Record<string, any>;
-    }): Promise<void>;
-    load<R>({ tableName, keys }: {
-        tableName: TABLE_NAMES;
-        keys: Record<string, string>;
-    }): Promise<R | null>;
-    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
-        workflowName: string;
-        runId: string;
-        snapshot: WorkflowRunState;
-    }): Promise<void>;
-    loadWorkflowSnapshot({ workflowName, runId, }: {
-        workflowName: string;
-        runId: string;
-    }): Promise<WorkflowRunState | null>;
-    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
-        workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
-        limit?: number;
-        offset?: number;
-        resourceId?: string;
-    }): Promise<WorkflowRuns>;
-    getWorkflowRunById({ runId, workflowName, }: {
-        runId: string;
-        workflowName?: string;
-    }): Promise<WorkflowRun | null>;
-    getTraces(args: StorageGetTracesArg): Promise<any[]>;
-    getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
-        traces: Trace[];
-    }>;
-    batchTraceInsert(args: {
-        records: Trace[];
-    }): Promise<void>;
-    getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    getThreadsByResourceId({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    saveThread({ thread }: {
-        thread: StorageThreadType;
-    }): Promise<StorageThreadType>;
-    updateThread({ id, title, metadata, }: {
-        id: string;
-        title: string;
-        metadata: Record<string, unknown>;
-    }): Promise<StorageThreadType>;
-    deleteThread({ threadId }: {
-        threadId: string;
-    }): Promise<void>;
-    getThreadsByResourceIdPaginated(args: {
-        resourceId: string;
-        page: number;
-        perPage: number;
-    }): Promise<PaginationInfo & {
-        threads: StorageThreadType[];
-    }>;
-    getMessages(args: StorageGetMessagesArg & {
-        format?: 'v1';
-    }): Promise<MastraMessageV1[]>;
-    getMessages(args: StorageGetMessagesArg & {
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    saveMessages(args: {
-        messages: MastraMessageV1[];
-        format?: undefined | 'v1';
-    }): Promise<MastraMessageV1[]>;
-    saveMessages(args: {
-        messages: MastraMessageV2[];
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    getMessagesPaginated(args: StorageGetMessagesArg & {
-        format?: 'v1' | 'v2';
-    }): Promise<PaginationInfo & {
-        messages: MastraMessageV1[] | MastraMessageV2[];
-    }>;
-    updateMessages(args: {
-        messages: (Partial<Omit<MastraMessageV2, 'createdAt'>> & {
-            id: string;
-            threadId?: string;
-            content?: {
-                metadata?: MastraMessageContentV2['metadata'];
-                content?: MastraMessageContentV2['content'];
-            };
-        })[];
-    }): Promise<MastraMessageV2[]>;
-    getResourceById({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageResourceType | null>;
-    saveResource({ resource }: {
-        resource: StorageResourceType;
-    }): Promise<StorageResourceType>;
-    updateResource({ resourceId, workingMemory, metadata, }: {
-        resourceId: string;
-        workingMemory?: string;
-        metadata?: Record<string, unknown>;
-    }): Promise<StorageResourceType>;
-    getScoreById({ id }: {
-        id: string;
-    }): Promise<ScoreRowData | null>;
-    saveScore(_score: ScoreRowData): Promise<{
-        score: ScoreRowData;
-    }>;
-    getScoresByRunId({ runId, pagination, }: {
-        runId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByEntityId({ entityId, entityType, pagination, }: {
-        pagination: StoragePagination;
-        entityId: string;
-        entityType: string;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByScorerId({ scorerId, pagination, }: {
-        scorerId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    close(): Promise<void>;
-}
-export { ClickhouseStore }
-export { ClickhouseStore as ClickhouseStore_alias_1 }
-
-export declare const COLUMN_TYPES: Record<StorageColumn['type'], string>;
-
-export declare type IntervalUnit = 'NANOSECOND' | 'MICROSECOND' | 'MILLISECOND' | 'SECOND' | 'MINUTE' | 'HOUR' | 'DAY' | 'WEEK' | 'MONTH' | 'QUARTER' | 'YEAR';
-
-declare type IntervalUnit_2 = 'NANOSECOND' | 'MICROSECOND' | 'MILLISECOND' | 'SECOND' | 'MINUTE' | 'HOUR' | 'DAY' | 'WEEK' | 'MONTH' | 'QUARTER' | 'YEAR';
-
-export declare class LegacyEvalsStorageClickhouse extends LegacyEvalsStorage {
-    protected client: ClickHouseClient;
-    protected operations: StoreOperationsClickhouse;
-    constructor({ client, operations }: {
-        client: ClickHouseClient;
-        operations: StoreOperationsClickhouse;
-    });
-    private transformEvalRow;
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getEvals(options?: {
-        agentName?: string;
-        type?: 'test' | 'live';
-    } & PaginationArgs): Promise<PaginationInfo & {
-        evals: EvalRow[];
-    }>;
-}
-
-export declare class MemoryStorageClickhouse extends MemoryStorage {
-    protected client: ClickHouseClient;
-    protected operations: StoreOperationsClickhouse;
-    constructor({ client, operations }: {
-        client: ClickHouseClient;
-        operations: StoreOperationsClickhouse;
-    });
-    getMessages(args: StorageGetMessagesArg & {
-        format?: 'v1';
-    }): Promise<MastraMessageV1[]>;
-    getMessages(args: StorageGetMessagesArg & {
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    saveMessages(args: {
-        messages: MastraMessageV1[];
-        format?: undefined | 'v1';
-    }): Promise<MastraMessageV1[]>;
-    saveMessages(args: {
-        messages: MastraMessageV2[];
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    getThreadsByResourceId({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    saveThread({ thread }: {
-        thread: StorageThreadType;
-    }): Promise<StorageThreadType>;
-    updateThread({ id, title, metadata, }: {
-        id: string;
-        title: string;
-        metadata: Record<string, unknown>;
-    }): Promise<StorageThreadType>;
-    deleteThread({ threadId }: {
-        threadId: string;
-    }): Promise<void>;
-    getThreadsByResourceIdPaginated(args: {
-        resourceId: string;
-        page?: number;
-        perPage?: number;
-    }): Promise<PaginationInfo & {
-        threads: StorageThreadType[];
-    }>;
-    getMessagesPaginated(args: StorageGetMessagesArg & {
-        format?: 'v1' | 'v2';
-    }): Promise<PaginationInfo & {
-        messages: MastraMessageV1[] | MastraMessageV2[];
-    }>;
-    updateMessages(args: {
-        messages: (Partial<Omit<MastraMessageV2, 'createdAt'>> & {
-            id: string;
-            threadId?: string;
-            content?: {
-                metadata?: MastraMessageContentV2['metadata'];
-                content?: MastraMessageContentV2['content'];
-            };
-        })[];
-    }): Promise<MastraMessageV2[]>;
-    getResourceById({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageResourceType | null>;
-    saveResource({ resource }: {
-        resource: StorageResourceType;
-    }): Promise<StorageResourceType>;
-    updateResource({ resourceId, workingMemory, metadata, }: {
-        resourceId: string;
-        workingMemory?: string;
-        metadata?: Record<string, unknown>;
-    }): Promise<StorageResourceType>;
-}
-
-export declare class ScoresStorageClickhouse extends ScoresStorage {
-    protected client: ClickHouseClient;
-    protected operations: StoreOperationsClickhouse;
-    constructor({ client, operations }: {
-        client: ClickHouseClient;
-        operations: StoreOperationsClickhouse;
-    });
-    private transformScoreRow;
-    getScoreById({ id }: {
-        id: string;
-    }): Promise<ScoreRowData | null>;
-    saveScore(score: ScoreRowData): Promise<{
-        score: ScoreRowData;
-    }>;
-    getScoresByRunId({ runId, pagination, }: {
-        runId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByScorerId({ scorerId, pagination, }: {
-        scorerId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByEntityId({ entityId, entityType, pagination, }: {
-        pagination: StoragePagination;
-        entityId: string;
-        entityType: string;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-}
-
-export declare class StoreOperationsClickhouse extends StoreOperations {
-    protected ttl: ClickhouseConfig_alias_2['ttl'];
-    protected client: ClickHouseClient;
-    constructor({ client, ttl }: {
-        client: ClickHouseClient;
-        ttl: ClickhouseConfig_alias_2['ttl'];
-    });
-    hasColumn(table: string, column: string): Promise<boolean>;
-    protected getSqlType(type: StorageColumn['type']): string;
-    createTable({ tableName, schema, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-    }): Promise<void>;
-    alterTable({ tableName, schema, ifNotExists, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-        ifNotExists: string[];
-    }): Promise<void>;
-    clearTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    dropTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    insert({ tableName, record }: {
-        tableName: TABLE_NAMES;
-        record: Record<string, any>;
-    }): Promise<void>;
-    batchInsert({ tableName, records }: {
-        tableName: TABLE_NAMES;
-        records: Record<string, any>[];
-    }): Promise<void>;
-    load<R>({ tableName, keys }: {
-        tableName: TABLE_NAMES;
-        keys: Record<string, string>;
-    }): Promise<R | null>;
-}
-
-export declare const TABLE_ENGINES: Record<TABLE_NAMES, string>;
-
-export declare class TracesStorageClickhouse extends TracesStorage {
-    protected client: ClickHouseClient;
-    protected operations: StoreOperationsClickhouse;
-    constructor({ client, operations }: {
-        client: ClickHouseClient;
-        operations: StoreOperationsClickhouse;
-    });
-    getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
-        traces: Trace[];
-    }>;
-    getTraces({ name, scope, page, perPage, attributes, filters, fromDate, toDate, }: StorageGetTracesArg): Promise<any[]>;
-    batchTraceInsert(args: {
-        records: Trace[];
-    }): Promise<void>;
-}
-
-export declare function transformRow<R>(row: any): R;
-
-export declare function transformRows<R>(rows: any[]): R[];
-
-export declare class WorkflowsStorageClickhouse extends WorkflowsStorage {
-    protected client: ClickHouseClient;
-    protected operations: StoreOperationsClickhouse;
-    constructor({ client, operations }: {
-        client: ClickHouseClient;
-        operations: StoreOperationsClickhouse;
-    });
-    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
-        workflowName: string;
-        runId: string;
-        snapshot: WorkflowRunState;
-    }): Promise<void>;
-    loadWorkflowSnapshot({ workflowName, runId, }: {
-        workflowName: string;
-        runId: string;
-    }): Promise<WorkflowRunState | null>;
-    private parseWorkflowRun;
-    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
-        workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
-        limit?: number;
-        offset?: number;
-        resourceId?: string;
-    }): Promise<WorkflowRuns>;
-    getWorkflowRunById({ runId, workflowName, }: {
-        runId: string;
-        workflowName?: string;
-    }): Promise<WorkflowRun | null>;
-}
-
-export { }

package/dist/index.d.cts

@@ -1,2 +0,0 @@
-export { ClickhouseConfig } from './_tsup-dts-rollup.cjs';
-export { ClickhouseStore } from './_tsup-dts-rollup.cjs';