From 80783439b184125f40b6edacd35a9816a049625a Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Mon, 25 Aug 2025 00:28:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=E6=94=B9=E6=94=B9events?= =?UTF-8?q?=E7=B3=BB=E7=BB=9F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugin_system/base/base_event.py | 38 +++++++++++++++++++ src/plugin_system/base/base_events_handler.py | 2 +- src/plugin_system/base/component_types.py | 6 ++- src/plugin_system/core/events_manager.py | 7 ++-- src/plugin_system/core/to_do_event.md | 14 +++++++ 5 files changed, 62 insertions(+), 5 deletions(-) create mode 100644 src/plugin_system/base/base_event.py create mode 100644 src/plugin_system/core/to_do_event.md diff --git a/src/plugin_system/base/base_event.py b/src/plugin_system/base/base_event.py new file mode 100644 index 00000000..6adb333c --- /dev/null +++ b/src/plugin_system/base/base_event.py @@ -0,0 +1,38 @@ +from typing import TYPE_CHECKING, List, Type + +from src.common.logger import get_logger +from src.plugin_system.base.component_types import EventType, MaiMessages + +if TYPE_CHECKING: + from .base_events_handler import BaseEventHandler + +logger = get_logger("base_event") + +class BaseEvent: + def __init__(self, event_type: EventType | str) -> None: + self.event_type = event_type + self.subscribers: List["BaseEventHandler"] = [] + + def register_handler_to_event(self, handler: "BaseEventHandler") -> bool: + if handler not in self.subscribers: + self.subscribers.append(handler) + return True + logger.warning(f"Handler {handler.handler_name} 已经注册，不可多次注册") + return False + + def remove_handler_from_event(self, handler_class: Type["BaseEventHandler"]) -> bool: + for handler in self.subscribers: + if isinstance(handler, handler_class): + self.subscribers.remove(handler) + return True + logger.warning(f"Handler {handler_class.__name__} 未注册，无法移除") + return False + + def trigger_event(self, message: MaiMessages): + copied_message = message.deepcopy() + for handler in self.subscribers: + result = handler.execute(copied_message) + + # TODO: Unfinished Events Handler + + \ No newline at end of file diff --git a/src/plugin_system/base/base_events_handler.py b/src/plugin_system/base/base_events_handler.py index 80c07ac4..630b1ef2 100644 --- a/src/plugin_system/base/base_events_handler.py +++ b/src/plugin_system/base/base_events_handler.py @@ -13,7 +13,7 @@ class BaseEventHandler(ABC): 所有事件处理器都应该继承这个基类，提供事件处理的基本接口 """ - event_type: EventType = EventType.UNKNOWN + event_type: EventType | str = EventType.UNKNOWN """事件类型，默认为未知""" handler_name: str = "" """处理器名称""" diff --git a/src/plugin_system/base/component_types.py b/src/plugin_system/base/component_types.py index 643f42b7..d02ad1ef 100644 --- a/src/plugin_system/base/component_types.py +++ b/src/plugin_system/base/component_types.py @@ -1,3 +1,4 @@ +import copy from enum import Enum from typing import Dict, Any, List, Optional, Tuple from dataclasses import dataclass, field @@ -165,7 +166,7 @@ class ToolInfo(ComponentInfo): class EventHandlerInfo(ComponentInfo): """事件处理器组件信息""" - event_type: EventType = EventType.ON_MESSAGE # 监听事件类型 + event_type: EventType | str = EventType.ON_MESSAGE # 监听事件类型 intercept_message: bool = False # 是否拦截消息处理（默认不拦截） weight: int = 0 # 事件处理器权重，决定执行顺序 @@ -281,3 +282,6 @@ class MaiMessages: def __post_init__(self): if self.message_segments is None: self.message_segments = [] + + def deepcopy(self): + return copy.deepcopy(self) diff --git a/src/plugin_system/core/events_manager.py b/src/plugin_system/core/events_manager.py index 2472d831..b00dfd6f 100644 --- a/src/plugin_system/core/events_manager.py +++ b/src/plugin_system/core/events_manager.py @@ -1,6 +1,6 @@ import asyncio import contextlib -from typing import List, Dict, Optional, Type, Tuple, Any, TYPE_CHECKING +from typing import List, Dict, Optional, Type, Tuple, TYPE_CHECKING from src.chat.message_receive.message import MessageRecv from src.chat.message_receive.chat_stream import get_chat_manager @@ -18,7 +18,7 @@ logger = get_logger("events_manager") class EventsManager: def __init__(self): # 有权重的 events 订阅者注册表 - self._events_subscribers: Dict[EventType, List[BaseEventHandler]] = {event: [] for event in EventType} + self._events_subscribers: Dict[EventType | str, List[BaseEventHandler]] = {event: [] for event in EventType} self._handler_mapping: Dict[str, Type[BaseEventHandler]] = {} # 事件处理器映射表 self._handler_tasks: Dict[str, List[asyncio.Task]] = {} # 事件处理器正在处理的任务 @@ -152,7 +152,8 @@ class EventsManager: if handler_class.event_type == EventType.UNKNOWN: logger.error(f"事件处理器 {handler_class.__name__} 的事件类型未知，无法注册") return False - + if handler_class.event_type not in self._events_subscribers: + self._events_subscribers[handler_class.event_type] = [] handler_instance = handler_class() handler_instance.set_plugin_name(handler_info.plugin_name or "unknown") self._events_subscribers[handler_class.event_type].append(handler_instance) diff --git a/src/plugin_system/core/to_do_event.md b/src/plugin_system/core/to_do_event.md new file mode 100644 index 00000000..11923ff0 --- /dev/null +++ b/src/plugin_system/core/to_do_event.md @@ -0,0 +1,14 @@ +- [x] 自定义事件 +- [ ] 允许handler随时订阅 +- [ ] 允许handler随时取消订阅 +- [ ] 允许其他组件给handler增加订阅 +- [ ] 允许其他组件给handler取消订阅 +- [ ] 允许一个handler订阅多个事件 +- [ ] event激活时给handler传递参数 +- [ ] handler能拿到所有handlers的结果（按照处理权重） +- [x] 随时注册 +- [ ] 删除event + - [ ] 必要性？ +- [ ] 能够更改prompt +- [ ] 能够更改llm_response +- [ ] 能够更改message \ No newline at end of file