config list for test (microsoft#395)

whiskyboy · Oct 24, 2023 · 6520adb · 6520adb
1 parent 275023b
commit 6520adb
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,