From 678ad6b7eb2e80aaed7ceeea6da4fbed93d3544b Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Thu, 11 Jul 2024 17:01:03 +0800 Subject: [PATCH] Fix/file stream azure blob (#6196) --- api/extensions/storage/azure_storage.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/api/extensions/storage/azure_storage.py b/api/extensions/storage/azure_storage.py index 3403bc6171..af3e7ef849 100644 --- a/api/extensions/storage/azure_storage.py +++ b/api/extensions/storage/azure_storage.py @@ -38,9 +38,8 @@ class AzureStorage(BaseStorage): def generate(filename: str = filename) -> Generator: blob = client.get_blob_client(container=self.bucket_name, blob=filename) blob_data = blob.download_blob() - for chunk in blob_data.chunks(): - yield from chunk - return generate() + yield from blob_data.chunks() + return generate(filename) def download(self, filename, target_filepath): client = self._sync_client()