diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py
index 1fdae65..9d6c4c1 100644
--- a/bot/chatgpt/chat_gpt_bot.py
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -11,8 +11,8 @@ from common.log import logger
 from common.token_bucket import TokenBucket
 from common.expired_dict import ExpiredDict
 import openai
+import openai.error
 import time
-
 # OpenAI对话模型API (可用)
 class ChatGPTBot(Bot,OpenAIImage):
     def __init__(self):
diff --git a/bot/openai/open_ai_bot.py b/bot/openai/open_ai_bot.py
index fde673d..b039987 100644
--- a/bot/openai/open_ai_bot.py
+++ b/bot/openai/open_ai_bot.py
@@ -9,6 +9,7 @@ from bridge.reply import Reply, ReplyType
 from config import conf
 from common.log import logger
 import openai
+import openai.error
 import time
 
 user_session = dict()
diff --git a/bot/openai/open_ai_image.py b/bot/openai/open_ai_image.py
index 4fa02de..2fae243 100644
--- a/bot/openai/open_ai_image.py
+++ b/bot/openai/open_ai_image.py
@@ -1,5 +1,6 @@
 import time
 import openai
+import openai.error
 from common.token_bucket import TokenBucket
 from common.log import logger
 from config import conf