Update bot/chatgpt/chat_gpt_bot.py

bot/chatgpt/chat_gpt_bot.py

@@ -12,7 +12,7 @@ user_session = dict()
 class ChatGPTBot(Bot):
     def __init__(self):
         openai.api_key = conf().get('open_ai_api_key')
-        # openai.api_base="https://api.realgpt.ink/v1"
+        openai.api_base="https://api.realgpt.ink/v1"
 
     def reply(self, query, context=None):
         # acquire reply content
@@ -57,7 +57,7 @@ class ChatGPTBot(Bot):
         except openai.error.RateLimitError as e:
             # rate limit exception
             logger.warn(e)
-            if retry_count < 1:
+            if retry_count < 3:
                 time.sleep(5)
                 logger.warn("[OPEN_AI] RateLimit exceed, 第{}次重试".format(retry_count+1))
                 return self.reply_text(query, user_id, retry_count+1)
@@ -82,7 +82,7 @@ class ChatGPTBot(Bot):
             return image_url
         except openai.error.RateLimitError as e:
             logger.warn(e)
-            if retry_count < 1:
+            if retry_count < 3:
                 time.sleep(5)
                 logger.warn("[OPEN_AI] ImgCreate RateLimit exceed, 第{}次重试".format(retry_count+1))
                 return self.reply_text(query, retry_count+1)