fix(azure_dalle): use openai.api_base

zhayujie · Apr 28, 2023 · b0a401a · b0a401a
1 parent 6b4dc37
commit b0a401a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py
@@ -158,9 +158,8 @@ def __init__(self):
         self.args["deployment_id"] = conf().get("azure_deployment_id")
 
     def create_img(self, query, retry_count=0, api_key=None):
-        api_base = "https://a-wxf.openai.azure.com/"
         api_version = "2022-08-03-preview"
-        url = "{}dalle/text-to-image?api-version={}".format(api_base, api_version)
+        url = "{}dalle/text-to-image?api-version={}".format(openai.api_base, api_version)
         api_key = api_key or openai.api_key
         headers = {"api-key": api_key, "Content-Type": "application/json"}
         try: