From 42356fc379687cbc1c26e67b01d8c7d73023ccc8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 30 Jan 2024 17:17:36 -0300 Subject: [PATCH] Add build_full_path method to LocalStorageService and stub method to StorageService --- src/backend/langflow/services/storage/local.py | 4 ++++ src/backend/langflow/services/storage/service.py | 3 +++ 2 files changed, 7 insertions(+) diff --git a/src/backend/langflow/services/storage/local.py b/src/backend/langflow/services/storage/local.py index 667a7aeeb..a3fa6887b 100644 --- a/src/backend/langflow/services/storage/local.py +++ b/src/backend/langflow/services/storage/local.py @@ -15,6 +15,10 @@ class LocalStorageService(StorageService): self.data_dir = Path(settings_service.settings.CONFIG_DIR) self.set_ready() + def build_full_path(self, flow_id: str, file_name: str) -> str: + """Build the full path of a file in the local storage.""" + return str(self.data_dir / flow_id / file_name) + async def save_file(self, flow_id: str, file_name: str, data: bytes): """ Save a file in the local storage. diff --git a/src/backend/langflow/services/storage/service.py b/src/backend/langflow/services/storage/service.py index 8d30ede61..830557dbf 100644 --- a/src/backend/langflow/services/storage/service.py +++ b/src/backend/langflow/services/storage/service.py @@ -16,6 +16,9 @@ class StorageService(Service): self.session_service = session_service self.set_ready() + def build_full_path(self, flow_id: str, file_name: str) -> str: + pass + def set_ready(self): self.ready = True