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