瀏覽代碼

Update open_ai_image.py

Haikui Yang 2 年之前
父節點
當前提交
76a95c0226
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      bot/openai/open_ai_image.py

+ 1 - 0
bot/openai/open_ai_image.py

@@ -12,6 +12,7 @@ from config import conf
 class OpenAIImage(object):
 class OpenAIImage(object):
     def __init__(self):
     def __init__(self):
         openai.api_key = conf().get("open_ai_api_key")
         openai.api_key = conf().get("open_ai_api_key")
+        openai.api_base = conf().get("open_ai_api_base")
         if conf().get("rate_limit_dalle"):
         if conf().get("rate_limit_dalle"):
             self.tb4dalle = TokenBucket(conf().get("rate_limit_dalle", 50))
             self.tb4dalle = TokenBucket(conf().get("rate_limit_dalle", 50))