diff --git a/examples/readers/src/custom-simple-directory-reader.ts b/examples/readers/src/custom-simple-directory-reader.ts index ff5bb726a0c25db1d85feff9a64bf48001dfd714..817425fbc8b9f04bff301f1a5b12e0ce888f60c2 100644 --- a/examples/readers/src/custom-simple-directory-reader.ts +++ b/examples/readers/src/custom-simple-directory-reader.ts @@ -1,12 +1,13 @@ -import type { BaseReader, Document, Metadata } from "llamaindex"; +import type { Document, Metadata } from "llamaindex"; +import { FileReader } from "llamaindex"; import { FILE_EXT_TO_READER, SimpleDirectoryReader, } from "llamaindex/readers/SimpleDirectoryReader"; import { TextFileReader } from "llamaindex/readers/TextFileReader"; -class ZipReader implements BaseReader { - loadData(...args: any[]): Promise<Document<Metadata>[]> { +class ZipReader extends FileReader { + loadDataAsContent(fileContent: Buffer): Promise<Document<Metadata>[]> { throw new Error("Implement me"); } }