From b975df81da48fc244969bce8f949302bf343db75 Mon Sep 17 00:00:00 2001 From: Li Jiang Date: Wed, 31 May 2023 20:07:57 +0800 Subject: [PATCH] Support more azure openai api_type (#1059) --- flaml/autogen/oai/completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flaml/autogen/oai/completion.py b/flaml/autogen/oai/completion.py index f98ccd8a8..e7b4a5031 100644 --- a/flaml/autogen/oai/completion.py +++ b/flaml/autogen/oai/completion.py @@ -236,7 +236,7 @@ class Completion(openai_Completion): request_timeout = min(request_timeout, time_left) sleep(cls.retry_time) except InvalidRequestError: - if "azure" == config.get("api_type", openai.api_type) and "model" in config: + if "azure" in config.get("api_type", openai.api_type) and "model" in config: # azure api uses "engine" instead of "model" config["engine"] = config.pop("model").replace("gpt-3.5-turbo", "gpt-35-turbo") else: