ishaq101 commited on
Commit
c98bbd4
·
1 Parent(s): ff2d5ce

Update Azure Blob Service Config

Browse files
Files changed (1) hide show
  1. externals/storages/azure_blob.py +14 -5
externals/storages/azure_blob.py CHANGED
@@ -12,18 +12,27 @@ from utils.logger import get_logger
12
  logger = get_logger("azure-blob")
13
 
14
  # ---------- SINGLETONS ----------
15
- _blob_service_client: BlobServiceClient | None = None
16
- _credential: DefaultAzureCredential | None = None
 
 
 
 
 
 
 
 
 
17
 
18
 
19
  async def get_blob_service_client() -> BlobServiceClient:
20
- global _blob_service_client, _credential
21
 
22
  if _blob_service_client is None:
23
- _credential = DefaultAzureCredential()
24
  _blob_service_client = BlobServiceClient(
25
  account_url=AzureBlobConstants.ENDPOINT,
26
- credential=_credential,
27
  )
28
  logger.info("✅ Azure BlobServiceClient initialized")
29
 
 
12
  logger = get_logger("azure-blob")
13
 
14
  # ---------- SINGLETONS ----------
15
+ # _blob_service_client: BlobServiceClient | None = None
16
+ # _credential: DefaultAzureCredential | None = None
17
+
18
+ if "azureai__container__key" not in os.environ:
19
+ credential=DefaultAzureCredential()
20
+ logger.info(f"✅ Initialized Azure AI Cred: Using Default Credential")
21
+ else:
22
+ from azure.core.credentials import AzureKeyCredential
23
+ # credential = AzureKeyCredential(os.environ.get('azureai__container__key'))
24
+ credential = AzureKeyCredential(AzureBlobConstants.SAS_KEY)
25
+ logger.info(f"✅ Initialized Azure AI Cred: Using Azure Key Credential")
26
 
27
 
28
  async def get_blob_service_client() -> BlobServiceClient:
29
+ # global _blob_service_client, _credential
30
 
31
  if _blob_service_client is None:
32
+ # _credential = DefaultAzureCredential()
33
  _blob_service_client = BlobServiceClient(
34
  account_url=AzureBlobConstants.ENDPOINT,
35
+ credential=credential,
36
  )
37
  logger.info("✅ Azure BlobServiceClient initialized")
38