feat: 在插件系统中添加 Chatter 组件

- 在 component_types.py 中新增了用于 CHATTER 的 ComponentType。 - 实现了 ChatterInfo 类，用于存储 Chatter 组件的相关信息。 - 增强了 ComponentRegistry，以支持 Chatter 组件的注册与管理。 - 创建了 ChatterManager，用于管理 Chatter 实例并处理聊天流。 - 开发了 BaseChatter 抽象类，用于定义 Chatter 的行为规范。 - 实现了 AffinityChatter，作为具备兴趣评分与关系构建功能的具体 Chatter 组件。 - 添加了一个内置的 Chatter 插件，并附带完整文档与使用示例。 - 更新了 PluginManager，在插件概览中加入 Chatter 组件的统计信息。
2025-09-23 00:17:32 +08:00
parent e382b2ffd9
commit e1683ee9e6
15 changed files with 806 additions and 60 deletions
--- a/src/chat/chatter_manager.py
+++ b/src/chat/chatter_manager.py
@@ -0,0 +1,136 @@
+from typing import Dict, List, Optional, Any
+import time
+from src.plugin_system.base.base_chatter import BaseChatter
+from src.common.data_models.message_manager_data_model import StreamContext
+from src.chat.planner_actions.planner import ActionPlanner
+from src.chat.planner_actions.action_manager import ActionManager
+from src.plugin_system.base.component_types import ChatType, ComponentType
+from src.common.logger import get_logger
+
+logger = get_logger("chatter_manager")
+
+class ChatterManager:
+    def __init__(self, action_manager: ActionManager):
+        self.action_manager = action_manager
+        self.chatter_classes: Dict[ChatType, List[type]] = {}
+        self.instances: Dict[str, BaseChatter] = {}
+
+        # 管理器统计
+        self.stats = {
+            "chatters_registered": 0,
+            "streams_processed": 0,
+            "successful_executions": 0,
+            "failed_executions": 0,
+        }
+
+    def _auto_register_from_component_registry(self):
+        """从组件注册表自动注册已注册的chatter组件"""
+        try:
+            from src.plugin_system.core.component_registry import component_registry
+            # 获取所有CHATTER类型的组件
+            chatter_components = component_registry.get_enabled_chatter_registry()
+            for chatter_name, chatter_class in chatter_components.items():
+                self.register_chatter(chatter_class)
+                logger.info(f"自动注册chatter组件: {chatter_name}")
+        except Exception as e:
+            logger.warning(f"自动注册chatter组件时发生错误: {e}")
+
+    def register_chatter(self, chatter_class: type):
+        """注册聊天处理器类"""
+        for chat_type in chatter_class.chat_types:
+            if chat_type not in self.chatter_classes:
+                self.chatter_classes[chat_type] = []
+            self.chatter_classes[chat_type].append(chatter_class)
+            logger.info(f"注册聊天处理器 {chatter_class.__name__} 支持 {chat_type.value} 聊天类型")
+
+        self.stats["chatters_registered"] += 1
+
+    def get_chatter_class(self, chat_type: ChatType) -> Optional[type]:
+        """获取指定聊天类型的聊天处理器类"""
+        if chat_type in self.chatter_classes:
+            return self.chatter_classes[chat_type][0]
+        return None
+
+    def get_supported_chat_types(self) -> List[ChatType]:
+        """获取支持的聊天类型列表"""
+        return list(self.chatter_classes.keys())
+
+    def get_registered_chatters(self) -> Dict[ChatType, List[type]]:
+        """获取已注册的聊天处理器"""
+        return self.chatter_classes.copy()
+
+    def get_stream_instance(self, stream_id: str) -> Optional[BaseChatter]:
+        """获取指定流的聊天处理器实例"""
+        return self.instances.get(stream_id)
+
+    def cleanup_inactive_instances(self, max_inactive_minutes: int = 60):
+        """清理不活跃的实例"""
+        current_time = time.time()
+        max_inactive_seconds = max_inactive_minutes * 60
+
+        inactive_streams = []
+        for stream_id, instance in self.instances.items():
+            if hasattr(instance, 'get_activity_time'):
+                activity_time = instance.get_activity_time()
+                if (current_time - activity_time) > max_inactive_seconds:
+                    inactive_streams.append(stream_id)
+
+        for stream_id in inactive_streams:
+            del self.instances[stream_id]
+            logger.info(f"清理不活跃聊天流实例: {stream_id}")
+
+    async def process_stream_context(self, stream_id: str, context: StreamContext) -> dict:
+        """处理流上下文"""
+        chat_type = context.chat_type
+        logger.debug(f"处理流 {stream_id}，聊天类型: {chat_type.value}")
+        if not self.chatter_classes:
+            self._auto_register_from_component_registry()
+
+        # 获取适合该聊天类型的chatter
+        chatter_class = self.get_chatter_class(chat_type)
+        if not chatter_class:
+            # 如果没有找到精确匹配，尝试查找支持ALL类型的chatter
+            from src.plugin_system.base.component_types import ChatType
+            all_chatter_class = self.get_chatter_class(ChatType.ALL)
+            if all_chatter_class:
+                chatter_class = all_chatter_class
+                logger.info(f"流 {stream_id} 使用通用chatter (类型: {chat_type.value})")
+            else:
+                raise ValueError(f"No chatter registered for chat type {chat_type}")
+
+        if stream_id not in self.instances:
+            planner = ActionPlanner(stream_id, self.action_manager)
+            self.instances[stream_id] = chatter_class(stream_id=stream_id, planner=planner, action_manager=self.action_manager)
+            logger.info(f"创建新的聊天流实例: {stream_id} 使用 {chatter_class.__name__} (类型: {chat_type.value})")
+
+        self.stats["streams_processed"] += 1
+        try:
+            result = await self.instances[stream_id].execute(context)
+            self.stats["successful_executions"] += 1
+
+            # 记录处理结果
+            success = result.get("success", False)
+            actions_count = result.get("actions_count", 0)
+            logger.debug(f"流 {stream_id} 处理完成: 成功={success}, 动作数={actions_count}")
+
+            return result
+        except Exception as e:
+            self.stats["failed_executions"] += 1
+            logger.error(f"处理流 {stream_id} 时发生错误: {e}")
+            raise
+
+    def get_stats(self) -> Dict[str, Any]:
+        """获取管理器统计信息"""
+        stats = self.stats.copy()
+        stats["active_instances"] = len(self.instances)
+        stats["registered_chatter_types"] = len(self.chatter_classes)
+        return stats
+
+    def reset_stats(self):
+        """重置统计信息"""
+        self.stats = {
+            "chatters_registered": 0,
+            "streams_processed": 0,
+            "successful_executions": 0,
+            "failed_executions": 0,
+        }