/ modules / loaders.py
loaders.py
 1  from llama_index.readers.file import (
 2      DocxReader,
 3      PDFReader,
 4      MarkdownReader,
 5      PptxReader,
 6      UnstructuredReader,
 7      XMLReader,
 8      CSVReader
 9  )
10  
11  from llama_index.readers.json import (
12      JSONReader
13  )
14  
15  from restai.loaders.excel import PandasExcelReader
16  
17  LOADERS = {
18      ".csv": (CSVReader, {}),
19      ".docx": (DocxReader, {}),
20      ".eml": (UnstructuredReader, {}),
21      ".epub": (UnstructuredReader, {}),
22      ".html": (UnstructuredReader, {}),
23      ".md": (MarkdownReader, {}),
24      ".odt": (PandasExcelReader, {}),
25      ".pdf": (PDFReader, {}),
26      ".pptx": (PptxReader, {}),
27      ".txt": (UnstructuredReader, {}),
28      ".json": (JSONReader, {}),
29      ".xls": (PandasExcelReader, {}),
30      ".xlsx": (PandasExcelReader, {}),
31      ".xml": (XMLReader, {}),
32  }