J1147
/
ai-on-wechat
Vērot 1
Pievienot zvaigznīti 0
Atdalīts 0
Kods Problēmas 0 Izmaiņu pieprasījumi 0 Laidieni 0 Vikivietne Aktivitāte
Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.
1194 Revīzijas
2 Atzari
4.4MB
Koks: f4f55d5dfd
Atzari Tagi
${ item.name }
Izveidot atzaru ${ searchTerm }
no 'f4f55d5dfd'
${ noResults }
ai-on-wechat/plugins/role/role.py

203 rindas
8.8KB
Neapstrādāts Vainot Vēsture 

  1. # encoding:utf-8

  2. 

  3. import json

  4. import os

  5. 

  6. import plugins

  7. from bridge.bridge import Bridge

  8. from bridge.context import ContextType

  9. from bridge.reply import Reply, ReplyType

  10. from common import const

  11. from common.log import logger

  12. from config import conf

  13. from plugins import *

  14. 

  15. 

  16. class RolePlay:

  17.     def __init__(self, bot, sessionid, desc, wrapper=None):

  18.         self.bot = bot

  19.         self.sessionid = sessionid

  20.         self.wrapper = wrapper or "%s"  # 用于包装用户输入

  21.         self.desc = desc

  22.         self.bot.sessions.build_session(self.sessionid, system_prompt=self.desc)

  23. 

  24.     def reset(self):

  25.         self.bot.sessions.clear_session(self.sessionid)

  26. 

  27.     def action(self, user_action):

  28.         session = self.bot.sessions.build_session(self.sessionid)

  29.         if session.system_prompt != self.desc:  # 目前没有触发session过期事件，这里先简单判断，然后重置

  30.             session.set_system_prompt(self.desc)

  31.         prompt = self.wrapper % user_action

  32.         return prompt

  33. 

  34. 

  35. @plugins.register(

  36.     name="Role",

  37.     desire_priority=0,

  38.     namecn="角色扮演",

  39.     desc="为你的Bot设置预设角色",

  40.     version="1.0",

  41.     author="lanvent",

  42. )

  43. class Role(Plugin):

  44.     def __init__(self):

  45.         super().__init__()

  46.         curdir = os.path.dirname(__file__)

  47.         config_path = os.path.join(curdir, "roles.json")

  48.         try:

  49.             with open(config_path, "r", encoding="utf-8") as f:

  50.                 config = json.load(f)

  51.                 self.tags = {tag: (desc, []) for tag, desc in config["tags"].items()}

  52.                 self.roles = {}

  53.                 for role in config["roles"]:

  54.                     self.roles[role["title"].lower()] = role

  55.                     for tag in role["tags"]:

  56.                         if tag not in self.tags:

  57.                             logger.warning(f"[Role] unknown tag {tag} ")

  58.                             self.tags[tag] = (tag, [])

  59.                         self.tags[tag][1].append(role)

  60.                 for tag in list(self.tags.keys()):

  61.                     if len(self.tags[tag][1]) == 0:

  62.                         logger.debug(f"[Role] no role found for tag {tag} ")

  63.                         del self.tags[tag]

  64. 

  65.             if len(self.roles) == 0:

  66.                 raise Exception("no role found")

  67.             self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context

  68.             self.roleplays = {}

  69.             logger.info("[Role] inited")

  70.         except Exception as e:

  71.             if isinstance(e, FileNotFoundError):

  72.                 logger.warn(f"[Role] init failed, {config_path} not found, ignore or see https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/role .")

  73.             else:

  74.                 logger.warn("[Role] init failed, ignore or see https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/role .")

  75.             raise e

  76. 

  77.     def get_role(self, name, find_closest=True, min_sim=0.35):

  78.         name = name.lower()

  79.         found_role = None

  80.         if name in self.roles:

  81.             found_role = name

  82.         elif find_closest:

  83.             import difflib

  84. 

  85.             def str_simularity(a, b):

  86.                 return difflib.SequenceMatcher(None, a, b).ratio()

  87. 

  88.             max_sim = min_sim

  89.             max_role = None

  90.             for role in self.roles:

  91.                 sim = str_simularity(name, role)

  92.                 if sim >= max_sim:

  93.                     max_sim = sim

  94.                     max_role = role

  95.             found_role = max_role

  96.         return found_role

  97. 

  98.     def on_handle_context(self, e_context: EventContext):

  99.         if e_context["context"].type != ContextType.TEXT:

  100.             return

  101.         btype = Bridge().get_bot_type("chat")

  102.         if btype not in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE, const.QWEN_DASHSCOPE, const.XUNFEI, const.BAIDU, const.ZHIPU_AI, const.MOONSHOT, const.MiniMax, const.LINKAI]:

  103.             logger.debug(f'不支持的bot: {btype}')

  104.             return

  105.         bot = Bridge().get_bot("chat")

  106.         content = e_context["context"].content[:]

  107.         clist = e_context["context"].content.split(maxsplit=1)

  108.         desckey = None

  109.         customize = False

  110.         sessionid = e_context["context"]["session_id"]

  111.         trigger_prefix = conf().get("plugin_trigger_prefix", "$")

  112.         if clist[0] == f"{trigger_prefix}停止扮演":

  113.             if sessionid in self.roleplays:

  114.                 self.roleplays[sessionid].reset()

  115.                 del self.roleplays[sessionid]

  116.             reply = Reply(ReplyType.INFO, "角色扮演结束!")

  117.             e_context["reply"] = reply

  118.             e_context.action = EventAction.BREAK_PASS

  119.             return

  120.         elif clist[0] == f"{trigger_prefix}角色":

  121.             desckey = "descn"

  122.         elif clist[0].lower() == f"{trigger_prefix}role":

  123.             desckey = "description"

  124.         elif clist[0] == f"{trigger_prefix}设定扮演":

  125.             customize = True

  126.         elif clist[0] == f"{trigger_prefix}角色类型":

  127.             if len(clist) > 1:

  128.                 tag = clist[1].strip()

  129.                 help_text = "角色列表：\n"

  130.                 for key, value in self.tags.items():

  131.                     if value[0] == tag:

  132.                         tag = key

  133.                         break

  134.                 if tag == "所有":

  135.                     for role in self.roles.values():

  136.                         help_text += f"{role['title']}: {role['remark']}\n"

  137.                 elif tag in self.tags:

  138.                     for role in self.tags[tag][1]:

  139.                         help_text += f"{role['title']}: {role['remark']}\n"

  140.                 else:

  141.                     help_text = f"未知角色类型。\n"

  142.                     help_text += "目前的角色类型有: \n"

  143.                     help_text += "，".join([self.tags[tag][0] for tag in self.tags]) + "\n"

  144.             else:

  145.                 help_text = f"请输入角色类型。\n"

  146.                 help_text += "目前的角色类型有: \n"

  147.                 help_text += "，".join([self.tags[tag][0] for tag in self.tags]) + "\n"

  148.             reply = Reply(ReplyType.INFO, help_text)

  149.             e_context["reply"] = reply

  150.             e_context.action = EventAction.BREAK_PASS

  151.             return

  152.         elif sessionid not in self.roleplays:

  153.             return

  154.         logger.debug("[Role] on_handle_context. content: %s" % content)

  155.         if desckey is not None:

  156.             if len(clist) == 1 or (len(clist) > 1 and clist[1].lower() in ["help", "帮助"]):

  157.                 reply = Reply(ReplyType.INFO, self.get_help_text(verbose=True))

  158.                 e_context["reply"] = reply

  159.                 e_context.action = EventAction.BREAK_PASS

  160.                 return

  161.             role = self.get_role(clist[1])

  162.             if role is None:

  163.                 reply = Reply(ReplyType.ERROR, "角色不存在")

  164.                 e_context["reply"] = reply

  165.                 e_context.action = EventAction.BREAK_PASS

  166.                 return

  167.             else:

  168.                 self.roleplays[sessionid] = RolePlay(

  169.                     bot,

  170.                     sessionid,

  171.                     self.roles[role][desckey],

  172.                     self.roles[role].get("wrapper", "%s"),

  173.                 )

  174.                 reply = Reply(ReplyType.INFO, f"预设角色为 {role}:\n" + self.roles[role][desckey])

  175.                 e_context["reply"] = reply

  176.                 e_context.action = EventAction.BREAK_PASS

  177.         elif customize == True:

  178.             self.roleplays[sessionid] = RolePlay(bot, sessionid, clist[1], "%s")

  179.             reply = Reply(ReplyType.INFO, f"角色设定为:\n{clist[1]}")

  180.             e_context["reply"] = reply

  181.             e_context.action = EventAction.BREAK_PASS

  182.         else:

  183.             prompt = self.roleplays[sessionid].action(content)

  184.             e_context["context"].type = ContextType.TEXT

  185.             e_context["context"].content = prompt

  186.             e_context.action = EventAction.BREAK

  187. 

  188.     def get_help_text(self, verbose=False, **kwargs):

  189.         help_text = "让机器人扮演不同的角色。\n"

  190.         if not verbose:

  191.             return help_text

  192.         trigger_prefix = conf().get("plugin_trigger_prefix", "$")

  193.         help_text = f"使用方法:\n{trigger_prefix}角色" + " 预设角色名: 设定角色为{预设角色名}。\n" + f"{trigger_prefix}role" + " 预设角色名: 同上，但使用英文设定。\n"

  194.         help_text += f"{trigger_prefix}设定扮演" + " 角色设定: 设定自定义角色人设为{角色设定}。\n"

  195.         help_text += f"{trigger_prefix}停止扮演: 清除设定的角色。\n"

  196.         help_text += f"{trigger_prefix}角色类型" + " 角色类型: 查看某类{角色类型}的所有预设角色，为所有时输出所有预设角色。\n"

  197.         help_text += "\n目前的角色类型有: \n"

  198.         help_text += "，".join([self.tags[tag][0] for tag in self.tags]) + "。\n"

  199.         help_text += f"\n命令例子: \n{trigger_prefix}角色 写作助理\n"

  200.         help_text += f"{trigger_prefix}角色类型 所有\n"

  201.         help_text += f"{trigger_prefix}停止扮演\n"

  202.         return help_text