Spaces:
Runtime error
Runtime error
File size: 3,166 Bytes
bb41dbe ef58f28 bb41dbe ef58f28 bb41dbe ef58f28 bb41dbe ef58f28 bb41dbe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | import logging
import logging.config
from pathlib import Path
from typing import Optional
def setup_logging(
log_level: str = "INFO",
log_file: Optional[str] = None,
log_format: Optional[str] = None
) -> None:
"""
Set up logging configuration for the Pokemon agent project.
Args:
log_level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
log_file: Optional file path to write logs to
log_format: Optional custom log format
"""
if log_format is None:
log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
# Create logs directory if logging to file
if log_file:
log_path = Path(log_file)
log_path.parent.mkdir(parents=True, exist_ok=True)
# Configure logging
config = {
"version": 1,
"disable_existing_loggers": False,
"formatters": {
"standard": {
"format": log_format,
"datefmt": "%Y-%m-%d %H:%M:%S"
},
"detailed": {
"format": "%(asctime)s - %(name)s - %(levelname)s - %(funcName)s:%(lineno)d - %(message)s",
"datefmt": "%Y-%m-%d %H:%M:%S"
}
},
"handlers": {
"console": {
"class": "logging.StreamHandler",
"level": log_level,
"formatter": "standard",
"stream": "ext://sys.stdout"
}
},
"loggers": {
"agent": {
"level": log_level,
"handlers": ["console"],
"propagate": True
},
"langgraph": {
"level": "WARNING",
"handlers": ["console"],
"propagate": False
},
"langchain": {
"level": "WARNING",
"handlers": ["console"],
"propagate": False
}
},
"root": {
"level": log_level,
"handlers": ["console"]
}
}
# Add file handler if log_file is specified
if log_file:
config["handlers"]["file"] = {
"class": "logging.handlers.RotatingFileHandler",
"level": log_level,
"formatter": "detailed",
"filename": log_file,
"maxBytes": 10485760, # 10MB
"backupCount": 5,
"encoding": "utf-8"
}
# Add file handler to all loggers
config["loggers"]["agent"]["handlers"].append("file")
# Add specific configuration for __main__ case
config["loggers"]["agent.__main__"] = {
"level": log_level,
"handlers": ["console", "file"],
"propagate": False
}
config["root"]["handlers"].append("file")
logging.config.dictConfig(config)
def get_logger(name: str) -> logging.Logger:
"""
Get a logger instance for a specific module.
Args:
name: Logger name (typically __name__)
Returns:
Configured logger instance
"""
return logging.getLogger(f"pokemon_agent.{name}") |