config list for test

sonichi · sonichi · commit 0c202d32ee9b · 2023-10-24T02:43:28.000Z
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,

Original file line number	Diff line number	Diff line change
`@@ -376,7 +376,7 @@ def test_math(num_samples=-1):`
`376`	`376`	`]`
`377`	`377`
`378`	`378`	`autogen.Completion.set_cache(seed)`
`379`		`- config_list = autogen.config_list_openai_aoai(KEY_LOC, exclude="aoai")`
	`379`	`+ config_list = autogen.config_list_openai_aoai(KEY_LOC)[:2]`
`380`	`380`	`vanilla_config = {`
`381`	`381`	`"model": "text-davinci-003",`
`382`	`382`	`"temperature": 1,`