Add support for markdown files (#270)

Quansight · Jan 16, 2024 · 18a6a25 · 18a6a25
1 parent 20ecade
commit 18a6a25
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/ragna/core/__init__.py b/ragna/core/__init__.py
@@ -17,7 +17,7 @@
     "Requirement",
     "Source",
     "SourceStorage",
-    "TxtDocumentHandler",
+    "PlainTextDocumentHandler",
 ]
 
 from ._utils import (
@@ -36,7 +36,7 @@
     LocalDocument,
     Page,
     PdfDocumentHandler,
-    TxtDocumentHandler,
+    PlainTextDocumentHandler,
 )
 
 # isort: split

diff --git a/ragna/core/_document.py b/ragna/core/_document.py
@@ -229,12 +229,14 @@ def load_if_available(self, cls: Type[T]) -> Type[T]:
 
 
 @DOCUMENT_HANDLERS.load_if_available
-class TxtDocumentHandler(DocumentHandler):
-    """Document handler for `.txt` documents."""
+class PlainTextDocumentHandler(DocumentHandler):
+    """Document handler for plain-text documents.
+    Currently supports `.txt` and `.md` extensions.
+    """
 
     @classmethod
     def supported_suffixes(cls) -> list[str]:
-        return [".txt"]
+        return [".txt", ".md"]
 
     def extract_pages(self, document: Document) -> Iterator[Page]:
         yield Page(text=document.read().decode())