From f232efe179e3fb21f41206c74a4323b38ef21f8b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 29 Jan 2024 22:49:43 -0300
Subject: [PATCH] Convert save_file, get_file, list_files, delete_file, and
 teardown methods to async functions in S3StorageService

---
 src/backend/langflow/services/storage/s3.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/backend/langflow/services/storage/s3.py b/src/backend/langflow/services/storage/s3.py
index 8a6a24f3f..afd39760a 100644
--- a/src/backend/langflow/services/storage/s3.py
+++ b/src/backend/langflow/services/storage/s3.py
@@ -8,14 +8,14 @@ from .service import StorageService
 class S3StorageService(StorageService):
     """A service class for handling operations with AWS S3 storage."""
 
-    def __init__(self, session_service, settings_service):
+    async def __init__(self, session_service, settings_service):
         """Initialize the S3 storage service with session and settings services."""
         super().__init__(session_service, settings_service)
         self.bucket = "langflow"
         self.s3_client = boto3.client("s3")
         self.set_ready()
 
-    def save_file(self, folder: str, file_name: str, data):
+    async def save_file(self, folder: str, file_name: str, data):
         """
         Save a file to the S3 bucket.
 
@@ -34,7 +34,7 @@ class S3StorageService(StorageService):
             logger.error(f"Error saving file {file_name} in folder {folder}: {e}")
             raise
 
-    def get_file(self, folder: str, file_name: str):
+    async def get_file(self, folder: str, file_name: str):
         """
         Retrieve a file from the S3 bucket.
 
@@ -51,7 +51,7 @@ class S3StorageService(StorageService):
             logger.error(f"Error retrieving file {file_name} from folder {folder}: {e}")
             raise
 
-    def list_files(self, folder: str):
+    async def list_files(self, folder: str):
         """
         List all files in a specified folder of the S3 bucket.
 
@@ -68,7 +68,7 @@ class S3StorageService(StorageService):
             logger.error(f"Error listing files in folder {folder}: {e}")
             raise
 
-    def delete_file(self, folder: str, file_name: str):
+    async def delete_file(self, folder: str, file_name: str):
         """
         Delete a file from the S3 bucket.
 
@@ -83,7 +83,7 @@ class S3StorageService(StorageService):
             logger.error(f"Error deleting file {file_name} from folder {folder}: {e}")
             raise
 
-    def teardown(self):
+    async def teardown(self):
         """Perform any cleanup operations when the service is being torn down."""
         # No specific teardown actions required for S3 storage at the moment.
         pass