Skip to content

Commit 6449318

Browse files
refactor: update credential retrieval logic based on environment
1 parent d478cd6 commit 6449318

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

src/mcp_server/services/image_service.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
import base64
1010
import logging
11+
import os
1112
import uuid
1213

1314

@@ -28,10 +29,11 @@
2829

2930

3031
def _get_credential():
31-
"""Return a credential, preferring user-assigned MI when a client id is set."""
32-
if config.azure_client_id:
33-
return ManagedIdentityCredential(client_id=config.azure_client_id)
34-
return DefaultAzureCredential()
32+
"""Return a credential based on environment (dev vs deployed)."""
33+
app_env = os.environ.get("APP_ENV", "prod").lower()
34+
if app_env == "dev":
35+
return DefaultAzureCredential(require_envvar=True)
36+
return ManagedIdentityCredential(client_id=config.azure_client_id)
3537

3638

3739
def _ensure_container(blob_service: BlobServiceClient, container_name: str) -> None:

0 commit comments

Comments
 (0)