From e6876e2ab0106233679e3fef936863d3a6bde161 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 26 Jan 2024 22:50:03 -0300 Subject: [PATCH] Update save_file and get_file methods to handle binary data --- src/backend/langflow/services/storage/local.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/services/storage/local.py b/src/backend/langflow/services/storage/local.py index d13f3fd7f..ddb65b4d5 100644 --- a/src/backend/langflow/services/storage/local.py +++ b/src/backend/langflow/services/storage/local.py @@ -12,14 +12,14 @@ class LocalStorageService(StorageService): self.set_ready() - def save_file(self, folder: str, file_name: str, data): + def save_file(self, folder: str, file_name: str, data: bytes): folder_path = Path(f"{self.data_dir}/{folder}") folder_path.mkdir(parents=True, exist_ok=True) - with open(f"{self.data_dir}/{folder}/{file_name}", "w") as f: + with open(f"{self.data_dir}/{folder}/{file_name}", "wb") as f: f.write(data) - def get_file(self, folder: str, file_name: str): - with open(f"{self.data_dir}/{folder}/{file_name}", "r") as f: + def get_file(self, folder: str, file_name: str) -> bytes: + with open(f"{self.data_dir}/{folder}/{file_name}", "rb") as f: return f.read() def list_files(self, folder: str):