azure model name

flaml/autogen/oai/completion.py

@@ -58,7 +58,7 @@ class Completion:
     chat_models = {
         "gpt-3.5-turbo",
         "gpt-3.5-turbo-0301",
-        "chatgpt-35-turbo-0301",
+        "gpt-35-turbo",
         "gpt-4",
         "gpt-4-32k",
         "gpt-4-32k-0314",
@@ -76,7 +76,7 @@ class Completion:
         "text-davinci-003": 0.02,
         "gpt-3.5-turbo": 0.002,
         "gpt-3.5-turbo-0301": 0.002,
-        "chatgpt-35-turbo-0301": 0.002,
+        "gpt-35-turbo": 0.002,
         "gpt-4": (0.03, 0.06),
         "gpt-4-0314": (0.03, 0.06),
         "gpt-4-32k": (0.06, 0.12),

notebook/autogen_chatgpt.ipynb

@@ -442,7 +442,7 @@
     "    # num_samples can further limit the number of trials for different hyperparameter configurations;\n",
     "    # -1 means decided by the optimization budget only\n",
     "    num_samples=-1,\n",
-    "    # model=\"chatgpt-35-turbo-0301\",  # uncomment if using Azure OpenAI\n",
+    "    # model=\"gpt-35-turbo\",  # the gpt-3.5-turbo model name in Azure OpenAI\n",
     "    # model=\"gpt-3.5-turbo\",  # uncomment if you don't have access to gpt-4\n",
     "    prompt=prompts,  # the prompt templates to choose from\n",
     "    # stop=\"###\",  # the stop sequence\n",