fix: replace gpt-35-turbo in model name to gpt-3.5-turbo so the name … (

microsoft#138) * fix: replace gpt-35-turbo in model name to gpt-3.5-turbo so the name string is in the current chat_model list * ref: reformatted with black --------- Co-authored-by: Chi Wang <[email protected]> Co-authored-by: Qingyun Wu <[email protected]>
kinddevil · Oct 9, 2023 · fbe82da · fbe82da
1 parent 71ac774
commit fbe82da
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/autogen/oai/completion.py b/autogen/oai/completion.py
@@ -194,7 +194,8 @@ def _get_response(cls, config: Dict, raise_on_ratelimit_or_timeout=False, use_ca
                 return response
         openai_completion = (
             openai.ChatCompletion
-            if config["model"] in cls.chat_models or issubclass(cls, ChatCompletion)
+            if config["model"].replace("gpt-35-turbo", "gpt-3.5-turbo") in cls.chat_models
+            or issubclass(cls, ChatCompletion)
             else openai.Completion
         )
         start_time = time.time()