import { RAGApplicationBuilder } from '@llm-tools/embedjs';
import { OpenAiEmbeddings } from '@llm-tools/embedjs-openai';
import { WebLoader } from '@llm-tools/embedjs-loader-web';
import { LibSqlDb } from '@llm-tools/embedjs-libsql';
// set OPENAI_API_KEY in your env
process.env.OPENAI_API_KEY = "sk-xxx";
const app = await new RAGApplicationBuilder()
.setVectorDatabase(new LibSqlDb({ path: './data.db' }))
.setEmbeddingModel(new OpenAiEmbeddings())
.setModel(SIMPLE_MODELS.OPENAI_GPT4_O)
.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');