better:不好意思刚刚不行,现在可以了
This commit is contained in:
@@ -122,6 +122,23 @@ SENDER_STYLE_CONFIG = {
|
||||
},
|
||||
}
|
||||
|
||||
HEARTFLOW_STYLE_CONFIG = {
|
||||
"advanced": {
|
||||
"console_format": (
|
||||
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
|
||||
"<level>{level: <8}</level> | "
|
||||
"<cyan>{extra[module]: <12}</cyan> | "
|
||||
"<light-yellow>麦麦大脑袋</light-yellow> | "
|
||||
"<level>{message}</level>"
|
||||
),
|
||||
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 麦麦大脑袋 | {message}"),
|
||||
},
|
||||
"simple": {
|
||||
"console_format": ("<green>{time:MM-DD HH:mm}</green> | <light-green>麦麦大脑袋</light-green> | <light-green>{message}</light-green>"), # noqa: E501
|
||||
"file_format": ("{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 麦麦大脑袋 | {message}"),
|
||||
},
|
||||
}
|
||||
|
||||
SCHEDULE_STYLE_CONFIG = {
|
||||
"advanced": {
|
||||
"console_format": (
|
||||
@@ -201,6 +218,7 @@ LLM_STYLE_CONFIG = LLM_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else LLM_STYLE_CO
|
||||
CHAT_STYLE_CONFIG = CHAT_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_STYLE_CONFIG["advanced"]
|
||||
MOOD_STYLE_CONFIG = MOOD_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MOOD_STYLE_CONFIG["advanced"]
|
||||
SCHEDULE_STYLE_CONFIG = SCHEDULE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else SCHEDULE_STYLE_CONFIG["advanced"]
|
||||
HEARTFLOW_STYLE_CONFIG = HEARTFLOW_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else HEARTFLOW_STYLE_CONFIG["advanced"]
|
||||
|
||||
def is_registered_module(record: dict) -> bool:
|
||||
"""检查是否为已注册的模块"""
|
||||
|
||||
@@ -76,8 +76,9 @@ class ScheduleGenerator:
|
||||
self.print_schedule()
|
||||
|
||||
# 执行当前活动
|
||||
current_activity = await self.move_doing(mind_thinking="")
|
||||
logger.info(f"当前活动: {current_activity}")
|
||||
# mind_thinking = subheartflow_manager.current_state.current_mind
|
||||
|
||||
await self.move_doing()
|
||||
|
||||
await asyncio.sleep(self.schedule_doing_update_interval)
|
||||
|
||||
@@ -190,12 +191,19 @@ class ScheduleGenerator:
|
||||
|
||||
async def move_doing(self,mind_thinking: str = ""):
|
||||
current_time = datetime.datetime.now()
|
||||
doing_prompt = self.construct_doing_prompt(current_time,mind_thinking)
|
||||
if mind_thinking:
|
||||
doing_prompt = self.construct_doing_prompt(current_time,mind_thinking)
|
||||
else:
|
||||
doing_prompt = self.construct_doing_prompt(current_time)
|
||||
|
||||
print(doing_prompt)
|
||||
doing_response,_ = await self.llm_scheduler_doing.generate_response_async(doing_prompt)
|
||||
self.today_done_list.append((current_time,doing_response))
|
||||
|
||||
await self.update_today_done_list()
|
||||
|
||||
logger.info(f"当前活动: {doing_response}")
|
||||
|
||||
return doing_response
|
||||
|
||||
async def get_task_from_time_to_time(self, start_time: str, end_time: str):
|
||||
|
||||
@@ -4,6 +4,14 @@ from src.plugins.models.utils_model import LLM_request
|
||||
from src.plugins.chat.config import global_config
|
||||
from src.plugins.schedule.schedule_generator import bot_schedule
|
||||
import asyncio
|
||||
from src.common.logger import get_module_logger, LogConfig, HEARTFLOW_STYLE_CONFIG # noqa: E402
|
||||
|
||||
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)
|
||||
|
||||
class CuttentState:
|
||||
def __init__(self):
|
||||
@@ -32,10 +40,10 @@ class Heartflow:
|
||||
async def heartflow_start_working(self):
|
||||
while True:
|
||||
await self.do_a_thinking()
|
||||
await asyncio.sleep(900)
|
||||
await asyncio.sleep(100)
|
||||
|
||||
async def do_a_thinking(self):
|
||||
print("麦麦大脑袋转起来了")
|
||||
logger.info("麦麦大脑袋转起来了")
|
||||
self.current_state.update_current_state_info()
|
||||
|
||||
personality_info = open("src/think_flow_demo/personality_info.txt", "r", encoding="utf-8").read()
|
||||
@@ -61,7 +69,10 @@ class Heartflow:
|
||||
self.update_current_mind(reponse)
|
||||
|
||||
self.current_mind = reponse
|
||||
print(f"麦麦的总体脑内状态:{self.current_mind}")
|
||||
logger.info(f"麦麦的总体脑内状态:{self.current_mind}")
|
||||
logger.info("麦麦想了想,当前活动:")
|
||||
await bot_schedule.move_doing(self.current_mind)
|
||||
|
||||
|
||||
for _, subheartflow in self._subheartflows.items():
|
||||
subheartflow.main_heartflow_info = reponse
|
||||
|
||||
Reference in New Issue
Block a user