Mirrors/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2026-02-23 19:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: update empty loader check

Browse Source
This commit is contained in:
trducng
2024-11-13 06:34:59 +00:00
parent f701a5eefd
commit e3c4e39db5
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/ktem/ktem/index/file/pipelines.py
View File

@@ -545,6 +545,13 @@ class IndexPipeline(BaseComponent):
def stream(
self, file_path: str | Path, reindex: bool, **kwargs
) -> Generator[Document, None, tuple[str, list[Document]]]:
if self.loader is None:
raise NotImplementedError(
f"No supported pipeline to index {Path(file_path).name}. "
"Please specify "
"the suitable pipeline for this file type in the settings."
)
# check for duplication
file_path = Path(file_path).resolve()
file_id = self.get_id_if_exists(file_path)
@@ -617,11 +624,6 @@ class IndexDocumentPipeline(BaseFileIndexIndexing):
ext = file_path.suffix
reader = readers.get(ext, KH_DEFAULT_FILE_EXTRACTORS.get(ext, None))
if reader is None:
raise NotImplementedError(
f"No supported pipeline to index {file_path.name}. Please specify "
"the suitable pipeline for this file type in the settings."
)
pipeline: IndexPipeline = IndexPipeline(
loader=reader,