73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
"""
|
||
使用 structlog 的结构化日志配置。
|
||
"""
|
||
import logging
|
||
import sys
|
||
from typing import Any
|
||
|
||
import structlog
|
||
from structlog.types import EventDict, Processor
|
||
|
||
|
||
def add_app_context(logger: Any, method_name: str, event_dict: EventDict) -> EventDict:
|
||
"""
|
||
向日志事件添加应用程序上下文。
|
||
|
||
Args:
|
||
logger: 日志记录器实例
|
||
method_name: 方法名
|
||
event_dict: 事件字典
|
||
|
||
Returns:
|
||
EventDict: 修改后的事件字典
|
||
"""
|
||
event_dict["app"] = "langchain-learning-kit"
|
||
return event_dict
|
||
|
||
|
||
def configure_logging(log_level: str = "INFO") -> None:
|
||
"""
|
||
配置结构化日志。
|
||
|
||
Args:
|
||
log_level: 日志级别（DEBUG、INFO、WARNING、ERROR、CRITICAL）
|
||
"""
|
||
# 配置标准日志
|
||
logging.basicConfig(
|
||
format="%(message)s",
|
||
stream=sys.stdout,
|
||
level=getattr(logging, log_level.upper()),
|
||
)
|
||
|
||
# 配置 structlog
|
||
structlog.configure(
|
||
processors=[
|
||
structlog.contextvars.merge_contextvars,
|
||
structlog.processors.add_log_level,
|
||
structlog.processors.StackInfoRenderer(),
|
||
structlog.dev.set_exc_info,
|
||
structlog.processors.TimeStamper(fmt="iso"),
|
||
add_app_context,
|
||
structlog.processors.JSONRenderer(),
|
||
],
|
||
wrapper_class=structlog.make_filtering_bound_logger(
|
||
getattr(logging, log_level.upper())
|
||
),
|
||
context_class=dict,
|
||
logger_factory=structlog.PrintLoggerFactory(),
|
||
cache_logger_on_first_use=True,
|
||
)
|
||
|
||
|
||
def get_logger(name: str = __name__) -> structlog.BoundLogger:
|
||
"""
|
||
获取结构化日志记录器实例。
|
||
|
||
Args:
|
||
name: 日志记录器名称
|
||
|
||
Returns:
|
||
structlog.BoundLogger: 日志记录器实例
|
||
"""
|
||
return structlog.get_logger(name)
|