Skip to content

File loader

FileLoader

Bases: BaseLoader

Source code in griptape/griptape/loaders/file_loader.py
@define
class FileLoader(BaseLoader):
    encoding: str | None = field(default=None, kw_only=True)

    def load(self, path: str | Path) -> TextArtifact | BlobArtifact | ErrorArtifact:
        return self.file_to_artifact(path)

    def load_collection(self, paths: list[str | Path]) -> dict[str, TextArtifact | BlobArtifact | ErrorArtifact]:
        return utils.execute_futures_dict(
            {utils.str_to_hash(str(path)): self.futures_executor.submit(self.file_to_artifact, path) for path in paths}
        )

    def file_to_artifact(self, path: str | Path) -> TextArtifact | BlobArtifact | ErrorArtifact:
        file_name = os.path.basename(path)

        try:
            with open(path, "rb") as file:
                if self.encoding:
                    return TextArtifact(file.read().decode(self.encoding), name=file_name)
                else:
                    return BlobArtifact(file.read(), name=file_name, dir_name=os.path.dirname(path))
        except FileNotFoundError:
            return ErrorArtifact(f"file {file_name} not found")
        except Exception as e:
            return ErrorArtifact(f"error loading file: {e}")

encoding: str | None = field(default=None, kw_only=True) class-attribute instance-attribute

file_to_artifact(path)

Source code in griptape/griptape/loaders/file_loader.py
def file_to_artifact(self, path: str | Path) -> TextArtifact | BlobArtifact | ErrorArtifact:
    file_name = os.path.basename(path)

    try:
        with open(path, "rb") as file:
            if self.encoding:
                return TextArtifact(file.read().decode(self.encoding), name=file_name)
            else:
                return BlobArtifact(file.read(), name=file_name, dir_name=os.path.dirname(path))
    except FileNotFoundError:
        return ErrorArtifact(f"file {file_name} not found")
    except Exception as e:
        return ErrorArtifact(f"error loading file: {e}")

load(path)

Source code in griptape/griptape/loaders/file_loader.py
def load(self, path: str | Path) -> TextArtifact | BlobArtifact | ErrorArtifact:
    return self.file_to_artifact(path)

load_collection(paths)

Source code in griptape/griptape/loaders/file_loader.py
def load_collection(self, paths: list[str | Path]) -> dict[str, TextArtifact | BlobArtifact | ErrorArtifact]:
    return utils.execute_futures_dict(
        {utils.str_to_hash(str(path)): self.futures_executor.submit(self.file_to_artifact, path) for path in paths}
    )