config list for test (#395)

microsoft · Oct 24, 2023 · 9ed6c8e · 9ed6c8e
1 parent f9acb69
commit 9ed6c8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/oai/test_completion.py b/test/oai/test_completion.py
@@ -376,7 +376,7 @@ def test_math(num_samples=-1):
     ]
 
     autogen.Completion.set_cache(seed)
-    config_list = autogen.config_list_openai_aoai(KEY_LOC, exclude="aoai")
+    config_list = autogen.config_list_openai_aoai(KEY_LOC)[:2]
     vanilla_config = {
         "model": "text-davinci-003",
         "temperature": 1,