Spaces:
Running
Running
Update Azure Blob Service Config
Browse files
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=
|
| 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 |
|