ebc4feebd9
- 移除ChatterManager中不必要的ActionPlanner实例化 - 在MessageManager中设置默认聊天模式为FOCUS - 重构BaseChatter构造函数，移除planner依赖 - 统一ChatMode枚举定义，移除GROUP/PRIVATE模式 - 重构AffinityChatter内部planner初始化逻辑 - 大幅简化兴趣评分系统的日志输出 - 修复plan_filter中的动作解析逻辑，支持新格式 - 更新planner_prompts提示词模板，移除私聊限制 - 优化关系追踪器的错误处理和日志输出
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from abc import ABC, abstractmethod
|
||
from typing import List, Optional, TYPE_CHECKING
|
||
from src.common.data_models.message_manager_data_model import StreamContext
|
||
from .component_types import ChatType
|
||
from src.plugin_system.base.component_types import ChatterInfo, ComponentType
|
||
|
||
if TYPE_CHECKING:
|
||
from src.chat.planner_actions.action_manager import ChatterActionManager
|
||
from src.plugins.built_in.affinity_flow_chatter.planner import ChatterActionPlanner as ActionPlanner
|
||
|
||
class BaseChatter(ABC):
|
||
chatter_name: str = ""
|
||
"""Chatter组件的名称"""
|
||
chatter_description: str = ""
|
||
"""Chatter组件的描述"""
|
||
chat_types: List[ChatType] = [ChatType.PRIVATE, ChatType.GROUP]
|
||
|
||
def __init__(self, stream_id: str, action_manager: 'ChatterActionManager'):
|
||
"""
|
||
初始化聊天处理器
|
||
|
||
Args:
|
||
stream_id: 聊天流ID
|
||
action_manager: 动作管理器
|
||
"""
|
||
self.stream_id = stream_id
|
||
self.action_manager = action_manager
|
||
|
||
@abstractmethod
|
||
async def execute(self, context: StreamContext) -> dict:
|
||
"""
|
||
执行聊天处理流程
|
||
|
||
Args:
|
||
context: StreamContext对象，包含聊天流的所有消息信息
|
||
|
||
Returns:
|
||
处理结果字典
|
||
"""
|
||
pass
|
||
|
||
@classmethod
|
||
def get_chatter_info(cls) -> "ChatterInfo":
|
||
"""从类属性生成ChatterInfo
|
||
Returns:
|
||
ChatterInfo对象
|
||
"""
|
||
|
||
return ChatterInfo(
|
||
name=cls.chatter_name,
|
||
description=cls.chatter_description or "No description provided.",
|
||
chat_type_allow=cls.chat_types[0],
|
||
component_type=ComponentType.CHATTER,
|
||
)
|
||
|