Merge branch 'dev' of https://github.com/MaiM-with-u/MaiBot into groupnickname

bot.py

@@ -13,12 +13,16 @@ from src.common.logger_manager import get_logger
 # from src.common.logger import LogConfig, CONFIRM_STYLE_CONFIG
 from src.common.crash_logger import install_crash_handler
 from src.main import MainSystem
+from rich.traceback import install
 from src.plugins.group_nickname.nickname_processor import (
     start_nickname_processor,
     stop_nickname_processor,
-)  # <--- 添加这行导入
+)
 import atexit
 
+install(show_locals=True, extra_lines=3)
+
+
 logger = get_logger("main")
 confirm_logger = get_logger("confirm")
 # 获取没有加载env时的环境变量

src/common/database.py

@@ -1,6 +1,9 @@
 import os
 from pymongo import MongoClient
 from pymongo.database import Database
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 _client = None
 _db = None

src/common/log_decorators.py

@@ -2,6 +2,9 @@ import functools
 import inspect
 from typing import Callable, Any
 from .logger import logger, add_custom_style_handler
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 def use_log_style(

src/common/server.py

@@ -2,6 +2,9 @@ from fastapi import FastAPI, APIRouter
 from typing import Optional
 from uvicorn import Config, Server as UvicornServer
 import os
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 class Server:

src/config/config.py

@@ -14,6 +14,9 @@ from packaging.version import Version, InvalidVersion
 from packaging.specifiers import SpecifierSet, InvalidSpecifier
 
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 # 配置主程序日志格式

src/do_tool/tool_can_use/base_tool.py

@@ -4,6 +4,9 @@ import importlib
 import pkgutil
 import os
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("base_tool")
 

src/individuality/individuality.py

@@ -2,6 +2,9 @@ from typing import Optional
 from .personality import Personality
 from .identity import Identity
 import random
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 class Individuality:

src/individuality/offline_llm.py

@@ -6,6 +6,9 @@ from typing import Tuple, Union
 import aiohttp
 import requests
 from src.common.logger import get_module_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_module_logger("offline_llm")
 

src/main.py

@@ -17,6 +17,9 @@ from .common.logger_manager import get_logger
 from .plugins.remote import heartbeat_thread  # noqa: F401
 from .individuality.individuality import Individuality
 from .common.server import global_server
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("main")
 

src/plugins/PFC/chat_observer.py

@@ -7,6 +7,9 @@ from maim_message import UserInfo
 from ...config.config import global_config
 from .chat_states import NotificationManager, create_new_message_notification, create_cold_chat_notification
 from .message_storage import MongoDBMessageStorage
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_module_logger("chat_observer")
 

src/plugins/PFC/conversation.py

@@ -23,6 +23,9 @@ from .pfc_KnowledgeFetcher import KnowledgeFetcher
 from .waiter import Waiter
 
 import traceback
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("pfc")
 

src/plugins/PFC/message_sender.py

@@ -8,6 +8,9 @@ from src.plugins.chat.message import MessageSending, MessageSet
 from src.plugins.chat.message_sender import message_manager
 from ..storage.storage import MessageStorage
 from ...config.config import global_config
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 logger = get_module_logger("message_sender")

src/plugins/PFC/pfc.py

@@ -8,6 +8,9 @@ from src.individuality.individuality import Individuality
 from .conversation_info import ConversationInfo
 from .observation_info import ObservationInfo
 from src.plugins.utils.chat_message_builder import build_readable_messages
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 if TYPE_CHECKING:
     pass

src/plugins/chat/chat_stream.py

@@ -9,6 +9,9 @@ from ...common.database import db
 from maim_message import GroupInfo, UserInfo
 
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 logger = get_logger("chat_stream")

src/plugins/chat/message.py

@@ -9,6 +9,9 @@ from src.common.logger_manager import get_logger
 from .chat_stream import ChatStream
 from .utils_image import image_manager
 from maim_message import Seg, UserInfo, BaseMessageInfo, MessageBase
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("chat_message")
 

src/plugins/chat/message_sender.py

@@ -13,6 +13,9 @@ from ...config.config import global_config
 from .utils import truncate_message, calculate_typing_time, count_messages_between
 
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 logger = get_logger("sender")

src/plugins/chat/utils_image.py

@@ -13,6 +13,9 @@ from ...config.config import global_config
 from ..models.utils_model import LLMRequest
 
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("chat_image")
 

src/plugins/config_reload/api.py

@@ -1,4 +1,7 @@
 from fastapi import APIRouter, HTTPException
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 # 创建APIRouter而不是FastAPI实例
 router = APIRouter()

src/plugins/emoji_system/emoji_manager.py

@@ -15,6 +15,9 @@ from ...config.config import global_config
 from ..chat.utils_image import image_path_to_base64, image_manager
 from ..models.utils_model import LLMRequest
 from src.common.logger_manager import get_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_logger("emoji")
 

src/plugins/heartFC_chat/heartFC_chat.py

@@ -27,8 +27,12 @@ from src.plugins.chat.utils import process_llm_response
 from src.plugins.respon_info_catcher.info_catcher import info_catcher_manager
 from src.plugins.moods.moods import MoodManager
 from src.heart_flow.utils_chat import get_chat_type_and_target_info
+from rich.traceback import install
 from src.plugins.group_nickname.nickname_utils import trigger_nickname_analysis_if_needed
 
+install(show_locals=True, extra_lines=3)
+
+
 
 WAITING_TIME_THRESHOLD = 300  # 等待新消息时间阈值，单位秒
 

src/plugins/heartFC_chat/heartFC_sender.py

@@ -9,6 +9,9 @@ from ..storage.storage import MessageStorage
 from ..chat.utils import truncate_message
 from src.common.logger_manager import get_logger
 from src.plugins.chat.utils import calculate_typing_time
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 logger = get_logger("sender")

src/plugins/knowledge/src/embedding_store.py

@@ -12,6 +12,9 @@ from .llm_client import LLMClient
 from .lpmmconfig import ENT_NAMESPACE, PG_NAMESPACE, REL_NAMESPACE, global_config
 from .utils.hash import get_sha256
 from .global_logger import logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 @dataclass

src/plugins/memory_system/Hippocampus.py

@@ -20,6 +20,9 @@ from ..utils.chat_message_builder import (
 )  # 导入 build_readable_messages
 from ..chat.utils import translate_timestamp_to_human_readable
 from .memory_config import MemoryConfig
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 def calculate_information_content(text):

src/plugins/memory_system/debug_memory.py

@@ -8,6 +8,9 @@ import os
 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))))
 from src.plugins.memory_system.Hippocampus import HippocampusManager
 from src.config.config import global_config
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 async def test_memory_system():

src/plugins/memory_system/manually_alter_memory.py

@@ -9,6 +9,9 @@ from Hippocampus import Hippocampus  # 海马体和记忆图
 
 
 from dotenv import load_dotenv
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 """

src/plugins/memory_system/offline_llm.py

@@ -6,6 +6,9 @@ from typing import Tuple, Union
 import aiohttp
 import requests
 from src.common.logger import get_module_logger
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_module_logger("offline_llm")
 

src/plugins/memory_system/sample_distribution.py

@@ -1,6 +1,9 @@
 import numpy as np
 from scipy import stats
 from datetime import datetime, timedelta
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 
 class DistributionVisualizer:

src/plugins/models/utils_model.py

@@ -14,6 +14,9 @@ import io
 import os
 from ...common.database import db
 from ...config.config import global_config
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_module_logger("model_utils")
 

src/plugins/utils/prompt_builder.py

@@ -3,7 +3,11 @@ import re
 from contextlib import asynccontextmanager
 import asyncio
 from src.common.logger import get_module_logger
+
 # import traceback
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 logger = get_module_logger("prompt_build")
 

src/plugins/utils/timer_calculator.py

@@ -2,6 +2,9 @@ from time import perf_counter
 from functools import wraps
 from typing import Optional, Dict, Callable
 import asyncio
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 """
 # 更好的计时器

src/plugins/willing/willing_manager.py

@@ -8,6 +8,9 @@ from abc import ABC, abstractmethod
 import importlib
 from typing import Dict, Optional
 import asyncio
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 """
 基类方法概览：

src/plugins/zhishi/knowledge_library.py

@@ -7,6 +7,9 @@ from datetime import datetime
 from tqdm import tqdm
 from rich.console import Console
 from rich.table import Table
+from rich.traceback import install
+
+install(show_locals=True, extra_lines=3)
 
 # 添加项目根目录到 Python 路径
 root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
@@ -15,6 +18,7 @@ sys.path.append(root_path)
 # 现在可以导入src模块
 from src.common.database import db  # noqa E402
 
+
 # 加载根目录下的env.edv文件
 env_path = os.path.join(root_path, ".env")
 if not os.path.exists(env_path):