f22e6365cc
引入新的 Action 激活机制，允许通过重写 go_activate() 方法来自定义激活逻辑。提供了三个工具函数： - _random_activation(): 随机概率激活 - _keyword_match(): 关键词匹配激活 - _llm_judge_activation(): LLM 智能判断激活 主要变更： - 在 BaseAction 中添加 go_activate() 抽象方法和相关工具函数 - 更新 ActionModifier 使用新的激活判断逻辑 - 在 hello_world_plugin 中添加新的激活方式示例 - 更新文档说明新的激活机制 - 保持向后兼容，旧的激活类型配置仍然可用 BREAKING CHANGE: Action 激活判断现在通过 go_activate() 方法进行，旧的激活类型字段已标记为废弃但仍然兼容
213 lines
7.5 KiB
Python
213 lines
7.5 KiB
Python
import logging
|
||
import random
|
||
from typing import Any
|
||
|
||
from src.plugin_system import (
|
||
ActionActivationType,
|
||
BaseAction,
|
||
BaseEventHandler,
|
||
BasePlugin,
|
||
BaseTool,
|
||
ChatType,
|
||
CommandArgs,
|
||
ComponentInfo,
|
||
ConfigField,
|
||
EventType,
|
||
PlusCommand,
|
||
register_plugin,
|
||
)
|
||
from src.plugin_system.base.base_event import HandlerResult
|
||
|
||
|
||
class StartupMessageHandler(BaseEventHandler):
|
||
"""启动时打印消息的事件处理器。"""
|
||
|
||
handler_name = "hello_world_startup_handler"
|
||
handler_description = "在机器人启动时打印一条日志。"
|
||
init_subscribe = [EventType.ON_START]
|
||
|
||
async def execute(self, params: dict) -> HandlerResult:
|
||
logging.info("🎉 Hello World 插件已启动，准备就绪！")
|
||
return HandlerResult(success=True, continue_process=True)
|
||
|
||
|
||
class GetSystemInfoTool(BaseTool):
|
||
"""一个提供系统信息的示例工具。"""
|
||
|
||
name = "get_system_info"
|
||
description = "获取当前系统的模拟版本和状态信息。"
|
||
available_for_llm = True
|
||
parameters = []
|
||
|
||
async def execute(self, function_args: dict[str, Any]) -> dict[str, Any]:
|
||
return {"name": self.name, "content": "系统版本: 1.0.1, 状态: 运行正常"}
|
||
|
||
|
||
class HelloCommand(PlusCommand):
|
||
"""一个简单的 /hello 命令，使用配置文件中的问候语。"""
|
||
|
||
command_name = "hello"
|
||
command_description = "向机器人发送一个简单的问候。"
|
||
command_aliases = ["hi", "你好"]
|
||
chat_type_allow = ChatType.ALL
|
||
|
||
async def execute(self, args: CommandArgs) -> tuple[bool, str | None, bool]:
|
||
greeting = str(self.get_config("greeting.message", "Hello, World! 我是一个由 MoFox_Bot 驱动的插件。"))
|
||
await self.send_text(greeting)
|
||
return True, "成功发送问候", True
|
||
|
||
|
||
# ==================================================================================
|
||
# 新的激活方式示例 Actions
|
||
# ==================================================================================
|
||
|
||
|
||
class KeywordActivationExampleAction(BaseAction):
|
||
"""关键词激活示例
|
||
|
||
此示例展示如何使用关键词匹配来激活 Action。
|
||
"""
|
||
|
||
action_name = "keyword_example"
|
||
action_description = "当检测到特定关键词时发送回应"
|
||
action_require = ["用户提到了问候语"]
|
||
associated_types = ["text"]
|
||
|
||
async def go_activate(self, chat_content: str = "", llm_judge_model=None) -> bool:
|
||
"""关键词激活：检测到"你好"、"hello"或"hi"时激活"""
|
||
return await self._keyword_match(
|
||
chat_content,
|
||
keywords=["你好", "hello", "hi", "嗨"],
|
||
case_sensitive=False # 不区分大小写
|
||
)
|
||
|
||
async def execute(self) -> tuple[bool, str]:
|
||
await self.send_text("检测到问候语，我也向你问好！👋")
|
||
return True, "发送了问候回应"
|
||
|
||
|
||
class LLMJudgeExampleAction(BaseAction):
|
||
"""LLM 判断激活示例
|
||
|
||
此示例展示如何使用 LLM 来智能判断是否激活 Action。
|
||
"""
|
||
|
||
action_name = "llm_judge_example"
|
||
action_description = "当用户表达情绪低落时提供安慰"
|
||
action_require = ["用户情绪低落", "需要情感支持"]
|
||
associated_types = ["text"]
|
||
|
||
async def go_activate(self, chat_content: str = "", llm_judge_model=None) -> bool:
|
||
"""LLM 判断激活：判断用户是否情绪低落"""
|
||
return await self._llm_judge_activation(
|
||
chat_content=chat_content,
|
||
judge_prompt="""
|
||
判断用户是否表达了以下情绪或需求：
|
||
1. 感到难过、沮丧或失落
|
||
2. 表达了负面情绪
|
||
3. 需要安慰或鼓励
|
||
|
||
如果用户表达了上述情绪或需求，回答"是"，否则回答"否"。
|
||
""",
|
||
llm_judge_model=llm_judge_model
|
||
)
|
||
|
||
async def execute(self) -> tuple[bool, str]:
|
||
await self.send_text("看起来你心情不太好，希望能让你开心一点！🤗💕")
|
||
return True, "发送了安慰消息"
|
||
|
||
|
||
class CombinedActivationExampleAction(BaseAction):
|
||
"""组合激活条件示例
|
||
|
||
此示例展示如何组合多种激活条件。
|
||
"""
|
||
|
||
action_name = "combined_example"
|
||
action_description = "展示如何组合多种激活条件"
|
||
action_require = ["展示灵活的激活逻辑"]
|
||
associated_types = ["text"]
|
||
|
||
async def go_activate(self, chat_content: str = "", llm_judge_model=None) -> bool:
|
||
"""组合激活：随机 20% 概率，或者匹配特定关键词"""
|
||
# 先尝试随机激活
|
||
if await self._random_activation(0.2):
|
||
return True
|
||
|
||
# 如果随机未激活，尝试关键词匹配
|
||
if await self._keyword_match(chat_content, ["表情", "emoji", "😊"], case_sensitive=False):
|
||
return True
|
||
|
||
# 都不满足则不激活
|
||
return False
|
||
|
||
async def execute(self) -> tuple[bool, str]:
|
||
await self.send_text("这是一个组合激活条件的示例！✨")
|
||
return True, "发送了示例消息"
|
||
|
||
|
||
class RandomEmojiAction(BaseAction):
|
||
"""一个随机发送表情的动作。
|
||
|
||
此示例展示了如何使用新的 go_activate() 方法来实现随机激活。
|
||
"""
|
||
|
||
action_name = "random_emoji"
|
||
action_description = "随机发送一个表情符号，增加聊天的趣味性。"
|
||
action_require = ["当对话气氛轻松时", "可以用来回应简单的情感表达"]
|
||
associated_types = ["text"]
|
||
|
||
async def go_activate(self, llm_judge_model=None) -> bool:
|
||
"""使用新的激活方式：10% 的概率激活
|
||
|
||
注意：不需要传入 chat_content，会自动从实例属性中获取
|
||
"""
|
||
return await self._random_activation(0.1)
|
||
|
||
async def execute(self) -> tuple[bool, str]:
|
||
emojis = ["😊", "😂", "👍", "🎉", "🤔", "🤖"]
|
||
await self.send_text(random.choice(emojis))
|
||
return True, "成功发送了一个随机表情"
|
||
|
||
|
||
@register_plugin
|
||
class HelloWorldPlugin(BasePlugin):
|
||
"""一个包含四大核心组件和高级配置功能的入门示例插件。"""
|
||
|
||
plugin_name = "hello_world_plugin"
|
||
enable_plugin = True
|
||
dependencies = []
|
||
python_dependencies = []
|
||
config_file_name = "config.toml"
|
||
enable_plugin = False
|
||
|
||
config_schema = {
|
||
"meta": {
|
||
"config_version": ConfigField(type=int, default=1, description="配置文件版本，请勿手动修改。"),
|
||
},
|
||
"greeting": {
|
||
"message": ConfigField(
|
||
type=str, default="这是来自配置文件的问候！👋", description="HelloCommand 使用的问候语。"
|
||
),
|
||
},
|
||
"components": {
|
||
"hello_command_enabled": ConfigField(type=bool, default=True, description="是否启用 /hello 命令。"),
|
||
"random_emoji_action_enabled": ConfigField(type=bool, default=True, description="是否启用随机表情动作。"),
|
||
},
|
||
}
|
||
|
||
def get_plugin_components(self) -> list[tuple[ComponentInfo, type]]:
|
||
"""根据配置文件动态注册插件的功能组件。"""
|
||
components: list[tuple[ComponentInfo, type]] = []
|
||
|
||
components.append((StartupMessageHandler.get_handler_info(), StartupMessageHandler))
|
||
components.append((GetSystemInfoTool.get_tool_info(), GetSystemInfoTool))
|
||
|
||
if self.get_config("components.hello_command_enabled", True):
|
||
components.append((HelloCommand.get_plus_command_info(), HelloCommand))
|
||
|
||
if self.get_config("components.random_emoji_action_enabled", True):
|
||
components.append((RandomEmojiAction.get_action_info(), RandomEmojiAction))
|
||
|
||
return components
|