Refactor storage service factory and service

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-27 23:27:19 -03:00
commit 3f1316b595
2 changed files with 8 additions and 9 deletions

View file

@ -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)

View file

@ -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):