AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 17:49:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: mypy checks violation in AzureBlobStorage (#13215)

Browse Source
This commit is contained in:
Bowen Liang 2025-02-05 15:56:23 +08:00 committed by GitHub
parent fac83e14bc
commit 9e1457c2c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/extensions/storage/azure_blob_storage.py
View File

@ -1,7 +1,8 @@
from collections.abc import Generator from collections.abc import Generator
from datetime import UTC, datetime, timedelta from datetime import UTC, datetime, timedelta
from typing import Optional
from azure.identity import DefaultAzureCredential from azure.identity import ChainedTokenCredential, DefaultAzureCredential
from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas
from configs import dify_config from configs import dify_config
@ -19,6 +20,7 @@ class AzureBlobStorage(BaseStorage):
self.account_name = dify_config.AZURE_BLOB_ACCOUNT_NAME self.account_name = dify_config.AZURE_BLOB_ACCOUNT_NAME
self.account_key = dify_config.AZURE_BLOB_ACCOUNT_KEY self.account_key = dify_config.AZURE_BLOB_ACCOUNT_KEY
self.credential: Optional[ChainedTokenCredential] = None
if self.account_key == "managedidentity": if self.account_key == "managedidentity":
self.credential = DefaultAzureCredential() self.credential = DefaultAzureCredential()
else: else:
@ -64,7 +66,7 @@ class AzureBlobStorage(BaseStorage):
def _sync_client(self): def _sync_client(self):
if self.account_key == "managedidentity": if self.account_key == "managedidentity":
return BlobServiceClient(account_url=self.account_url, credential=self.credential) return BlobServiceClient(account_url=self.account_url, credential=self.credential) # type: ignore
cache_key = "azure_blob_sas_token_{}_{}".format(self.account_name, self.account_key) cache_key = "azure_blob_sas_token_{}_{}".format(self.account_name, self.account_key)
cache_result = redis_client.get(cache_key) cache_result = redis_client.get(cache_key)