src/cloud_storage/__init__.py

from .google_cloud_storage import GoogleCloudStorage

storage_map = {
    "gcp": GoogleCloudStorage,
}


def get_cloud_storage(ingestion_mode):
    if ingestion_mode not in storage_map:
        raise ValueError(f"Invalid storage mode: {ingestion_mode}")
    return storage_map[ingestion_mode]()