| """日志工具""" |
|
|
| import logging |
| import sys |
| from typing import Optional |
|
|
| def setup_logger( |
| name: str = "hello_agents", |
| level: str = "INFO", |
| format_string: Optional[str] = None |
| ) -> logging.Logger: |
| """ |
| 设置日志记录器 |
| |
| Args: |
| name: 日志记录器名称 |
| level: 日志级别 |
| format_string: 日志格式 |
| |
| Returns: |
| 配置好的日志记录器 |
| """ |
| logger = logging.getLogger(name) |
| logger.setLevel(getattr(logging, level.upper())) |
| |
| if not logger.handlers: |
| handler = logging.StreamHandler(sys.stdout) |
| formatter = logging.Formatter( |
| format_string or |
| '%(asctime)s - %(name)s - %(levelname)s - %(message)s' |
| ) |
| handler.setFormatter(formatter) |
| logger.addHandler(handler) |
| |
| return logger |
|
|
| def get_logger(name: str = "hello_agents") -> logging.Logger: |
| """获取日志记录器""" |
| return logging.getLogger(name) |
