mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-12 18:51:33 +00:00
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import json
|
|
import os
|
|
import autogen
|
|
from test_completion import KEY_LOC, OAI_CONFIG_LIST
|
|
|
|
|
|
def test_config_list_from_json():
|
|
config_list = autogen.config_list_gpt4_gpt35(key_file_path=KEY_LOC)
|
|
json_file = os.path.join(KEY_LOC, "config_list_test.json")
|
|
with open(json_file, "w") as f:
|
|
json.dump(config_list, f, indent=4)
|
|
config_list_1 = autogen.config_list_from_json(json_file)
|
|
assert config_list == config_list_1
|
|
os.environ["config_list_test"] = json.dumps(config_list)
|
|
config_list_2 = autogen.config_list_from_json("config_list_test")
|
|
assert config_list == config_list_2
|
|
config_list_3 = autogen.config_list_from_json(
|
|
OAI_CONFIG_LIST, file_location=KEY_LOC, filter_dict={"model": ["gpt4", "gpt-4-32k"]}
|
|
)
|
|
assert all(config.get("model") in ["gpt4", "gpt-4-32k"] for config in config_list_3)
|
|
del os.environ["config_list_test"]
|
|
os.remove(json_file)
|
|
|
|
|
|
def test_config_list_openai_aoai():
|
|
config_list = autogen.config_list_openai_aoai(key_file_path=KEY_LOC)
|
|
assert all(config.get("api_type") in [None, "open_ai", "azure"] for config in config_list)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_config_list_from_json()
|