135 lines
5.6 KiB
Python
135 lines
5.6 KiB
Python
from model.command.command import Command
|
|
from model.provider.rev_chatgpt import ProviderRevChatGPT
|
|
from model.platform.qq import QQ
|
|
from cores.qqbot.personality import personalities
|
|
from cores.qqbot.global_object import GlobalObject
|
|
|
|
class CommandRevChatGPT(Command):
|
|
def __init__(self, provider: ProviderRevChatGPT, global_object: GlobalObject):
|
|
self.provider = provider
|
|
self.cached_plugins = {}
|
|
self.global_object = global_object
|
|
self.personality_str = ""
|
|
super().__init__(provider, global_object)
|
|
|
|
def check_command(self,
|
|
message: str,
|
|
session_id: str,
|
|
role: str,
|
|
platform: str,
|
|
message_obj):
|
|
self.platform = platform
|
|
hit, res = super().check_command(
|
|
message,
|
|
session_id,
|
|
role,
|
|
platform,
|
|
message_obj
|
|
)
|
|
|
|
if hit:
|
|
return True, res
|
|
if self.command_start_with(message, "help", "帮助"):
|
|
return True, self.help()
|
|
elif self.command_start_with(message, "reset"):
|
|
return True, self.reset(session_id, message)
|
|
elif self.command_start_with(message, "update"):
|
|
return True, self.update(message, role)
|
|
elif self.command_start_with(message, "set"):
|
|
return True, self.set(message, session_id)
|
|
elif self.command_start_with(message, "switch"):
|
|
return True, self.switch(message, session_id)
|
|
return False, None
|
|
|
|
def reset(self, session_id, message: str):
|
|
l = message.split(" ")
|
|
if len(l) == 1:
|
|
self.provider.forget(session_id)
|
|
return True, "重置完毕。", "reset"
|
|
if len(l) == 2 and l[1] == "p":
|
|
self.provider.forget(session_id)
|
|
ret = self.provider.text_chat(self.personality_str)
|
|
return True, f"重置完毕（保留人格）。\n\n{ret}", "reset"
|
|
|
|
def set(self, message: str, session_id: str):
|
|
l = message.split(" ")
|
|
if len(l) == 1:
|
|
return True, f"设置人格: \n/set 人格名或人格文本。例如/set 编剧\n人格列表: /set list\n人格详细信息: \
|
|
/set view 人格名\n重置会话(清除人格): /reset\n重置会话(保留人格): /reset p", "set"
|
|
elif l[1] == "list":
|
|
msg = "人格列表：\n"
|
|
for key in personalities.keys():
|
|
msg += f" |-{key}\n"
|
|
msg += '\n\n*输入/set view 人格名查看人格详细信息'
|
|
msg += '\n*不定时更新人格库，请及时更新本项目。'
|
|
return True, msg, "set"
|
|
elif l[1] == "view":
|
|
if len(l) == 2:
|
|
return True, "请输入/set view 人格名", "set"
|
|
ps = l[2].strip()
|
|
if ps in personalities:
|
|
msg = f"人格【{ps}】详细信息：\n"
|
|
msg += f"{personalities[ps]}\n"
|
|
else:
|
|
msg = f"人格【{ps}】不存在。"
|
|
return True, msg, "set"
|
|
else:
|
|
ps = l[1].strip()
|
|
if ps in personalities:
|
|
self.reset(session_id, "reset")
|
|
self.personality_str = personalities[ps]
|
|
ret = self.provider.text_chat(self.personality_str, session_id)
|
|
return True, f"人格【{ps}】已设置。\n\n{ret}", "set"
|
|
else:
|
|
self.reset(session_id, "reset")
|
|
self.personality_str = ps
|
|
ret = self.provider.text_chat(ps, session_id)
|
|
return True, f"人格信息已设置。\n\n{ret}", "set"
|
|
|
|
def switch(self, message: str, session_id: str):
|
|
'''
|
|
切换账号
|
|
'''
|
|
l = message.split(" ")
|
|
rev_chatgpt = self.provider.get_revchatgpt()
|
|
if len(l) == 1:
|
|
ret = "当前账号：\n"
|
|
index = 0
|
|
curr_ = None
|
|
for revstat in rev_chatgpt:
|
|
index += 1
|
|
ret += f"[{index}]. {revstat['id']}\n"
|
|
# if session_id in revstat['user']:
|
|
# curr_ = revstat['id']
|
|
for user in revstat['user']:
|
|
if session_id == user['id']:
|
|
curr_ = revstat['id']
|
|
break
|
|
if curr_ is None:
|
|
ret += "当前您未选择账号。输入/switch <账号序号>切换账号。"
|
|
else:
|
|
ret += f"当前您选择的账号为：{curr_}。输入/switch <账号序号>切换账号。"
|
|
return True, ret, "switch"
|
|
elif len(l) == 2:
|
|
try:
|
|
index = int(l[1])
|
|
if index > len(self.provider.rev_chatgpt) or index < 1:
|
|
return True, "账号序号不合法。", "switch"
|
|
else:
|
|
# pop
|
|
for revstat in self.provider.rev_chatgpt:
|
|
if session_id in revstat['user']:
|
|
revstat['user'].remove(session_id)
|
|
# append
|
|
self.provider.rev_chatgpt[index - 1]['user'].append(session_id)
|
|
return True, f"切换账号成功。当前账号为：{self.provider.rev_chatgpt[index - 1]['id']}", "switch"
|
|
except BaseException:
|
|
return True, "账号序号不合法。", "switch"
|
|
else:
|
|
return True, "参数过多。", "switch"
|
|
|
|
def help(self):
|
|
commands = super().general_commands()
|
|
commands['set'] = '设置人格'
|
|
return True, super().help_messager(commands, self.platform, self.global_object.cached_plugins), "help"
|