diff --git a/api/extensions/storage/azure_storage.py b/api/extensions/storage/azure_storage.py index a712a8bbdb..3403bc6171 100644 --- a/api/extensions/storage/azure_storage.py +++ b/api/extensions/storage/azure_storage.py @@ -1,5 +1,4 @@ from collections.abc import Generator -from contextlib import closing from datetime import datetime, timedelta, timezone from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas @@ -38,10 +37,9 @@ class AzureStorage(BaseStorage): def generate(filename: str = filename) -> Generator: blob = client.get_blob_client(container=self.bucket_name, blob=filename) - with closing(blob.download_blob()) as blob_stream: - while chunk := blob_stream.readall(): - yield chunk - + blob_data = blob.download_blob() + for chunk in blob_data.chunks(): + yield from chunk return generate() def download(self, filename, target_filepath):