diff --git a/autogen/oai/completion.py b/autogen/oai/completion.py index 55ca278f8..fa0161612 100644 --- a/autogen/oai/completion.py +++ b/autogen/oai/completion.py @@ -194,7 +194,8 @@ class Completion(openai_Completion): 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()