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.
"""
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)