diff --git a/examples/package.json b/examples/package.json index d044cb7240ba6e0234986e4143bc0d021f716d6c..e0bf64435d9b56ff4d11c4b1b01cf55e57dba5cc 100644 --- a/examples/package.json +++ b/examples/package.json @@ -33,7 +33,8 @@ "typescript": "^5.7.2" }, "scripts": { - "lint": "eslint ." + "lint": "eslint .", + "start": "tsx ./starter.ts" }, "stackblitz": { "startCommand": "npm start" diff --git a/examples/starter.ts b/examples/starter.ts index 371575a9a336e34b8ead5927c70cc6f5c0e7e521..e73e8dee8aa06b6059b1ed8601ace16f55b044ca 100644 --- a/examples/starter.ts +++ b/examples/starter.ts @@ -1,16 +1,22 @@ -import { SimpleDirectoryReader, VectorStoreIndex } from "llamaindex"; +import { Document, VectorStoreIndex } from "llamaindex"; +import fs from "node:fs/promises"; import { createInterface } from "node:readline/promises"; async function main() { - const reader = new SimpleDirectoryReader(); + const path = "node_modules/llamaindex/examples/abramov.txt"; + const essay = await fs.readFile(path, "utf-8"); + const document = new Document({ text: essay, id_: path }); - const documents = await reader.loadData("./data"); - - const index = await VectorStoreIndex.fromDocuments(documents); + const index = await VectorStoreIndex.fromDocuments([document]); const queryEngine = index.asQueryEngine(); const rl = createInterface({ input: process.stdin, output: process.stdout }); + console.log( + "Try asking a question about the essay: https://github.com/run-llama/LlamaIndexTS/blob/main/packages/llamaindex/examples/abramov.txt", + "\nExample: When did the author graduate from high school?", + "\n==============================\n", + ); while (true) { const query = await rl.question("Query: "); const response = await queryEngine.query({