829ff4cd4f
引入了更完善的睡眠唤醒和重新入睡逻辑，以处理在睡眠期间被消息打扰的情况。 - **唤醒机制**: 当在睡眠时间内收到消息并达到唤醒阈值时，角色会被唤醒并进入愤怒状态。唤醒后，将保持清醒状态处理消息，而不是立即重新入睡。 - **状态持久化**: 新增 `_is_woken_up` 状态到 `schedule_manager`，并将其持久化，以确保在重启后能记住唤醒状态。 - **重新入睡**: 如果角色被吵醒后，在配置的一段时间内（`re_sleep_delay_minutes`）没有收到新消息，系统将自动尝试重新进入睡眠状态，以模拟更自然的行为。 - **上下文同步**: 在唤醒时，`wakeup_manager` 会通知 `schedule_manager` 更新其内部状态，确保系统各模块之间的睡眠状态一致。
100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
from typing import List, Optional, TYPE_CHECKING
|
||
import time
|
||
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
|
||
from src.common.logger import get_logger
|
||
from src.manager.local_store_manager import local_storage
|
||
from src.person_info.relationship_builder_manager import RelationshipBuilder
|
||
from src.chat.express.expression_learner import ExpressionLearner
|
||
from src.plugin_system.base.component_types import ChatMode
|
||
from src.chat.planner_actions.action_manager import ActionManager
|
||
from src.chat.chat_loop.hfc_utils import CycleDetail
|
||
|
||
if TYPE_CHECKING:
|
||
from .wakeup_manager import WakeUpManager
|
||
from .energy_manager import EnergyManager
|
||
|
||
class HfcContext:
|
||
def __init__(self, chat_id: str):
|
||
"""
|
||
初始化HFC聊天上下文
|
||
|
||
Args:
|
||
chat_id: 聊天ID标识符
|
||
|
||
功能说明:
|
||
- 存储和管理单个聊天会话的所有状态信息
|
||
- 包含聊天流、关系构建器、表达学习器等核心组件
|
||
- 管理聊天模式、能量值、时间戳等关键状态
|
||
- 提供循环历史记录和当前循环详情的存储
|
||
- 集成唤醒度管理器，处理休眠状态下的唤醒机制
|
||
|
||
Raises:
|
||
ValueError: 如果找不到对应的聊天流
|
||
"""
|
||
self.stream_id: str = chat_id
|
||
self.chat_stream: Optional[ChatStream] = get_chat_manager().get_stream(self.stream_id)
|
||
if not self.chat_stream:
|
||
raise ValueError(f"无法找到聊天流: {self.stream_id}")
|
||
|
||
self.log_prefix = f"[{get_chat_manager().get_stream_name(self.stream_id) or self.stream_id}]"
|
||
|
||
self.relationship_builder: Optional[RelationshipBuilder] = None
|
||
self.expression_learner: Optional[ExpressionLearner] = None
|
||
|
||
self.loop_mode = ChatMode.NORMAL
|
||
self.energy_value = 5.0
|
||
self.sleep_pressure = 0.0
|
||
self.was_sleeping = False # 用于检测睡眠状态的切换
|
||
|
||
# 失眠状态
|
||
self.is_in_insomnia: bool = False
|
||
self.insomnia_end_time: float = 0.0
|
||
self.last_wakeup_time: float = 0.0 # 被吵醒的时间
|
||
|
||
self.last_message_time = time.time()
|
||
self.last_read_time = time.time() - 10
|
||
|
||
self.action_manager = ActionManager()
|
||
|
||
self.running: bool = False
|
||
|
||
self.history_loop: List[CycleDetail] = []
|
||
self.cycle_counter = 0
|
||
self.current_cycle_detail: Optional[CycleDetail] = None
|
||
|
||
# 唤醒度管理器 - 延迟初始化以避免循环导入
|
||
self.wakeup_manager: Optional['WakeUpManager'] = None
|
||
self.energy_manager: Optional['EnergyManager'] = None
|
||
|
||
self._load_context_state()
|
||
|
||
def _get_storage_key(self) -> str:
|
||
"""获取当前聊天流的本地存储键"""
|
||
return f"hfc_context_state_{self.stream_id}"
|
||
|
||
def _load_context_state(self):
|
||
"""从本地存储加载状态"""
|
||
state = local_storage[self._get_storage_key()]
|
||
if state and isinstance(state, dict):
|
||
self.energy_value = state.get("energy_value", 5.0)
|
||
self.sleep_pressure = state.get("sleep_pressure", 0.0)
|
||
self.is_in_insomnia = state.get("is_in_insomnia", False)
|
||
self.insomnia_end_time = state.get("insomnia_end_time", 0.0)
|
||
logger = get_logger("hfc_context")
|
||
logger.info(f"{self.log_prefix} 成功从本地存储加载HFC上下文状态: {state}")
|
||
else:
|
||
logger = get_logger("hfc_context")
|
||
logger.info(f"{self.log_prefix} 未找到本地HFC上下文状态，将使用默认值初始化。")
|
||
|
||
def save_context_state(self):
|
||
"""将当前状态保存到本地存储"""
|
||
state = {
|
||
"energy_value": self.energy_value,
|
||
"sleep_pressure": self.sleep_pressure,
|
||
"is_in_insomnia": self.is_in_insomnia,
|
||
"insomnia_end_time": self.insomnia_end_time,
|
||
"last_wakeup_time": self.last_wakeup_time,
|
||
}
|
||
local_storage[self._get_storage_key()] = state
|
||
logger = get_logger("hfc_context")
|
||
logger.debug(f"{self.log_prefix} 已将HFC上下文状态保存到本地存储: {state}") |