Skip to content

Commit 69e1f9e

Browse files
committed
test: update API key names to avoid conflicts
- Changed `NG_OPENAI_API_KEY` to `DUMMY_OPENAI_API_KEY`. - Changed `NVIDIA_API_KEY` to `DUMMY_NVIDIA_API_KEY`. - Updated test cases to use dummy key names to prevent conflicts with user defined environment variables when running tests locally.
1 parent 112497e commit 69e1f9e

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

tests/test_rails_config.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from nemoguardrails.llm.prompts import TaskPrompt
2424
from nemoguardrails.rails.llm.config import Model, RailsConfig
2525

26-
TEST_API_KEY_NAME = "NG_OPENAI_API_KEY"
26+
TEST_API_KEY_NAME = "DUMMY_OPENAI_API_KEY"
2727
TEST_API_KEY_VALUE = "sk-svcacct-abcdefGHIJKlmnoPQRSTuvXYZ1234567890"
2828

2929

@@ -213,7 +213,10 @@ def test_model_api_key_value_valid_string():
213213

214214
@mock.patch.dict(
215215
os.environ,
216-
{TEST_API_KEY_NAME: TEST_API_KEY_VALUE, "NVIDIA_API_KEY": "nvapi-abcdef12345"},
216+
{
217+
TEST_API_KEY_NAME: TEST_API_KEY_VALUE,
218+
"DUMMY_NVIDIA_API_KEY": "nvapi-abcdef12345",
219+
},
217220
)
218221
def test_model_api_key_value_multiple_strings():
219222
"""Check if we reference a valid api_key_env_var we can create the Model"""
@@ -230,20 +233,20 @@ def test_model_api_key_value_multiple_strings():
230233
type="content_safety",
231234
engine="nim",
232235
model="nvidia/llama-3.1-nemoguard-8b-content-safety",
233-
api_key_env_var="NVIDIA_API_KEY",
236+
api_key_env_var="DUMMY_NVIDIA_API_KEY",
234237
),
235238
]
236239
)
237240
assert config.models[0].api_key_env_var == TEST_API_KEY_NAME
238-
assert config.models[1].api_key_env_var == "NVIDIA_API_KEY"
241+
assert config.models[1].api_key_env_var == "DUMMY_NVIDIA_API_KEY"
239242

240243

241244
@mock.patch.dict(os.environ, {TEST_API_KEY_NAME: TEST_API_KEY_VALUE})
242245
def test_model_api_key_value_multiple_strings_one_missing():
243246
"""Check if we have multiple models and one references an invalid api_key_env_var we throw error"""
244247
with pytest.raises(
245248
ValueError,
246-
match=f"Model API Key environment variable 'NVIDIA_API_KEY' not set.",
249+
match=f"Model API Key environment variable 'DUMMY_NVIDIA_API_KEY' not set.",
247250
):
248251
_ = RailsConfig(
249252
models=[
@@ -257,20 +260,20 @@ def test_model_api_key_value_multiple_strings_one_missing():
257260
type="content_safety",
258261
engine="nim",
259262
model="nvidia/llama-3.1-nemoguard-8b-content-safety",
260-
api_key_env_var="NVIDIA_API_KEY",
263+
api_key_env_var="DUMMY_NVIDIA_API_KEY",
261264
),
262265
]
263266
)
264267

265268

266269
@mock.patch.dict(
267-
os.environ, {TEST_API_KEY_NAME: TEST_API_KEY_VALUE, "NVIDIA_API_KEY": ""}
270+
os.environ, {TEST_API_KEY_NAME: TEST_API_KEY_VALUE, "DUMMY_NVIDIA_API_KEY": ""}
268271
)
269272
def test_model_api_key_value_multiple_strings_one_empty():
270273
"""Check if we have multiple models and one references an invalid api_key_env_var we throw error"""
271274
with pytest.raises(
272275
ValueError,
273-
match=f"Model API Key environment variable 'NVIDIA_API_KEY' not set.",
276+
match=f"Model API Key environment variable 'DUMMY_NVIDIA_API_KEY' not set.",
274277
):
275278
_ = RailsConfig(
276279
models=[
@@ -284,7 +287,7 @@ def test_model_api_key_value_multiple_strings_one_empty():
284287
type="content_safety",
285288
engine="nim",
286289
model="nvidia/llama-3.1-nemoguard-8b-content-safety",
287-
api_key_env_var="NVIDIA_API_KEY",
290+
api_key_env_var="DUMMY_NVIDIA_API_KEY",
288291
),
289292
]
290293
)

0 commit comments

Comments
 (0)