diff --git a/src/plugins/reminder_plugin/_manifest.json b/src/plugins/reminder_plugin/_manifest.json new file mode 100644 index 000000000..58c9fc9e4 --- /dev/null +++ b/src/plugins/reminder_plugin/_manifest.json @@ -0,0 +1,9 @@ +{ + "manifest_version": 1, + "name": "智能提醒插件", + "version": "1.0.0", + "description": "一个能从对话中智能识别并设置定时提醒的插件。", + "author": { + "name": "墨墨" + } +} \ No newline at end of file diff --git a/src/plugins/reminder_plugin/plugin.py b/src/plugins/reminder_plugin/plugin.py new file mode 100644 index 000000000..8a833f5be --- /dev/null +++ b/src/plugins/reminder_plugin/plugin.py @@ -0,0 +1,198 @@ +import asyncio +from datetime import datetime +from typing import List, Tuple, Type +from dateutil.parser import parse as parse_datetime + +from src.common.logger import get_logger +from src.manager.async_task_manager import AsyncTask, async_task_manager +from src.person_info.person_info import get_person_info_manager +from src.plugin_system import ( + BaseAction, + ActionInfo, + BasePlugin, + register_plugin, + ActionActivationType, +) +from src.plugin_system.apis import send_api +from src.plugin_system.base.component_types import ChatType + +logger = get_logger(__name__) + + +# ============================ AsyncTask ============================ + +class ReminderTask(AsyncTask): + def __init__(self, delay: float, stream_id: str, is_group: bool, target_user_id: str, target_user_name: str, event_details: str, creator_name: str): + super().__init__(task_name=f"ReminderTask_{target_user_id}_{datetime.now().timestamp()}") + self.delay = delay + self.stream_id = stream_id + self.is_group = is_group + self.target_user_id = target_user_id + self.target_user_name = target_user_name + self.event_details = event_details + self.creator_name = creator_name + + async def run(self): + try: + if self.delay > 0: + logger.info(f"等待 {self.delay:.2f} 秒后执行提醒...") + await asyncio.sleep(self.delay) + + logger.info(f"执行提醒任务: 给 {self.target_user_name} 发送关于 '{self.event_details}' 的提醒") + + reminder_text = f"叮咚！这是 {self.creator_name} 让我准时提醒你的事情：\n\n{self.event_details}" + + if self.is_group: + # 在群聊中，构造 @ 消息段并发送 + group_id = self.stream_id.split('_')[-1] if '_' in self.stream_id else self.stream_id + message_payload = [ + {"type": "at", "data": {"qq": self.target_user_id}}, + {"type": "text", "data": {"text": f" {reminder_text}"}} + ] + await send_api.adapter_command_to_stream( + action="send_group_msg", + params={"group_id": group_id, "message": message_payload}, + stream_id=self.stream_id + ) + else: + # 在私聊中，直接发送文本 + await send_api.text_to_stream(text=reminder_text, stream_id=self.stream_id) + + logger.info(f"提醒任务 {self.task_name} 成功完成。") + + except Exception as e: + logger.error(f"执行提醒任务 {self.task_name} 时出错: {e}", exc_info=True) + + +# =============================== Actions =============================== + +class RemindAction(BaseAction): + """一个能从对话中智能识别并设置定时提醒的动作。""" + + # === 基本信息 === + action_name = "set_reminder" + action_description = "根据用户的对话内容，智能地设置一个未来的提醒事项。" + activation_type = ActionActivationType.LLM_JUDGE + chat_type_allow = ChatType.ALL + + # === LLM 判断与参数提取 === + llm_judge_prompt = """ + 判断用户是否意图设置一个未来的提醒。 + - 必须包含明确的时间点或时间段（如“十分钟后”、“明天下午3点”、“周五”）。 + - 必须包含一个需要被提醒的事件。 + - 可能会包含需要提醒的特定人物。 + - 如果只是普通的聊天或询问时间，则不应触发。 + + 示例: + - "半小时后提醒我开会" -> 是 + - "明天下午三点叫张三来一下" -> 是 + - "别忘了周五把报告交了" -> 是 + - "现在几点了？" -> 否 + - "我明天下午有空" -> 否 + + 请只回答"是"或"否"。 + """ + action_parameters = { + "user_name": "需要被提醒的人的称呼或名字，如果没有明确指定给某人，则默认为'自己'", + "remind_time": "描述提醒时间的自然语言字符串，例如'十分钟后'或'明天下午3点'", + "event_details": "需要提醒的具体事件内容" + } + action_require = [ + "当用户请求在未来的某个时间点提醒他/她或别人某件事时使用", + "适用于包含明确时间信息和事件描述的对话", + "例如：'10分钟后提醒我收快递'、'明天早上九点喊一下李四参加晨会'" + ] + + async def execute(self) -> Tuple[bool, str]: + """执行设置提醒的动作""" + user_name = self.action_data.get("user_name") + remind_time_str = self.action_data.get("remind_time") + event_details = self.action_data.get("event_details") + + if not all([user_name, remind_time_str, event_details]): + missing_params = [p for p, v in {"user_name": user_name, "remind_time": remind_time_str, "event_details": event_details}.items() if not v] + error_msg = f"缺少必要的提醒参数: {', '.join(missing_params)}" + logger.warning(f"[ReminderPlugin] LLM未能提取完整参数: {error_msg}") + return False, error_msg + + # 1. 解析时间 + try: + assert isinstance(remind_time_str, str) + target_time = parse_datetime(remind_time_str, fuzzy=True) + except Exception as e: + logger.error(f"[ReminderPlugin] 无法解析时间字符串 '{remind_time_str}': {e}") + await self.send_text(f"抱歉，我无法理解您说的时间 '{remind_time_str}'，提醒设置失败。") + return False, f"无法解析时间 '{remind_time_str}'" + + now = datetime.now() + if target_time <= now: + await self.send_text("提醒时间必须是一个未来的时间点哦，提醒设置失败。") + return False, "提醒时间必须在未来" + + delay_seconds = (target_time - now).total_seconds() + + # 2. 解析用户 + person_manager = get_person_info_manager() + user_id_to_remind = None + user_name_to_remind = "" + + assert isinstance(user_name, str) + + if user_name.strip() in ["自己", "我", "me"]: + user_id_to_remind = self.user_id + user_name_to_remind = self.user_nickname + else: + user_info = await person_manager.get_person_info_by_name(user_name) + if not user_info or not user_info.get("user_id"): + logger.warning(f"[ReminderPlugin] 找不到名为 '{user_name}' 的用户") + await self.send_text(f"抱歉，我的联系人里找不到叫做 '{user_name}' 的人，提醒设置失败。") + return False, f"用户 '{user_name}' 不存在" + user_id_to_remind = user_info.get("user_id") + user_name_to_remind = user_name + + # 3. 创建并调度异步任务 + try: + assert user_id_to_remind is not None + assert event_details is not None + + reminder_task = ReminderTask( + delay=delay_seconds, + stream_id=self.chat_id, + is_group=self.is_group, + target_user_id=str(user_id_to_remind), + target_user_name=str(user_name_to_remind), + event_details=str(event_details), + creator_name=str(self.user_nickname) + ) + await async_task_manager.add_task(reminder_task) + + # 4. 发送确认消息 + confirm_message = f"好的，我记下了。\n将在 {target_time.strftime('%Y-%m-%d %H:%M:%S')} 提醒 {user_name_to_remind}：\n{event_details}" + await self.send_text(confirm_message) + + return True, "提醒设置成功" + except Exception as e: + logger.error(f"[ReminderPlugin] 创建提醒任务时出错: {e}", exc_info=True) + await self.send_text("抱歉，设置提醒时发生了一点内部错误。") + return False, "设置提醒时发生内部错误" + + +# =============================== Plugin =============================== + +@register_plugin +class ReminderPlugin(BasePlugin): + """一个能从对话中智能识别并设置定时提醒的插件。""" + + # --- 插件基础信息 --- + plugin_name = "reminder_plugin" + enable_plugin = True + dependencies = [] + python_dependencies = [] + config_file_name = "config.toml" + config_schema = {} + + def get_plugin_components(self) -> List[Tuple[ActionInfo, Type[BaseAction]]]: + """注册插件的所有功能组件。""" + return [ + (RemindAction.get_action_info(), RemindAction) + ]