2323from nemoguardrails .llm .prompts import TaskPrompt
2424from 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 "
2727TEST_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)
218221def 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 })
242245def 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)
269272def 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