109 lines
4.8 KiB
Python
109 lines
4.8 KiB
Python
from src.heart_flow.sub_heartflow import SubHeartflow
|
|
from src.plugins.models.utils_model import LLMRequest
|
|
from src.config.config import global_config
|
|
from src.plugins.schedule.schedule_generator import bot_schedule
|
|
from src.common.logger import get_module_logger, LogConfig, HEARTFLOW_STYLE_CONFIG
|
|
from typing import Any, Optional
|
|
from src.do_tool.tool_use import ToolUser
|
|
from src.plugins.person_info.relationship_manager import relationship_manager # Module instance
|
|
from src.heart_flow.mai_state_manager import MaiStateInfo, MaiStateManager
|
|
from src.heart_flow.subheartflow_manager import SubHeartflowManager
|
|
from src.heart_flow.mind import Mind
|
|
from src.heart_flow.interest_logger import InterestLogger # Import InterestLogger
|
|
from src.heart_flow.background_tasks import BackgroundTaskManager # Import BackgroundTaskManager
|
|
# --- End import ---
|
|
|
|
heartflow_config = LogConfig(
|
|
console_format=HEARTFLOW_STYLE_CONFIG["console_format"],
|
|
file_format=HEARTFLOW_STYLE_CONFIG["file_format"],
|
|
)
|
|
logger = get_module_logger("heartflow", config=heartflow_config)
|
|
|
|
|
|
# Task Intervals (should be in BackgroundTaskManager or config)
|
|
CLEANUP_INTERVAL_SECONDS = 1200
|
|
STATE_UPDATE_INTERVAL_SECONDS = 60
|
|
|
|
# Thresholds (should be in SubHeartflowManager or config)
|
|
INACTIVE_THRESHOLD_SECONDS = 1200
|
|
# --- End Constants --- #
|
|
|
|
|
|
class Heartflow:
|
|
"""主心流协调器，负责初始化并协调各个子系统:
|
|
- 状态管理 (MaiState)
|
|
- 子心流管理 (SubHeartflow)
|
|
- 思考过程 (Mind)
|
|
- 日志记录 (InterestLogger)
|
|
- 后台任务 (BackgroundTaskManager)
|
|
"""
|
|
|
|
def __init__(self):
|
|
# 核心状态
|
|
self.current_mind = "什么也没想" # 当前主心流想法
|
|
self.past_mind = [] # 历史想法记录
|
|
|
|
# 状态管理相关
|
|
self.current_state: MaiStateInfo = MaiStateInfo() # 当前状态信息
|
|
self.mai_state_manager: MaiStateManager = MaiStateManager() # 状态决策管理器
|
|
|
|
# 子心流管理 (在初始化时传入 current_state)
|
|
self.subheartflow_manager: SubHeartflowManager = SubHeartflowManager(self.current_state)
|
|
|
|
# LLM模型配置
|
|
self.llm_model = LLMRequest(
|
|
model=global_config.llm_heartflow, temperature=0.6, max_tokens=1000, request_type="heart_flow"
|
|
)
|
|
|
|
# 外部依赖模块
|
|
self.tool_user_instance = ToolUser() # 工具使用模块
|
|
self.relationship_manager_instance = relationship_manager # 关系管理模块
|
|
|
|
# 子系统初始化
|
|
self.mind: Mind = Mind(self.subheartflow_manager, self.llm_model) # 思考管理器
|
|
self.interest_logger: InterestLogger = InterestLogger(self.subheartflow_manager, self) # 兴趣日志记录器
|
|
|
|
# 后台任务管理器 (整合所有定时任务)
|
|
self.background_task_manager: BackgroundTaskManager = BackgroundTaskManager(
|
|
mai_state_info=self.current_state,
|
|
mai_state_manager=self.mai_state_manager,
|
|
subheartflow_manager=self.subheartflow_manager,
|
|
interest_logger=self.interest_logger,
|
|
update_interval=STATE_UPDATE_INTERVAL_SECONDS,
|
|
cleanup_interval=CLEANUP_INTERVAL_SECONDS,
|
|
log_interval=3, # Example: Using value directly, ideally get from config
|
|
inactive_threshold=INACTIVE_THRESHOLD_SECONDS,
|
|
)
|
|
|
|
async def get_or_create_subheartflow(self, subheartflow_id: Any) -> Optional["SubHeartflow"]:
|
|
"""获取或创建一个新的SubHeartflow实例 - 委托给 SubHeartflowManager"""
|
|
# 不再需要传入 self.current_state
|
|
return await self.subheartflow_manager.get_or_create_subheartflow(subheartflow_id)
|
|
|
|
async def heartflow_start_working(self):
|
|
"""启动后台任务"""
|
|
await self.background_task_manager.start_tasks()
|
|
logger.info("[Heartflow] 后台任务已启动")
|
|
|
|
# 根本不会用到这个函数吧，那样麦麦直接死了
|
|
async def stop_working(self):
|
|
"""停止所有任务和子心流"""
|
|
logger.info("[Heartflow] 正在停止任务和子心流...")
|
|
await self.background_task_manager.stop_tasks()
|
|
await self.subheartflow_manager.deactivate_all_subflows()
|
|
logger.info("[Heartflow] 所有任务和子心流已停止")
|
|
|
|
async def do_a_thinking(self):
|
|
"""执行一次主心流思考过程"""
|
|
schedule_info = bot_schedule.get_current_num_task(num=4, time_info=True)
|
|
new_mind = await self.mind.do_a_thinking(
|
|
current_main_mind=self.current_mind, mai_state_info=self.current_state, schedule_info=schedule_info
|
|
)
|
|
self.past_mind.append(self.current_mind)
|
|
self.current_mind = new_mind
|
|
logger.info(f"麦麦的总体脑内状态更新为：{self.current_mind[:100]}...")
|
|
self.mind.update_subflows_with_main_mind(new_mind)
|
|
|
|
|
|
heartflow = Heartflow()
|