From 484fc20983058d48111c52a63084e224f2c75584 Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Mon, 21 Jul 2025 01:23:23 +0800 Subject: [PATCH 1/4] typing and plugins --- changes.md | 7 + plugins/hello_world_plugin/plugin.py | 7 +- src/chat/message_receive/bot.py | 23 +-- src/chat/replyer/default_generator.py | 84 +++++----- src/main.py | 3 +- src/plugin_system/base/base_events_handler.py | 49 +++++- src/plugin_system/core/component_registry.py | 4 +- src/plugin_system/core/events_manager.py | 146 ++++++++++++------ 8 files changed, 215 insertions(+), 108 deletions(-) diff --git a/changes.md b/changes.md index 0d6b507b9..407537d28 100644 --- a/changes.md +++ b/changes.md @@ -41,6 +41,13 @@ - 仅在插件 import 失败时会如此，正常注册过程中失败的插件不会显示包名，而是显示插件内部标识符。（这是特性，但是基本上不可能出现这个情况） 7. 现在不支持单文件插件了，加载方式已经完全删除。 8. 把`BaseEventPlugin`合并到了`BasePlugin`中，所有插件都应该继承自`BasePlugin`。 +9. `BaseEventHandler`现在有了`get_config`方法了。 +10. 修正了`main.py`中的错误输出。 +11. 修正了`command`所编译的`Pattern`注册时的错误输出。 +12. `events_manager`有了task相关逻辑了。 + +### TODO +把这个看起来就很别扭的config获取方式改一下 # 吐槽 diff --git a/plugins/hello_world_plugin/plugin.py b/plugins/hello_world_plugin/plugin.py index 2d645616b..14a9d16c5 100644 --- a/plugins/hello_world_plugin/plugin.py +++ b/plugins/hello_world_plugin/plugin.py @@ -102,11 +102,12 @@ class PrintMessage(BaseEventHandler): handler_name = "print_message_handler" handler_description = "打印接收到的消息" - async def execute(self, message: MaiMessages) -> Tuple[bool, str | None]: + async def execute(self, message: MaiMessages) -> Tuple[bool, bool, str | None]: """执行打印消息事件处理""" # 打印接收到的消息 - print(f"接收到消息: {message.raw_message}") - return True, "消息已打印" + if self.get_config("print_message.enabled", False): + print(f"接收到消息: {message.raw_message}") + return True, True, "消息已打印" # ===== 插件注册 ===== diff --git a/src/chat/message_receive/bot.py b/src/chat/message_receive/bot.py index 7bea9987d..87c9a9422 100644 --- a/src/chat/message_receive/bot.py +++ b/src/chat/message_receive/bot.py @@ -13,8 +13,8 @@ from src.chat.message_receive.message import MessageRecv, MessageRecvS4U from src.chat.message_receive.storage import MessageStorage from src.chat.heart_flow.heartflow_message_processor import HeartFCMessageReceiver from src.chat.utils.prompt_builder import Prompt, global_prompt_manager -from src.plugin_system.core.component_registry import component_registry # 导入新插件系统 -from src.plugin_system.base.base_command import BaseCommand +from src.plugin_system.core import component_registry, events_manager # 导入新插件系统 +from src.plugin_system.base import BaseCommand, EventType from src.mais4u.mais4u_chat.s4u_msg_processor import S4UMessageProcessor @@ -140,24 +140,22 @@ class ChatBot: message = MessageRecvS4U(message_data) group_info = message.message_info.group_info user_info = message.message_info.user_info - - + get_chat_manager().register_message(message) chat = await get_chat_manager().get_or_create_stream( platform=message.message_info.platform, # type: ignore user_info=user_info, # type: ignore group_info=group_info, ) - + message.update_chat_stream(chat) # 处理消息内容 await message.process() - - await self.s4u_message_processor.process_message(message) - - return + await self.s4u_message_processor.process_message(message) + + return async def message_process(self, message_data: Dict[str, Any]) -> None: """处理转化后的统一格式消息 @@ -176,9 +174,9 @@ class ChatBot: try: # 确保所有任务已启动 await self._ensure_started() - + platform = message_data["message_info"].get("platform") - + if platform == "amaidesu_default": await self.do_s4u(message_data) return @@ -202,6 +200,9 @@ class ChatBot: await MessageStorage.update_message(message) return + if not await events_manager.handle_mai_events(EventType.ON_MESSAGE, message): + return + get_chat_manager().register_message(message) chat = await get_chat_manager().get_or_create_stream( diff --git a/src/chat/replyer/default_generator.py b/src/chat/replyer/default_generator.py index 464681cb7..cc5cad827 100644 --- a/src/chat/replyer/default_generator.py +++ b/src/chat/replyer/default_generator.py @@ -508,7 +508,7 @@ class DefaultReplyer: for msg_dict in message_list_before_now: try: msg_user_id = str(msg_dict.get("user_id")) - if msg_user_id == bot_id or msg_user_id == target_user_id: + if msg_user_id in [bot_id, target_user_id]: # bot 和目标用户的对话 core_dialogue_list.append(msg_dict) else: @@ -553,7 +553,7 @@ class DefaultReplyer: available_actions: Optional[Dict[str, ActionInfo]] = None, enable_timeout: bool = False, enable_tool: bool = True, - ) -> str: # sourcery skip: merge-else-if-into-elif, remove-redundant-if + ) -> str: # sourcery skip: merge-else-if-into-elif, remove-redundant-if """ 构建回复器上下文 @@ -724,47 +724,7 @@ class DefaultReplyer: # 根据sender通过person_info_manager反向查找person_id，再获取user_id person_id = person_info_manager.get_person_id_by_person_name(sender) - # 根据配置选择使用哪种 prompt 构建模式 - if global_config.chat.use_s4u_prompt_mode and person_id: - # 使用 s4u 对话构建模式：分离当前对话对象和其他对话 - try: - user_id_value = await person_info_manager.get_value(person_id, "user_id") - if user_id_value: - target_user_id = str(user_id_value) - except Exception as e: - logger.warning(f"无法从person_id {person_id} 获取user_id: {e}") - target_user_id = "" - - # 构建分离的对话 prompt - core_dialogue_prompt, background_dialogue_prompt = self.build_s4u_chat_history_prompts( - message_list_before_now_long, target_user_id - ) - - # 使用 s4u 风格的模板 - template_name = "s4u_style_prompt" - - return await global_prompt_manager.format_prompt( - template_name, - expression_habits_block=expression_habits_block, - tool_info_block=tool_info_block, - knowledge_prompt=prompt_info, - memory_block=memory_block, - relation_info_block=relation_info, - extra_info_block=extra_info_block, - identity=identity_block, - action_descriptions=action_descriptions, - sender_name=sender, - mood_state=mood_prompt, - background_dialogue_prompt=background_dialogue_prompt, - time_block=time_block, - core_dialogue_prompt=core_dialogue_prompt, - reply_target_block=reply_target_block, - message_txt=target, - config_expression_style=global_config.expression.expression_style, - keywords_reaction_prompt=keywords_reaction_prompt, - moderation_prompt=moderation_prompt_block, - ) - else: + if not global_config.chat.use_s4u_prompt_mode or not person_id: # 使用原有的模式 return await global_prompt_manager.format_prompt( template_name, @@ -788,6 +748,44 @@ class DefaultReplyer: chat_target_2=chat_target_2, mood_state=mood_prompt, ) + # 使用 s4u 对话构建模式：分离当前对话对象和其他对话 + try: + user_id_value = await person_info_manager.get_value(person_id, "user_id") + if user_id_value: + target_user_id = str(user_id_value) + except Exception as e: + logger.warning(f"无法从person_id {person_id} 获取user_id: {e}") + target_user_id = "" + + # 构建分离的对话 prompt + core_dialogue_prompt, background_dialogue_prompt = self.build_s4u_chat_history_prompts( + message_list_before_now_long, target_user_id + ) + + # 使用 s4u 风格的模板 + template_name = "s4u_style_prompt" + + return await global_prompt_manager.format_prompt( + template_name, + expression_habits_block=expression_habits_block, + tool_info_block=tool_info_block, + knowledge_prompt=prompt_info, + memory_block=memory_block, + relation_info_block=relation_info, + extra_info_block=extra_info_block, + identity=identity_block, + action_descriptions=action_descriptions, + sender_name=sender, + mood_state=mood_prompt, + background_dialogue_prompt=background_dialogue_prompt, + time_block=time_block, + core_dialogue_prompt=core_dialogue_prompt, + reply_target_block=reply_target_block, + message_txt=target, + config_expression_style=global_config.expression.expression_style, + keywords_reaction_prompt=keywords_reaction_prompt, + moderation_prompt=moderation_prompt_block, + ) async def build_prompt_rewrite_context( self, diff --git a/src/main.py b/src/main.py index dbd12f1a4..3cd2107d1 100644 --- a/src/main.py +++ b/src/main.py @@ -78,8 +78,7 @@ class MainSystem: # logger.info("API服务器启动成功") # 加载所有actions，包括默认的和插件的 - plugin_count, component_count = plugin_manager.load_all_plugins() - logger.info(f"插件系统加载成功: {plugin_count} 个插件，{component_count} 个组件") + plugin_manager.load_all_plugins() # 初始化表情管理器 get_emoji_manager().initialize() diff --git a/src/plugin_system/base/base_events_handler.py b/src/plugin_system/base/base_events_handler.py index db6c20b62..b6c9e965d 100644 --- a/src/plugin_system/base/base_events_handler.py +++ b/src/plugin_system/base/base_events_handler.py @@ -1,5 +1,5 @@ from abc import ABC, abstractmethod -from typing import Tuple, Optional +from typing import Tuple, Optional, Dict from src.common.logger import get_logger from .component_types import MaiMessages, EventType, EventHandlerInfo, ComponentType @@ -21,15 +21,17 @@ class BaseEventHandler(ABC): def __init__(self): self.log_prefix = "[EventHandler]" + self.plugin_name = "" # 对应插件名 + self.plugin_config: Optional[Dict] = None # 插件配置字典 if self.event_type == EventType.UNKNOWN: raise NotImplementedError("事件处理器必须指定 event_type") @abstractmethod - async def execute(self, message: MaiMessages) -> Tuple[bool, Optional[str]]: + async def execute(self, message: MaiMessages) -> Tuple[bool, bool, Optional[str]]: """执行事件处理的抽象方法，子类必须实现 Returns: - Tuple[bool, Optional[str]]: (是否执行成功, 可选的返回消息) + Tuple[bool, bool, Optional[str]]: (是否执行成功, 是否需要继续处理, 可选的返回消息) """ raise NotImplementedError("子类必须实现 execute 方法") @@ -49,3 +51,44 @@ class BaseEventHandler(ABC): weight=cls.weight, intercept_message=cls.intercept_message, ) + + def set_plugin_config(self, plugin_config: Dict) -> None: + """设置插件配置 + + Args: + plugin_config (dict): 插件配置字典 + """ + self.plugin_config = plugin_config + + def set_plugin_name(self, plugin_name: str) -> None: + """设置插件名称 + + Args: + plugin_name (str): 插件名称 + """ + self.plugin_name = plugin_name + + def get_config(self, key: str, default=None): + """获取插件配置值，支持嵌套键访问 + + Args: + key: 配置键名，支持嵌套访问如 "section.subsection.key" + default: 默认值 + + Returns: + Any: 配置值或默认值 + """ + if not self.plugin_config: + return default + + # 支持嵌套键访问 + keys = key.split(".") + current = self.plugin_config + + for k in keys: + if isinstance(current, dict) and k in current: + current = current[k] + else: + return default + + return current diff --git a/src/plugin_system/core/component_registry.py b/src/plugin_system/core/component_registry.py index 29a45c604..7283cf9eb 100644 --- a/src/plugin_system/core/component_registry.py +++ b/src/plugin_system/core/component_registry.py @@ -159,8 +159,8 @@ class ComponentRegistry: pattern = re.compile(command_info.command_pattern, re.IGNORECASE | re.DOTALL) if pattern not in self._command_patterns: self._command_patterns[pattern] = command_name - - logger.warning(f"'{command_name}' 对应的命令模式与 '{self._command_patterns[pattern]}' 重复，忽略此命令") + else: + logger.warning(f"'{command_name}' 对应的命令模式与 '{self._command_patterns[pattern]}' 重复，忽略此命令") return True diff --git a/src/plugin_system/core/events_manager.py b/src/plugin_system/core/events_manager.py index 2c48f9d6d..6352c4a09 100644 --- a/src/plugin_system/core/events_manager.py +++ b/src/plugin_system/core/events_manager.py @@ -1,5 +1,6 @@ import asyncio -from typing import List, Dict, Optional, Type +import contextlib +from typing import List, Dict, Optional, Type, Tuple from src.chat.message_receive.message import MessageRecv from src.common.logger import get_logger @@ -12,8 +13,9 @@ logger = get_logger("events_manager") class EventsManager: def __init__(self): # 有权重的 events 订阅者注册表 - self.events_subscribers: Dict[EventType, List[BaseEventHandler]] = {event: [] for event in EventType} - self.handler_mapping: Dict[str, Type[BaseEventHandler]] = {} # 事件处理器映射表 + self._events_subscribers: Dict[EventType, List[BaseEventHandler]] = {event: [] for event in EventType} + self._handler_mapping: Dict[str, Type[BaseEventHandler]] = {} # 事件处理器映射表 + self._handler_tasks: Dict[str, List[asyncio.Task]] = {} # 事件处理器正在处理的任务 def register_event_subscriber(self, handler_info: EventHandlerInfo, handler_class: Type[BaseEventHandler]) -> bool: """注册事件处理器 @@ -29,7 +31,7 @@ class EventsManager: plugin_name = getattr(handler_info, "plugin_name", "unknown") namespace_name = f"{plugin_name}.{handler_name}" - if namespace_name in self.handler_mapping: + if namespace_name in self._handler_mapping: logger.warning(f"事件处理器 {namespace_name} 已存在，跳过注册") return False @@ -37,50 +39,73 @@ class EventsManager: logger.error(f"类 {handler_class.__name__} 不是 BaseEventHandler 的子类") return False - self.handler_mapping[namespace_name] = handler_class + self._handler_mapping[namespace_name] = handler_class + return self._insert_event_handler(handler_class, handler_info) - return self._insert_event_handler(handler_class) - - async def handler_mai_events( + async def handle_mai_events( self, event_type: EventType, message: MessageRecv, llm_prompt: Optional[str] = None, llm_response: Optional[str] = None, - ) -> None: + ) -> bool: """处理 events""" - transformed_message = self._transform_event_message(message, llm_prompt, llm_response) - for handler in self.events_subscribers.get(event_type, []): - if handler.intercept_message: - await handler.execute(transformed_message) - else: - asyncio.create_task(handler.execute(transformed_message)) + from src.plugin_system.core import component_registry - def _insert_event_handler(self, handler_class: Type[BaseEventHandler]) -> bool: - """插入事件处理器到对应的事件类型列表中""" + continue_flag = True + transformed_message = self._transform_event_message(message, llm_prompt, llm_response) + for handler in self._events_subscribers.get(event_type, []): + handler.set_plugin_config(component_registry.get_plugin_config(handler.plugin_name) or {}) + if handler.intercept_message: + try: + success, continue_processing, result = await handler.execute(transformed_message) + if not success: + logger.error(f"EventHandler {handler.handler_name} 执行失败: {result}") + else: + logger.debug(f"EventHandler {handler.handler_name} 执行成功: {result}") + continue_flag = continue_flag and continue_processing + except Exception as e: + logger.error(f"EventHandler {handler.handler_name} 发生异常: {e}") + continue + else: + try: + handler_task = asyncio.create_task(handler.execute(transformed_message)) + handler_task.add_done_callback(self._task_done_callback) + handler_task.set_name(f"EventHandler-{handler.handler_name}-{event_type.name}") + self._handler_tasks[handler.handler_name].append(handler_task) + except Exception as e: + logger.error(f"创建事件处理器任务 {handler.handler_name} 时发生异常: {e}") + continue + return continue_flag + + def _insert_event_handler(self, handler_class: Type[BaseEventHandler], handler_info: EventHandlerInfo) -> bool: + """插入事件处理器到对应的事件类型列表中并设置其插件配置""" if handler_class.event_type == EventType.UNKNOWN: logger.error(f"事件处理器 {handler_class.__name__} 的事件类型未知，无法注册") return False - self.events_subscribers[handler_class.event_type].append(handler_class()) - self.events_subscribers[handler_class.event_type].sort(key=lambda x: x.weight, reverse=True) + 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) + self._events_subscribers[handler_class.event_type].sort(key=lambda x: x.weight, reverse=True) return True def _remove_event_handler(self, handler_class: Type[BaseEventHandler]) -> bool: """从事件类型列表中移除事件处理器""" + display_handler_name = handler_class.handler_name or handler_class.__name__ if handler_class.event_type == EventType.UNKNOWN: - logger.warning(f"事件处理器 {handler_class.__name__} 的事件类型未知，不存在于处理器列表中") + logger.warning(f"事件处理器 {display_handler_name} 的事件类型未知，不存在于处理器列表中") return False - handlers = self.events_subscribers[handler_class.event_type] + handlers = self._events_subscribers[handler_class.event_type] for i, handler in enumerate(handlers): if isinstance(handler, handler_class): del handlers[i] - logger.debug(f"事件处理器 {handler_class.__name__} 已移除") + logger.debug(f"事件处理器 {display_handler_name} 已移除") return True - logger.warning(f"未找到事件处理器 {handler_class.__name__}，无法移除") + logger.warning(f"未找到事件处理器 {display_handler_name}，无法移除") return False def _transform_event_message( @@ -102,35 +127,68 @@ class EventsManager: transformed_message.message_segments = [message.message_segment] # stream_id 处理 - if hasattr(message, "chat_stream"): + if hasattr(message, "chat_stream") and message.chat_stream: transformed_message.stream_id = message.chat_stream.stream_id # 处理后文本 transformed_message.plain_text = message.processed_plain_text # 基本信息 - if message.message_info.platform: - transformed_message.message_base_info["platform"] = message.message_info.platform - if message.message_info.group_info: - transformed_message.is_group_message = True - transformed_message.message_base_info.update( - { - "group_id": message.message_info.group_info.group_id, - "group_name": message.message_info.group_info.group_name, - } - ) - if message.message_info.user_info: - if not transformed_message.is_group_message: - transformed_message.is_private_message = True - transformed_message.message_base_info.update( - { - "user_id": message.message_info.user_info.user_id, - "user_cardname": message.message_info.user_info.user_cardname, # 用户群昵称 - "user_nickname": message.message_info.user_info.user_nickname, # 用户昵称（用户名） - } - ) + if hasattr(message, "message_info") and message.message_info: + if message.message_info.platform: + transformed_message.message_base_info["platform"] = message.message_info.platform + if message.message_info.group_info: + transformed_message.is_group_message = True + transformed_message.message_base_info.update( + { + "group_id": message.message_info.group_info.group_id, + "group_name": message.message_info.group_info.group_name, + } + ) + if message.message_info.user_info: + if not transformed_message.is_group_message: + transformed_message.is_private_message = True + transformed_message.message_base_info.update( + { + "user_id": message.message_info.user_info.user_id, + "user_cardname": message.message_info.user_info.user_cardname, # 用户群昵称 + "user_nickname": message.message_info.user_info.user_nickname, # 用户昵称（用户名） + } + ) return transformed_message + def _task_done_callback(self, task: asyncio.Task[Tuple[bool, bool, str | None]]): + """任务完成回调""" + task_name = task.get_name() or "Unknown Task" + try: + success, _, result = task.result() # 忽略是否继续的标志，因为消息本身未被拦截 + if success: + logger.debug(f"事件处理任务 {task_name} 已成功完成: {result}") + else: + logger.error(f"事件处理任务 {task_name} 执行失败: {result}") + except asyncio.CancelledError: + pass + except Exception as e: + logger.error(f"事件处理任务 {task_name} 发生异常: {e}") + finally: + with contextlib.suppress(ValueError, KeyError): + self._handler_tasks[task_name].remove(task) + + async def cancel_handler_tasks(self, handler_name: str) -> None: + tasks_to_be_cancelled = self._handler_tasks.get(handler_name, []) + remaining_tasks = [task for task in tasks_to_be_cancelled if not task.done()] + for task in remaining_tasks: + task.cancel() + try: + await asyncio.wait_for(asyncio.gather(*remaining_tasks, return_exceptions=True), timeout=5) + logger.info(f"已取消事件处理器 {handler_name} 的所有任务") + except asyncio.TimeoutError: + logger.warning(f"取消事件处理器 {handler_name} 的任务超时，开始强制取消") + except Exception as e: + logger.error(f"取消事件处理器 {handler_name} 的任务时发生异常: {e}") + finally: + del self._handler_tasks[handler_name] + events_manager = EventsManager() From 610e9e5617d16c8663f916f5a0029f55ef849960 Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Mon, 21 Jul 2025 01:27:00 +0800 Subject: [PATCH 2/4] soft reset changes to avoid chaos --- src/chat/replyer/default_generator.py | 50 ++++++++------------------- 1 file changed, 14 insertions(+), 36 deletions(-) diff --git a/src/chat/replyer/default_generator.py b/src/chat/replyer/default_generator.py index 88b7a3a56..36ce63a63 100644 --- a/src/chat/replyer/default_generator.py +++ b/src/chat/replyer/default_generator.py @@ -508,7 +508,7 @@ class DefaultReplyer: for msg_dict in message_list_before_now: try: msg_user_id = str(msg_dict.get("user_id")) - if msg_user_id in [bot_id, target_user_id]: + if msg_user_id == bot_id or msg_user_id == target_user_id: # bot 和目标用户的对话 core_dialogue_list.append(msg_dict) else: @@ -553,7 +553,7 @@ class DefaultReplyer: available_actions: Optional[Dict[str, ActionInfo]] = None, enable_timeout: bool = False, enable_tool: bool = True, - ) -> str: # sourcery skip: merge-else-if-into-elif, remove-redundant-if + ) -> str: # sourcery skip: merge-else-if-into-elif, remove-redundant-if """ 构建回复器上下文 @@ -724,38 +724,16 @@ class DefaultReplyer: # 根据sender通过person_info_manager反向查找person_id，再获取user_id person_id = person_info_manager.get_person_id_by_person_name(sender) - if not global_config.chat.use_s4u_prompt_mode or not person_id: - # 使用原有的模式 - return await global_prompt_manager.format_prompt( - template_name, - expression_habits_block=expression_habits_block, - chat_target=chat_target_1, - chat_info=chat_talking_prompt, - memory_block=memory_block, - tool_info_block=tool_info_block, - knowledge_prompt=prompt_info, - extra_info_block=extra_info_block, - relation_info_block=relation_info, - time_block=time_block, - reply_target_block=reply_target_block, - moderation_prompt=moderation_prompt_block, - keywords_reaction_prompt=keywords_reaction_prompt, - identity=identity_block, - target_message=target, - sender_name=sender, - config_expression_style=global_config.expression.expression_style, - action_descriptions=action_descriptions, - chat_target_2=chat_target_2, - mood_state=mood_prompt, - ) - # 使用 s4u 对话构建模式：分离当前对话对象和其他对话 - try: - user_id_value = await person_info_manager.get_value(person_id, "user_id") - if user_id_value: - target_user_id = str(user_id_value) - except Exception as e: - logger.warning(f"无法从person_id {person_id} 获取user_id: {e}") - target_user_id = "" + # 根据配置选择使用哪种 prompt 构建模式 + if global_config.chat.use_s4u_prompt_mode and person_id: + # 使用 s4u 对话构建模式：分离当前对话对象和其他对话 + try: + user_id_value = await person_info_manager.get_value(person_id, "user_id") + if user_id_value: + target_user_id = str(user_id_value) + except Exception as e: + logger.warning(f"无法从person_id {person_id} 获取user_id: {e}") + target_user_id = "" # 构建分离的对话 prompt core_dialogue_prompt, background_dialogue_prompt = self.build_s4u_chat_history_prompts( @@ -782,8 +760,8 @@ class DefaultReplyer: {core_dialogue_prompt}""" - # 使用 s4u 风格的模板 - template_name = "s4u_style_prompt" + # 使用 s4u 风格的模板 + template_name = "s4u_style_prompt" return await global_prompt_manager.format_prompt( template_name, From d1877089fcbf2ad8e08f6e594bfda28c62d27742 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=85=E8=AF=BA=E7=8B=90?= <212194964+foxcyber907@users.noreply.github.com> Date: Mon, 21 Jul 2025 09:33:42 +0800 Subject: [PATCH 3/4] Update mode_custom.py --- src/chat/willing/mode_custom.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/chat/willing/mode_custom.py b/src/chat/willing/mode_custom.py index 36334df43..01a85acb4 100644 --- a/src/chat/willing/mode_custom.py +++ b/src/chat/willing/mode_custom.py @@ -1,21 +1,22 @@ from .willing_manager import BaseWillingManager +text = "你丫的不配置你选什么custom模式，给你退了快点给你麦爹配置\n注：以上内容由gemini生成，如有不满请投诉gemini" class CustomWillingManager(BaseWillingManager): async def async_task_starter(self) -> None: - pass + raise NotImplementedError(text) async def before_generate_reply_handle(self, message_id: str): - pass + raise NotImplementedError(text) async def after_generate_reply_handle(self, message_id: str): - pass + raise NotImplementedError(text) async def not_reply_handle(self, message_id: str): - pass + raise NotImplementedError(text) async def get_reply_probability(self, message_id: str): - pass + raise NotImplementedError(text) def __init__(self): super().__init__() From 22c7f667e9421b352a11c62d008c36f38818579b Mon Sep 17 00:00:00 2001 From: UnCLAS-Prommer Date: Mon, 21 Jul 2025 10:17:37 +0800 Subject: [PATCH 4/4] not implemented --- src/chat/willing/mode_custom.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/chat/willing/mode_custom.py b/src/chat/willing/mode_custom.py index 01a85acb4..9987ba942 100644 --- a/src/chat/willing/mode_custom.py +++ b/src/chat/willing/mode_custom.py @@ -1,22 +1,23 @@ from .willing_manager import BaseWillingManager -text = "你丫的不配置你选什么custom模式，给你退了快点给你麦爹配置\n注：以上内容由gemini生成，如有不满请投诉gemini" +NOT_IMPLEMENTED_MESSAGE = "\ncustom模式你实现了吗？没自行实现不要选custom。给你退了快点给你麦爹配置\n注：以上内容由gemini生成，如有不满请投诉gemini" class CustomWillingManager(BaseWillingManager): async def async_task_starter(self) -> None: - raise NotImplementedError(text) + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE) async def before_generate_reply_handle(self, message_id: str): - raise NotImplementedError(text) + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE) async def after_generate_reply_handle(self, message_id: str): - raise NotImplementedError(text) + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE) async def not_reply_handle(self, message_id: str): - raise NotImplementedError(text) + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE) async def get_reply_probability(self, message_id: str): - raise NotImplementedError(text) + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE) def __init__(self): super().__init__() + raise NotImplementedError(NOT_IMPLEMENTED_MESSAGE)