import {
Document,
Settings,
SimpleNodeParser,
SummaryIndex,
SummaryRetrieverMode,
} from "llamaindex";
import essay from "./essay";
Settings.nodeParser = new SimpleNodeParser({
chunkSize: 40,
});
async function main() {
const document = new Document({ text: essay, id_: "essay" });
const index = await SummaryIndex.fromDocuments([document]);
const queryEngine = index.asQueryEngine({
retriever: index.asRetriever({ mode: SummaryRetrieverMode.LLM }),
});
const response = await queryEngine.query({
query: "What did the author do growing up?",
});
console.log(response.toString());
}
main().catch((e: Error) => {
console.error(e, e.stack);
});