QChatGPT/res/wiki/插件开发.md
2023-04-06 15:07:25 +08:00

11 KiB
Raw Blame History

QChatGPT 插件开发Wiki

请先阅读插件使用页
请先阅读技术信息页
建议先阅读本项目源码,了解项目架构

问题、需求请到仓库issue发起
提问前请先靠自己尝试

💬简介

尽管“为一个基于OpenAI API的QQ机器人开发插件支持”这事看起来有点小题大做但萌生此想法后的几天内好几个人提出了这个需求最终促使此项目正式支持插件。

🧱实现

基于importlib库加载模块的方法动态加载额外Python程序文件以便实现插件加载插件均存放在plugins文件夹,其中的所有.py文件都将被加载(除了所有__init__.py)

📚示例代码

请查看代码目录tests/plugin_examples中的插件目录

💻快速开始

按照文档部署此项目,并使其正常运行。
plugins目录下新建目录hello,在其中新建空文件__init__.py以标记此目录为软件包,继续新建文件main.py

您也可以使用hello_plugin作为模板直接生成插件代码仓库

编辑main.py输入以下内容:

from pkg.plugin.models import *
from pkg.plugin.host import EventContext, PluginHost

"""
在收到私聊或群聊消息"hello"时,回复"hello, <发送者id>!"或"hello, everyone!"
"""


# 注册插件
@register(name="Hello", description="hello world", version="0.1", author="RockChinQ")
class HelloPlugin(Plugin):

    # 插件加载时触发
    # plugin_host (pkg.plugin.host.PluginHost) 提供了与主程序交互的一些方法,详细请查看其源码
    def __init__(self, plugin_host: PluginHost):
        pass

    # 当收到个人消息时触发
    @on(PersonNormalMessageReceived)
    def person_normal_message_received(self, event: EventContext, **kwargs):
        msg = kwargs['text_message']
        if msg == "hello":  # 如果消息为hello

            # 输出调试信息
            logging.debug("hello, {}".format(kwargs['sender_id']))

            # 回复消息 "hello, <发送者id>!"
            event.add_return("reply", ["hello, {}!".format(kwargs['sender_id'])])

            # 阻止该事件默认行为(向接口获取回复)
            event.prevent_default()

    # 当收到群消息时触发
    @on(GroupNormalMessageReceived)
    def group_normal_message_received(self, event: EventContext, **kwargs):
        msg = kwargs['text_message']
        if msg == "hello":  # 如果消息为hello

            # 输出调试信息
            logging.debug("hello, {}".format(kwargs['sender_id']))

            # 回复消息 "hello, everyone!"
            event.add_return("reply", ["hello, everyone!"])

            # 阻止该事件默认行为(向接口获取回复)
            event.prevent_default()

    # 插件卸载时触发
    def __del__(self):
        pass

此插件将实现:私聊收到hello消息时回复hello, <发送者QQ号>!,群聊收到hello消息时回复hello, everyone!

解读此插件程序

  • importpkg.plugin引入models模块的所有字段(此程序使用了其中的register函数on函数Plugin类PersonNormalMessageReceived事件GroupNormalMessageReceived事件
  • @register()将类HelloPlugin标记为一个插件类,声明插件名称为Hello以及插件简介、版本、作者
  • 声明类HelloPlugin继承于Plugin,此类可以随意命名,插件名称只与register调用时的参数有关
  • 声明此类的__init__方法,此方法是可选的,其中的代码将在主程序启动时加载插件的时候被执行
  • @on将方法person_normal_message_received标记为一个事件处理器,处理PersonNormalMessageReceived收到私聊消息并在获取OpenAI回复前触发事件此方法可以随意命名绑定的事件只与on中的参数有关,更多支持的事件可到pkg.plugin.models.py文件中查看或查看下方API
  • 输出调试信息,程序中可通过logging将日志输出到控制台和qchatgpt.log文件
  • 方法内部从参数中取出text_message参数,判断是否为hello,如果是就将返回值reply设置为["hello, {}!".format(kwargs['sender_id'])],接下来调用event对象的prevent_default方法,阻止原程序默认行为
    • 每个事件提供的参数支持的返回值请查看pkg.plugin.models中的每个事件的注释或查看下方API
    • event对象提供的方法请查看pkg.plugin.host中的EventContext类或查看下方API
  • 用相似的程序注册GroupNormalMessageReceived事件处理群消息

编写完毕保存后,重新启动主程序,查看到输出中包含以下内容,即为加载成功:

[2023-01-16 18:29:47.193] host.py (43) - [INFO] : 加载模块: hello.main
[2023-01-16 18:29:47.194] models.py (209) - [INFO] : 插件注册完成: n='Hello', d='hello world', v=0.1, a='RockChinQ' (<class 'plugins.hello.main.HelloPlugin'>)

建议在config.py中设置logging_level = logging.DEBUG以便开启调试输出

规范(重要)

  • 请每个插件独立一个目录以便管理建议在Github上创建一个仓库储存单个插件以便获取和更新
  • 插件名使用大驼峰命名法,如HelloExamplePluginChineseCommands
  • 一个目录内可以存放多个Python程序文件以独立出插件的各个功能便于开发者管理但不建议在一个目录内注册多个插件
  • 插件需要的依赖库请在插件目录下的requirements.txt中指定,程序从储存库获取此插件时将自动安装依赖

📄API参考

说明

事件处理函数将会获得一系列参数,可以在kwargs中取出。
其中host参数(pkg.plugin.host.PluginHost类的实例)是插件宿主,提供与主程序各个模块交互的一些方法。
event参数(pkg.plugin.host.EventContext类的实例)是事件执行期间的上下文,提供对此次事件执行的一些操作方法。

事件返回值均为可选的,可以通过调用event.add_return(key: str, ret)来提交返回值

事件

所有事件参数均有hostevent,以下仅展示其他参数
关于YiriMirai支持的消息链组件,请查看 YiriMirai的文档

PersonMessageReceived = "person_message_received"
"""收到私聊消息时,在判断是否应该响应前触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        message_chain: mirai.models.message.MessageChain 消息链
"""

GroupMessageReceived = "group_message_received"
"""收到群聊消息时,在判断是否应该响应前触发(所有群消息)
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        message_chain: mirai.models.message.MessageChain 消息链
"""

PersonNormalMessageReceived = "person_normal_message_received"
"""判断为应该处理的私聊普通消息时触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        text_message: str 消息文本
        
    returns (optional):
        alter: str 修改后的消息文本
        reply: list 回复消息组件列表元素为YiriMirai支持的消息组件
"""

PersonCommandSent = "person_command_sent"
"""判断为应该处理的私聊指令时触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        command: str 指令
        params: list[str] 参数列表
        text_message: str 完整指令文本
        is_admin: bool 是否为管理员
    
    returns (optional):
        alter: str 修改后的完整指令文本
        reply: list 回复消息组件列表元素为YiriMirai支持的消息组件
"""

GroupNormalMessageReceived = "group_normal_message_received"
"""判断为应该处理的群聊普通消息时触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        text_message: str 消息文本
        
    returns (optional):
        alter: str 修改后的消息文本
        reply: list 回复消息组件列表元素为YiriMirai支持的消息组件
"""

GroupCommandSent = "group_command_sent"
"""判断为应该处理的群聊指令时触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        command: str 指令
        params: list[str] 参数列表
        text_message: str 完整指令文本
        is_admin: bool 是否为管理员
    
    returns (optional):
        alter: str 修改后的完整指令文本
        reply: list 回复消息组件列表元素为YiriMirai支持的消息组件
"""

NormalMessageResponded = "normal_message_responded"
"""获取到对普通消息的文字响应时触发
    kwargs:
        launcher_type: str 发起对象类型(group/person)
        launcher_id: int 发起对象ID(群号/QQ号)
        sender_id: int 发送者ID(QQ号)
        session: pkg.openai.session.Session 会话对象
        prefix: str 回复文字消息的前缀
        response_text: str 响应文本
    
    returns (optional):
        prefix: str 修改后的回复文字消息的前缀
        reply: list 替换回复消息组件列表元素为YiriMirai支持的消息组件
"""

SessionFirstMessageReceived = "session_first_message_received"
"""会话被第一次交互时触发
    kwargs:
        session_name: str 会话名称(<launcher_type>_<launcher_id>)
        session: pkg.openai.session.Session 会话对象
        default_prompt: str 预设值
"""

SessionExplicitReset = "session_reset"
"""会话被用户手动重置时触发,此事件不支持阻止默认行为
    kwargs:
        session_name: str 会话名称(<launcher_type>_<launcher_id>)
        session: pkg.openai.session.Session 会话对象
"""

SessionExpired = "session_expired"
"""会话过期时触发
    kwargs:
        session_name: str 会话名称(<launcher_type>_<launcher_id>)
        session: pkg.openai.session.Session 会话对象
        session_expire_time: int 已设置的会话过期时间(秒)   
"""

KeyExceeded = "key_exceeded"
"""api-key超额时触发
    kwargs:
        key_name: str 超额的api-key名称
        usage: dict 超额的api-key使用情况
        exceeded_keys: list[str] 超额的api-key列表
"""

KeySwitched = "key_switched"
"""api-key超额切换成功时触发此事件不支持阻止默认行为
    kwargs:
        key_name: str 切换成功的api-key名称
        key_list: list[str] api-key列表
"""

host: PluginHost 详解

提供与主程序各个模块交互的一些方法,具体查看pkg.plugin.host中的PluginHost

event: EventContext 详解

提供对此次事件执行的一些操作方法,具体查看pkg.plugin.host中的EventContext