2d1f23af88
将原先仅检查日程表首尾项的休眠判断方式，重构为通过关键词匹配所有日程项。 这样可以更灵活地在一天中的任何时段安排休眠，而不仅限于一天的开始或结束。 - 引入 `sleep_keywords` 列表，用于识别表示休眠的活动。 - 遍历当天的所有日程，检查活动名称是否包含休眠关键词。 - 如果关键词匹配，则进一步检查当前时间是否处于该活动的时间范围内。 - 保留了对唤醒状态的检查，确保在被唤醒时不会错误地判断为休眠。 同时，对 `hfc_context.py` 中的 `last_read_time` 初始值进行了微调，以改善相关逻辑的初始行为。
56 lines
2.2 KiB
Python
56 lines
2.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.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
|
|
|
|
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.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 |