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

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

import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
import { ChatOpenAI } from '@langchain/openai';
import { getApiKey } from './config.js';
export function createLLM(config) {
    const apiKey = getApiKey(config.llm.provider);
    if (config.llm.provider === 'gemini') {
        return new ChatGoogleGenerativeAI({
            apiKey,
            model: config.llm.model,
            temperature: config.llm.temperature,
        });
    }
    else {
        return new ChatOpenAI({
            apiKey,
            model: config.llm.model,
            temperature: config.llm.temperature,
        });
    }
}
//# sourceMappingURL=llm.js.map