diff --git a/autogen/oai/completion.py b/autogen/oai/completion.py index 55ca278f88a3..fa01616128ad 100644 --- 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()