File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change 1818from llama_index .core .node_parser import SemanticSplitterNodeParser
1919from llama_index .readers .github import GithubRepositoryReader , GithubClient # type: ignore
2020from llama_index .readers .file import PDFReader # type: ignore
21+ from llama_index .readers .pdf_marker import PDFMarkerReader # type: ignore
2122from llama_index .core .postprocessor import SentenceTransformerRerank
2223from dotenv import load_dotenv
2324import tiktoken
@@ -220,6 +221,9 @@ def __init__(
220221 # pdf_loader = download_loader("PDFReader")
221222 pdf_loader = PDFReader ()
222223 paper_documents = pdf_loader .load_data (file = Path (self ._file_path ))
224+ case "PDFMarker" :
225+ pdf_loader = PDFMarkerReader ()
226+ paper_documents = pdf_loader .load_data (file = Path (self ._file_path ))
223227
224228 documents = paper_documents # type: ignore
225229 if self ._git_data is not None :
You can’t perform that action at this time.
0 commit comments