Merge pull request #151 from BroKun/brokun_dev

fix: override _load_data instead of load_data
antgroup · Aug 27, 2024 · c07f50b · c07f50b
2 parents 4ff3a25 + 4b75cab
commit c07f50b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agentuniverse/agent/action/knowledge/reader/file/txt_reader.py b/agentuniverse/agent/action/knowledge/reader/file/txt_reader.py
@@ -8,7 +8,7 @@
 
 class LineTxtReader(Reader):
 
-    def load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
+    def _load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
         dlist = []
 
         with open(fpath, 'r', encoding='utf-8') as file:
@@ -26,7 +26,7 @@ def load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Docume
 class TxtReader(Reader):
     """Txt reader."""
 
-    def load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
+    def _load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
 
         with open(fpath, 'r', encoding='utf-8') as file: