mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
491 lines
21 KiB
Python
491 lines
21 KiB
Python
from typing import Any, List, Dict, Tuple
|
||
|
||
from app.core.config import settings
|
||
from app.core.event import eventmanager, Event
|
||
from app.log import logger
|
||
from app.plugins import _PluginBase
|
||
from app.plugins.chatgpt.openai import OpenAi
|
||
from app.schemas.types import EventType, ChainEventType
|
||
from app.schemas import NotificationType
|
||
|
||
|
||
class ChatGPT(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "ChatGPT"
|
||
# 插件描述
|
||
plugin_desc = "消息交互支持与ChatGPT对话。"
|
||
# 插件图标
|
||
plugin_icon = "Chatgpt_A.png"
|
||
# 插件版本
|
||
plugin_version = "2.1.6"
|
||
# 插件作者
|
||
plugin_author = "jxxghp"
|
||
# 作者主页
|
||
author_url = "https://github.com/jxxghp"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "chatgpt_"
|
||
# 加载顺序
|
||
plugin_order = 15
|
||
# 可使用的用户级别
|
||
auth_level = 1
|
||
|
||
# 私有属性
|
||
openai = None
|
||
_enabled = False
|
||
_proxy = False
|
||
_compatible = False
|
||
_recognize = False
|
||
_openai_url = None
|
||
_openai_key = None
|
||
_model = None
|
||
# 存储多个API密钥
|
||
_api_keys = []
|
||
# 当前使用的密钥索引
|
||
_current_key_index = 0
|
||
# 密钥失效状态
|
||
_key_status = {}
|
||
# 是否发送通知
|
||
_notify = False
|
||
# 自定义提示词
|
||
_customize_prompt = '接下来我会给你一个电影或电视剧的文件名，你需要识别文件名中的名称、版本、分段、年份、分瓣率、季集等信息，并按以下JSON格式返回：{"name":string,"version":string,"part":string,"year":string,"resolution":string,"season":number|null,"episode":number|null}，特别注意返回结果需要严格附合JSON格式，不需要有任何其它的字符。如果中文电影或电视剧的文件名中存在谐音字或字母替代的情况，请还原最有可能的结果。'
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._proxy = config.get("proxy")
|
||
self._compatible = config.get("compatible")
|
||
self._recognize = config.get("recognize")
|
||
self._openai_url = config.get("openai_url")
|
||
self._openai_key = config.get("openai_key")
|
||
self._model = config.get("model")
|
||
self._notify = config.get("notify")
|
||
self._customize_prompt = config.get("customize_prompt")
|
||
# 处理多个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), customize_prompt=self._customize_prompt)
|
||
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}")
|
||
success = self.init_openai(next_key)
|
||
return success, ""
|
||
|
||
def get_state(self) -> bool:
|
||
return self._enabled
|
||
|
||
@staticmethod
|
||
def get_command() -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_api(self) -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
|
||
"""
|
||
拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构
|
||
"""
|
||
return [
|
||
{
|
||
'component': 'VForm',
|
||
'content': [
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'enabled',
|
||
'label': '启用插件',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'proxy',
|
||
'label': '使用代理服务器',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'compatible',
|
||
'label': '兼容模式',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'recognize',
|
||
'label': '辅助识别',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'notify',
|
||
'label': '开启通知',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'openai_url',
|
||
'label': 'OpenAI API Url',
|
||
'placeholder': 'https://api.openai.com',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'openai_key',
|
||
'label': 'API密钥 (多个密钥以逗号分隔)',
|
||
'placeholder': 'sk-xxx,sk-yyy'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'model',
|
||
'label': '自定义模型',
|
||
'placeholder': 'gpt-3.5-turbo',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {'cols': 12},
|
||
'content': [
|
||
{
|
||
'component': 'VTextarea',
|
||
'props': {
|
||
'rows': 2,
|
||
'auto-grow': True,
|
||
'model': 'customize_prompt',
|
||
'label': '辅助识别提示词',
|
||
'hint': '在辅助识别时的给AI的提示词',
|
||
'clearable': True,
|
||
'persistent-hint': True,
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '开启插件后，消息交互时使用请[问帮你]开头，或者以？号结尾，或者超过10个汉字/单词，则会触发ChatGPT回复。'
|
||
'开启辅助识别后，内置识别功能无法正常识别种子/文件名称时，将使用ChatGTP进行AI辅助识别，可以提升动漫等非规范命名的识别成功率。'
|
||
'支持输入多个API密钥（以逗号分隔），在密钥调用失败时将自动切换到下一个可用密钥。'
|
||
'开启通知选项后，将在API密钥调用失败时发送系统通知。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
"proxy": False,
|
||
"compatible": False,
|
||
"recognize": False,
|
||
"notify": False,
|
||
"openai_url": "https://api.openai.com",
|
||
"openai_key": "",
|
||
"model": "gpt-3.5-turbo",
|
||
"customize_prompt": '接下来我会给你一个电影或电视剧的文件名，你需要识别文件名中的名称、版本、分段、年份、分瓣率、季集等信息，并按以下JSON格式返回：{"name":string, '
|
||
'"version":string,"part":string,"year":string,"resolution":string,"season":number|null,"episode":number|null}，特别注意返回结果需要严格附合JSON格式，不需要有任何其它的字符。如果中文电影或电视剧的文件名中存在谐音字或字母替代的情况，请还原最有可能的结果。'
|
||
}
|
||
|
||
def get_page(self) -> List[dict]:
|
||
pass
|
||
|
||
@staticmethod
|
||
def is_api_error(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):
|
||
"""
|
||
监听用户消息，获取ChatGPT回复
|
||
"""
|
||
if not self._enabled:
|
||
return
|
||
if not self.openai:
|
||
return
|
||
text = event.event_data.get("text")
|
||
userid = event.event_data.get("userid")
|
||
channel = event.event_data.get("channel")
|
||
if not text:
|
||
return
|
||
if text.startswith("http") or text.startswith("magnet") or text.startswith("ftp"):
|
||
return
|
||
|
||
# 尝试获取响应，失败时切换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} 调用失败: {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):
|
||
"""
|
||
监听识别事件，使用ChatGPT辅助识别名称
|
||
"""
|
||
if not self.openai:
|
||
return
|
||
if not self._recognize:
|
||
return
|
||
if not event.event_data:
|
||
return
|
||
title = event.event_data.get("title")
|
||
if not title:
|
||
return
|
||
|
||
# 尝试获取媒体名称，失败时切换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} 调用失败: {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):
|
||
"""
|
||
退出插件
|
||
"""
|
||
pass
|