diff --git a/src/chat/frequency_control/frequency_control.py b/src/chat/frequency_control/frequency_control.py index 63fb0e33..95c691aa 100644 --- a/src/chat/frequency_control/frequency_control.py +++ b/src/chat/frequency_control/frequency_control.py @@ -13,13 +13,19 @@ logger = get_logger("frequency_control") class FrequencyControl: """ 频率控制类，可以根据最近时间段的发言数量和发言人数动态调整频率 + + 特点： + - 发言频率调整：基于最近10分钟的数据，评估单位为"消息数/10分钟" + - 专注度调整：基于最近10分钟的数据，评估单位为"消息数/10分钟" + - 历史基准值：基于最近一周的数据，按小时统计，每小时都有独立的基准值 + - 统一标准：两个调整都使用10分钟窗口，确保逻辑一致性和响应速度 """ def __init__(self, chat_id: str): self.chat_id = chat_id self.chat_stream: ChatStream = get_chat_manager().get_stream(self.chat_id) if not self.chat_stream: - raise ValueError(f"无法找到聊天流: {self.chat_id}") + raise ValueError(f"无法找到聊天流: {chat_id}") self.log_prefix = f"[{get_chat_manager().get_stream_name(self.chat_id) or self.chat_id}]" # 发言频率调整值 self.talk_frequency_adjust: float = 1.0 @@ -42,13 +48,139 @@ class FrequencyControl: self.min_adjust = 0.3 # 最小调整值 self.max_adjust = 2.0 # 最大调整值 - # 基准值（可根据实际情况调整） - self.base_message_count = 5 # 基准消息数量 - self.base_user_count = 3 # 基准用户数量 + # 动态基准值（将根据历史数据计算） + self.base_message_count = 5 # 默认基准消息数量，将被动态更新 + self.base_user_count = 3 # 默认基准用户数量，将被动态更新 # 平滑因子 self.smoothing_factor = 0.3 + + # 历史数据相关参数 + self._last_historical_update = 0 + self._historical_update_interval = 3600 # 每小时更新一次历史基准值 + self._historical_days = 7 # 使用最近7天的数据计算基准值 + + # 按小时统计的历史基准值 + self._hourly_baseline = { + 'messages': {}, # {0-23: 平均消息数} + 'users': {} # {0-23: 平均用户数} + } + + # 初始化24小时的默认基准值 + for hour in range(24): + self._hourly_baseline['messages'][hour] = 5.0 + self._hourly_baseline['users'][hour] = 3.0 + def _update_historical_baseline(self): + """ + 更新基于历史数据的基准值 + 使用最近一周的数据，按小时统计平均消息数量和用户数量 + """ + current_time = time.time() + + # 检查是否需要更新历史基准值 + if current_time - self._last_historical_update < self._historical_update_interval: + return + + try: + # 计算一周前的时间戳 + week_ago = current_time - (self._historical_days * 24 * 3600) + + # 获取最近一周的消息数据 + historical_messages = message_api.get_messages_by_time_in_chat( + chat_id=self.chat_stream.stream_id, + start_time=week_ago, + end_time=current_time, + filter_mai=True, + filter_command=True + ) + + if historical_messages: + # 按小时统计消息数和用户数 + hourly_stats = {hour: {'messages': [], 'users': set()} for hour in range(24)} + + for msg in historical_messages: + # 获取消息的小时（UTC时间） + msg_time = time.localtime(msg.time) + msg_hour = msg_time.tm_hour + + # 统计消息数 + hourly_stats[msg_hour]['messages'].append(msg) + + # 统计用户数 + if msg.user_info and msg.user_info.user_id: + hourly_stats[msg_hour]['users'].add(msg.user_info.user_id) + + # 计算每个小时的平均值（基于一周的数据） + for hour in range(24): + # 计算该小时的平均消息数（一周内该小时的总消息数 / 7天） + total_messages = len(hourly_stats[hour]['messages']) + avg_messages = total_messages / self._historical_days + + # 计算该小时的平均用户数（一周内该小时的总用户数 / 7天） + total_users = len(hourly_stats[hour]['users']) + avg_users = total_users / self._historical_days + + # 使用平滑更新更新基准值 + self._hourly_baseline['messages'][hour] = ( + self._hourly_baseline['messages'][hour] * 0.7 + avg_messages * 0.3 + ) + self._hourly_baseline['users'][hour] = ( + self._hourly_baseline['users'][hour] * 0.7 + avg_users * 0.3 + ) + + # 确保基准值不为0 + self._hourly_baseline['messages'][hour] = max(1.0, self._hourly_baseline['messages'][hour]) + self._hourly_baseline['users'][hour] = max(1.0, self._hourly_baseline['users'][hour]) + + # 更新整体基准值（用于兼容性） + overall_avg_messages = sum(self._hourly_baseline['messages'].values()) / 24 + overall_avg_users = sum(self._hourly_baseline['users'].values()) / 24 + + self.base_message_count = overall_avg_messages + self.base_user_count = overall_avg_users + + logger.info( + f"{self.log_prefix} 历史基准值更新完成: " + f"整体平均消息数={overall_avg_messages:.2f}, 整体平均用户数={overall_avg_users:.2f}" + ) + + # 记录几个关键时段的基准值 + key_hours = [8, 12, 18, 22] # 早、中、晚、夜 + for hour in key_hours: + # 计算该小时平均每10分钟的消息数和用户数 + hourly_10min_messages = self._hourly_baseline['messages'][hour] / 6 # 1小时 = 6个10分钟 + hourly_10min_users = self._hourly_baseline['users'][hour] / 6 + logger.info( + f"{self.log_prefix} {hour}时基准值: " + f"消息数={self._hourly_baseline['messages'][hour]:.2f}/小时 " + f"({hourly_10min_messages:.2f}/10分钟), " + f"用户数={self._hourly_baseline['users'][hour]:.2f}/小时 " + f"({hourly_10min_users:.2f}/10分钟)" + ) + + else: + # 如果没有历史数据，使用默认值 + logger.info(f"{self.log_prefix} 无历史数据，使用默认基准值") + + except Exception as e: + logger.error(f"{self.log_prefix} 更新历史基准值时出错: {e}") + # 出错时保持原有基准值不变 + + self._last_historical_update = current_time + + def _get_current_hour_baseline(self) -> tuple[float, float]: + """ + 获取当前小时的基准值 + + Returns: + tuple: (基准消息数, 基准用户数) + """ + current_hour = time.localtime().tm_hour + return ( + self._hourly_baseline['messages'][current_hour], + self._hourly_baseline['users'][current_hour] + ) def get_dynamic_talk_frequency_adjust(self) -> float: """ @@ -81,11 +213,14 @@ class FrequencyControl: if current_time - self.last_update_time < self.update_interval: return + # 先更新历史基准值 + self._update_historical_baseline() + try: - # 获取最近30分钟的数据（发言频率更敏感） + # 获取最近10分钟的数据（发言频率更敏感） recent_messages = message_api.get_messages_by_time_in_chat( chat_id=self.chat_stream.stream_id, - start_time=current_time - 1800, # 30分钟前 + start_time=current_time - 600, # 10分钟前 end_time=current_time, filter_mai=True, filter_command=True @@ -99,12 +234,19 @@ class FrequencyControl: user_ids.add(msg.user_info.user_id) user_count = len(user_ids) + # 获取当前小时的基准值 + current_hour_base_messages, current_hour_base_users = self._get_current_hour_baseline() + + # 计算当前小时平均每10分钟的基准值 + current_hour_10min_messages = current_hour_base_messages / 6 # 1小时 = 6个10分钟 + current_hour_10min_users = current_hour_base_users / 6 + # 发言频率调整逻辑：人少话多时提高回复频率 if user_count > 0: - # 计算人均消息数 + # 计算人均消息数（10分钟窗口） messages_per_user = message_count / user_count - # 基准人均消息数 - base_messages_per_user = self.base_message_count / self.base_user_count if self.base_user_count > 0 else 1.0 + # 使用当前小时每10分钟的基准人均消息数 + base_messages_per_user = current_hour_10min_messages / current_hour_10min_users if current_hour_10min_users > 0 else 1.0 # 如果人均消息数高，说明活跃度高，提高回复频率 if messages_per_user > base_messages_per_user: @@ -126,9 +268,11 @@ class FrequencyControl: ) logger.info( - f"{self.log_prefix} 发言频率调整更新: " + f"{self.log_prefix} 发言频率调整更新(10分钟窗口): " f"消息数={message_count}, 用户数={user_count}, " f"人均消息数={message_count/user_count if user_count > 0 else 0:.2f}, " + f"当前小时基准值(消息:{current_hour_base_messages:.2f}/小时, {current_hour_10min_messages:.2f}/10分钟, " + f"用户:{current_hour_base_users:.2f}/小时, {current_hour_10min_users:.2f}/10分钟), " f"调整值={self.talk_frequency_adjust:.2f}" ) @@ -147,10 +291,10 @@ class FrequencyControl: return try: - # 获取最近1小时的数据 + # 获取最近10分钟的数据（与发言频率保持一致） recent_messages = message_api.get_messages_by_time_in_chat( chat_id=self.chat_stream.stream_id, - start_time=current_time - 3600, # 1小时前 + start_time=current_time - 600, # 10分钟前 end_time=current_time, filter_mai=True, filter_command=True @@ -164,12 +308,19 @@ class FrequencyControl: user_ids.add(msg.user_info.user_id) user_count = len(user_ids) + # 获取当前小时的基准值 + current_hour_base_messages, current_hour_base_users = self._get_current_hour_baseline() + + # 计算当前小时平均每10分钟的基准值 + current_hour_10min_messages = current_hour_base_messages / 6 # 1小时 = 6个10分钟 + current_hour_10min_users = current_hour_base_users / 6 + # 专注度调整逻辑：人多话多时提高专注度 - if user_count > 0 and self.base_user_count > 0: - # 计算用户活跃度比率 - user_ratio = user_count / self.base_user_count - # 计算消息活跃度比率 - message_ratio = message_count / self.base_message_count if self.base_message_count > 0 else 1.0 + if user_count > 0 and current_hour_10min_users > 0: + # 计算用户活跃度比率（基于10分钟数据） + user_ratio = user_count / current_hour_10min_users + # 计算消息活跃度比率（基于10分钟数据） + message_ratio = message_count / current_hour_10min_messages if current_hour_10min_messages > 0 else 1.0 # 如果用户多且消息多，提高专注度 if user_ratio > 1.2 and message_ratio > 1.2: @@ -193,11 +344,17 @@ class FrequencyControl: target_focus_adjust * self.smoothing_factor ) + # 计算当前小时平均每10分钟的基准值 + current_hour_10min_messages = current_hour_base_messages / 6 # 1小时 = 6个10分钟 + current_hour_10min_users = current_hour_base_users / 6 + logger.info( - f"{self.log_prefix} 专注度调整更新: " + f"{self.log_prefix} 专注度调整更新(10分钟窗口): " f"消息数={message_count}, 用户数={user_count}, " - f"用户比率={user_count/self.base_user_count if self.base_user_count > 0 else 0:.2f}, " - f"消息比率={message_count/self.base_message_count if self.base_message_count > 0 else 0:.2f}, " + f"用户比率={user_count/current_hour_10min_users if current_hour_10min_users > 0 else 0:.2f}, " + f"消息比率={message_count/current_hour_10min_messages if current_hour_10min_messages > 0 else 0:.2f}, " + f"当前小时基准值(消息:{current_hour_base_messages:.2f}/小时, {current_hour_10min_messages:.2f}/10分钟, " + f"用户:{current_hour_base_users:.2f}/小时, {current_hour_10min_users:.2f}/10分钟), " f"调整值={self.focus_value_adjust:.2f}" ) @@ -218,7 +375,9 @@ class FrequencyControl: base_message_count: Optional[int] = None, base_user_count: Optional[int] = None, smoothing_factor: Optional[float] = None, - update_interval: Optional[int] = None + update_interval: Optional[int] = None, + historical_update_interval: Optional[int] = None, + historical_days: Optional[int] = None ): """ 设置调整参数 @@ -243,6 +402,10 @@ class FrequencyControl: self.smoothing_factor = max(0.0, min(1.0, smoothing_factor)) if update_interval is not None: self.update_interval = max(10, update_interval) + if historical_update_interval is not None: + self._historical_update_interval = max(300, historical_update_interval) # 最少5分钟 + if historical_days is not None: + self._historical_days = max(1, min(30, historical_days)) # 1-30天之间 class FrequencyControlManager: diff --git a/src/chat/heart_flow/heartFC_chat.py b/src/chat/heart_flow/heartFC_chat.py index 33b73427..289752d0 100644 --- a/src/chat/heart_flow/heartFC_chat.py +++ b/src/chat/heart_flow/heartFC_chat.py @@ -322,12 +322,11 @@ class HeartFChatting: logger.info(f"{self.log_prefix} 开始第{self._cycle_counter}次思考") # 第一步：动作检查 - with Timer("动作检查", cycle_timers): - try: - await self.action_modifier.modify_actions() - available_actions = self.action_manager.get_using_actions() - except Exception as e: - logger.error(f"{self.log_prefix} 动作修改失败: {e}") + try: + await self.action_modifier.modify_actions() + available_actions = self.action_manager.get_using_actions() + except Exception as e: + logger.error(f"{self.log_prefix} 动作修改失败: {e}") # 执行planner is_group_chat, chat_target_info, _ = self.action_planner.get_necessary_info()