From cbbe28f40db088c5c6ae00b7c7f1820b24cf6401 Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Mon, 8 Jul 2024 17:13:16 +0800
Subject: [PATCH] fix azure stream download (#6063)

---
 api/extensions/storage/azure_storage.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

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