import { RAGApplicationBuilder } from '@llm-tools/embedjs';
import { OpenAiEmbeddings } from '@llm-tools/embedjs-openai';
import { LanceDb } from '@llm-tools/embedjs-lancedb';
import { WebLoader } from '@llm-tools/embedjs-loader-web';
// set OPENAI_API_KEY in your env
process.env.OPENAI_API_KEY = "sk-xxx";
const app = await new RAGApplicationBuilder()
.setEmbeddingModel(new OpenAiEmbeddings())
.setModel(SIMPLE_MODELS.OPENAI_GPT4_O)
.setVectorDatabase(
    new LanceDb({
        path: './lmdb',
    }),
)
.build();
//add data source and start query it
await app.addLoader(new WebLoader({ urlOrContent: 'https://www.forbes.com/profile/elon-musk' }));
await app.query('Tell me about Elon Musk');