jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix/file stream azure blob (#6196)

Browse source
This commit is contained in:
Jyong 2024-07-11 17:01:03 +08:00 • committed by GitHub
commit 678ad6b7eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/extensions/storage/azure_storage.py
View file

@ -38,9 +38,8 @@ 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)
blob_data = blob.download_blob() blob_data = blob.download_blob()
for chunk in blob_data.chunks(): yield from blob_data.chunks()
yield from chunk return generate(filename)
return generate()
def download(self, filename, target_filepath): def download(self, filename, target_filepath):
client = self._sync_client() client = self._sync_client()