diff --git a/package.v2.json b/package.v2.json index d91db8c..d8cf9a1 100644 --- a/package.v2.json +++ b/package.v2.json @@ -99,6 +99,7 @@ "author": "jxxghp", "level": 1, "history": { + "v2.1.2": "支持传入多个api key", "v2.1.1": "兼容/v1后仍有路径的接口", "v2.1.0": "优化辅助识别提示词", "v2.0.1": "修复辅助识别", diff --git a/plugins.v2/chatgpt/__init__.py b/plugins.v2/chatgpt/__init__.py index d205a04..a366e18 100644 --- a/plugins.v2/chatgpt/__init__.py +++ b/plugins.v2/chatgpt/__init__.py @@ -16,7 +16,7 @@ class ChatGPT(_PluginBase): # 插件图标 plugin_icon = "Chatgpt_A.png" # 插件版本 - plugin_version = "2.1.1" + plugin_version = "2.1.2" # 插件作者 plugin_author = "jxxghp" # 作者主页 @@ -37,6 +37,14 @@ class ChatGPT(_PluginBase): _openai_url = None _openai_key = None _model = None + # 存储多个API密钥 + _api_keys = [] + # 当前使用的密钥索引 + _current_key_index = 0 + # 密钥失效状态 + _key_status = {} + # 是否发送通知 + _notify = False def init_plugin(self, config: dict = None): if config: @@ -47,10 +55,60 @@ class ChatGPT(_PluginBase): self._openai_url = config.get("openai_url") self._openai_key = config.get("openai_key") self._model = config.get("model") - if self._openai_url and self._openai_key: - self.openai = OpenAi(api_key=self._openai_key, api_url=self._openai_url, - proxy=settings.PROXY if self._proxy else None, - model=self._model, compatible=bool(self._compatible)) + self._notify = config.get("notify") + + # 处理多个API密钥 + if self._openai_key: + self._api_keys = [key.strip() for key in self._openai_key.split(',') if key.strip()] + # 初始化密钥状态 + self._key_status = {key: True for key in self._api_keys} + logger.info(f"ChatGPT插件加载了 {len(self._api_keys)} 个API密钥") + + if self._openai_url and self._api_keys: + # 使用第一个密钥初始化 + self._current_key_index = 0 + self.init_openai(self._api_keys[self._current_key_index]) + + def init_openai(self, api_key): + """ + 初始化OpenAI客户端 + """ + if self._openai_url and api_key: + self.openai = OpenAi(api_key=api_key, api_url=self._openai_url, + proxy=settings.PROXY if self._proxy else None, + model=self._model, compatible=bool(self._compatible)) + logger.info(f"ChatGPT插件初始化API客户端成功") + return True + return False + + def switch_to_next_key(self, failed_key): + """ + 切换到下一个可用的API密钥 + :return: (is_switched, error_message) 元组，表示是否切换成功及错误信息 + """ + # 标记当前密钥为失效 + self._key_status[failed_key] = False + + # 寻找下一个可用的密钥 + original_index = self._current_key_index + while True: + self._current_key_index = (self._current_key_index + 1) % len(self._api_keys) + next_key = self._api_keys[self._current_key_index] + + # 如果密钥标记为可用或者已经尝试了所有密钥，则使用该密钥 + if self._key_status.get(next_key, True) or self._current_key_index == original_index: + break + + # 检查是否所有密钥都失效 + if all(not status for status in self._key_status.values()): + logger.error("所有API密钥均已失效") + return False, "所有API密钥均已失效，请检查配置" + + # 使用新密钥重新初始化客户端 + next_key = self._api_keys[self._current_key_index] + logger.info(f"切换到下一个API密钥 {next_key[:5]}...") + success = self.init_openai(next_key) + return success, "" def get_state(self) -> bool: return self._enabled @@ -136,6 +194,22 @@ class ChatGPT(_PluginBase): } } ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'notify', + 'label': '开启通知', + } + } + ] } ] }, @@ -170,7 +244,8 @@ class ChatGPT(_PluginBase): 'component': 'VTextField', 'props': { 'model': 'openai_key', - 'label': 'sk-xxx' + 'label': 'API密钥 (多个密钥以逗号分隔)', + 'placeholder': 'sk-xxx,sk-yyy' } } ] @@ -210,6 +285,8 @@ class ChatGPT(_PluginBase): 'variant': 'tonal', 'text': '开启插件后，消息交互时使用请[问帮你]开头，或者以？号结尾，或者超过10个汉字/单词，则会触发ChatGPT回复。' '开启辅助识别后，内置识别功能无法正常识别种子/文件名称时，将使用ChatGTP进行AI辅助识别，可以提升动漫等非规范命名的识别成功率。' + '支持输入多个API密钥（以逗号分隔），在密钥调用失败时将自动切换到下一个可用密钥。' + '开启通知选项后，将在API密钥调用失败时发送系统通知。' } } ] @@ -223,6 +300,7 @@ class ChatGPT(_PluginBase): "proxy": False, "compatible": False, "recognize": False, + "notify": False, "openai_url": "https://api.openai.com", "openai_key": "", "model": "gpt-3.5-turbo" @@ -231,6 +309,24 @@ class ChatGPT(_PluginBase): def get_page(self) -> List[dict]: pass + def is_api_error(self, response): + """ + 判断响应是否表示API错误 + :param response: API响应 + :return: (is_error, error_message) 元组，表示是否错误及错误信息 + """ + + # 检查响应是否为字典且包含errorMsg + if isinstance(response, dict) and response.get("errorMsg"): + return True, response.get("errorMsg") + + # 检查响应是否为字符串且包含错误信息 + if isinstance(response, str) and "请求ChatGPT出现错误" in response: + return True, response + + # 如果没有错误信息，则表示调用成功 + return False, "" + @eventmanager.register(EventType.UserMessage) def talk(self, event: Event): """ @@ -245,9 +341,47 @@ class ChatGPT(_PluginBase): channel = event.event_data.get("channel") if not text: return - response = self.openai.get_response(text=text, userid=userid) - if response: - self.post_message(channel=channel, title=response, userid=userid) + + # 尝试获取响应，失败时切换API密钥 + retry_count = 0 + max_retries = len(self._api_keys) + + while retry_count < max_retries: + response = self.openai.get_response(text=text, userid=userid) + + # 判断响应是否正常 + is_error, error_msg = self.is_api_error(response) + logger.info(f"ChatGPT返回结果：{response}") + + if is_error: + current_key = self._api_keys[self._current_key_index] + switched, switch_error = self.switch_to_next_key(current_key) + + # 发送密钥失效通知 + if self._notify: + message = f"API密钥 {current_key[:5]}... 调用失败: {error_msg}" + self.post_message(channel=channel, title=message, userid=userid) + + # 如果所有密钥都失效，发送额外通知 + if not switched: + message = switch_error + self.post_message(mtype=NotificationType.Plugin, title="ChatGpt", text=message) + + if not switched: + # 所有密钥都失效，发送消息并退出 + return + + retry_count += 1 + else: + # 成功获取响应 + self.post_message(channel=channel, title=response, userid=userid) + return + + # 所有重试都失败 + if self._notify: + self.post_message(channel=channel, + title="无法获取ChatGPT响应，所有API密钥都已失效", + userid=userid) @eventmanager.register(ChainEventType.NameRecognize) def recognize(self, event: Event): @@ -263,17 +397,60 @@ class ChatGPT(_PluginBase): title = event.event_data.get("title") if not title: return - # 调用ChatGPT - response = self.openai.get_media_name(filename=title) - logger.info(f"ChatGPT返回结果：{response}") - if response and response.get("name"): - event.event_data = { - 'title': title, - 'name': response.get("name"), - 'year': response.get("year"), - 'season': response.get("season"), - 'episode': response.get("episode") - } + + # 尝试获取媒体名称，失败时切换API密钥 + retry_count = 0 + max_retries = len(self._api_keys) + + while retry_count < max_retries: + response = self.openai.get_media_name(filename=title) + logger.info(f"ChatGPT返回结果：{response}") + + # 判断响应是否正常 + is_error, error_msg = self.is_api_error(response) + + # 如果不是错误但返回字典中没有name字段，也视为错误 + if not is_error and isinstance(response, dict) and not response.get("name"): + is_error = True + error_msg = "未返回有效识别结果" + + if is_error: + # 发生错误，尝试切换密钥 + current_key = self._api_keys[self._current_key_index] + switched, switch_error = self.switch_to_next_key(current_key) + + # 发送密钥失效通知 (通过系统通知，因为这里没有用户交互) + if self._notify: + message = f"API密钥 {current_key[:5]}... 调用失败: {error_msg}" + self.post_message(mtype=NotificationType.Plugin, title="ChatGpt", text=message) + + # 如果所有密钥都失效，发送额外通知 + if not switched: + message = switch_error + self.post_message(mtype=NotificationType.Plugin, title="ChatGpt", text=message) + + if not switched: + # 所有密钥都失效 + return + + retry_count += 1 + else: + # 成功获取结果 + event.event_data = { + 'title': title, + 'name': response.get("name"), + 'year': response.get("year"), + 'season': response.get("season"), + 'episode': response.get("episode") + } + return + + # 所有重试都失败 + if self._notify: + logger.error(f"无法识别标题 {title}，所有API密钥都已失效") + self.post_message(mtype=NotificationType.Plugin, + title="ChatGpt", + text=f"无法识别标题 {title}，所有API密钥都已失效") def stop_service(self): """