diff --git a/src/backend/langflow/services/storage/factory.py b/src/backend/langflow/services/storage/factory.py index 1616c69b9..be407d751 100644 --- a/src/backend/langflow/services/storage/factory.py +++ b/src/backend/langflow/services/storage/factory.py @@ -1,9 +1,8 @@ from typing import TYPE_CHECKING -from loguru import logger - from langflow.services.factory import ServiceFactory from langflow.services.storage.service import StorageService +from loguru import logger if TYPE_CHECKING: from langflow.services.session.service import SessionService @@ -19,13 +18,13 @@ class StorageServiceFactory(ServiceFactory): if storage_type.lower() == "local": from .local import LocalStorageService - return LocalStorageService(session_service) + return LocalStorageService(session_service, settings_service) elif storage_type.lower() == "s3": from .s3 import S3StorageService - return S3StorageService(session_service) + return S3StorageService(session_service, settings_service) else: logger.warning(f"Storage type {storage_type} not supported. Using local storage.") from .local import LocalStorageService - return LocalStorageService(session_service) + return LocalStorageService(session_service, settings_service) diff --git a/src/backend/langflow/services/storage/service.py b/src/backend/langflow/services/storage/service.py index d8fc2f476..f59fa2aa1 100644 --- a/src/backend/langflow/services/storage/service.py +++ b/src/backend/langflow/services/storage/service.py @@ -20,19 +20,19 @@ class StorageService(Service): self.ready = True @abstractmethod - def save_file(self, folder: str, file_name: str, data) -> None: + def save_file(self, flow_id: str, file_name: str, data) -> None: pass @abstractmethod - def get_file(self, folder: str, file_name: str) -> bytes: + def get_file(self, flow_id: str, file_name: str) -> bytes: pass @abstractmethod - def list_files(self, folder: str) -> list[str]: + def list_files(self, flow_id: str) -> list[str]: pass @abstractmethod - def delete_file(self, folder: str, file_name: str) -> bool: + def delete_file(self, flow_id: str, file_name: str) -> bool: pass def teardown(self):