mirror of https://github.com/Mai-with-u/MaiBot.git
parent
57af092308
commit
6a3e58b383
|
|
@ -9,6 +9,7 @@ from src.common.logger import (
|
|||
RELATION_STYLE_CONFIG,
|
||||
CONFIG_STYLE_CONFIG,
|
||||
HEARTFLOW_STYLE_CONFIG,
|
||||
SCHEDULE_STYLE_CONFIG,
|
||||
LLM_STYLE_CONFIG,
|
||||
CHAT_STYLE_CONFIG,
|
||||
EMOJI_STYLE_CONFIG,
|
||||
|
|
@ -55,6 +56,7 @@ MODULE_LOGGER_CONFIGS = {
|
|||
"relation": RELATION_STYLE_CONFIG, # 关系
|
||||
"config": CONFIG_STYLE_CONFIG, # 配置
|
||||
"heartflow": HEARTFLOW_STYLE_CONFIG, # 麦麦大脑袋
|
||||
"L_schedule": SCHEDULE_STYLE_CONFIG, # 在干嘛
|
||||
"llm": LLM_STYLE_CONFIG, # 麦麦组织语言
|
||||
"chat": CHAT_STYLE_CONFIG, # 见闻
|
||||
"emoji": EMOJI_STYLE_CONFIG, # 表情包
|
||||
|
|
|
|||
|
|
@ -40,7 +40,7 @@ EMOJI_SEND_PRO = 0.3 # 设置一个概率,比如 30% 才真的发
|
|||
CONSECUTIVE_NO_REPLY_THRESHOLD = 3 # 连续不回复的阈值
|
||||
|
||||
|
||||
logger = get_logger("L_hfc") # Logger Name Changed
|
||||
logger = get_logger("hfc") # Logger Name Changed
|
||||
|
||||
|
||||
# 默认动作定义
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@ from rich.traceback import install
|
|||
install(extra_lines=3)
|
||||
|
||||
|
||||
logger = get_logger("L_sender")
|
||||
logger = get_logger("sender")
|
||||
|
||||
|
||||
async def send_message(message: MessageSending) -> None:
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ from .subheartflow_manager import SubHeartflowManager
|
|||
from .interest_logger import InterestLogger
|
||||
|
||||
|
||||
logger = get_logger("L_background_tasks")
|
||||
logger = get_logger("background_tasks")
|
||||
|
||||
|
||||
# 新增兴趣评估间隔
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@ from .mind import Mind
|
|||
from .interest_logger import InterestLogger # Import InterestLogger
|
||||
from .background_tasks import BackgroundTaskManager # Import BackgroundTaskManager
|
||||
|
||||
logger = get_logger("L_heartflow")
|
||||
logger = get_logger("heartflow")
|
||||
|
||||
|
||||
class Heartflow:
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ import math
|
|||
# 定义常量 (从 interest.py 移动过来)
|
||||
MAX_INTEREST = 15.0
|
||||
|
||||
logger = get_logger("L_interest_chatting")
|
||||
logger = get_logger("interest_chatting")
|
||||
|
||||
PROBABILITY_INCREASE_RATE_PER_SECOND = 0.1
|
||||
PROBABILITY_DECREASE_RATE_PER_SECOND = 0.1
|
||||
|
|
|
|||
|
|
@ -16,7 +16,7 @@ if TYPE_CHECKING:
|
|||
from .heartflow import Heartflow # 导入 Heartflow 类型
|
||||
|
||||
|
||||
logger = get_logger("L_interest")
|
||||
logger = get_logger("interest")
|
||||
|
||||
# Consider moving log directory/filename constants here
|
||||
LOG_DIRECTORY = "logs/interest"
|
||||
|
|
|
|||
|
|
@ -6,7 +6,7 @@ from src.common.logger_manager import get_logger
|
|||
from src.manager.mood_manager import mood_manager
|
||||
from src.config.config import global_config
|
||||
|
||||
logger = get_logger("L_mai_state")
|
||||
logger = get_logger("mai_state")
|
||||
|
||||
|
||||
# -- 状态相关的可配置参数 (可以从 glocal_config 加载) --
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ if TYPE_CHECKING:
|
|||
from .mai_state_manager import MaiStateInfo
|
||||
|
||||
|
||||
logger = get_logger("L_sub_heartflow_mind")
|
||||
logger = get_logger("sub_heartflow_mind")
|
||||
|
||||
|
||||
class Mind:
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@ from src.chat.message_receive.message import MessageRecv # 添加 MessageRecv
|
|||
# Import the new utility function
|
||||
from .utils_chat import get_chat_type_and_target_info
|
||||
|
||||
logger = get_logger("L_observation")
|
||||
logger = get_logger("observation")
|
||||
|
||||
# --- Define Prompt Templates for Chat Summary ---
|
||||
Prompt(
|
||||
|
|
|
|||
|
|
@ -15,7 +15,7 @@ from .utils_chat import get_chat_type_and_target_info
|
|||
from .interest_chatting import InterestChatting
|
||||
|
||||
|
||||
logger = get_logger("L_sub_heartflow")
|
||||
logger = get_logger("sub_heartflow")
|
||||
|
||||
|
||||
class SubHeartflow:
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@ from src.chat.memory_system.Hippocampus import HippocampusManager
|
|||
import jieba
|
||||
|
||||
|
||||
logger = get_logger("L_sub_heartflow")
|
||||
logger = get_logger("sub_heartflow")
|
||||
|
||||
|
||||
def init_prompt():
|
||||
|
|
|
|||
|
|
@ -25,7 +25,7 @@ import traceback
|
|||
|
||||
# 初始化日志记录器
|
||||
|
||||
logger = get_logger("L_subheartflow_manager")
|
||||
logger = get_logger("subheartflow_manager")
|
||||
|
||||
# 子心流管理相关常量
|
||||
INACTIVE_THRESHOLD_SECONDS = 3600 # 子心流不活跃超时时间(秒)
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ from src.common.logger_manager import get_logger
|
|||
from src.chat.message_receive.chat_stream import chat_manager
|
||||
from src.chat.person_info.person_info import person_info_manager
|
||||
|
||||
logger = get_logger("L_heartflow_utils")
|
||||
logger = get_logger("heartflow_utils")
|
||||
|
||||
|
||||
async def get_chat_type_and_target_info(chat_id: str) -> Tuple[bool, Optional[Dict]]:
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@ from src.chat.utils.timer_calculator import Timer
|
|||
from src.chat.person_info.relationship_manager import relationship_manager
|
||||
from typing import Optional, Tuple, Dict, Any
|
||||
|
||||
logger = get_logger("L_chat")
|
||||
logger = get_logger("chat")
|
||||
|
||||
|
||||
async def _handle_error(error: Exception, context: str, message: Optional[MessageRecv] = None) -> None:
|
||||
|
|
|
|||
|
|
@ -18,7 +18,7 @@ from src.plugins.group_nickname.nickname_manager import nickname_manager
|
|||
import traceback
|
||||
from .heartFC_Cycleinfo import CycleInfo
|
||||
|
||||
logger = get_logger("L_prompt")
|
||||
logger = get_logger("prompt")
|
||||
|
||||
|
||||
def init_prompt():
|
||||
|
|
|
|||
|
|
@ -24,7 +24,7 @@ from ...config.config import global_config
|
|||
from src.plugins.group_nickname.nickname_manager import nickname_manager
|
||||
|
||||
|
||||
logger = get_logger("L_chat")
|
||||
logger = get_logger("chat")
|
||||
|
||||
|
||||
class NormalChat:
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ from src.common.logger_manager import get_logger
|
|||
from src.chat.utils.info_catcher import info_catcher_manager
|
||||
|
||||
|
||||
logger = get_logger("L_llm")
|
||||
logger = get_logger("llm")
|
||||
|
||||
|
||||
class NormalChatGenerator:
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@ schedule_config = LogConfig(
|
|||
console_format=SCHEDULE_STYLE_CONFIG["console_format"],
|
||||
file_format=SCHEDULE_STYLE_CONFIG["file_format"],
|
||||
)
|
||||
logger = get_module_logger("L_scheduler", config=schedule_config)
|
||||
logger = get_module_logger("scheduler", config=schedule_config)
|
||||
|
||||
|
||||
class ScheduleGenerator:
|
||||
|
|
|
|||
10
src/main.py
10
src/main.py
|
|
@ -9,6 +9,7 @@ from .chat.emoji_system.emoji_manager import emoji_manager
|
|||
from .chat.person_info.person_info import person_info_manager
|
||||
from .chat.normal_chat.willing.willing_manager import willing_manager
|
||||
from .chat.message_receive.chat_stream import chat_manager
|
||||
from src.experimental.Legacy_HFC.schedule.schedule_generator import bot_schedule
|
||||
from src.chat.heart_flow.heartflow import heartflow
|
||||
from src.experimental.Legacy_HFC.heart_flow.heartflow import heartflow as legacy_heartflow
|
||||
from .chat.memory_system.Hippocampus import HippocampusManager
|
||||
|
|
@ -80,6 +81,15 @@ class MainSystem:
|
|||
# 启动愿望管理器
|
||||
await willing_manager.async_task_starter()
|
||||
|
||||
# 初始化日程
|
||||
bot_schedule.initialize(
|
||||
name=global_config.BOT_NICKNAME,
|
||||
personality=global_config.personality_core,
|
||||
behavior=global_config.PROMPT_SCHEDULE_GEN,
|
||||
interval=global_config.SCHEDULE_DOING_UPDATE_INTERVAL,
|
||||
)
|
||||
asyncio.create_task(bot_schedule.mai_schedule_start())
|
||||
|
||||
# 初始化聊天管理器
|
||||
await chat_manager._initialize()
|
||||
asyncio.create_task(chat_manager._auto_save_task())
|
||||
|
|
|
|||
Loading…
Reference in New Issue