Skip to content

Commit

Permalink
update default aoai version (#1229)
Browse files Browse the repository at this point in the history
* update default aoai version

* disable notebook test that is going to be outdated

* bump version to 0.2.7
  • Loading branch information
sonichi authored and joshkyh committed Jan 17, 2024
1 parent 79cacb7 commit 405f5c0
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 11 deletions.
14 changes: 7 additions & 7 deletions autogen/oai/openai_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
Assistant = object

NON_CACHE_KEY = ["api_key", "base_url", "api_type", "api_version"]
DEFAULT_AZURE_API_VERSION = "2023-08-01-preview"
DEFAULT_AZURE_API_VERSION = "2023-12-01-preview"
OAI_PRICE1K = {
"text-ada-001": 0.0004,
"text-babbage-001": 0.0005,
Expand Down Expand Up @@ -96,7 +96,7 @@ def get_config_list(
# Optionally, define the API type and version if they are common for all keys
api_type = 'azure'
api_version = '2023-08-01-preview'
api_version = '2023-12-01-preview'
# Call the get_config_list function to get a list of configuration dictionaries
config_list = get_config_list(api_keys, base_urls, api_type, api_version)
Expand Down Expand Up @@ -305,11 +305,11 @@ def config_list_from_models(
```
"""
config_list = config_list_openai_aoai(
key_file_path,
openai_api_key_file,
aoai_api_key_file,
aoai_api_base_file,
exclude,
key_file_path=key_file_path,
openai_api_key_file=openai_api_key_file,
aoai_api_key_file=aoai_api_key_file,
aoai_api_base_file=aoai_api_base_file,
exclude=exclude,
)
if model_list:
config_list = [{**config, "model": model} for model in model_list for config in config_list]
Expand Down
2 changes: 1 addition & 1 deletion autogen/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.2.6"
__version__ = "0.2.7"
2 changes: 1 addition & 1 deletion test/agentchat/test_function_call.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
@pytest.mark.skipif(skip, reason="openai not installed OR requested to skip")
def test_eval_math_responses():
config_list = autogen.config_list_from_models(
KEY_LOC, exclude="aoai", model_list=["gpt-4-0613", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k"]
KEY_LOC, model_list=["gpt-4-0613", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k"]
)
functions = [
{
Expand Down
6 changes: 4 additions & 2 deletions test/test_notebook.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,8 @@ def _test_oai_chatgpt_gpt4(save=False):
skip or not sys.version.startswith("3.10"),
reason="do not run if openai is not installed or py!=3.10",
)
def test_hierarchy_flow_using_select_speaker(save=False):
def _test_hierarchy_flow_using_select_speaker(save=False):
# TODO: recover this test after rewriting after the new group chat api
run_notebook("agentchat_hierarchy_flow_using_select_speaker.ipynb", save=save)


Expand All @@ -131,4 +132,5 @@ def test_oai_client_cost(save=False):
# test_oai_completion(save=True)
# test_agentchat_MathChat(save=True)
# test_agentchat_function_call(save=True)
test_graph_modelling_language_using_select_speaker(save=True)
# test_graph_modelling_language_using_select_speaker(save=True)
test_agentchat_function_call_async(save=True)

0 comments on commit 405f5c0

Please sign in to comment.