hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-13 04:46:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix azure stream download (#6063)

Browse Source
This commit is contained in:
Jyong
2024-07-08 17:13:16 +08:00
committed by GitHub
parent 603187393a
commit cbbe28f40d
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/extensions/storage/azure_storage.py
View File

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