gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(chat): 实现动态消息分发系统和消息打断机制

Browse Source 
添加基于focus_energy的动态消息分发周期调整功能，根据聊天流兴趣度智能调整检查间隔
实现消息打断系统，允许高优先级消息打断正在处理的任务
重构ChatStream类，引入动态兴趣度计算系统，包括消息兴趣度统计和用户关系评分
扩展数据库模型和配置系统以支持新功能，增加相关配置项
更新版本号至0.11.0-alpha-1以反映重大功能更新
This commit is contained in:
Windpicker-owo
2025-09-25 17:14:01 +08:00
parent 4cbd86aa96
commit a2baec088e
15 changed files with 703 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/common/data_models/message_manager_data_model.py
View File

@@ -11,10 +11,13 @@ from typing import List, Optional, TYPE_CHECKING
from . import BaseDataModel
from src.plugin_system.base.component_types import ChatMode, ChatType
from src.common.logger import get_logger
if TYPE_CHECKING:
from .database_data_model import DatabaseMessages
logger = get_logger("stream_context")
class MessageStatus(Enum):
"""消息状态枚举"""
@@ -36,6 +39,13 @@ class StreamContext(BaseDataModel):
last_check_time: float = field(default_factory=time.time)
is_active: bool = True
processing_task: Optional[asyncio.Task] = None
interruption_count: int = 0 # 打断计数器
last_interruption_time: float = 0.0 # 上次打断时间
afc_threshold_adjustment: float = 0.0 # afc阈值调整量
# 独立分发周期字段
next_check_time: float = field(default_factory=time.time) # 下次检查时间
distribution_interval: float = 5.0 # 当前分发周期（秒）
def add_message(self, message: "DatabaseMessages"):
"""添加消息到上下文"""
@@ -50,9 +60,9 @@ class StreamContext(BaseDataModel):
# 只有在第一次添加消息时才检测聊天类型，避免后续消息改变类型
if len(self.unread_messages) == 1: # 只有这条消息
# 如果消息包含群组信息，则为群聊
if hasattr(message, 'chat_info_group_id') and message.chat_info_group_id:
if hasattr(message, "chat_info_group_id") and message.chat_info_group_id:
self.chat_type = ChatType.GROUP
elif hasattr(message, 'chat_info_group_name') and message.chat_info_group_name:
elif hasattr(message, "chat_info_group_name") and message.chat_info_group_name:
self.chat_type = ChatType.GROUP
else:
self.chat_type = ChatType.PRIVATE
@@ -64,7 +74,7 @@ class StreamContext(BaseDataModel):
def set_chat_mode(self, chat_mode: ChatMode):
"""设置聊天模式"""
self.chat_mode = chat_mode
def is_group_chat(self) -> bool:
"""检查是否为群聊"""
return self.chat_type == ChatType.GROUP
@@ -82,10 +92,6 @@ class StreamContext(BaseDataModel):
else:
return "未知类型"
def get_unread_messages(self) -> List["DatabaseMessages"]:
"""获取未读消息"""
return [msg for msg in self.unread_messages if not msg.is_read]
def mark_message_as_read(self, message_id: str):
"""标记消息为已读"""
for msg in self.unread_messages:
@@ -94,13 +100,56 @@ class StreamContext(BaseDataModel):
self.history_messages.append(msg)
self.unread_messages.remove(msg)
break
def get_unread_messages(self) -> List["DatabaseMessages"]:
"""获取未读消息"""
return [msg for msg in self.unread_messages if not msg.is_read]
def get_history_messages(self, limit: int = 20) -> List["DatabaseMessages"]:
"""获取历史消息"""
# 优先返回最近的历史消息和所有未读消息
recent_history = self.history_messages[-limit:] if len(self.history_messages) > limit else self.history_messages
return recent_history
def calculate_interruption_probability(self, max_limit: int, probability_factor: float) -> float:
"""计算打断概率"""
if max_limit <= 0:
return 0.0
# 计算打断比例
interruption_ratio = self.interruption_count / max_limit
# 如果超过概率因子，概率下降
if interruption_ratio > probability_factor:
# 使用指数衰减，超过限制越多，概率越低
excess_ratio = interruption_ratio - probability_factor
probability = 1.0 * (0.5**excess_ratio) # 基础概率0.5，指数衰减
else:
# 在限制内，保持较高概率
probability = 0.8
return max(0.0, min(1.0, probability))
def increment_interruption_count(self):
"""增加打断计数"""
self.interruption_count += 1
self.last_interruption_time = time.time()
def reset_interruption_count(self):
"""重置打断计数和afc阈值调整"""
self.interruption_count = 0
self.last_interruption_time = 0.0
self.afc_threshold_adjustment = 0.0
def apply_interruption_afc_reduction(self, reduction_value: float):
"""应用打断导致的afc阈值降低"""
self.afc_threshold_adjustment += reduction_value
logger.debug(f"应用afc阈值降低: {reduction_value}, 总调整量: {self.afc_threshold_adjustment}")
def get_afc_threshold_adjustment(self) -> float:
"""获取当前的afc阈值调整量"""
return self.afc_threshold_adjustment
@dataclass
class MessageManagerStats(BaseDataModel):

10
src/common/database/sqlalchemy_models.py
View File

@@ -53,7 +53,15 @@ class ChatStreams(Base):
user_cardname = Column(Text, nullable=True)
energy_value = Column(Float, nullable=True, default=5.0)
sleep_pressure = Column(Float, nullable=True, default=0.0)
focus_energy = Column(Float, nullable=True, default=1.0)
focus_energy = Column(Float, nullable=True, default=0.5)
# 动态兴趣度系统字段
base_interest_energy = Column(Float, nullable=True, default=0.5)
message_interest_total = Column(Float, nullable=True, default=0.0)
message_count = Column(Integer, nullable=True, default=0)
action_count = Column(Integer, nullable=True, default=0)
reply_count = Column(Integer, nullable=True, default=0)
last_interaction_time = Column(Float, nullable=True, default=None)
consecutive_no_reply = Column(Integer, nullable=True, default=0)
__table_args__ = (
Index("idx_chatstreams_stream_id", "stream_id"),

8
src/common/logger.py
View File

@@ -350,6 +350,10 @@ MODULE_COLORS = {
"memory": "\033[38;5;117m", # 天蓝色
"hfc": "\033[38;5;81m", # 稍微暗一些的青色，保持可读
"action_manager": "\033[38;5;208m", # 橙色，不与replyer重复
"message_manager": "\033[38;5;27m", # 深蓝色，消息管理器
"chatter_manager": "\033[38;5;129m", # 紫色，聊天管理器
"chatter_interest_scoring": "\033[38;5;214m", # 橙黄色，兴趣评分
"plan_executor": "\033[38;5;172m", # 橙褐色，计划执行器
# 关系系统
"relation": "\033[38;5;139m", # 柔和的紫色，不刺眼
# 聊天相关模块
@@ -551,6 +555,10 @@ MODULE_ALIASES = {
"llm_models": "模型",
"person_info": "人物",
"chat_stream": "聊天流",
"message_manager": "消息管理",
"chatter_manager": "聊天管理",
"chatter_interest_scoring": "兴趣评分",
"plan_executor": "计划执行",
"planner": "规划器",
"replyer": "言语",
"config": "配置",