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 = {
|
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:
|
||||||
"""检查是否为已注册的模块"""
|
"""检查是否为已注册的模块"""
|
||||||
|
|||||||
@@ -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()
|
||||||
|
if mind_thinking:
|
||||||
doing_prompt = self.construct_doing_prompt(current_time,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):
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user