AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 13:39:04 +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
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 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):