Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 03:32:23 +08:00
Code Issues Actions 4 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats 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 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):