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

feat: mypy for all type check (#10921)

Browse source
This commit is contained in:
yihong 2024-12-24 18:38:51 +08:00 • committed by GitHub
commit 56e15d09a9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
584 changed files with 3975 additions and 2826 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/extensions/storage/azure_blob_storage.py
View file

@ -27,7 +27,7 @@ class AzureBlobStorage(BaseStorage):
client = self._sync_client()
blob = client.get_container_client(container=self.bucket_name)
blob = blob.get_blob_client(blob=filename)
data = blob.download_blob().readall()
data: bytes = blob.download_blob().readall()
return data
def load_stream(self, filename: str) -> Generator:
@ -63,11 +63,11 @@ class AzureBlobStorage(BaseStorage):
sas_token = cache_result.decode("utf-8")
else:
sas_token = generate_account_sas(
account_name=self.account_name,
account_key=self.account_key,
account_name=self.account_name or "",
account_key=self.account_key or "",
resource_types=ResourceTypes(service=True, container=True, object=True),
permission=AccountSasPermissions(read=True, write=True, delete=True, list=True, add=True, create=True),
expiry=datetime.now(UTC).replace(tzinfo=None) + timedelta(hours=1),
)
redis_client.set(cache_key, sas_token, ex=3000)
return BlobServiceClient(account_url=self.account_url, credential=sas_token)
return BlobServiceClient(account_url=self.account_url or "", credential=sas_token)