@mastra/pg 0.13.0 → 0.13.1

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

@@ -1,786 +0,0 @@
-import { BaseFilterTranslator } from '@mastra/core/vector/filter';
-import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
-import type { CreateIndexParams } from '@mastra/core/vector';
-import type { DeleteIndexParams } from '@mastra/core/vector';
-import type { DeleteVectorParams } from '@mastra/core/vector';
-import type { DescribeIndexParams } from '@mastra/core/vector';
-import type { EvalRow } from '@mastra/core/storage';
-import type { IDatabase } from 'pg-promise';
-import type { IndexStats } from '@mastra/core/vector';
-import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
-import { LegacyEvalsStorage } from '@mastra/core/storage';
-import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
-import type { MastraMessageContentV2 } from '@mastra/core/agent';
-import type { MastraMessageV1 } from '@mastra/core/memory';
-import type { MastraMessageV2 } from '@mastra/core/agent';
-import type { MastraMessageV2 as MastraMessageV2_2 } from '@mastra/core/memory';
-import { MastraStorage } from '@mastra/core/storage';
-import { MastraVector } from '@mastra/core/vector';
-import { MemoryStorage } from '@mastra/core/storage';
-import type { OperatorSupport } from '@mastra/core/vector/filter';
-import type { OperatorValueMap } from '@mastra/core/vector/filter';
-import type { PaginationArgs } from '@mastra/core/storage';
-import type { PaginationInfo } from '@mastra/core/storage';
-import type { PaginationInfo as PaginationInfo_2 } from '@mastra/core';
-import pg from 'pg';
-import pgPromise from 'pg-promise';
-import type { QueryResult } from '@mastra/core/vector';
-import type { QueryVectorParams } from '@mastra/core/vector';
-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 { StoragePagination as StoragePagination_2 } from '@mastra/core';
-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 { ThreadSortOptions } from '@mastra/core/storage';
-import type { Trace } from '@mastra/core/telemetry';
-import { TracesStorage } from '@mastra/core/storage';
-import type { UpdateVectorParams } from '@mastra/core/vector';
-import type { UpsertVectorParams } from '@mastra/core/vector';
-import type { VectorFieldValue } from '@mastra/core/vector/filter';
-import type { VectorFilter } from '@mastra/core/vector/filter';
-import type { WorkflowRun } from '@mastra/core/storage';
-import type { WorkflowRun as WorkflowRun_2 } from '@mastra/core';
-import type { WorkflowRuns } from '@mastra/core/storage';
-import type { WorkflowRuns as WorkflowRuns_2 } from '@mastra/core';
-import type { WorkflowRunState } from '@mastra/core/workflows';
-import type { WorkflowRunState as WorkflowRunState_2 } from '@mastra/core';
-import { WorkflowsStorage } from '@mastra/core/storage';
-
-export declare const baseTestConfigs: {
-    smokeTests: {
-        dimension: number;
-        size: number;
-        k: number;
-        queryCount: number;
-    }[];
-    '64': {
-        dimension: number;
-        size: number;
-        k: number;
-        queryCount: number;
-    }[];
-    '384': {
-        dimension: number;
-        size: number;
-        k: number;
-        queryCount: number;
-    }[];
-    '1024': {
-        dimension: number;
-        size: number;
-        k: number;
-        queryCount: number;
-    }[];
-    stressTests: {
-        dimension: number;
-        size: number;
-        k: number;
-        queryCount: number;
-    }[];
-};
-
-export declare function buildFilterQuery(filter: PGVectorFilter, minScore: number, topK: number): FilterResult;
-
-export declare const calculateRecall: (actual: number[], expected: number[], k: number) => number;
-
-export declare const calculateTimeout: (dimension: number, size: number, k: number) => number;
-
-export declare const connectionString: string;
-
-export declare function cosineSimilarity(a: number[], b: number[]): number;
-
-declare interface FilterResult {
-    sql: string;
-    values: any[];
-}
-
-export declare const findNearestBruteForce: (query: number[], vectors: number[][], k: number) => number[];
-
-export declare const formatTable: (data: any[], columns: string[]) => string;
-
-export declare const generateClusteredVectors: (count: number, dim: number, numClusters?: number) => number[][];
-
-export declare const generateRandomVectors: (count: number, dim: number) => number[][];
-
-export declare const generateSkewedVectors: (count: number, dim: number) => number[][];
-
-export declare const getHNSWConfig: (indexConfig: IndexConfig) => {
-    m: number;
-    efConstruction: number;
-};
-
-export declare function getIndexDescription({ type, hnsw, }: {
-    type: IndexType;
-    hnsw: {
-        m: number;
-        efConstruction: number;
-    };
-}): string;
-
-export declare const getListCount: (indexConfig: IndexConfig, size: number) => number | undefined;
-
-export declare function getSchemaName(schema?: string): string | undefined;
-
-export declare function getSearchEf(k: number, m: number): {
-    default: number;
-    lower: number;
-    higher: number;
-};
-
-export declare function getTableName({ indexName, schemaName }: {
-    indexName: string;
-    schemaName?: string;
-}): string;
-
-export declare const groupBy: <T, K extends keyof T>(array: T[], key: K | ((item: T) => string), reducer?: (group: T[]) => any) => Record<string, any>;
-
-declare interface HNSWConfig {
-    m?: number;
-    efConstruction?: number;
-}
-
-export declare interface IndexConfig {
-    type?: IndexType;
-    ivf?: IVFConfig;
-    hnsw?: HNSWConfig;
-}
-
-export declare type IndexType = 'ivfflat' | 'hnsw' | 'flat';
-
-declare interface IVFConfig {
-    lists?: number;
-}
-
-export declare class LegacyEvalsPG extends LegacyEvalsStorage {
-    private client;
-    private schema;
-    constructor({ client, schema }: {
-        client: IDatabase<{}>;
-        schema: string;
-    });
-    /** @deprecated use getEvals instead */
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getEvals(options?: {
-        agentName?: string;
-        type?: 'test' | 'live';
-    } & PaginationArgs): Promise<PaginationInfo & {
-        evals: EvalRow[];
-    }>;
-}
-
-export declare function measureLatency<T>(fn: () => Promise<T>): Promise<[number, T]>;
-
-export declare class MemoryPG extends MemoryStorage {
-    private client;
-    private schema;
-    private operations;
-    constructor({ client, schema, operations, }: {
-        client: IDatabase<{}>;
-        schema: string;
-        operations: StoreOperationsPG;
-    });
-    getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    /**
-     * @deprecated use getThreadsByResourceIdPaginated instead
-     */
-    getThreadsByResourceId(args: {
-        resourceId: string;
-    } & ThreadSortOptions): Promise<StorageThreadType[]>;
-    getThreadsByResourceIdPaginated(args: {
-        resourceId: string;
-        page: number;
-        perPage: number;
-    } & ThreadSortOptions): Promise<PaginationInfo & {
-        threads: 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>;
-    private _getIncludedMessages;
-    /**
-     * @deprecated use getMessagesPaginated instead
-     */
-    getMessages(args: StorageGetMessagesArg & {
-        format?: 'v1';
-    }): Promise<MastraMessageV1[]>;
-    getMessages(args: StorageGetMessagesArg & {
-        format: 'v2';
-    }): Promise<MastraMessageV2_2[]>;
-    getMessagesPaginated(args: StorageGetMessagesArg & {
-        format?: 'v1' | 'v2';
-    }): Promise<PaginationInfo & {
-        messages: MastraMessageV1[] | MastraMessageV2_2[];
-    }>;
-    saveMessages(args: {
-        messages: MastraMessageV1[];
-        format?: undefined | 'v1';
-    }): Promise<MastraMessageV1[]>;
-    saveMessages(args: {
-        messages: MastraMessageV2_2[];
-        format: 'v2';
-    }): Promise<MastraMessageV2_2[]>;
-    updateMessages({ messages, }: {
-        messages: (Partial<Omit<MastraMessageV2_2, 'createdAt'>> & {
-            id: string;
-            content?: {
-                metadata?: MastraMessageContentV2['metadata'];
-                content?: MastraMessageContentV2['content'];
-            };
-        })[];
-    }): Promise<MastraMessageV2_2[]>;
-    deleteMessages(messageIds: string[]): Promise<void>;
-    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>;
-}
-
-declare type PGBlacklisted = BlacklistedRootOperators | '$contains' | '$size';
-
-declare interface PgCreateIndexParams extends CreateIndexParams {
-    indexConfig?: IndexConfig;
-    buildIndex?: boolean;
-}
-
-declare interface PgDefineIndexParams {
-    indexName: string;
-    metric: 'cosine' | 'euclidean' | 'dotproduct';
-    indexConfig: IndexConfig;
-}
-
-/**
- * Translates MongoDB-style filters to PG compatible filters.
- *
- * Key differences from MongoDB:
- *
- * Logical Operators ($and, $or, $nor):
- * - Can be used at the top level or nested within fields
- * - Can take either a single condition or an array of conditions
- *
- */
-export declare class PGFilterTranslator extends BaseFilterTranslator<PGVectorFilter> {
-    protected getSupportedOperators(): OperatorSupport;
-    translate(filter?: PGVectorFilter): PGVectorFilter;
-    private translateNode;
-    private translateRegexPattern;
-}
-
-declare type PGFilterValue = VectorFieldValue | RegExp;
-
-declare interface PGIndexStats extends IndexStats {
-    type: IndexType;
-    config: {
-        m?: number;
-        efConstruction?: number;
-        lists?: number;
-        probes?: number;
-    };
-}
-export { PGIndexStats }
-export { PGIndexStats as PGIndexStats_alias_1 }
-
-declare type PGOperatorValueMap = Omit<OperatorValueMap, '$in' | '$all' | '$nin' | '$eq' | '$ne'> & {
-    $size: number;
-    $contains: VectorFieldValue | Record<string, unknown>;
-    $all: VectorFieldValue;
-    $in: VectorFieldValue;
-    $nin: VectorFieldValue;
-    $eq: VectorFieldValue;
-    $ne: VectorFieldValue;
-};
-
-declare interface PgQueryVectorParams extends QueryVectorParams<PGVectorFilter> {
-    minScore?: number;
-    /**
-     * HNSW search parameter. Controls the size of the dynamic candidate
-     * list during search. Higher values improve accuracy at the cost of speed.
-     */
-    ef?: number;
-    /**
-     * IVFFlat probe parameter. Number of cells to visit during search.
-     * Higher values improve accuracy at the cost of speed.
-     */
-    probes?: number;
-}
-
-export declare function pgTests(): void;
-
-declare class PgVector extends MastraVector<PGVectorFilter> {
-    pool: pg.Pool;
-    private describeIndexCache;
-    private createdIndexes;
-    private mutexesByName;
-    private schema?;
-    private setupSchemaPromise;
-    private installVectorExtensionPromise;
-    private vectorExtensionInstalled;
-    private schemaSetupComplete;
-    constructor({ connectionString, schemaName, pgPoolOptions, }: {
-        connectionString: string;
-        schemaName?: string;
-        pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
-    });
-    private getMutexByName;
-    private getTableName;
-    private getSchemaName;
-    transformFilter(filter?: PGVectorFilter): PGVectorFilter;
-    getIndexInfo({ indexName }: DescribeIndexParams): Promise<PGIndexStats>;
-    query({ indexName, queryVector, topK, filter, includeVector, minScore, ef, probes, }: PgQueryVectorParams): Promise<QueryResult[]>;
-    upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
-    private hasher;
-    private getIndexCacheKey;
-    private cachedIndexExists;
-    private setupSchema;
-    createIndex({ indexName, dimension, metric, indexConfig, buildIndex, }: PgCreateIndexParams): Promise<void>;
-    buildIndex({ indexName, metric, indexConfig }: PgDefineIndexParams): Promise<void>;
-    private setupIndex;
-    private installVectorExtension;
-    listIndexes(): Promise<string[]>;
-    /**
-     * Retrieves statistics about a vector index.
-     *
-     * @param {string} indexName - The name of the index to describe
-     * @returns A promise that resolves to the index statistics including dimension, count and metric
-     */
-    describeIndex({ indexName }: DescribeIndexParams): Promise<PGIndexStats>;
-    deleteIndex({ indexName }: DeleteIndexParams): Promise<void>;
-    truncateIndex({ indexName }: DeleteIndexParams): Promise<void>;
-    disconnect(): Promise<void>;
-    /**
-     * Updates a vector by its ID with the provided vector and/or metadata.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to update.
-     * @param update - An object containing the vector and/or metadata to update.
-     * @param update.vector - An optional array of numbers representing the new vector.
-     * @param update.metadata - An optional record containing the new metadata.
-     * @returns A promise that resolves when the update is complete.
-     * @throws Will throw an error if no updates are provided or if the update operation fails.
-     */
-    updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void>;
-    /**
-     * Deletes a vector by its ID.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to delete.
-     * @returns A promise that resolves when the deletion is complete.
-     * @throws Will throw an error if the deletion operation fails.
-     */
-    deleteVector({ indexName, id }: DeleteVectorParams): Promise<void>;
-}
-export { PgVector }
-export { PgVector as PgVector_alias_1 }
-
-/**
- * Vector store specific prompt that details supported operators and examples.
- * This prompt helps users construct valid filters for PG Vector.
- */
-declare const PGVECTOR_PROMPT = "When querying PG Vector, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n- $elemMatch: Match array elements that meet all specified conditions\n  Example: { \"items\": { \"$elemMatch\": { \"price\": { \"$gt\": 100 } } } }\n- $contains: Check if array contains value\n  Example: { \"tags\": { \"$contains\": \"premium\" } }\n\nLogical Operators:\n- $and: Logical AND (implicit when using multiple conditions)\n  Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n- $not: Logical NOT\n  Example: { \"$not\": { \"category\": \"electronics\" } }\n- $nor: Logical NOR\n  Example: { \"$nor\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nSpecial Operators:\n- $size: Array length check\n  Example: { \"tags\": { \"$size\": 2 } }\n\nRestrictions:\n- Regex patterns are not supported\n- Direct RegExp patterns will throw an error\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Array operations work on array fields only\n- Basic operators handle array values as JSON strings\n- Empty arrays in conditions are handled gracefully\n- Only logical operators ($and, $or, $not, $nor) can be used at the top level\n- All other operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n  Invalid: { \"$contains\": \"value\" }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- $not operator:\n  - Must be an object\n  - Cannot be empty\n  - Can be used at field level or top level\n  - Valid: { \"$not\": { \"field\": \"value\" } }\n  - Valid: { \"field\": { \"$not\": { \"$eq\": \"value\" } } }\n- Other logical operators ($and, $or, $nor):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n- $elemMatch requires an object with conditions\n  Valid: { \"array\": { \"$elemMatch\": { \"field\": \"value\" } } }\n  Invalid: { \"array\": { \"$elemMatch\": \"value\" } }\n\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}";
-export { PGVECTOR_PROMPT }
-export { PGVECTOR_PROMPT as PGVECTOR_PROMPT_alias_1 }
-
-export declare type PGVectorFilter = VectorFilter<keyof PGOperatorValueMap, PGOperatorValueMap, LogicalOperatorValueMap, PGBlacklisted, PGFilterValue>;
-
-declare type PostgresConfig = {
-    schemaName?: string;
-} & ({
-    host: string;
-    port: number;
-    database: string;
-    user: string;
-    password: string;
-    ssl?: boolean | ISSLConfig;
-} | {
-    connectionString: string;
-});
-export { PostgresConfig }
-export { PostgresConfig as PostgresConfig_alias_1 }
-
-declare class PostgresStore extends MastraStorage {
-    db: pgPromise.IDatabase<{}>;
-    pgp: pgPromise.IMain;
-    private client;
-    private schema?;
-    stores: StorageDomains;
-    constructor(config: PostgresConfig);
-    get supports(): {
-        selectByIncludeResourceScope: boolean;
-        resourceWorkingMemory: boolean;
-        hasColumn: boolean;
-        createTable: boolean;
-        deleteMessages: boolean;
-    };
-    /** @deprecated use getEvals instead */
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
-    getEvals(options?: {
-        agentName?: string;
-        type?: 'test' | 'live';
-    } & PaginationArgs): Promise<PaginationInfo & {
-        evals: EvalRow[];
-    }>;
-    /**
-     * @deprecated use getTracesPaginated instead
-     */
-    getTraces(args: StorageGetTracesArg): Promise<Trace[]>;
-    getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
-        traces: Trace[];
-    }>;
-    batchTraceInsert({ records }: {
-        records: Record<string, any>[];
-    }): Promise<void>;
-    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>;
-    /**
-     * Memory
-     */
-    getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    /**
-     * @deprecated use getThreadsByResourceIdPaginated instead
-     */
-    getThreadsByResourceId(args: {
-        resourceId: string;
-    } & ThreadSortOptions): Promise<StorageThreadType[]>;
-    getThreadsByResourceIdPaginated(args: {
-        resourceId: string;
-        page: number;
-        perPage: number;
-    } & ThreadSortOptions): Promise<PaginationInfo & {
-        threads: 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>;
-    /**
-     * @deprecated use getMessagesPaginated instead
-     */
-    getMessages(args: StorageGetMessagesArg & {
-        format?: 'v1';
-    }): Promise<MastraMessageV1[]>;
-    getMessages(args: StorageGetMessagesArg & {
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    getMessagesPaginated(args: StorageGetMessagesArg & {
-        format?: 'v1' | 'v2';
-    }): Promise<PaginationInfo & {
-        messages: MastraMessageV1[] | MastraMessageV2[];
-    }>;
-    saveMessages(args: {
-        messages: MastraMessageV1[];
-        format?: undefined | 'v1';
-    }): Promise<MastraMessageV1[]>;
-    saveMessages(args: {
-        messages: MastraMessageV2[];
-        format: 'v2';
-    }): Promise<MastraMessageV2[]>;
-    updateMessages({ messages, }: {
-        messages: (Partial<Omit<MastraMessageV2, 'createdAt'>> & {
-            id: string;
-            content?: {
-                metadata?: MastraMessageContentV2['metadata'];
-                content?: MastraMessageContentV2['content'];
-            };
-        })[];
-    }): Promise<MastraMessageV2[]>;
-    deleteMessages(messageIds: string[]): Promise<void>;
-    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>;
-    /**
-     * Workflows
-     */
-    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>;
-    close(): Promise<void>;
-    /**
-     * Scorers
-     */
-    getScoreById({ id: _id }: {
-        id: string;
-    }): Promise<ScoreRowData | null>;
-    getScoresByScorerId({ scorerId: _scorerId, pagination: _pagination, }: {
-        scorerId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    saveScore(_score: ScoreRowData): Promise<{
-        score: ScoreRowData;
-    }>;
-    getScoresByRunId({ runId: _runId, pagination: _pagination, }: {
-        runId: string;
-        pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByEntityId({ entityId: _entityId, entityType: _entityType, pagination: _pagination, }: {
-        pagination: StoragePagination;
-        entityId: string;
-        entityType: string;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-}
-export { PostgresStore }
-export { PostgresStore as PostgresStore_alias_1 }
-
-export declare class ScoresPG extends ScoresStorage {
-    client: IDatabase<{}>;
-    private operations;
-    constructor({ client, operations }: {
-        client: IDatabase<{}>;
-        operations: StoreOperationsPG;
-    });
-    getScoreById({ id }: {
-        id: string;
-    }): Promise<ScoreRowData | null>;
-    getScoresByScorerId({ scorerId, pagination, }: {
-        scorerId: string;
-        pagination: StoragePagination_2;
-    }): Promise<{
-        pagination: PaginationInfo_2;
-        scores: ScoreRowData[];
-    }>;
-    saveScore(score: Omit<ScoreRowData, 'createdAt' | 'updatedAt'>): Promise<{
-        score: ScoreRowData;
-    }>;
-    getScoresByRunId({ runId, pagination, }: {
-        runId: string;
-        pagination: StoragePagination_2;
-    }): Promise<{
-        pagination: PaginationInfo_2;
-        scores: ScoreRowData[];
-    }>;
-    getScoresByEntityId({ entityId, entityType, pagination, }: {
-        pagination: StoragePagination_2;
-        entityId: string;
-        entityType: string;
-    }): Promise<{
-        pagination: PaginationInfo_2;
-        scores: ScoreRowData[];
-    }>;
-}
-
-export declare class StoreOperationsPG extends StoreOperations {
-    client: IDatabase<{}>;
-    schemaName?: string;
-    private setupSchemaPromise;
-    private schemaSetupComplete;
-    constructor({ client, schemaName }: {
-        client: IDatabase<{}>;
-        schemaName?: string;
-    });
-    hasColumn(table: string, column: string): Promise<boolean>;
-    private setupSchema;
-    insert({ tableName, record }: {
-        tableName: TABLE_NAMES;
-        record: Record<string, any>;
-    }): Promise<void>;
-    clearTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    protected getDefaultValue(type: StorageColumn['type']): string;
-    createTable({ tableName, schema, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-    }): Promise<void>;
-    /**
-     * Alters table schema to add columns if they don't exist
-     * @param tableName Name of the table
-     * @param schema Schema of the table
-     * @param ifNotExists Array of column names to add if they don't exist
-     */
-    alterTable({ tableName, schema, ifNotExists, }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-        ifNotExists: string[];
-    }): Promise<void>;
-    load<R>({ tableName, keys }: {
-        tableName: TABLE_NAMES;
-        keys: Record<string, string>;
-    }): Promise<R | null>;
-    batchInsert({ tableName, records }: {
-        tableName: TABLE_NAMES;
-        records: Record<string, any>[];
-    }): Promise<void>;
-    dropTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-}
-
-export declare const TEST_CONFIG: PostgresConfig;
-
-export declare interface TestConfig {
-    dimension: number;
-    size: number;
-    k: number;
-    queryCount: number;
-}
-
-export declare interface TestResult {
-    distribution: string;
-    dimension: number;
-    type: IndexType;
-    size: number;
-    k?: number;
-    metrics: {
-        recall?: number;
-        minRecall?: number;
-        maxRecall?: number;
-        latency?: {
-            p50: number;
-            p95: number;
-            lists?: number;
-            vectorsPerList?: number;
-            m?: number;
-            ef?: number;
-        };
-        clustering?: {
-            numLists?: number;
-            avgVectorsPerList?: number;
-            recommendedLists?: number;
-            distribution?: string;
-        };
-    };
-}
-
-export declare class TracesPG extends TracesStorage {
-    client: IDatabase<{}>;
-    private operations;
-    private schema?;
-    constructor({ client, operations, schema, }: {
-        client: IDatabase<{}>;
-        operations: StoreOperationsPG;
-        schema?: string;
-    });
-    getTraces(args: StorageGetTracesArg): Promise<Trace[]>;
-    getTracesPaginated(args: StorageGetTracesPaginatedArg): Promise<PaginationInfo & {
-        traces: Trace[];
-    }>;
-    batchTraceInsert({ records }: {
-        records: Record<string, any>[];
-    }): Promise<void>;
-}
-
-export declare function warmupQuery(vectorDB: PgVector, indexName: string, dimension: number, k: number): Promise<void>;
-
-export declare class WorkflowsPG extends WorkflowsStorage {
-    client: IDatabase<{}>;
-    private operations;
-    private schema;
-    constructor({ client, operations, schema, }: {
-        client: IDatabase<{}>;
-        operations: StoreOperationsPG;
-        schema: string;
-    });
-    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
-        workflowName: string;
-        runId: string;
-        snapshot: WorkflowRunState_2;
-    }): Promise<void>;
-    loadWorkflowSnapshot({ workflowName, runId, }: {
-        workflowName: string;
-        runId: string;
-    }): Promise<WorkflowRunState_2 | null>;
-    getWorkflowRunById({ runId, workflowName, }: {
-        runId: string;
-        workflowName?: string;
-    }): Promise<WorkflowRun_2 | null>;
-    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId, }?: {
-        workflowName?: string;
-        fromDate?: Date;
-        toDate?: Date;
-        limit?: number;
-        offset?: number;
-        resourceId?: string;
-    }): Promise<WorkflowRuns_2>;
-}
-
-export { }