Set up logging configuration.
(log_file: Optional[str] = None)
| 209 | return response,user_prompt,raw_context |
| 210 | |
| 211 | def setup_logger(log_file: Optional[str] = None) -> logging.Logger: |
| 212 | """Set up logging configuration.""" |
| 213 | logger = logging.getLogger('locomo_eval') |
| 214 | logger.setLevel(logging.INFO) |
| 215 | formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s') |
| 216 | |
| 217 | # Console handler |
| 218 | console_handler = logging.StreamHandler() |
| 219 | console_handler.setFormatter(formatter) |
| 220 | logger.addHandler(console_handler) |
| 221 | |
| 222 | # File handler if log_file is specified |
| 223 | if log_file: |
| 224 | file_handler = logging.FileHandler(log_file) |
| 225 | file_handler.setFormatter(formatter) |
| 226 | logger.addHandler(file_handler) |
| 227 | |
| 228 | return logger |
| 229 | |
| 230 | def process_single_sample(sample_data: tuple, model: str, backend: str, retrieve_k: int, |
| 231 | temperature_c5: float, sglang_host: str, sglang_port: int, |