AtomdffAI commited on
Commit
b5f5c08
·
1 Parent(s): cd2279d

Update bot/chatgpt/chat_gpt_bot.py

Browse files
Files changed (1) hide show
  1. bot/chatgpt/chat_gpt_bot.py +4 -4
bot/chatgpt/chat_gpt_bot.py CHANGED
@@ -57,8 +57,8 @@ class ChatGPTBot(Bot):
57
  except openai.error.RateLimitError as e:
58
  # rate limit exception
59
  logger.warn(e)
60
- if retry_count < 1:
61
- time.sleep(5)
62
  logger.warn("[OPEN_AI] RateLimit exceed, 第{}次重试".format(retry_count+1))
63
  return self.reply_text(query, user_id, retry_count+1)
64
  else:
@@ -82,8 +82,8 @@ class ChatGPTBot(Bot):
82
  return image_url
83
  except openai.error.RateLimitError as e:
84
  logger.warn(e)
85
- if retry_count < 1:
86
- time.sleep(5)
87
  logger.warn("[OPEN_AI] ImgCreate RateLimit exceed, 第{}次重试".format(retry_count+1))
88
  return self.reply_text(query, retry_count+1)
89
  else:
 
57
  except openai.error.RateLimitError as e:
58
  # rate limit exception
59
  logger.warn(e)
60
+ if retry_count < 5:
61
+ time.sleep(8)
62
  logger.warn("[OPEN_AI] RateLimit exceed, 第{}次重试".format(retry_count+1))
63
  return self.reply_text(query, user_id, retry_count+1)
64
  else:
 
82
  return image_url
83
  except openai.error.RateLimitError as e:
84
  logger.warn(e)
85
+ if retry_count < 5:
86
+ time.sleep(8)
87
  logger.warn("[OPEN_AI] ImgCreate RateLimit exceed, 第{}次重试".format(retry_count+1))
88
  return self.reply_text(query, retry_count+1)
89
  else: