Skip to content

Commit caf0091

Browse files
rename env name from local to dev
1 parent d7386d7 commit caf0091

File tree

4 files changed

+17
-17
lines changed

4 files changed

+17
-17
lines changed

infra/scripts/index_scripts/azure_credential_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from azure.identity import ManagedIdentityCredential, DefaultAzureCredential
22

3-
APP_ENV = 'prod' # Change to 'local' for local development
3+
APP_ENV = 'prod' # Change to 'dev' for local development
44

55
def get_azure_credential(client_id=None):
66
"""
@@ -16,7 +16,7 @@ def get_azure_credential(client_id=None):
1616
azure.identity.DefaultAzureCredential or azure.identity.ManagedIdentityCredential:
1717
The Azure credential object.
1818
"""
19-
if APP_ENV == 'local':
19+
if APP_ENV == 'dev':
2020
return DefaultAzureCredential() # CodeQL [SM05139] Okay use of DefaultAzureCredential as it is only used in development
2121
else:
2222
return ManagedIdentityCredential(client_id=client_id)

src/api/.env.sample

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,4 @@ SQLDB_USER_MID=
2323
SQLDB_USERNAME=
2424
USE_AI_PROJECT_CLIENT="False"
2525
USE_CHAT_HISTORY_ENABLED="True"
26-
APP_ENV="local"
26+
APP_ENV="dev"

src/api/helpers/azure_credential_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ async def get_azure_credential_async(client_id=None):
77
"""
88
Returns an Azure credential asynchronously based on the application environment.
99
10-
If the environment is 'local', it uses AioDefaultAzureCredential.
10+
If the environment is 'dev', it uses AioDefaultAzureCredential.
1111
Otherwise, it uses AioManagedIdentityCredential.
1212
1313
Args:
@@ -17,7 +17,7 @@ async def get_azure_credential_async(client_id=None):
1717
Credential object: Either AioDefaultAzureCredential or AioManagedIdentityCredential.
1818
"""
1919
config = Config()
20-
if config.app_env == 'local':
20+
if config.app_env == 'dev':
2121
return AioDefaultAzureCredential() # CodeQL [SM05139] Okay use of DefaultAzureCredential as it is only used in development
2222
else:
2323
return AioManagedIdentityCredential(client_id=client_id)
@@ -27,7 +27,7 @@ def get_azure_credential(client_id=None):
2727
"""
2828
Returns an Azure credential based on the application environment.
2929
30-
If the environment is 'local', it uses DefaultAzureCredential.
30+
If the environment is 'dev', it uses DefaultAzureCredential.
3131
Otherwise, it uses ManagedIdentityCredential.
3232
3333
Args:
@@ -37,7 +37,7 @@ def get_azure_credential(client_id=None):
3737
Credential object: Either DefaultAzureCredential or ManagedIdentityCredential.
3838
"""
3939
config = Config()
40-
if config.app_env == 'local':
40+
if config.app_env == 'dev':
4141
return DefaultAzureCredential() # CodeQL [SM05139] Okay use of DefaultAzureCredential as it is only used in development
4242
else:
4343
return ManagedIdentityCredential(client_id=client_id)

src/tests/api/helpers/test_azure_credential_utils.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,11 @@ class TestAzureCredentialUtils:
1111
@patch("helpers.azure_credential_utils.Config")
1212
@patch("helpers.azure_credential_utils.DefaultAzureCredential")
1313
@patch("helpers.azure_credential_utils.ManagedIdentityCredential")
14-
def test_get_azure_credential_local_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
15-
"""Test get_azure_credential in local environment."""
14+
def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
15+
"""Test get_azure_credential in dev environment."""
1616
# Arrange
1717
mock_config_instance = MagicMock()
18-
mock_config_instance.app_env = "local"
18+
mock_config_instance.app_env = "dev"
1919
mock_config.return_value = mock_config_instance
2020

2121
mock_default_credential = MagicMock()
@@ -33,8 +33,8 @@ def test_get_azure_credential_local_env(self, mock_managed_identity_credential,
3333
@patch("helpers.azure_credential_utils.Config")
3434
@patch("helpers.azure_credential_utils.DefaultAzureCredential")
3535
@patch("helpers.azure_credential_utils.ManagedIdentityCredential")
36-
def test_get_azure_credential_non_local_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
37-
"""Test get_azure_credential in non-local environment."""
36+
def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
37+
"""Test get_azure_credential in non-dev environment."""
3838
# Arrange
3939
mock_config_instance = MagicMock()
4040
mock_config_instance.app_env = "Prod"
@@ -56,11 +56,11 @@ def test_get_azure_credential_non_local_env(self, mock_managed_identity_credenti
5656
@patch("helpers.azure_credential_utils.Config")
5757
@patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
5858
@patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
59-
async def test_get_azure_credential_async_local_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
60-
"""Test get_azure_credential_async in local environment."""
59+
async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
60+
"""Test get_azure_credential_async in dev environment."""
6161
# Arrange
6262
mock_config_instance = MagicMock()
63-
mock_config_instance.app_env = "local"
63+
mock_config_instance.app_env = "dev"
6464
mock_config.return_value = mock_config_instance
6565

6666
mock_aio_default_credential = MagicMock()
@@ -79,8 +79,8 @@ async def test_get_azure_credential_async_local_env(self, mock_aio_managed_ident
7979
@patch("helpers.azure_credential_utils.Config")
8080
@patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
8181
@patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
82-
async def test_get_azure_credential_async_non_local_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
83-
"""Test get_azure_credential_async in non-local environment."""
82+
async def test_get_azure_credential_async_non_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
83+
"""Test get_azure_credential_async in non-dev environment."""
8484
# Arrange
8585
mock_config_instance = MagicMock()
8686
mock_config_instance.app_env = "Prod"

0 commit comments

Comments
 (0)