From 400ffd0b5355d3bdb8f907cd735a6e0080a7bd73 Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Sat, 19 Jul 2025 00:46:04 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=A0=B7=E4=BE=8B=E6=8F=92?= =?UTF-8?q?=E4=BB=B6=EF=BC=8C=E4=BF=AE=E5=A4=8D=E7=BB=9F=E8=AE=A1=E6=95=B0?= =?UTF-8?q?=E6=8D=AE(=E9=83=A8=E5=88=86)=EF=BC=8C=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E4=B8=80=E4=B8=AAbug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- plugins/hello_world_plugin/plugin.py | 40 +++++++++++++++++++ src/chat/focus_chat/heartFC_chat.py | 4 +- src/llm_models/utils_model.py | 4 +- src/plugin_system/apis/plugin_register_api.py | 3 +- src/plugin_system/base/base_events_handler.py | 18 ++++++++- src/plugin_system/core/plugin_manager.py | 12 +++++- 6 files changed, 73 insertions(+), 8 deletions(-) diff --git a/plugins/hello_world_plugin/plugin.py b/plugins/hello_world_plugin/plugin.py index dc9b8571c..0376cbf2a 100644 --- a/plugins/hello_world_plugin/plugin.py +++ b/plugins/hello_world_plugin/plugin.py @@ -7,7 +7,11 @@ from src.plugin_system import ( ComponentInfo, ActionActivationType, ConfigField, + BaseEventPlugin, + BaseEventHandler, + EventType, ) +from src.plugin_system.base.component_types import MaiMessages # ===== Action组件 ===== @@ -93,6 +97,20 @@ class TimeCommand(BaseCommand): return True, f"显示了当前时间: {time_str}" +class PrintMessage(BaseEventHandler): + """打印消息事件处理器 - 处理打印消息事件""" + + event_type = EventType.ON_MESSAGE + handler_name = "print_message_handler" + handler_description = "打印接收到的消息" + + async def execute(self, message: MaiMessages) -> Tuple[bool, str | None]: + """执行打印消息事件处理""" + # 打印接收到的消息 + print(f"接收到消息: {message.raw_message}") + return True, "消息已打印" + + # ===== 插件注册 ===== @@ -130,3 +148,25 @@ class HelloWorldPlugin(BasePlugin): (ByeAction.get_action_info(), ByeAction), # 添加告别Action (TimeCommand.get_command_info(), TimeCommand), ] + + +@register_plugin +class HelloWorldEventPlugin(BaseEventPlugin): + """Hello World事件插件 - 处理问候和告别事件""" + + plugin_name = "hello_world_event_plugin" + enable_plugin = False + dependencies = [] + python_dependencies = [] + config_file_name = "event_config.toml" + + config_schema = { + "plugin": { + "name": ConfigField(type=str, default="hello_world_event_plugin", description="插件名称"), + "version": ConfigField(type=str, default="1.0.0", description="插件版本"), + "enabled": ConfigField(type=bool, default=True, description="是否启用插件"), + }, + } + + def get_plugin_components(self) -> List[Tuple[ComponentInfo, Type]]: + return [(PrintMessage.get_handler_info(), PrintMessage)] diff --git a/src/chat/focus_chat/heartFC_chat.py b/src/chat/focus_chat/heartFC_chat.py index 934991afe..738479754 100644 --- a/src/chat/focus_chat/heartFC_chat.py +++ b/src/chat/focus_chat/heartFC_chat.py @@ -179,8 +179,7 @@ class HeartFChatting: await asyncio.sleep(10) if self.loop_mode == ChatMode.NORMAL: self.energy_value -= 0.3 - if self.energy_value <= 0.3: - self.energy_value = 0.3 + self.energy_value = max(self.energy_value, 0.3) def print_cycle_info(self, cycle_timers): # 记录循环信息和计时器结果 @@ -257,6 +256,7 @@ class HeartFChatting: return f"{person_name}:{message_data.get('processed_plain_text')}" async def _observe(self, message_data: Optional[Dict[str, Any]] = None): + # sourcery skip: hoist-statement-from-if, merge-comparisons, reintroduce-else if not message_data: message_data = {} action_type = "no_action" diff --git a/src/llm_models/utils_model.py b/src/llm_models/utils_model.py index 1f90a730a..2e1d426a6 100644 --- a/src/llm_models/utils_model.py +++ b/src/llm_models/utils_model.py @@ -629,7 +629,7 @@ class LLMRequest: ) # 安全地检查和记录请求详情 handled_payload = await _safely_record(request_content, payload) - logger.critical(f"请求头: {await self._build_headers(no_key=True)} 请求体: {handled_payload[:100]}") + logger.critical(f"请求头: {await self._build_headers(no_key=True)} 请求体: {str(handled_payload)[:100]}") raise RuntimeError( f"模型 {self.model_name} API请求失败: 状态码 {exception.status}, {exception.message}" ) @@ -643,7 +643,7 @@ class LLMRequest: logger.critical(f"模型 {self.model_name} 请求失败: {str(exception)}") # 安全地检查和记录请求详情 handled_payload = await _safely_record(request_content, payload) - logger.critical(f"请求头: {await self._build_headers(no_key=True)} 请求体: {handled_payload[:100]}") + logger.critical(f"请求头: {await self._build_headers(no_key=True)} 请求体: {str(handled_payload)[:100]}") raise RuntimeError(f"模型 {self.model_name} API请求失败: {str(exception)}") async def _transform_parameters(self, params: dict) -> dict: diff --git a/src/plugin_system/apis/plugin_register_api.py b/src/plugin_system/apis/plugin_register_api.py index 78f5e50dd..4dafa3f72 100644 --- a/src/plugin_system/apis/plugin_register_api.py +++ b/src/plugin_system/apis/plugin_register_api.py @@ -8,6 +8,7 @@ logger = get_logger("plugin_register") def register_plugin(cls): from src.plugin_system.core.plugin_manager import plugin_manager from src.plugin_system.base.base_plugin import BasePlugin + from src.plugin_system.base.base_event_plugin import BaseEventPlugin """插件注册装饰器 @@ -18,7 +19,7 @@ def register_plugin(cls): plugin_description = "我的插件" ... """ - if not issubclass(cls, BasePlugin): + if not issubclass(cls, BasePlugin) and not issubclass(cls, BaseEventPlugin): logger.error(f"类 {cls.__name__} 不是 BasePlugin 的子类") return cls diff --git a/src/plugin_system/base/base_events_handler.py b/src/plugin_system/base/base_events_handler.py index 2541d9abb..24d577a13 100644 --- a/src/plugin_system/base/base_events_handler.py +++ b/src/plugin_system/base/base_events_handler.py @@ -2,7 +2,7 @@ from abc import ABC, abstractmethod from typing import Tuple, Optional from src.common.logger import get_logger -from .component_types import MaiMessages, EventType +from .component_types import MaiMessages, EventType, EventHandlerInfo, ComponentType logger = get_logger("base_event_handler") @@ -14,7 +14,7 @@ class BaseEventHandler(ABC): """ event_type: EventType = EventType.UNKNOWN # 事件类型，默认为未知 - handler_name: str = "" + handler_name: str = "" # 处理器名称 handler_description: str = "" weight: int = 0 # 权重，数值越大优先级越高 intercept_message: bool = False # 是否拦截消息，默认为否 @@ -32,3 +32,17 @@ class BaseEventHandler(ABC): Tuple[bool, Optional[str]]: (是否执行成功, 可选的返回消息) """ raise NotImplementedError("子类必须实现 execute 方法") + + @classmethod + def get_handler_info(cls) -> "EventHandlerInfo": + """获取事件处理器的信息""" + # 从类属性读取名称，如果没有定义则使用类名自动生成 + name: str = getattr(cls, "handler_name", cls.__name__.lower().replace("handler", "")) + return EventHandlerInfo( + name=name, + component_type=ComponentType.LISTENER, + description=getattr(cls, "handler_description", "events处理器"), + event_type=cls.event_type, + weight=cls.weight, + intercept_message=cls.intercept_message, + ) diff --git a/src/plugin_system/core/plugin_manager.py b/src/plugin_system/core/plugin_manager.py index aa9324f1b..be91d9292 100644 --- a/src/plugin_system/core/plugin_manager.py +++ b/src/plugin_system/core/plugin_manager.py @@ -512,6 +512,12 @@ class PluginManager: config_status = "✅" if self.plugin_paths.get(plugin_name) else "❌" logger.info(f" ⚙️ 配置: {plugin_info.config_file} {config_status}") + root_path = Path(__file__) + + # 查找项目根目录 + while not (root_path / "pyproject.toml").exists() and root_path.parent != root_path: + root_path = root_path.parent + # 显示目录统计 logger.info("📂 加载目录统计:") for directory in self.plugin_directories: @@ -519,7 +525,11 @@ class PluginManager: plugins_in_dir = [] for plugin_name in self.loaded_plugins.keys(): plugin_path = self.plugin_paths.get(plugin_name, "") - if plugin_path.startswith(directory): + if ( + Path(plugin_path) + .resolve() + .is_relative_to(Path(os.path.join(str(root_path), directory)).resolve()) + ): plugins_in_dir.append(plugin_name) if plugins_in_dir: