dnrops
/
KAG
mirror of https://github.com/OpenSPG/KAG
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix mix reader (#270)

This commit is contained in:
zhuzhongshu123 2025-01-14 10:18:38 +08:00 committed by GitHub
parent c2056ef2f6
commit 671a9a016c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kag/builder/component/reader/mix_reader.py
View File

@ -52,7 +52,7 @@ class MixReader(ReaderABC):
dict_reader (DictReader, optional): Reader for dictionary inputs. Defaults to None. dict_reader (DictReader, optional): Reader for dictionary inputs. Defaults to None.
""" """
super().__init__() super().__init__()
self.parse_map = { self.reader_map = {
"txt": txt_reader, "txt": txt_reader,
"pdf": pdf_reader, "pdf": pdf_reader,
"docx": docx_reader, "docx": docx_reader,
@ -83,11 +83,11 @@ class MixReader(ReaderABC):
reader_type = "dict" reader_type = "dict"
else: else:
if os.path.exists(input): if not os.path.exists(input):
raise FileNotFoundError(f"File {input} not found.") raise FileNotFoundError(f"File {input} not found.")
file_suffix = input.split(".")[-1] file_suffix = input.split(".")[-1]
if file_suffix not in self.parse_map: if file_suffix not in self.reader_map:
raise NotImplementedError( raise NotImplementedError(
f"File suffix {file_suffix} not supported yet." f"File suffix {file_suffix} not supported yet."
) )
@ -96,4 +96,4 @@ class MixReader(ReaderABC):
reader = self.reader_map[reader_type] reader = self.reader_map[reader_type]
if reader is None: if reader is None:
raise KeyError(f"{reader_type} reader not correctly configured.") raise KeyError(f"{reader_type} reader not correctly configured.")
return self.parse_map[file_suffix]._invoke(input) return reader._invoke(input)