@masuidrive/bloom-local-rag/dist/lib/embeddings.js

RAG (Retrieval-Augmented Generation) system for local directories - Index and search your documents with AI-powered answers

import { GoogleGenerativeAIEmbeddings } from '@langchain/google-genai';
import { OpenAIEmbeddings } from '@langchain/openai';
import { getApiKey } from './config.js';
export function createEmbeddings(config) {
    const apiKey = getApiKey(config.embedding.provider);
    if (config.embedding.provider === 'gemini') {
        return new GoogleGenerativeAIEmbeddings({
            apiKey,
            model: config.embedding.model,
        });
    }
    else {
        return new OpenAIEmbeddings({
            apiKey,
            model: config.embedding.model,
        });
    }
}
//# sourceMappingURL=embeddings.js.map