jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update save_file and get_file methods to handle binary data

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-26 22:50:03 -03:00
commit e6876e2ab0
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/services/storage/local.py
View file

@ -12,14 +12,14 @@ class LocalStorageService(StorageService):
self.set_ready() 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 = Path(f"{self.data_dir}/{folder}")
folder_path.mkdir(parents=True, exist_ok=True) 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) f.write(data)
def get_file(self, folder: str, file_name: str): def get_file(self, folder: str, file_name: str) -> bytes:
with open(f"{self.data_dir}/{folder}/{file_name}", "r") as f: with open(f"{self.data_dir}/{folder}/{file_name}", "rb") as f:
return f.read() return f.read()
def list_files(self, folder: str): def list_files(self, folder: str):