import { EmbeddingFunction, EmbeddingFunctionSpace } from 'chromadb';

interface SentenceTransformersConfig {
    model_name: string;
    device: string;
    normalize_embeddings: boolean;
    kwargs?: Record<string, any>;
}
interface SentenceTransformersArgs {
    modelName?: string;
    device?: string;
    normalizeEmbeddings?: boolean;
    kwargs?: Record<string, any>;
}
declare class SentenceTransformersEmbeddingFunction implements EmbeddingFunction {
    readonly name = "sentence_transformer";
    private readonly modelName;
    private readonly device;
    private readonly normalizeEmbeddings;
    private readonly kwargs;
    private pipelinePromise;
    private pipeline;
    constructor(args?: SentenceTransformersArgs);
    private getPipeline;
    generate(texts: string[]): Promise<number[][]>;
    dispose(): Promise<void>;
    defaultSpace(): EmbeddingFunctionSpace;
    supportedSpaces(): EmbeddingFunctionSpace[];
    static buildFromConfig(config: SentenceTransformersConfig): SentenceTransformersEmbeddingFunction;
    getConfig(): SentenceTransformersConfig;
    validateConfigUpdate(newConfig: Record<string, any>): void;
    static validateConfig(config: SentenceTransformersConfig): void;
}

export { type SentenceTransformersArgs, type SentenceTransformersConfig, SentenceTransformersEmbeddingFunction };