sun
feb1a0b280
- 实现了基于LangChain的MCP Agent，支持连接MCP服务器调用工具 - 添加了环境配置文件(.env)，包含LLM模型和API配置信息 - 创建了完整的工具系统，包括BaseTool基类和Bash、Terminate、Add等工具 - 集成了天气查询工具，支持通过中国气象局API获取天气预报信息 - 实现了交互式对话功能，支持多轮工具调用和结果处理 - 添加了详细的CLAUDE.md开发指导文档
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
import logging
|
||
import sys
|
||
from loguru import logger
|
||
|
||
|
||
class InterceptHandler(logging.Handler):
|
||
def emit(self, record):
|
||
# 获取对应的 Loguru 级别
|
||
try:
|
||
level = logger.level(record.levelname).name
|
||
except ValueError:
|
||
level = record.levelno
|
||
|
||
# 从 logging 记录中找到调用位置
|
||
frame, depth = logging.currentframe(), 2
|
||
while frame.f_code.co_filename == logging.__file__:
|
||
frame = frame.f_back
|
||
depth += 1
|
||
|
||
logger.opt(depth=depth, exception=record.exc_info).log(
|
||
level, record.getMessage()
|
||
)
|
||
|
||
|
||
def setup_logger():
|
||
"""
|
||
配置 Loguru 日志记录器
|
||
- 移除默认的 handler
|
||
- 添加一个用于 INFO 级别日志的文件 sink
|
||
- 添加一个用于 ERROR 级别日志的文件 sink
|
||
- 保持控制台输出（级别为 DEBUG）
|
||
"""
|
||
# 检查是否已经初始化，防止重复初始化
|
||
if hasattr(setup_logger, '_initialized') and setup_logger._initialized:
|
||
return
|
||
|
||
# 1. 移除默认的控制台 handler
|
||
logger.remove()
|
||
|
||
# 2. 添加一个新的控制台 handler，设置级别为 DEBUG
|
||
logger.add(
|
||
sys.stderr,
|
||
level="INFO",
|
||
format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | "
|
||
"<level>{level: <8}</level> | "
|
||
"<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
|
||
colorize=True, # 开启颜色
|
||
)
|
||
|
||
# 3. 添加 INFO 日志文件 sink
|
||
# - level="INFO": 只记录 INFO 及以上级别的日志
|
||
# - filter: 确保 ERROR 级别的日志不会写入这个文件
|
||
# - rotation: 每天午夜创建一个新文件
|
||
# - retention: 只保留最近 14 天的日志
|
||
# - enqueue=True: 异步写入，提升性能
|
||
# - encoding="utf-8": 确保中文不会乱码
|
||
logger.add(
|
||
"logs/app.info.log",
|
||
level="INFO",
|
||
filter=lambda record: record["level"].name in ["INFO", "WARNING"],
|
||
rotation="00:00", # 每天午夜轮转
|
||
retention="14 days", # 保留 14 天
|
||
enqueue=True, # 异步写入
|
||
backtrace=True, # 记录堆栈信息
|
||
diagnose=True, # 诊断信息
|
||
encoding="utf-8",
|
||
)
|
||
|
||
# 4. 添加 ERROR 日志文件 sink
|
||
# - level="ERROR": 只记录 ERROR 及以上级别的日志 (ERROR, CRITICAL)
|
||
# - rotation: 文件大小超过 10 MB 时轮转
|
||
# - retention: 只保留最近 30 天的日志
|
||
logger.add(
|
||
"logs/app.error.log",
|
||
level="ERROR",
|
||
rotation="10 MB", # 按文件大小轮转
|
||
retention="30 days", # 保留 30 天
|
||
enqueue=True, # 异步写入
|
||
backtrace=True,
|
||
diagnose=True,
|
||
encoding="utf-8",
|
||
)
|
||
|
||
# 拦截 logging，以便将所有日志都转发到 Loguru
|
||
logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True)
|
||
|
||
# 设置已初始化标志
|
||
setup_logger._initialized = True
|
||
logger.info("日志系统配置完成。")
|