diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py index 1fdae659..9d6c4c1c 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 fde673d6..b0399875 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 4fa02de6..2fae243f 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