Method set_reader

KVCOMM/tools/reader/readers.py:307–309

(self, suffix)

class FileReader:
    def set_reader(self, suffix) -> None:
        self.reader = READER_MAP[suffix]
        logger.info(f"Setting Reader to {type(self.reader).__name__}")

    def read_file(self, file_path: Path, task="describe the file") -> str:
        suffix = '.' + file_path.split(".")[-1]