better:不好意思刚刚不行,现在可以了

This commit is contained in:
SengokuCola
2025-03-26 23:19:19 +08:00
parent 572bffc273
commit 67291f1b49
3 changed files with 43 additions and 6 deletions

View File

@@ -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 = { SCHEDULE_STYLE_CONFIG = {
"advanced": { "advanced": {
"console_format": ( "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"] 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"] 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"] 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: def is_registered_module(record: dict) -> bool:
"""检查是否为已注册的模块""" """检查是否为已注册的模块"""

View File

@@ -76,8 +76,9 @@ class ScheduleGenerator:
self.print_schedule() self.print_schedule()
# 执行当前活动 # 执行当前活动
current_activity = await self.move_doing(mind_thinking="") # mind_thinking = subheartflow_manager.current_state.current_mind
logger.info(f"当前活动: {current_activity}")
await self.move_doing()
await asyncio.sleep(self.schedule_doing_update_interval) await asyncio.sleep(self.schedule_doing_update_interval)
@@ -190,12 +191,19 @@ class ScheduleGenerator:
async def move_doing(self,mind_thinking: str = ""): async def move_doing(self,mind_thinking: str = ""):
current_time = datetime.datetime.now() 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) doing_response,_ = await self.llm_scheduler_doing.generate_response_async(doing_prompt)
self.today_done_list.append((current_time,doing_response)) self.today_done_list.append((current_time,doing_response))
await self.update_today_done_list() await self.update_today_done_list()
logger.info(f"当前活动: {doing_response}")
return doing_response return doing_response
async def get_task_from_time_to_time(self, start_time: str, end_time: str): async def get_task_from_time_to_time(self, start_time: str, end_time: str):

View File

@@ -4,6 +4,14 @@ from src.plugins.models.utils_model import LLM_request
from src.plugins.chat.config import global_config from src.plugins.chat.config import global_config
from src.plugins.schedule.schedule_generator import bot_schedule from src.plugins.schedule.schedule_generator import bot_schedule
import asyncio 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: class CuttentState:
def __init__(self): def __init__(self):
@@ -32,10 +40,10 @@ class Heartflow:
async def heartflow_start_working(self): async def heartflow_start_working(self):
while True: while True:
await self.do_a_thinking() await self.do_a_thinking()
await asyncio.sleep(900) await asyncio.sleep(100)
async def do_a_thinking(self): async def do_a_thinking(self):
print("麦麦大脑袋转起来了") logger.info("麦麦大脑袋转起来了")
self.current_state.update_current_state_info() self.current_state.update_current_state_info()
personality_info = open("src/think_flow_demo/personality_info.txt", "r", encoding="utf-8").read() 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.update_current_mind(reponse)
self.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(): for _, subheartflow in self._subheartflows.items():
subheartflow.main_heartflow_info = reponse subheartflow.main_heartflow_info = reponse