From 0c202d32ee9b7aa36d23c4ff543c6250e526afbf Mon Sep 17 00:00:00 2001
From: Chi Wang <wang.chi@microsoft.com>
Date: Tue, 24 Oct 2023 02:43:28 +0000
Subject: [PATCH] config list for test

---
 test/oai/test_completion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/oai/test_completion.py b/test/oai/test_completion.py
index 934dbadc901..f33533b7bf8 100644
--- 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,