From 9cf62f983e00f9f6a7c1fdc005de2250c9d6762c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com> Date: Fri, 2 May 2025 03:18:24 +0800 Subject: [PATCH 1/2] =?UTF-8?q?feat=EF=BC=9A=E4=B8=BA=E5=A4=9A=E4=B8=AA?= =?UTF-8?q?=E6=A8=A1=E5=9D=97=E6=B7=BB=E5=8A=A0=E4=B8=B0=E5=AF=8C=E7=9A=84?= =?UTF-8?q?=E8=BF=BD=E8=B8=AA=E4=BF=A1=E6=81=AF=E4=BB=A5=E5=A2=9E=E5=BC=BA?= =?UTF-8?q?=E8=B0=83=E8=AF=95=E8=83=BD=E5=8A=9B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot.py | 2 ++ src/common/database.py | 2 ++ src/common/log_decorators.py | 2 ++ src/common/server.py | 2 ++ src/config/config.py | 2 ++ src/do_tool/tool_can_use/base_tool.py | 2 ++ src/individuality/individuality.py | 2 ++ src/individuality/offline_llm.py | 2 ++ src/main.py | 2 ++ src/plugins/PFC/chat_observer.py | 2 ++ src/plugins/PFC/conversation.py | 2 ++ src/plugins/PFC/message_sender.py | 2 ++ src/plugins/PFC/pfc.py | 2 ++ src/plugins/chat/chat_stream.py | 2 ++ src/plugins/chat/message.py | 2 ++ src/plugins/chat/message_sender.py | 2 ++ src/plugins/chat/utils_image.py | 2 ++ src/plugins/config_reload/api.py | 2 ++ src/plugins/emoji_system/emoji_manager.py | 2 ++ src/plugins/heartFC_chat/heartFC_chat.py | 2 ++ src/plugins/heartFC_chat/heartFC_sender.py | 2 ++ src/plugins/knowledge/src/embedding_store.py | 2 ++ src/plugins/memory_system/Hippocampus.py | 2 ++ src/plugins/memory_system/debug_memory.py | 2 ++ src/plugins/memory_system/manually_alter_memory.py | 2 ++ src/plugins/memory_system/offline_llm.py | 2 ++ src/plugins/memory_system/sample_distribution.py | 2 ++ src/plugins/models/utils_model.py | 2 ++ src/plugins/utils/prompt_builder.py | 2 ++ src/plugins/utils/timer_calculator.py | 2 ++ src/plugins/willing/willing_manager.py | 2 ++ src/plugins/zhishi/knowledge_library.py | 3 +++ 32 files changed, 65 insertions(+) diff --git a/bot.py b/bot.py index 9a4a5000..ecdbd2b9 100644 --- a/bot.py +++ b/bot.py @@ -13,6 +13,8 @@ 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 +install(show_locals=True, extra_lines=3) logger = get_logger("main") diff --git a/src/common/database.py b/src/common/database.py index ee0ead0b..846c4ee7 100644 --- a/src/common/database.py +++ b/src/common/database.py @@ -1,6 +1,8 @@ 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 diff --git a/src/common/log_decorators.py b/src/common/log_decorators.py index 9838717f..4e881984 100644 --- a/src/common/log_decorators.py +++ b/src/common/log_decorators.py @@ -2,6 +2,8 @@ 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( diff --git a/src/common/server.py b/src/common/server.py index 51799629..0511f443 100644 --- a/src/common/server.py +++ b/src/common/server.py @@ -2,6 +2,8 @@ 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: diff --git a/src/config/config.py b/src/config/config.py index f5be7ec6..29c28c83 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -14,6 +14,8 @@ 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) # 配置主程序日志格式 diff --git a/src/do_tool/tool_can_use/base_tool.py b/src/do_tool/tool_can_use/base_tool.py index 42570884..298bc15a 100644 --- a/src/do_tool/tool_can_use/base_tool.py +++ b/src/do_tool/tool_can_use/base_tool.py @@ -4,6 +4,8 @@ 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") diff --git a/src/individuality/individuality.py b/src/individuality/individuality.py index 9ffdfdaa..6befe3e5 100644 --- a/src/individuality/individuality.py +++ b/src/individuality/individuality.py @@ -2,6 +2,8 @@ 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: diff --git a/src/individuality/offline_llm.py b/src/individuality/offline_llm.py index 2b5b6dc2..2059d7fd 100644 --- a/src/individuality/offline_llm.py +++ b/src/individuality/offline_llm.py @@ -6,6 +6,8 @@ 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") diff --git a/src/main.py b/src/main.py index c0e743d6..1f6ed188 100644 --- a/src/main.py +++ b/src/main.py @@ -17,6 +17,8 @@ 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") diff --git a/src/plugins/PFC/chat_observer.py b/src/plugins/PFC/chat_observer.py index 1a17db90..675e9963 100644 --- a/src/plugins/PFC/chat_observer.py +++ b/src/plugins/PFC/chat_observer.py @@ -7,6 +7,8 @@ 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") diff --git a/src/plugins/PFC/conversation.py b/src/plugins/PFC/conversation.py index 9f744c30..f7702e1b 100644 --- a/src/plugins/PFC/conversation.py +++ b/src/plugins/PFC/conversation.py @@ -23,6 +23,8 @@ 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") diff --git a/src/plugins/PFC/message_sender.py b/src/plugins/PFC/message_sender.py index 53c20374..709150c7 100644 --- a/src/plugins/PFC/message_sender.py +++ b/src/plugins/PFC/message_sender.py @@ -8,6 +8,8 @@ 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") diff --git a/src/plugins/PFC/pfc.py b/src/plugins/PFC/pfc.py index 6cb1fe83..4ce98a52 100644 --- a/src/plugins/PFC/pfc.py +++ b/src/plugins/PFC/pfc.py @@ -8,6 +8,8 @@ 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 diff --git a/src/plugins/chat/chat_stream.py b/src/plugins/chat/chat_stream.py index 14d02a81..d3095f3b 100644 --- a/src/plugins/chat/chat_stream.py +++ b/src/plugins/chat/chat_stream.py @@ -9,6 +9,8 @@ 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") diff --git a/src/plugins/chat/message.py b/src/plugins/chat/message.py index a0ff33ab..4c11dd98 100644 --- a/src/plugins/chat/message.py +++ b/src/plugins/chat/message.py @@ -9,6 +9,8 @@ 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") diff --git a/src/plugins/chat/message_sender.py b/src/plugins/chat/message_sender.py index 61e2dd49..aaa48ca8 100644 --- a/src/plugins/chat/message_sender.py +++ b/src/plugins/chat/message_sender.py @@ -13,6 +13,8 @@ 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") diff --git a/src/plugins/chat/utils_image.py b/src/plugins/chat/utils_image.py index f567c527..285b6394 100644 --- a/src/plugins/chat/utils_image.py +++ b/src/plugins/chat/utils_image.py @@ -13,6 +13,8 @@ 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") diff --git a/src/plugins/config_reload/api.py b/src/plugins/config_reload/api.py index 327451e2..93b5d1cb 100644 --- a/src/plugins/config_reload/api.py +++ b/src/plugins/config_reload/api.py @@ -1,4 +1,6 @@ from fastapi import APIRouter, HTTPException +from rich.traceback import install +install(show_locals=True, extra_lines=3) # 创建APIRouter而不是FastAPI实例 router = APIRouter() diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index f789f906..67075e29 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -15,6 +15,8 @@ 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") diff --git a/src/plugins/heartFC_chat/heartFC_chat.py b/src/plugins/heartFC_chat/heartFC_chat.py index f203926f..077d623b 100644 --- a/src/plugins/heartFC_chat/heartFC_chat.py +++ b/src/plugins/heartFC_chat/heartFC_chat.py @@ -27,6 +27,8 @@ 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 +install(show_locals=True, extra_lines=3) WAITING_TIME_THRESHOLD = 300 # 等待新消息时间阈值，单位秒 diff --git a/src/plugins/heartFC_chat/heartFC_sender.py b/src/plugins/heartFC_chat/heartFC_sender.py index ee4e86ee..001f578a 100644 --- a/src/plugins/heartFC_chat/heartFC_sender.py +++ b/src/plugins/heartFC_chat/heartFC_sender.py @@ -9,6 +9,8 @@ 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") diff --git a/src/plugins/knowledge/src/embedding_store.py b/src/plugins/knowledge/src/embedding_store.py index 9e60b8e1..98ab757c 100644 --- a/src/plugins/knowledge/src/embedding_store.py +++ b/src/plugins/knowledge/src/embedding_store.py @@ -12,6 +12,8 @@ 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 diff --git a/src/plugins/memory_system/Hippocampus.py b/src/plugins/memory_system/Hippocampus.py index ccc4d4fa..789da590 100644 --- a/src/plugins/memory_system/Hippocampus.py +++ b/src/plugins/memory_system/Hippocampus.py @@ -20,6 +20,8 @@ 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): diff --git a/src/plugins/memory_system/debug_memory.py b/src/plugins/memory_system/debug_memory.py index 4e357557..03c8fcac 100644 --- a/src/plugins/memory_system/debug_memory.py +++ b/src/plugins/memory_system/debug_memory.py @@ -8,6 +8,8 @@ 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(): diff --git a/src/plugins/memory_system/manually_alter_memory.py b/src/plugins/memory_system/manually_alter_memory.py index 1452d3d5..4054265b 100644 --- a/src/plugins/memory_system/manually_alter_memory.py +++ b/src/plugins/memory_system/manually_alter_memory.py @@ -9,6 +9,8 @@ from Hippocampus import Hippocampus # 海马体和记忆图 from dotenv import load_dotenv +from rich.traceback import install +install(show_locals=True, extra_lines=3) """ diff --git a/src/plugins/memory_system/offline_llm.py b/src/plugins/memory_system/offline_llm.py index fc50b17b..f86ec411 100644 --- a/src/plugins/memory_system/offline_llm.py +++ b/src/plugins/memory_system/offline_llm.py @@ -6,6 +6,8 @@ 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") diff --git a/src/plugins/memory_system/sample_distribution.py b/src/plugins/memory_system/sample_distribution.py index 5dae2f26..f4fec12b 100644 --- a/src/plugins/memory_system/sample_distribution.py +++ b/src/plugins/memory_system/sample_distribution.py @@ -1,6 +1,8 @@ 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: diff --git a/src/plugins/models/utils_model.py b/src/plugins/models/utils_model.py index 6b9c0416..60b507b9 100644 --- a/src/plugins/models/utils_model.py +++ b/src/plugins/models/utils_model.py @@ -14,6 +14,8 @@ 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") diff --git a/src/plugins/utils/prompt_builder.py b/src/plugins/utils/prompt_builder.py index 578d9677..02347033 100644 --- a/src/plugins/utils/prompt_builder.py +++ b/src/plugins/utils/prompt_builder.py @@ -4,6 +4,8 @@ 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") diff --git a/src/plugins/utils/timer_calculator.py b/src/plugins/utils/timer_calculator.py index 13bc26f1..076a4fe5 100644 --- a/src/plugins/utils/timer_calculator.py +++ b/src/plugins/utils/timer_calculator.py @@ -2,6 +2,8 @@ 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) """ # 更好的计时器 diff --git a/src/plugins/willing/willing_manager.py b/src/plugins/willing/willing_manager.py index c26325b1..dc4abfae 100644 --- a/src/plugins/willing/willing_manager.py +++ b/src/plugins/willing/willing_manager.py @@ -8,6 +8,8 @@ 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) """ 基类方法概览： diff --git a/src/plugins/zhishi/knowledge_library.py b/src/plugins/zhishi/knowledge_library.py index f8914c2f..1a8d35a3 100644 --- a/src/plugins/zhishi/knowledge_library.py +++ b/src/plugins/zhishi/knowledge_library.py @@ -7,6 +7,8 @@ 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 +17,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): From c88a73597a6c07fab73cf5d49f4cc84b2b54e21e Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Thu, 1 May 2025 19:18:37 +0000 Subject: [PATCH 2/2] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8=E6=A0=BC?= =?UTF-8?q?=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot.py | 1 + src/common/database.py | 1 + src/common/log_decorators.py | 1 + src/common/server.py | 1 + src/config/config.py | 1 + src/do_tool/tool_can_use/base_tool.py | 1 + src/individuality/individuality.py | 1 + src/individuality/offline_llm.py | 1 + src/main.py | 1 + src/plugins/PFC/chat_observer.py | 1 + src/plugins/PFC/conversation.py | 1 + src/plugins/PFC/message_sender.py | 1 + src/plugins/PFC/pfc.py | 1 + src/plugins/chat/chat_stream.py | 1 + src/plugins/chat/message.py | 1 + src/plugins/chat/message_sender.py | 1 + src/plugins/chat/utils_image.py | 1 + src/plugins/config_reload/api.py | 1 + src/plugins/emoji_system/emoji_manager.py | 1 + src/plugins/heartFC_chat/heartFC_chat.py | 1 + src/plugins/heartFC_chat/heartFC_sender.py | 1 + src/plugins/knowledge/src/embedding_store.py | 1 + src/plugins/memory_system/Hippocampus.py | 1 + src/plugins/memory_system/debug_memory.py | 1 + src/plugins/memory_system/manually_alter_memory.py | 1 + src/plugins/memory_system/offline_llm.py | 1 + src/plugins/memory_system/sample_distribution.py | 1 + src/plugins/models/utils_model.py | 1 + src/plugins/utils/prompt_builder.py | 2 ++ src/plugins/utils/timer_calculator.py | 1 + src/plugins/willing/willing_manager.py | 1 + src/plugins/zhishi/knowledge_library.py | 1 + 32 files changed, 33 insertions(+) diff --git a/bot.py b/bot.py index ecdbd2b9..5d811d4e 100644 --- a/bot.py +++ b/bot.py @@ -14,6 +14,7 @@ from src.common.logger_manager import get_logger from src.common.crash_logger import install_crash_handler from src.main import MainSystem from rich.traceback import install + install(show_locals=True, extra_lines=3) diff --git a/src/common/database.py b/src/common/database.py index 846c4ee7..66a2dc16 100644 --- a/src/common/database.py +++ b/src/common/database.py @@ -2,6 +2,7 @@ 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 diff --git a/src/common/log_decorators.py b/src/common/log_decorators.py index 4e881984..a57fae79 100644 --- a/src/common/log_decorators.py +++ b/src/common/log_decorators.py @@ -3,6 +3,7 @@ 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) diff --git a/src/common/server.py b/src/common/server.py index 0511f443..c080e28a 100644 --- a/src/common/server.py +++ b/src/common/server.py @@ -3,6 +3,7 @@ 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) diff --git a/src/config/config.py b/src/config/config.py index 29c28c83..a067633b 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -15,6 +15,7 @@ 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) diff --git a/src/do_tool/tool_can_use/base_tool.py b/src/do_tool/tool_can_use/base_tool.py index 298bc15a..88da036d 100644 --- a/src/do_tool/tool_can_use/base_tool.py +++ b/src/do_tool/tool_can_use/base_tool.py @@ -5,6 +5,7 @@ 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") diff --git a/src/individuality/individuality.py b/src/individuality/individuality.py index 6befe3e5..963fae0e 100644 --- a/src/individuality/individuality.py +++ b/src/individuality/individuality.py @@ -3,6 +3,7 @@ from .personality import Personality from .identity import Identity import random from rich.traceback import install + install(show_locals=True, extra_lines=3) diff --git a/src/individuality/offline_llm.py b/src/individuality/offline_llm.py index 2059d7fd..0e1a446c 100644 --- a/src/individuality/offline_llm.py +++ b/src/individuality/offline_llm.py @@ -7,6 +7,7 @@ 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") diff --git a/src/main.py b/src/main.py index 1f6ed188..3de3e880 100644 --- a/src/main.py +++ b/src/main.py @@ -18,6 +18,7 @@ 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") diff --git a/src/plugins/PFC/chat_observer.py b/src/plugins/PFC/chat_observer.py index 675e9963..34b66316 100644 --- a/src/plugins/PFC/chat_observer.py +++ b/src/plugins/PFC/chat_observer.py @@ -8,6 +8,7 @@ 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") diff --git a/src/plugins/PFC/conversation.py b/src/plugins/PFC/conversation.py index f7702e1b..925fd7b5 100644 --- a/src/plugins/PFC/conversation.py +++ b/src/plugins/PFC/conversation.py @@ -24,6 +24,7 @@ from .waiter import Waiter import traceback from rich.traceback import install + install(show_locals=True, extra_lines=3) logger = get_logger("pfc") diff --git a/src/plugins/PFC/message_sender.py b/src/plugins/PFC/message_sender.py index 709150c7..f1085768 100644 --- a/src/plugins/PFC/message_sender.py +++ b/src/plugins/PFC/message_sender.py @@ -9,6 +9,7 @@ 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) diff --git a/src/plugins/PFC/pfc.py b/src/plugins/PFC/pfc.py index 4ce98a52..50f7bf4c 100644 --- a/src/plugins/PFC/pfc.py +++ b/src/plugins/PFC/pfc.py @@ -9,6 +9,7 @@ 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: diff --git a/src/plugins/chat/chat_stream.py b/src/plugins/chat/chat_stream.py index d3095f3b..a949247c 100644 --- a/src/plugins/chat/chat_stream.py +++ b/src/plugins/chat/chat_stream.py @@ -10,6 +10,7 @@ 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) diff --git a/src/plugins/chat/message.py b/src/plugins/chat/message.py index 4c11dd98..354082e1 100644 --- a/src/plugins/chat/message.py +++ b/src/plugins/chat/message.py @@ -10,6 +10,7 @@ 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") diff --git a/src/plugins/chat/message_sender.py b/src/plugins/chat/message_sender.py index aaa48ca8..8bfee44b 100644 --- a/src/plugins/chat/message_sender.py +++ b/src/plugins/chat/message_sender.py @@ -14,6 +14,7 @@ from .utils import truncate_message, calculate_typing_time, count_messages_betwe from src.common.logger_manager import get_logger from rich.traceback import install + install(show_locals=True, extra_lines=3) diff --git a/src/plugins/chat/utils_image.py b/src/plugins/chat/utils_image.py index 285b6394..1f734502 100644 --- a/src/plugins/chat/utils_image.py +++ b/src/plugins/chat/utils_image.py @@ -14,6 +14,7 @@ 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") diff --git a/src/plugins/config_reload/api.py b/src/plugins/config_reload/api.py index 93b5d1cb..ee0a5454 100644 --- a/src/plugins/config_reload/api.py +++ b/src/plugins/config_reload/api.py @@ -1,5 +1,6 @@ from fastapi import APIRouter, HTTPException from rich.traceback import install + install(show_locals=True, extra_lines=3) # 创建APIRouter而不是FastAPI实例 diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 67075e29..24266c08 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -16,6 +16,7 @@ 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") diff --git a/src/plugins/heartFC_chat/heartFC_chat.py b/src/plugins/heartFC_chat/heartFC_chat.py index 077d623b..712b6af5 100644 --- a/src/plugins/heartFC_chat/heartFC_chat.py +++ b/src/plugins/heartFC_chat/heartFC_chat.py @@ -28,6 +28,7 @@ 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 + install(show_locals=True, extra_lines=3) diff --git a/src/plugins/heartFC_chat/heartFC_sender.py b/src/plugins/heartFC_chat/heartFC_sender.py index 001f578a..6fab5d62 100644 --- a/src/plugins/heartFC_chat/heartFC_sender.py +++ b/src/plugins/heartFC_chat/heartFC_sender.py @@ -10,6 +10,7 @@ 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) diff --git a/src/plugins/knowledge/src/embedding_store.py b/src/plugins/knowledge/src/embedding_store.py index 98ab757c..72c6c7b5 100644 --- a/src/plugins/knowledge/src/embedding_store.py +++ b/src/plugins/knowledge/src/embedding_store.py @@ -13,6 +13,7 @@ from .lpmmconfig import ENT_NAMESPACE, PG_NAMESPACE, REL_NAMESPACE, global_confi from .utils.hash import get_sha256 from .global_logger import logger from rich.traceback import install + install(show_locals=True, extra_lines=3) diff --git a/src/plugins/memory_system/Hippocampus.py b/src/plugins/memory_system/Hippocampus.py index 789da590..11ba8f40 100644 --- a/src/plugins/memory_system/Hippocampus.py +++ b/src/plugins/memory_system/Hippocampus.py @@ -21,6 +21,7 @@ from ..utils.chat_message_builder import ( 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) diff --git a/src/plugins/memory_system/debug_memory.py b/src/plugins/memory_system/debug_memory.py index 03c8fcac..ae767c85 100644 --- a/src/plugins/memory_system/debug_memory.py +++ b/src/plugins/memory_system/debug_memory.py @@ -9,6 +9,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname( 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) diff --git a/src/plugins/memory_system/manually_alter_memory.py b/src/plugins/memory_system/manually_alter_memory.py index 4054265b..10a75738 100644 --- a/src/plugins/memory_system/manually_alter_memory.py +++ b/src/plugins/memory_system/manually_alter_memory.py @@ -10,6 +10,7 @@ from Hippocampus import Hippocampus # 海马体和记忆图 from dotenv import load_dotenv from rich.traceback import install + install(show_locals=True, extra_lines=3) diff --git a/src/plugins/memory_system/offline_llm.py b/src/plugins/memory_system/offline_llm.py index f86ec411..335a76d3 100644 --- a/src/plugins/memory_system/offline_llm.py +++ b/src/plugins/memory_system/offline_llm.py @@ -7,6 +7,7 @@ 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") diff --git a/src/plugins/memory_system/sample_distribution.py b/src/plugins/memory_system/sample_distribution.py index f4fec12b..76796728 100644 --- a/src/plugins/memory_system/sample_distribution.py +++ b/src/plugins/memory_system/sample_distribution.py @@ -2,6 +2,7 @@ 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) diff --git a/src/plugins/models/utils_model.py b/src/plugins/models/utils_model.py index 60b507b9..7c7fe713 100644 --- a/src/plugins/models/utils_model.py +++ b/src/plugins/models/utils_model.py @@ -15,6 +15,7 @@ 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") diff --git a/src/plugins/utils/prompt_builder.py b/src/plugins/utils/prompt_builder.py index 02347033..c4555a55 100644 --- a/src/plugins/utils/prompt_builder.py +++ b/src/plugins/utils/prompt_builder.py @@ -3,8 +3,10 @@ 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") diff --git a/src/plugins/utils/timer_calculator.py b/src/plugins/utils/timer_calculator.py index 076a4fe5..d66f21cc 100644 --- a/src/plugins/utils/timer_calculator.py +++ b/src/plugins/utils/timer_calculator.py @@ -3,6 +3,7 @@ from functools import wraps from typing import Optional, Dict, Callable import asyncio from rich.traceback import install + install(show_locals=True, extra_lines=3) """ diff --git a/src/plugins/willing/willing_manager.py b/src/plugins/willing/willing_manager.py index dc4abfae..a5884da2 100644 --- a/src/plugins/willing/willing_manager.py +++ b/src/plugins/willing/willing_manager.py @@ -9,6 +9,7 @@ import importlib from typing import Dict, Optional import asyncio from rich.traceback import install + install(show_locals=True, extra_lines=3) """ diff --git a/src/plugins/zhishi/knowledge_library.py b/src/plugins/zhishi/knowledge_library.py index 1a8d35a3..26af3bda 100644 --- a/src/plugins/zhishi/knowledge_library.py +++ b/src/plugins/zhishi/knowledge_library.py @@ -8,6 +8,7 @@ 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 路径