From 671a9a016ccc86bf5cb616518288c9a0c316adc3 Mon Sep 17 00:00:00 2001 From: zhuzhongshu123 <152354526+zhuzhongshu123@users.noreply.github.com> Date: Tue, 14 Jan 2025 10:18:38 +0800 Subject: [PATCH] fix mix reader (#270) --- kag/builder/component/reader/mix_reader.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/kag/builder/component/reader/mix_reader.py b/kag/builder/component/reader/mix_reader.py index 6af7380a..519d154f 100644 --- a/kag/builder/component/reader/mix_reader.py +++ b/kag/builder/component/reader/mix_reader.py @@ -52,7 +52,7 @@ class MixReader(ReaderABC): dict_reader (DictReader, optional): Reader for dictionary inputs. Defaults to None. """ super().__init__() - self.parse_map = { + self.reader_map = { "txt": txt_reader, "pdf": pdf_reader, "docx": docx_reader, @@ -83,11 +83,11 @@ class MixReader(ReaderABC): reader_type = "dict" else: - if os.path.exists(input): + if not os.path.exists(input): raise FileNotFoundError(f"File {input} not found.") file_suffix = input.split(".")[-1] - if file_suffix not in self.parse_map: + if file_suffix not in self.reader_map: raise NotImplementedError( f"File suffix {file_suffix} not supported yet." ) @@ -96,4 +96,4 @@ class MixReader(ReaderABC): reader = self.reader_map[reader_type] if reader is None: raise KeyError(f"{reader_type} reader not correctly configured.") - return self.parse_map[file_suffix]._invoke(input) + return reader._invoke(input)