@forge-ml/rag/dist/src/simple/embedder.d.ts

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

import { Chunk, Embedder, Embedding } from "../../types";
type EmbedderOptions = {
    type: "openai";
    apiKey: string;
};
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[]>;
    /**
     * Embed an array of chunks.
     * @param chunks The chunks to embed.
     * @returns The embeddings for the chunks.
     */
    embedChunks(chunks: Chunk[]): Promise<Embedding[]>;
}
export default OpenAIEmbedder;