feat: 为normal增加action_plan能力

This commit is contained in:
tcmofashi
2025-06-02 21:06:00 +08:00
parent 13c3a40ca5
commit d932b48444
11 changed files with 867 additions and 21 deletions

View File

@@ -1,6 +1,6 @@
from src.chat.heart_flow.sub_heartflow import SubHeartflow, ChatState
from src.common.logger_manager import get_logger
from typing import Any, Optional, List
from typing import Any, Optional, List, Dict
from src.chat.heart_flow.subheartflow_manager import SubHeartflowManager
from src.chat.heart_flow.background_tasks import BackgroundTaskManager # Import BackgroundTaskManager

View File

@@ -330,6 +330,27 @@ class SubHeartflow:
oldest_key = next(iter(self.interest_dict))
self.interest_dict.pop(oldest_key)
def get_normal_chat_action_manager(self):
"""获取NormalChat的ActionManager实例
Returns:
ActionManager: NormalChat的ActionManager实例如果不存在则返回None
"""
if self.normal_chat_instance:
return self.normal_chat_instance.get_action_manager()
return None
def set_normal_chat_planner_enabled(self, enabled: bool):
"""设置NormalChat的planner是否启用
Args:
enabled: 是否启用planner
"""
if self.normal_chat_instance:
self.normal_chat_instance.set_planner_enabled(enabled)
else:
logger.warning(f"{self.log_prefix} NormalChat实例不存在无法设置planner状态")
async def get_full_state(self) -> dict:
"""获取子心流的完整状态,包括兴趣、思维和聊天状态。"""
return {