diff --git a/api/extensions/storage/aws_s3_storage.py b/api/extensions/storage/aws_s3_storage.py index 507a303223..42ff434306 100644 --- a/api/extensions/storage/aws_s3_storage.py +++ b/api/extensions/storage/aws_s3_storage.py @@ -4,7 +4,6 @@ from collections.abc import Generator import boto3 from botocore.client import Config from botocore.exceptions import ClientError -from flask import Flask from extensions.storage.base_storage import BaseStorage @@ -14,8 +13,8 @@ logger = logging.getLogger(__name__) class AwsS3Storage(BaseStorage): """Implementation for Amazon Web Services S3 storage.""" - def __init__(self, app: Flask): - super().__init__(app) + def __init__(self): + super().__init__() app_config = self.app.config self.bucket_name = app_config.get("S3_BUCKET_NAME") if app_config.get("S3_USE_AWS_MANAGED_IAM"): diff --git a/api/extensions/storage/supabase_storage.py b/api/extensions/storage/supabase_storage.py index 1e399f87c8..1ec8c6b7f0 100644 --- a/api/extensions/storage/supabase_storage.py +++ b/api/extensions/storage/supabase_storage.py @@ -2,7 +2,6 @@ import io from collections.abc import Generator from pathlib import Path -from flask import Flask from supabase import Client from extensions.storage.base_storage import BaseStorage @@ -11,8 +10,8 @@ from extensions.storage.base_storage import BaseStorage class SupabaseStorage(BaseStorage): """Implementation for supabase obs storage.""" - def __init__(self, app: Flask): - super().__init__(app) + def __init__(self): + super().__init__() app_config = self.app.config self.bucket_name = app_config.get("SUPABASE_BUCKET_NAME") self.client = Client(