suffix in model name (microsoft#1206)

* suffix in model name * bump version to 2.0.3
whiskyboy · Sep 4, 2023 · bb57610 · bb57610
1 parent 8290417
commit bb57610
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 2 deletions.
diff --git a/flaml/autogen/oai/completion.py b/flaml/autogen/oai/completion.py
@@ -48,6 +48,7 @@ class Completion(openai_Completion):
         "gpt-3.5-turbo-0301",  # deprecate in Sep
         "gpt-3.5-turbo-0613",
         "gpt-3.5-turbo-16k",
+        "gpt-3.5-turbo-16k-0613",
         "gpt-35-turbo",
         "gpt-4",
         "gpt-4-32k",
@@ -70,6 +71,7 @@ class Completion(openai_Completion):
         "gpt-3.5-turbo-0301": (0.0015, 0.002),  # deprecate in Sep
         "gpt-3.5-turbo-0613": (0.0015, 0.002),
         "gpt-3.5-turbo-16k": (0.003, 0.004),
+        "gpt-3.5-turbo-16k-0613": (0.003, 0.004),
         "gpt-35-turbo": 0.002,
         "gpt-4": (0.03, 0.06),
         "gpt-4-32k": (0.06, 0.12),

diff --git a/flaml/version.py b/flaml/version.py
@@ -1 +1 @@
-__version__ = "2.0.2"
+__version__ = "2.0.3"
diff --git a/notebook/autogen_agentchat_chess.ipynb b/notebook/autogen_agentchat_chess.ipynb
@@ -79,6 +79,7 @@
     "#         \"model\": {\n",
     "#             \"gpt-3.5-turbo\",\n",
     "#             \"gpt-3.5-turbo-16k\",\n",
+    "#             \"gpt-3.5-turbo-16k-0613\",\n",
     "#             \"gpt-3.5-turbo-0301\",\n",
     "#             \"chatgpt-35-turbo-0301\",\n",
     "#             \"gpt-35-turbo-v0301\",\n",
@@ -1009,7 +1010,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.16"
+   "version": "3.9.17"
   },
   "orig_nbformat": 4
  },

diff --git a/test/autogen/agentchat/test_assistant_agent.py b/test/autogen/agentchat/test_assistant_agent.py
@@ -69,6 +69,7 @@ def test_gpt35(human_input_mode="NEVER", max_consecutive_auto_reply=5):
             "model": {
                 "gpt-3.5-turbo",
                 "gpt-3.5-turbo-16k",
+                "gpt-3.5-turbo-16k-0613",
                 "gpt-3.5-turbo-0301",
                 "chatgpt-35-turbo-0301",
                 "gpt-35-turbo-v0301",

diff --git a/test/autogen/oai/test_completion.py b/test/autogen/oai/test_completion.py
@@ -134,6 +134,7 @@ def test_nocontext():
                 "model": {
                     "gpt-3.5-turbo",
                     "gpt-3.5-turbo-16k",
+                    "gpt-3.5-turbo-16k-0613",
                     "gpt-3.5-turbo-0301",
                     "chatgpt-35-turbo-0301",
                     "gpt-35-turbo-v0301",
@@ -169,6 +170,7 @@ def test_humaneval(num_samples=1):
             "model": {
                 "gpt-3.5-turbo",
                 "gpt-3.5-turbo-16k",
+                "gpt-3.5-turbo-16k-0613",
                 "gpt-3.5-turbo-0301",
                 "chatgpt-35-turbo-0301",
                 "gpt-35-turbo-v0301",

diff --git a/test/autogen/test_code.py b/test/autogen/test_code.py
@@ -36,6 +36,7 @@
 #     #         "model": {
 #     #             "gpt-3.5-turbo",
 #     #             "gpt-3.5-turbo-16k",
+#     #             "gpt-3.5-turbo-16k-0613",
 #     #             "gpt-3.5-turbo-0301",
 #     #             "chatgpt-35-turbo-0301",
 #     #             "gpt-35-turbo-v0301",