use openai endpoint

microsoft · Nov 12, 2023 · d996eb5 · d996eb5
1 parent 486c2fe
commit d996eb5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/agentchat/contrib/test_gpt_assistant.py b/test/agentchat/contrib/test_gpt_assistant.py
@@ -13,6 +13,10 @@
 except ImportError:
     skip_test = True
 
+config_list = autogen.config_list_from_json(
+    OAI_CONFIG_LIST, file_location=KEY_LOC, filter_dict={"api_type": ["openai"]}
+)
+
 
 def ask_ossinsight(question):
     return f"That is a good question, but I don't know the answer yet. Please ask your human  developer friend to help you. \n\n{question}"
@@ -38,7 +42,6 @@ def test_gpt_assistant_chat():
         "description": "This is an API endpoint allowing users (analysts) to input question about GitHub in text format to retrieve the realted and structured data.",
     }
 
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, file_location=KEY_LOC)
     analyst = GPTAssistantAgent(
         name="Open_Source_Project_Analyst",
         llm_config={"tools": [{"type": "function", "function": ossinsight_api_schema}], "config_list": config_list},
@@ -73,7 +76,6 @@ def test_get_assistant_instructions():
     and assert that the retrieved instructions match the set instructions.
     """
 
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, file_location=KEY_LOC)
     assistant = GPTAssistantAgent(
         "assistant",
         instructions="This is a test",
@@ -107,7 +109,6 @@ def test_gpt_assistant_instructions_overwrite():
     instructions1 = "This is a test #1"
     instructions2 = "This is a test #2"
 
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, file_location=KEY_LOC)
     assistant = GPTAssistantAgent(
         "assistant",
         instructions=instructions1,
@@ -144,7 +145,6 @@ def test_gpt_assistant_existing_no_instructions():
     """
     instructions = "This is a test #1"
 
-    config_list = autogen.config_list_from_json(OAI_CONFIG_LIST, file_location=KEY_LOC)
     assistant = GPTAssistantAgent(
         "assistant",
         instructions=instructions,