use apu key function

microsoft · Mar 15, 2024 · a250c3d · a250c3d
1 parent cf02c1a
commit a250c3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/agentchat/contrib/capabilities/test_image_generation_capability.py b/test/agentchat/contrib/capabilities/test_image_generation_capability.py
@@ -61,7 +61,7 @@ def api_key():
 def dalle_config() -> Dict[str, Any]:
     config_list = openai_utils.config_list_from_models(model_list=["dall-e-2"], exclude="aoai")
     if not config_list:
-        config_list = [{"model": "dall-e-2", "api_key": MOCK_OPEN_AI_API_KEY}]
+        config_list = [{"model": "dall-e-2", "api_key": api_key()}]
     return {"config_list": config_list, "timeout": 120, "cache_seed": None}