[PDFMardownReader] Cleaned code

Author: YanSte

pymupdf4llm/pymupdf4llm/llama_index/pdf_markdown_reader.py

@@ -2,17 +2,16 @@
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Union
 
+import fitz
+from fitz import Document as FitzDocument
+
 try:
-    import fitz
-    from fitz import Document as FitzDocument
     from llama_index.core.readers.base import BaseReader
     from llama_index.core.schema import Document as LlamaIndexDocument
 
     print("All imports are successful.")
 except ImportError:
-    raise NotImplementedError(
-        "Please install 'llama_index' version 1.24.0 or later is needed."
-    )
+    raise NotImplementedError("Please install 'llama_index' is needed.")
 
 
 import pymupdf4llm

tests/pymupdf4llm/llama_index/test_pdf_markdown_reader.py

@@ -1,11 +1,18 @@
 import os
 
 import pytest
-from gary.services.rag.reader.pdf import LlamaIndexDocument
+from llama_index.core.schema import Document as LlamaIndexDocument
 
-from pymupdf4llm.pymupdf4llm.llama_index.pdf_markdown_reader import PDFMardownReader
+try:
+    from llama_index.core.readers.base import BaseReader
+    from llama_index.core.schema import Document as LlamaIndexDocument
+
+    print("All imports are successful.")
+except ImportError:
+    raise NotImplementedError("Please install 'llama_index' is needed.")
 
-# Temporay.
+
+from pymupdf4llm.pymupdf4llm.llama_index.pdf_markdown_reader import PDFMardownReader
 
 PDF = "input.pdf"