@forge-ml/rag/dist/src/embedders/openaiEmbedder.d.ts

A RAG (Retrieval-Augmented Generation) package for Forge ML

import { Embedder, EmbedderOptions, Chunk, Embedding } from "../../types";
declare class OpenAIEmbedder implements Embedder {
    private openai;
    constructor({ apiKey }: EmbedderOptions);
    /**
     * Generate an embedding for a given text.
     * @param text The text to generate an embedding for.
     * @returns The embedding for the text.
     */
    generateEmbedding(text: string): Promise<number[]>;
    generateEmbeddingBatch(texts: string[]): Promise<number[][]>;
    /**
     * Embed an array of chunks.
     * @param chunks The chunks to embed.
     * @returns The embeddings for the chunks.
     */
    embedChunks(chunks: Chunk[], documentId: string): Promise<Embedding[]>;
}
export default OpenAIEmbedder;